mirror of https://github.com/jumpserver/jumpserver
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
# ~*~ coding: utf-8 ~*~
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from audits.models import OperateLog
|
|
|
|
|
|
class OperateLogStore(object):
|
|
# 用不可见字符分割前后数据，节省存储-> diff: {'key': 'before\0after'}
|
|
SEP = '\0'
|
|
|
|
def __init__(self, config):
|
|
self.model = OperateLog
|
|
self.max_length = 2048
|
|
self.max_length_tip_msg = _(
|
|
'The text content is too long. Use Elasticsearch to store operation logs'
|
|
)
|
|
|
|
@staticmethod
|
|
def ping(timeout=None):
|
|
return True
|
|
|
|
@classmethod
|
|
def convert_before_after_to_diff(cls, before, after):
|
|
if not isinstance(before, dict):
|
|
before = dict()
|
|
if not isinstance(after, dict):
|
|
after = dict()
|
|
|
|
diff = dict()
|
|
keys = set(before.keys()) | set(after.keys())
|
|
for k in keys:
|
|
before_value = before.get(k, '')
|
|
after_value = after.get(k, '')
|
|
diff[k] = '%s%s%s' % (before_value, cls.SEP, after_value)
|
|
return diff
|
|
|
|
@classmethod
|
|
def convert_diff_to_before_after(cls, diff):
|
|
before, after = dict(), dict()
|
|
if not diff:
|
|
return before, after
|
|
|
|
for k, v in diff.items():
|
|
before_value, after_value = v.split(cls.SEP, 1)
|
|
before[k], after[k] = before_value, after_value
|
|
return before, after
|
|
|
|
@classmethod
|
|
def convert_diff_friendly(cls, raw_diff):
|
|
diff_list = list()
|
|
for k, v in raw_diff.items():
|
|
before, after = v.split(cls.SEP, 1)
|
|
diff_list.append({
|
|
'field': k,
|
|
'before': before if before else _('empty'),
|
|
'after': after if after else _('empty'),
|
|
})
|
|
return diff_list
|
|
|
|
def save(self, **kwargs):
|
|
log_id = kwargs.get('id', '')
|
|
before = kwargs.pop('before') or {}
|
|
after = kwargs.pop('after') or {}
|
|
|
|
op_log = self.model.objects.filter(pk=log_id).first()
|
|
if op_log is not None:
|
|
op_log_diff = op_log.diff or {}
|
|
op_before, op_after = self.convert_diff_to_before_after(op_log_diff)
|
|
before.update(op_before)
|
|
after.update(op_after)
|
|
else:
|
|
op_log = self.model(**kwargs)
|
|
|
|
diff = self.convert_before_after_to_diff(before, after)
|
|
if len(str(diff)) > self.max_length:
|
|
limit = {str(_('Tips')): self.max_length_tip_msg}
|
|
diff = self.convert_before_after_to_diff(limit, limit)
|
|
|
|
op_log.diff = diff
|
|
op_log.save()
|