Underground-Digital/Workflow-Engine

1from flask_restful import marshal_with2 3from controllers.common import fields4from controllers.common import helpers as controller_helpers5from controllers.console import api6from controllers.console.app.error import AppUnavailableError7from controllers.console.explore.wraps import InstalledAppResource8from models.model import AppMode, InstalledApp9from services.app_service import AppService10 11 12class AppParameterApi(InstalledAppResource):13    """Resource for app variables."""14 15    @marshal_with(fields.parameters_fields)16    def get(self, installed_app: InstalledApp):17        """Retrieve app parameters."""18        app_model = installed_app.app19 20        if app_model is None:21            raise AppUnavailableError()22 23        if app_model.mode in {AppMode.ADVANCED_CHAT.value, AppMode.WORKFLOW.value}:24            workflow = app_model.workflow25            if workflow is None:26                raise AppUnavailableError()27 28            features_dict = workflow.features_dict29            user_input_form = workflow.user_input_form(to_old_structure=True)30        else:31            app_model_config = app_model.app_model_config32            if app_model_config is None:33                raise AppUnavailableError()34 35            features_dict = app_model_config.to_dict()36 37            user_input_form = features_dict.get("user_input_form", [])38 39        return controller_helpers.get_parameters_from_feature_dict(40            features_dict=features_dict, user_input_form=user_input_form41        )42 43 44class ExploreAppMetaApi(InstalledAppResource):45    def get(self, installed_app: InstalledApp):46        """Get app meta"""47        app_model = installed_app.app48        return AppService().get_app_meta(app_model)49 50 51api.add_resource(52    AppParameterApi, "/installed-apps/<uuid:installed_app_id>/parameters", endpoint="installed_app_parameters"53)54api.add_resource(ExploreAppMetaApi, "/installed-apps/<uuid:installed_app_id>/meta", endpoint="installed_app_meta")55