mirror of https://github.com/jumpserver/jumpserver
perf: 优化 m2m json field
parent
a112d3c99d
commit
1ec4cbdf38
|
@ -37,7 +37,8 @@ class LoginAssetCheckAPI(CreateAPIView):
|
||||||
'account_username': self.serializer.validated_data.get('account_username'),
|
'account_username': self.serializer.validated_data.get('account_username'),
|
||||||
'action': LoginAssetACL.ActionChoices.review
|
'action': LoginAssetACL.ActionChoices.review
|
||||||
}
|
}
|
||||||
acl = LoginAssetACL.filter_queryset(**kwargs).valid().first()
|
acl = LoginAssetACL.objects.filter(**kwargs).valid().first()
|
||||||
|
|
||||||
if acl:
|
if acl:
|
||||||
need_review = True
|
need_review = True
|
||||||
response_data = self._get_response_data_of_need_review(acl)
|
response_data = self._get_response_data_of_need_review(acl)
|
||||||
|
|
|
@ -1,19 +1,13 @@
|
||||||
from django.core.validators import MinValueValidator, MaxValueValidator
|
from django.core.validators import MinValueValidator, MaxValueValidator
|
||||||
from django.db import models
|
from django.db import models
|
||||||
from django.db.models import Q
|
|
||||||
from django.utils.translation import gettext_lazy as _
|
from django.utils.translation import gettext_lazy as _
|
||||||
|
|
||||||
from common.db.fields import JSONManyToManyField
|
from common.db.fields import JSONManyToManyField
|
||||||
from common.db.models import JMSBaseModel
|
from common.db.models import JMSBaseModel
|
||||||
from common.utils import contains_ip
|
from orgs.mixins.models import OrgModelMixin
|
||||||
from orgs.mixins.models import OrgModelMixin, OrgManager
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
'ACLManager',
|
'BaseACL', 'UserAssetAccountBaseACL',
|
||||||
'BaseACL',
|
|
||||||
'BaseACLQuerySet',
|
|
||||||
'UserAssetAccountBaseACL',
|
|
||||||
'UserAssetAccountACLQuerySet'
|
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
@ -37,41 +31,6 @@ class BaseACLQuerySet(models.QuerySet):
|
||||||
return self.inactive()
|
return self.inactive()
|
||||||
|
|
||||||
|
|
||||||
class UserAssetAccountACLQuerySet(BaseACLQuerySet):
|
|
||||||
def filter_user(self, username):
|
|
||||||
q = Q(users__username_group__contains=username) | \
|
|
||||||
Q(users__username_group__contains='*')
|
|
||||||
return self.filter(q)
|
|
||||||
|
|
||||||
def filter_asset(self, name=None, address=None):
|
|
||||||
queryset = self.filter()
|
|
||||||
if name:
|
|
||||||
q = Q(assets__name_group__contains=name) | \
|
|
||||||
Q(assets__name_group__contains='*')
|
|
||||||
queryset = queryset.filter(q)
|
|
||||||
if address:
|
|
||||||
ids = [
|
|
||||||
q.id for q in queryset
|
|
||||||
if contains_ip(address, q.assets.get('address_group', []))
|
|
||||||
]
|
|
||||||
queryset = queryset.filter(id__in=ids)
|
|
||||||
return queryset
|
|
||||||
|
|
||||||
def filter_account(self, username):
|
|
||||||
q = Q(accounts__username_group__contains=username) | \
|
|
||||||
Q(accounts__username_group__contains='*')
|
|
||||||
return self.filter(q)
|
|
||||||
|
|
||||||
|
|
||||||
class ACLManager(models.Manager):
|
|
||||||
def valid(self):
|
|
||||||
return self.get_queryset().valid()
|
|
||||||
|
|
||||||
|
|
||||||
class OrgACLManager(OrgManager, ACLManager):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class BaseACL(JMSBaseModel):
|
class BaseACL(JMSBaseModel):
|
||||||
name = models.CharField(max_length=128, verbose_name=_('Name'))
|
name = models.CharField(max_length=128, verbose_name=_('Name'))
|
||||||
priority = models.IntegerField(
|
priority = models.IntegerField(
|
||||||
|
@ -84,7 +43,7 @@ class BaseACL(JMSBaseModel):
|
||||||
is_active = models.BooleanField(default=True, verbose_name=_("Active"))
|
is_active = models.BooleanField(default=True, verbose_name=_("Active"))
|
||||||
|
|
||||||
ActionChoices = ActionChoices
|
ActionChoices = ActionChoices
|
||||||
objects = ACLManager.from_queryset(BaseACLQuerySet)()
|
objects = BaseACLQuerySet.as_manager()
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
ordering = ('priority', 'date_updated', 'name')
|
ordering = ('priority', 'date_updated', 'name')
|
||||||
|
@ -99,8 +58,6 @@ class UserAssetAccountBaseACL(BaseACL, OrgModelMixin):
|
||||||
assets = JSONManyToManyField('assets.Asset', default=dict, verbose_name=_('Assets'))
|
assets = JSONManyToManyField('assets.Asset', default=dict, verbose_name=_('Assets'))
|
||||||
accounts = models.JSONField(default=list, verbose_name=_("Account"))
|
accounts = models.JSONField(default=list, verbose_name=_("Account"))
|
||||||
|
|
||||||
objects = OrgACLManager.from_queryset(UserAssetAccountACLQuerySet)()
|
|
||||||
|
|
||||||
class Meta(BaseACL.Meta):
|
class Meta(BaseACL.Meta):
|
||||||
unique_together = ('name', 'org_id')
|
unique_together = ('name', 'org_id')
|
||||||
abstract = True
|
abstract = True
|
||||||
|
|
|
@ -1,11 +1,9 @@
|
||||||
from django.utils.translation import ugettext_lazy as _
|
from django.utils.translation import ugettext_lazy as _
|
||||||
|
|
||||||
|
|
||||||
from .base import UserAssetAccountBaseACL
|
from .base import UserAssetAccountBaseACL
|
||||||
|
|
||||||
|
|
||||||
class LoginAssetACL(UserAssetAccountBaseACL):
|
class LoginAssetACL(UserAssetAccountBaseACL):
|
||||||
|
|
||||||
class Meta(UserAssetAccountBaseACL.Meta):
|
class Meta(UserAssetAccountBaseACL.Meta):
|
||||||
verbose_name = _('Login asset acl')
|
verbose_name = _('Login asset acl')
|
||||||
abstract = False
|
abstract = False
|
||||||
|
|
|
@ -10,7 +10,6 @@
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import uuid
|
import uuid
|
||||||
from functools import reduce
|
|
||||||
|
|
||||||
from django.db import models
|
from django.db import models
|
||||||
from django.db import transaction
|
from django.db import transaction
|
||||||
|
@ -55,7 +54,6 @@ def output_as_string(field_name):
|
||||||
|
|
||||||
|
|
||||||
class MultiTableChildQueryset(QuerySet):
|
class MultiTableChildQueryset(QuerySet):
|
||||||
|
|
||||||
def bulk_create(self, objs, batch_size=None):
|
def bulk_create(self, objs, batch_size=None):
|
||||||
assert batch_size is None or batch_size > 0
|
assert batch_size is None or batch_size > 0
|
||||||
if not objs:
|
if not objs:
|
||||||
|
|
Loading…
Reference in New Issue