rref #5367

rref #5366 ref zetaops/zengine#66 ref zetaops/zengine#65
zetaops · Jul 10, 2016 · 877e78e · 877e78e
1 parent cde1d94
commit 877e78e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/ulakbus/models/auth.py b/ulakbus/models/auth.py
@@ -224,6 +224,11 @@ class Unit(Model):
     uid = field.Integer(index=True)
     parent = LinkProxy('Unit', verbose_name='Üst Birim', reverse_name='alt_birimler')
 
+
+    @classmethod
+    def get_user_keys(cls, current, unit_key):
+        return cls(current).objects.filter(unit_id=unit_key).values_list('user_id', flatten=True)
+
     class Meta:
         app = 'Sistem'
         verbose_name = "Unit"

diff --git a/ulakbus/settings.py b/ulakbus/settings.py
@@ -26,6 +26,8 @@
 PERMISSION_MODEL = 'ulakbus.models.auth.Permission'
 USER_MODEL = 'ulakbus.models.auth.User'
 ROLE_MODEL = 'ulakbus.models.auth.Role'
+UNIT_MODEL = 'ulakbus.models.auth.Unit'
+
 # # left blank to use StreamHandler aka stderr
 # LOG_HANDLER = os.environ.get('LOG_HANDLER', 'file')
 #
@@ -231,3 +233,6 @@
 
 #: User search method of messaging subsystem will work on these fields
 MESSAGING_USER_SEARCH_FIELDS = ['username', ]
+
+#: Unit search method of messaging subsystem will work on these fields
+MESSAGING_UNIT_SEARCH_FIELDS = ['name',]