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.
89 lines
2.6 KiB
89 lines
2.6 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
import os
|
|
|
|
from django.core.cache import cache
|
|
from django.core.files.storage import default_storage
|
|
import jms_storage
|
|
|
|
from assets.models import Asset, SystemUser
|
|
from users.models import User
|
|
from common.utils import get_logger
|
|
from .const import USERS_CACHE_KEY, ASSETS_CACHE_KEY, SYSTEM_USER_CACHE_KEY
|
|
from .models import ReplayStorage
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def get_session_asset_list():
|
|
return Asset.objects.values_list('hostname', flat=True)
|
|
|
|
|
|
def get_session_user_list():
|
|
return User.objects.exclude(role=User.ROLE_APP).values_list('username', flat=True)
|
|
|
|
|
|
def get_session_system_user_list():
|
|
return SystemUser.objects.values_list('username', flat=True)
|
|
|
|
|
|
def get_user_list_from_cache():
|
|
return cache.get(USERS_CACHE_KEY)
|
|
|
|
|
|
def get_asset_list_from_cache():
|
|
return cache.get(ASSETS_CACHE_KEY)
|
|
|
|
|
|
def get_system_user_list_from_cache():
|
|
return cache.get(SYSTEM_USER_CACHE_KEY)
|
|
|
|
|
|
def find_session_replay_local(session):
|
|
# 新版本和老版本的文件后缀不同
|
|
session_path = session.get_rel_replay_path() # 存在外部存储上的路径
|
|
local_path = session.get_local_path()
|
|
local_path_v1 = session.get_local_path(version=1)
|
|
|
|
# 去default storage中查找
|
|
for _local_path in (local_path, local_path_v1, session_path):
|
|
if default_storage.exists(_local_path):
|
|
url = default_storage.url(_local_path)
|
|
return _local_path, url
|
|
return None, None
|
|
|
|
|
|
def download_session_replay(session):
|
|
session_path = session.get_rel_replay_path() # 存在外部存储上的路径
|
|
local_path = session.get_local_path()
|
|
replay_storages = ReplayStorage.objects.all()
|
|
configs = {
|
|
storage.name: storage.config
|
|
for storage in replay_storages
|
|
if not storage.in_defaults()
|
|
}
|
|
if not configs:
|
|
msg = "Not found replay file, and not remote storage set"
|
|
return None, msg
|
|
|
|
# 保存到storage的路径
|
|
target_path = os.path.join(default_storage.base_location, local_path)
|
|
target_dir = os.path.dirname(target_path)
|
|
if not os.path.isdir(target_dir):
|
|
os.makedirs(target_dir, exist_ok=True)
|
|
storage = jms_storage.get_multi_object_storage(configs)
|
|
ok, err = storage.download(session_path, target_path)
|
|
if not ok:
|
|
msg = "Failed download replay file: {}".format(err)
|
|
logger.error(msg)
|
|
return None, msg
|
|
url = default_storage.url(local_path)
|
|
return local_path, url
|
|
|
|
|
|
def get_session_replay_url(session):
|
|
local_path, url = find_session_replay_local(session)
|
|
if local_path is None:
|
|
local_path, url = download_session_replay(session)
|
|
return local_path, url
|