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

* commit 'c563dbcd':
  fix update company

app/entities/models.py

@@ -95,9 +95,12 @@ class Company(AuditClass):
 
     def save(self, *args, **kwargs):
         super(Company, self).save(*args, **kwargs)
-        code = enums.GenerateCode.APP.value
-        code = number_generator(enums.GenerateCode.COMPANY.value, self.id)
-        Company.objects.filter(id=self.id).update(code=code)
+        # print(self.code)
+        if self.code == 'com':
+            print(self.code)
+            code = enums.GenerateCode.APP.value
+            code = number_generator(enums.GenerateCode.COMPANY.value, self.id)
+            Company.objects.filter(id=self.id).update(code=code)
 
 
 class Department(AuditClass):
@@ -953,19 +956,19 @@ class PasswordReset(models.Model):
 class AllowedCompany(models.Model):
     id_number = models.ForeignKey(
         User,
-        on_delete=models.DO_NOTHING,
+        on_delete=models.PROTECT,
         to_field='code',
         related_name='allowed_company_id_number'
     )
     company_pivot = models.ForeignKey(
         Company,
-        on_delete=models.DO_NOTHING,
+        to_field='code',
         related_name='allowed_company_company_pivot',
-        to_field='code'
+        on_delete=models.PROTECT,
     )
     group_pivots = models.ForeignKey(
         Department,
-        on_delete=models.DO_NOTHING,
+        on_delete=models.PROTECT,
         related_name='allowed_company_group_pivots',
         to_field='code'
     )