浏览代码

fix: account check in runtime (#569)

tags/0.3.9
John Wang 2 年前
父节点
当前提交
24f3456990
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6
    1
      api/app.py

+ 6
- 1
api/app.py 查看文件

import os import os
from datetime import datetime from datetime import datetime


from werkzeug.exceptions import Forbidden

if not os.environ.get("DEBUG") or os.environ.get("DEBUG").lower() != 'true': if not os.environ.get("DEBUG") or os.environ.get("DEBUG").lower() != 'true':
from gevent import monkey from gevent import monkey
monkey.patch_all() monkey.patch_all()
import core import core
from config import Config, CloudEditionConfig from config import Config, CloudEditionConfig
from commands import register_commands from commands import register_commands
from models.account import TenantAccountJoin
from models.account import TenantAccountJoin, AccountStatus
from models.model import Account, EndUser, App from models.model import Account, EndUser, App


import warnings import warnings
account = db.session.query(Account).filter(Account.id == account_id).first() account = db.session.query(Account).filter(Account.id == account_id).first()


if account: if account:
if account.status == AccountStatus.BANNED.value or account.status == AccountStatus.CLOSED.value:
raise Forbidden('Account is banned or closed.')

workspace_id = session.get('workspace_id') workspace_id = session.get('workspace_id')
if workspace_id: if workspace_id:
tenant_account_join = db.session.query(TenantAccountJoin).filter( tenant_account_join = db.session.query(TenantAccountJoin).filter(

正在加载...
取消
保存