[Update] 修改外键为org_id

apps/assets/models/asset.py

@@ -225,7 +225,7 @@ class Asset(OrgModelMixin):
         return data
 
     class Meta:
-        unique_together = [('org', 'hostname'),]
+        unique_together = [('org_id', 'hostname')]
         verbose_name = _("Asset")
 
     @classmethod

apps/assets/models/domain.py

@@ -34,7 +34,7 @@ class Domain(OrgModelMixin):
         return random.choice(self.gateways)
 
 
-class Gateway(AssetUser, OrgModelMixin):
+class Gateway(AssetUser):
     SSH_PROTOCOL = 'ssh'
     RDP_PROTOCOL = 'rdp'
     PROTOCOL_CHOICES = (
@@ -52,4 +52,4 @@ class Gateway(AssetUser, OrgModelMixin):
         return self.name
 
     class Meta:
-        unique_together = [('name', 'org')]
+        unique_together = [('name', 'org_id')]

apps/assets/models/label.py

@@ -34,4 +34,4 @@ class Label(models.Model):
 
     class Meta:
         db_table = "assets_label"
-        unique_together = [('name', 'value'),]
+        unique_together = [('name', 'value')]

apps/assets/models/user.py

@@ -69,7 +69,7 @@ class AdminUser(AssetUser):
 
     class Meta:
         ordering = ['name']
-        unique_together = [('name', 'org'),]
+        unique_together = [('name', 'org_id')]
         verbose_name = _("Admin user")
 
     @classmethod
@@ -157,7 +157,7 @@ class SystemUser(AssetUser):
 
     class Meta:
         ordering = ['name']
-        unique_together = [('name', 'org'),]
+        unique_together = [('name', 'org_id')]
         verbose_name = _("System user")
 
     @classmethod

apps/orgs/mixins.py

@@ -27,9 +27,9 @@ class OrgManager(models.Manager):
         if not current_org:
             kwargs['id'] = None
         elif current_org.is_real():
-            kwargs['org'] = current_org
+            kwargs['org_id'] = current_org.id
         elif current_org.is_default():
-            kwargs['org'] = None
+            kwargs['org_id'] = None
         queryset = super(OrgManager, self).get_queryset()
         queryset = queryset.filter(**kwargs)
         # print(kwargs)
@@ -53,13 +53,13 @@ class OrgManager(models.Manager):
 
 
 class OrgModelMixin(models.Model):
-    org = models.ForeignKey('orgs.Organization', on_delete=models.PROTECT, null=True)
+    org_id = models.CharField(max_length=36, null=True)
     objects = OrgManager()
 
     def save(self, *args, **kwargs):
         current_org = get_current_org()
         if current_org and current_org.is_real():
-            self.org = current_org
+            self.org_id = current_org.id
         return super(OrgModelMixin, self).save(*args, **kwargs)
 
     class Meta:

apps/users/models/group.py

@@ -22,7 +22,7 @@ class UserGroup(OrgModelMixin):
 
     class Meta:
         ordering = ['name']
-        unique_together = [('org', 'name'),]
+        unique_together = [('org_id', 'name')]
         verbose_name = _("User group")
 
     @classmethod