Parcourir la source

Feat/add admin check (#11050)

Dr.MerdanBay il y a 5 mois
Parent
commit
41772c325f
1 fichiers modifiés avec 8 ajouts et 0 suppressions
  1. 8 0
      api/models/account.py

+ 8 - 0
api/models/account.py

@@ -108,6 +108,10 @@ class Account(UserMixin, db.Model):
     def is_admin_or_owner(self):
         return TenantAccountRole.is_privileged_role(self._current_tenant.current_role)
 
+    @property
+    def is_admin(self):
+        return TenantAccountRole.is_admin_role(self._current_tenant.current_role)
+
     @property
     def is_editor(self):
         return TenantAccountRole.is_editing_role(self._current_tenant.current_role)
@@ -147,6 +151,10 @@ class TenantAccountRole(enum.StrEnum):
     def is_privileged_role(role: str) -> bool:
         return role and role in {TenantAccountRole.OWNER, TenantAccountRole.ADMIN}
 
+    @staticmethod
+    def is_admin_role(role: str) -> bool:
+        return role and role == TenantAccountRole.ADMIN
+
     @staticmethod
     def is_non_owner_role(role: str) -> bool:
         return role and role in {