1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# ~*~ 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 = {}
date_from_default = timezone.now() - datetime.timedelta(days=7)
date_to_default = timezone.now()
date_from = date_from if date_from else date_from_default
date_to = date_to if date_to else date_to_default
filter_kwargs['timestamp__gte'] = int(date_from.timestamp())
filter_kwargs['timestamp__lte'] = int(date_to.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()