diff --git a/apps/accounts/models.py b/apps/accounts/models.py index 38cb039..4253431 100644 --- a/apps/accounts/models.py +++ b/apps/accounts/models.py @@ -25,6 +25,7 @@ class User(AbstractUser): introduction = models.TextField(null=True, blank=True, verbose_name='简介') # org = models.CharField(max_length=100, null=True, blank=True, verbose_name='组织') org = models.ForeignKey(Org, verbose_name='所在组织', on_delete='PROTECT') + parent_org = models.PositiveIntegerField(verbose_name="父组织ID", null=True, blank=True) 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) @@ -49,3 +50,7 @@ class User(AbstractUser): def __str__(self): return self.username + + def get_child(self): + print(self.org) + return [u.id for u in User.objects.filter(parent_org=self.org)]