github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: 修复 sqlserver 命令执行问题

pull/12138/head
Aaron3S 2023-11-16 16:08:22 +08:00 committed by Bryan
parent 44f029774d
commit d1391cb5d5
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/ops/models/job.py
View File

@ -298,10 +298,18 @@ class JobExecution(JMSOrgBaseModel):
db_module_name_map = {
'mysql': 'community.mysql.mysql_query',
'postgresql': 'community.postgresql.postgresql_query',
'sqlserver': 'community.general.mssql_script:',
'sqlserver': 'community.general.mssql_script',
}
extra_query_token_map = {
'sqlserver': 'script'
}
extra_login_db_token_map = {
'sqlserver': 'name'
}
if module in db_modules:
login_db_token = extra_login_db_token_map.get(module, 'login_db')
query_token = extra_query_token_map.get(module, 'query')
module = db_module_name_map.get(module, None)
if not module:
print('not support db module: {}'.format(module))
@ -311,8 +319,9 @@ class JobExecution(JMSOrgBaseModel):
"login_user={{login_user}} " \
"login_password={{login_password}} " \
"login_port={{login_port}} " \
"login_db={{login_db}}"
shell = "{} query=\"{}\" ".format(login_args, self.current_job.args)
"%s={{login_db}}" % login_db_token
print(login_args)
shell = "{} {}=\"{}\" ".format(login_args, query_token, self.current_job.args)
return module, shell
if module == 'win_shell':