__init__.py 580 B

12345678910111213141516171819202122
  1. from flask_login import current_user
  2. from werkzeug.exceptions import NotFound
  3. from controllers.console.app.error import AppUnavailableError
  4. from extensions.ext_database import db
  5. from models.model import App
  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