# up org

apps/accounts/models.py

@@ -20,6 +20,7 @@ class User(AbstractUser):
     department = models.CharField(max_length=20, null=True, blank=True, default='', verbose_name='部门')
     position = models.CharField(max_length=50, null=True, blank=True, default='', verbose_name='职位')
     introduction = models.TextField(verbose_name='简介')
+    org = models.CharField(max_length=100, null=True, blank=True, verbose_name='组织')
     role = models.CharField(max_length=20, default='admin', choices=ROLES, verbose_name='角色')
     avatar = models.ImageField(upload_to=avatar_path,
                                default='https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif', null=True, blank=True)

apps/mosquito/models.py

@@ -66,4 +66,22 @@ class DevicePostStatistic(models.Model):
         verbose_name_plural = verbose_name
 
     def __str__(self):
-        return '{0}: {1}'.format(self.device_id, self.total)
+        return '{0}: {1}'.format(self.device_id, self.total)
+
+
+class Org(models.Model):
+    name = models.CharField(max_length=100, verbose_name='组织名称')
+    code = models.CharField(max_length=10, verbose_name='组织代码')
+    desc = models.CharField(max_length=100, blank='True', verbose_name='描述')
+
+    class Meta:
+        verbose_name = '客户组织'
+        verbose_name_plural = verbose_name
+
+    def __str__(self):
+        return self.name
+
+
+class DeviceInfo(models.Model):
+    device_id = models.CharField(max_length=100, verbose_name='设备ID')
+    org = models.ForeignKey(Org, verbose_name='所在组织', on_delete='PROTECT')