diff --git a/account/models.py b/account/models.py index 807df44..0e53c36 100644 --- 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