mirror of
https://github.com/langgenius/dify.git
synced 2025-11-08 07:23:47 +00:00
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from collections.abc import Callable
|
|
from functools import wraps
|
|
from typing import ParamSpec, TypeVar, Union
|
|
|
|
from controllers.console.app.error import AppNotFoundError
|
|
from extensions.ext_database import db
|
|
from libs.login import current_user
|
|
from models import App, AppMode
|
|
from models.account import Account
|
|
|
|
P = ParamSpec("P")
|
|
R = TypeVar("R")
|
|
|
|
|
|
def _load_app_model(app_id: str) -> App | None:
|
|
assert isinstance(current_user, Account)
|
|
app_model = (
|
|
db.session.query(App)
|
|
.where(App.id == app_id, App.tenant_id == current_user.current_tenant_id, App.status == "normal")
|
|
.first()
|
|
)
|
|
return app_model
|
|
|
|
|
|
def get_app_model(view: Callable[P, R] | None = None, *, mode: Union[AppMode, list[AppMode], None] = None):
|
|
def decorator(view_func: Callable[P, R]):
|
|
@wraps(view_func)
|
|
def decorated_view(*args: P.args, **kwargs: P.kwargs):
|
|
if not kwargs.get("app_id"):
|
|
raise ValueError("missing app_id in path parameters")
|
|
|
|
app_id = kwargs.get("app_id")
|
|
app_id = str(app_id)
|
|
|
|
del kwargs["app_id"]
|
|
|
|
app_model = _load_app_model(app_id)
|
|
|
|
if not app_model:
|
|
raise AppNotFoundError()
|
|
|
|
app_mode = AppMode.value_of(app_model.mode)
|
|
|
|
if mode is not None:
|
|
if isinstance(mode, list):
|
|
modes = mode
|
|
else:
|
|
modes = [mode]
|
|
|
|
if app_mode not in modes:
|
|
mode_values = {m.value for m in modes}
|
|
raise AppNotFoundError(f"App mode is not in the supported list: {mode_values}")
|
|
|
|
kwargs["app_model"] = app_model
|
|
|
|
return view_func(*args, **kwargs)
|
|
|
|
return decorated_view
|
|
|
|
if view is None:
|
|
return decorator
|
|
else:
|
|
return decorator(view)
|