__init__.py 579 B

123456789101112131415161718192021
  1. from controllers.console.app.error import AppUnavailableError
  2. from extensions.ext_database import db
  3. from flask_login import current_user
  4. from models.model import App
  5. from werkzeug.exceptions import NotFound
  6. def _get_app(app_id, mode=None):
  7. app = db.session.query(App).filter(
  8. App.id == app_id,
  9. App.tenant_id == current_user.current_tenant_id,
  10. App.status == 'normal'
  11. ).first()
  12. if not app:
  13. raise NotFound("App not found")
  14. if mode and app.mode != mode:
  15. raise NotFound("The {} app not found".format(mode))
  16. return app