浏览代码

fix: bypass admin users to use dataset api with API key (#2072)

Bowen Liang 1 年之前
父节点
当前提交
e2c89a9487
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      api/controllers/service_api/wraps.py

+ 1 - 1
api/controllers/service_api/wraps.py

@@ -75,7 +75,7 @@ def validate_dataset_token(view=None):
             tenant_account_join = db.session.query(Tenant, TenantAccountJoin) \
                 .filter(Tenant.id == api_token.tenant_id) \
                 .filter(TenantAccountJoin.tenant_id == Tenant.id) \
-                .filter(TenantAccountJoin.role == 'owner') \
+                .filter(TenantAccountJoin.role.in_(['owner', 'admin'])) \
                 .one_or_none()
             if tenant_account_join:
                 tenant, ta = tenant_account_join