add is_admin method

2017-02-10 14:37:52 +08:00
parent 817e5aadae
commit 29aa457a2d
1 changed files with 3 additions and 0 deletions
--- a/account/models.py
+++ b/account/models.py
@@ -51,6 +51,9 @@ class User(AbstractBaseUser):

    objects = UserManager()

+    def is_admin(self):
+        return self.admin_type == AdminType.ADMIN
+
    def is_super_admin(self):
        return self.admin_type == AdminType.SUPER_ADMIN