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

fix: 修复获取类型为null的命令显示不支持的问题 

fix: 修复获取类型为null的命令显示不支持的问题
pull/8253/head v2.21.4
Jiangjie.Bai 2022-05-07 17:50:49 +08:00 committed by Jiangjie.Bai
parent a23bd4b3eb
commit 622bef07ad
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/terminal/models/storage.py
View File

@ -89,16 +89,20 @@ class CommandStorage(CommonStorageModelMixin, CommonModelMixin):
return Terminal.objects.filter(command_storage=self.name, is_deleted=False).exists() return Terminal.objects.filter(command_storage=self.name, is_deleted=False).exists()
def get_command_queryset(self): def get_command_queryset(self):
if self.type_null:
return Command.objects.none()
if self.type_server: if self.type_server:
qs = Command.objects.all() return Command.objects.all()
else:
if self.type not in TYPE_ENGINE_MAPPING: if self.type in TYPE_ENGINE_MAPPING:
logger.error(f'Command storage `{self.type}` not support')
return Command.objects.none()
engine_mod = import_module(TYPE_ENGINE_MAPPING[self.type]) engine_mod = import_module(TYPE_ENGINE_MAPPING[self.type])
qs = engine_mod.QuerySet(self.config) qs = engine_mod.QuerySet(self.config)
qs.model = Command qs.model = Command
return qs return qs
logger.error(f'Command storage `{self.type}` not support')
return Command.objects.none()
def save(self, force_insert=False, force_update=False, using=None, def save(self, force_insert=False, force_update=False, using=None,
update_fields=None): update_fields=None):