mirror of https://github.com/jumpserver/jumpserver
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import os
|
|
from .base import ObjectStorage, LogStorage
|
|
|
|
|
|
class JMSReplayStorage(ObjectStorage):
|
|
def __init__(self, config):
|
|
self.client = config.get("SERVICE")
|
|
|
|
def upload(self, src, target):
|
|
session_id = os.path.basename(target).split('.')[0]
|
|
ok = self.client.push_session_replay(src, session_id)
|
|
return ok, None
|
|
|
|
def delete(self, path):
|
|
return False, Exception("Not support not")
|
|
|
|
def exists(self, path):
|
|
return False
|
|
|
|
def download(self, src, target):
|
|
return False, Exception("Not support not")
|
|
|
|
@property
|
|
def type(self):
|
|
return 'jms'
|
|
|
|
|
|
class JMSCommandStorage(LogStorage):
|
|
def __init__(self, config):
|
|
self.client = config.get("SERVICE")
|
|
if not self.client:
|
|
raise Exception("Not found app service")
|
|
|
|
def save(self, command):
|
|
return self.client.push_session_command([command])
|
|
|
|
def bulk_save(self, command_set, raise_on_error=True):
|
|
return self.client.push_session_command(command_set)
|
|
|
|
def filter(self, date_from=None, date_to=None,
|
|
user=None, asset=None, account=None,
|
|
input=None, session=None):
|
|
pass
|
|
|
|
def count(self, date_from=None, date_to=None,
|
|
user=None, asset=None, account=None,
|
|
input=None, session=None):
|
|
pass
|