github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-10-22 08:19:04 +00:00
Code Issues Projects Releases Wiki Activity

[Update] 修改约束

Browse Source
This commit is contained in:
ibuler
2018-07-19 19:24:29 +08:00
parent fd5f562cbf
commit c24f1a0517
11 changed files with 39 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
apps/orgs/mixins.py
View File

@@ -8,6 +8,7 @@ from django.forms import ModelForm
from common.utils import get_logger
from .utils import get_current_org, set_current_org
from .models import Organization
logger = get_logger(__file__)
@@ -32,19 +33,21 @@ class OrgManager(models.Manager):
queryset = super(OrgManager, self).get_queryset()
queryset = queryset.filter(**kwargs)
# print(kwargs)
print(queryset.query)
# print(queryset.query)
return queryset
def all(self):
current_org = get_current_org()
if not current_org:
msg = 'You can `objects.set_current_org(org).all()` then run it'
warnings.warn(msg)
return self
else:
return super(OrgManager, self).all()
# def all(self):
# current_org = get_current_org()
# if not current_org:
# msg = 'You can `objects.set_current_org(org).all()` then run it'
# warnings.warn(msg)
# return self
# else:
# return super(OrgManager, self).all()
def set_current_org(self, org):
if isinstance(org, str):
org = Organization.objects.get(name=org)
set_current_org(org)
return self