diff --git a/apps/accounts/models.py b/apps/accounts/models.py index 4c1af16..7ff3ddb 100644 --- a/apps/accounts/models.py +++ b/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) diff --git a/apps/mosquito/models.py b/apps/mosquito/models.py index 20c05d6..62f8ffc 100644 --- a/apps/mosquito/models.py +++ b/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) \ No newline at end of file + 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')