From 31513c5c9935bd03eb41febaf34f126e74f54873 Mon Sep 17 00:00:00 2001 From: "VIJAY\\vijay" Date: Wed, 28 Sep 2022 17:25:26 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BA=86parent=5Forg?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/accounts/models.py | 5 +++++ 1 file changed, 5 insertions(+) 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)]