Merge pull request #146 in RMS/api-main-service from red-develop to RMSv2

* commit 'ccd3ed86':
  fixing the update of code

app/entities/models.py

@@ -73,9 +73,10 @@ class Module(AuditClass):
 
     def save(self, *args, **kwargs):
         super(Module, self).save(*args, **kwargs)
-        code = enums.GenerateCode.APP.value
-        code = number_generator(enums.GenerateCode.MODULE.value, self.id)
-        Module.objects.filter(id=self.id).update(code=code)
+        if self.code == 'mod':
+            code = enums.GenerateCode.APP.value
+            code = number_generator(enums.GenerateCode.MODULE.value, self.id)
+            Module.objects.filter(id=self.id).update(code=code)
 
 
 class Company(AuditClass):
@@ -95,7 +96,6 @@ class Company(AuditClass):
 
     def save(self, *args, **kwargs):
         super(Company, self).save(*args, **kwargs)
-        # print(self.code)
         if self.code == 'com':
             print(self.code)
             code = enums.GenerateCode.APP.value
@@ -268,9 +268,10 @@ class User(AbstractUser):
 
     def save(self, *args, **kwargs):
         super(User, self).save(*args, **kwargs)
-        code = enums.GenerateCode.APP.value
-        code = number_generator(enums.GenerateCode.USER.value, self.id)
-        User.objects.filter(id=self.id).update(code=code)
+        if self.code == 'acct':
+            code = enums.GenerateCode.APP.value
+            code = number_generator(enums.GenerateCode.USER.value, self.id)
+            User.objects.filter(id=self.id).update(code=code)
 
 
 class UserImage(AuditClass):