mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-06-25 06:21:58 +00:00
[Update] 修改节点full value
This commit is contained in:
parent
1877511acf
commit
cd2b88caee
@ -5,9 +5,10 @@ import uuid
|
|||||||
from django.db import models, transaction
|
from django.db import models, transaction
|
||||||
from django.db.models import Q
|
from django.db.models import Q
|
||||||
from django.utils.translation import ugettext_lazy as _
|
from django.utils.translation import ugettext_lazy as _
|
||||||
|
from django.core.cache import cache
|
||||||
|
|
||||||
from orgs.mixins import OrgModelMixin
|
from orgs.mixins import OrgModelMixin
|
||||||
from orgs.utils import current_org, set_current_org, get_current_org
|
from orgs.utils import set_current_org, get_current_org
|
||||||
from orgs.models import Organization
|
from orgs.models import Organization
|
||||||
|
|
||||||
__all__ = ['Node']
|
__all__ = ['Node']
|
||||||
@ -21,10 +22,10 @@ class Node(OrgModelMixin):
|
|||||||
date_create = models.DateTimeField(auto_now_add=True)
|
date_create = models.DateTimeField(auto_now_add=True)
|
||||||
|
|
||||||
is_node = True
|
is_node = True
|
||||||
|
_full_value_cache_key_prefix = '_NODE_VALUE_{}'
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return self.value
|
return self.full_value
|
||||||
# return self.full_value
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def __eq__(self, other):
|
||||||
return self.key == other.key
|
return self.key == other.key
|
||||||
@ -47,10 +48,29 @@ class Node(OrgModelMixin):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def full_value(self):
|
def full_value(self):
|
||||||
ancestor = [a.value for a in self.get_ancestor(with_self=True)]
|
key = self._full_value_cache_key_prefix.format(self.key)
|
||||||
|
cached = cache.get(key)
|
||||||
|
if cached:
|
||||||
|
return cached
|
||||||
|
value = self.get_full_value()
|
||||||
|
self.cache_full_value(value)
|
||||||
|
return value
|
||||||
|
|
||||||
|
def get_full_value(self):
|
||||||
|
# ancestor = [a.value for a in self.get_ancestor(with_self=True)]
|
||||||
if self.is_root():
|
if self.is_root():
|
||||||
return self.value
|
return self.value
|
||||||
return ' / '.join(ancestor)
|
parent_full_value = self.parent.full_value
|
||||||
|
value = parent_full_value + ' / ' + self.value
|
||||||
|
return value
|
||||||
|
|
||||||
|
def cache_full_value(self, value):
|
||||||
|
key = self._full_value_cache_key_prefix.format(self.key)
|
||||||
|
cache.set(key, value, 3600)
|
||||||
|
|
||||||
|
def expire_full_value(self):
|
||||||
|
key = self._full_value_cache_key_prefix.format(self.key)
|
||||||
|
cache.delete_pattern(key+'*')
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def level(self):
|
def level(self):
|
||||||
@ -204,6 +224,10 @@ class Node(OrgModelMixin):
|
|||||||
defaults = {'value': 'Default'}
|
defaults = {'value': 'Default'}
|
||||||
return cls.objects.get_or_create(defaults=defaults, key='0')
|
return cls.objects.get_or_create(defaults=defaults, key='0')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def get_tree_name_ref(cls):
|
||||||
|
pass
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def generate_fake(cls, count=100):
|
def generate_fake(cls, count=100):
|
||||||
import random
|
import random
|
||||||
@ -212,4 +236,3 @@ class Node(OrgModelMixin):
|
|||||||
node.create_child('Node {}'.format(i))
|
node.create_child('Node {}'.format(i))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -86,3 +86,9 @@ def on_node_assets_changed(sender, instance=None, **kwargs):
|
|||||||
system_users = SystemUser.objects.filter(nodes=instance)
|
system_users = SystemUser.objects.filter(nodes=instance)
|
||||||
for system_user in system_users:
|
for system_user in system_users:
|
||||||
system_user.assets.add(*tuple(assets))
|
system_user.assets.add(*tuple(assets))
|
||||||
|
|
||||||
|
|
||||||
|
@receiver(post_save, sender=Node)
|
||||||
|
def on_node_update_or_created(sender, instance=None, created=False, **kwargs):
|
||||||
|
if instance and not created:
|
||||||
|
instance.expire_full_value()
|
||||||
|
Loading…
Reference in New Issue
Block a user