mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.2 KiB
43 lines
1.2 KiB
from importlib import import_module
|
|
from django.conf import settings
|
|
from .command.serializers import SessionCommandSerializer
|
|
from ..const import COMMAND_STORAGE_TYPE_SERVER
|
|
|
|
|
|
TYPE_ENGINE_MAPPING = {
|
|
'elasticsearch': 'terminal.backends.command.es',
|
|
'es': 'terminal.backends.command.es',
|
|
}
|
|
|
|
|
|
def get_command_storage():
|
|
config = settings.COMMAND_STORAGE
|
|
engine_class = import_module(config['ENGINE'])
|
|
storage = engine_class.CommandStore(config)
|
|
return storage
|
|
|
|
|
|
def get_terminal_command_storages():
|
|
from ..models import CommandStorage
|
|
storage_list = {}
|
|
for s in CommandStorage.objects.all():
|
|
tp = s.type
|
|
if tp == COMMAND_STORAGE_TYPE_SERVER:
|
|
storage = get_command_storage()
|
|
else:
|
|
if not TYPE_ENGINE_MAPPING.get(tp):
|
|
continue
|
|
engine_class = import_module(TYPE_ENGINE_MAPPING[tp])
|
|
storage = engine_class.CommandStore(s.config)
|
|
storage_list[s.name] = storage
|
|
return storage_list
|
|
|
|
|
|
def get_multi_command_storage():
|
|
from .command.multi import CommandStore
|
|
storage_list = get_terminal_command_storages().values()
|
|
storage = CommandStore(storage_list)
|
|
return storage
|
|
|
|
|