mirror of https://github.com/jumpserver/jumpserver
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
|
# ~*~ coding: utf-8 ~*~
|
||
|
|
import datetime
|
||
|
|
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from .base import CommandBase
|
||
|
|
|
||
|
|
|
||
|
|
class CommandStore(CommandBase):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
from terminal.models import Command
|
||
|
|
self.model = Command
|
||
|
|
|
||
|
|
def save(self, command):
|
||
|
|
"""
|
||
|
|
保存命令到数据库
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.model.objects.create(
|
||
|
|
user=command["user"], asset=command["asset"],
|
||
|
|
system_user=command["system_user"], input=command["input"],
|
||
|
|
output=command["output"], session=command["session"],
|
||
|
|
timestamp=command["timestamp"]
|
||
|
|
)
|
||
|
|
|
||
|
|
def bulk_save(self, commands):
|
||
|
|
"""
|
||
|
|
批量保存命令到数据库, command的顺序和save中一致
|
||
|
|
"""
|
||
|
|
_commands = []
|
||
|
|
for c in commands:
|
||
|
|
_commands.append(self.model(
|
||
|
|
user=c["user"], asset=c["asset"], system_user=c["system_user"],
|
||
|
|
input=c["input"], output=c["output"], session=c["session"],
|
||
|
|
timestamp=c["timestamp"]
|
||
|
|
))
|
||
|
|
return self.model.objects.bulk_create(_commands)
|
||
|
|
|
||
|
|
def filter(self, date_from=None, date_to=None, user=None,
|
||
|
|
asset=None, system_user=None, _input=None, session=None):
|
||
|
|
filter_kwargs = {}
|
||
|
|
|
||
|
|
if date_from:
|
||
|
|
filter_kwargs['timestamp__gte'] = int(date_from.timestamp())
|
||
|
|
else:
|
||
|
|
week_ago = timezone.now() - datetime.timedelta(days=7)
|
||
|
|
filter_kwargs['timestamp__gte'] = int(week_ago.timestamp())
|
||
|
|
if date_to:
|
||
|
|
filter_kwargs['timestamp__lte'] = int(date_to.timestamp())
|
||
|
|
else:
|
||
|
|
filter_kwargs['timestamp__lte'] = int(timezone.now().timestamp())
|
||
|
|
if user:
|
||
|
|
filter_kwargs['user'] = user
|
||
|
|
if asset:
|
||
|
|
filter_kwargs['asset'] = asset
|
||
|
|
if system_user:
|
||
|
|
filter_kwargs['system_user'] = system_user
|
||
|
|
if _input:
|
||
|
|
filter_kwargs['input__icontains'] = _input
|
||
|
|
if session:
|
||
|
|
filter_kwargs['session'] = session
|
||
|
|
|
||
|
|
queryset = self.model.objects.filter(**filter_kwargs)
|
||
|
|
return queryset
|
||
|
|
|
||
|
|
def all(self):
|
||
|
|
"""返回所有数据"""
|
||
|
|
return self.model.objects.iterator()
|
||
|
|
|