model save actions

models/User.py

@@ -1,3 +1,4 @@
+from flask_jwt_extended import current_user
 from mongoengine import *
 
 from internal_lib.EncryptedField import EncryptedStringField
@@ -14,6 +15,7 @@ class User(Document):
         'auto_create_index': True,
         'can_query': True,
         "quyery": {},
+        'ignore_fields': ['password'],
         'methods': [Methods.Get, Methods.List, Methods.Create],
         "indexes": [
             ('union'),
@@ -105,6 +107,12 @@ class User(Document):
     user_group = ReferenceField(Group)
     payment_group = ReferenceField(Group)
 
+    def save(self,*args, **kwargs):
+        self.union = current_user.union
+        self.member_no = User.objects.filter(union=self.union).count() + 1
+        super(User, self).save(*args, **kwargs)
+
+