fix: multiple rows were found correctly (#2219)

2024-01-26 12:47:42 +08:00
parent 2068ae215e
commit 828822243a
13 changed files with 49 additions and 48 deletions
--- a/api/models/account.py
+++ b/api/models/account.py
@@ -101,7 +101,10 @@ class Account(UserMixin, db.Model):
        return db.session.query(ai).filter(
            ai.account_id == self.id
        ).all()
-
+    # check current_user.current_tenant.current_role in ['admin', 'owner']
+    @property
+    def is_admin_or_owner(self):
+        return self._current_tenant.current_role in ['admin', 'owner']

 class Tenant(db.Model):
    __tablename__ = 'tenants'