From 41772c325f71dc1a6a24f7276c5e260c431eb673 Mon Sep 17 00:00:00 2001 From: "Dr.MerdanBay" <110794035+KMerdan@users.noreply.github.com> Date: Mon, 25 Nov 2024 12:11:00 +0900 Subject: [PATCH] Feat/add admin check (#11050) --- api/models/account.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/api/models/account.py b/api/models/account.py index 6684e8dd6e..18be4be036 100644 --- a/api/models/account.py +++ b/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 {