fix: add marketplace switch

This commit is contained in:
Yeuoly
2024-10-16 14:47:48 +08:00
parent 43ffccc8fd
commit 31cca4a849
6 changed files with 40 additions and 0 deletions

View File

@ -248,6 +248,20 @@ class PluginFetchInstallTaskApi(Resource):
return jsonable_encoder({"task": PluginService.fetch_install_task(tenant_id, task_id)})
class PluginDeleteInstallTaskApi(Resource):
@setup_required
@login_required
@account_initialization_required
def post(self, task_id: str):
user = current_user
if not user.is_admin_or_owner:
raise Forbidden()
tenant_id = user.current_tenant_id
return {"success": PluginService.delete_install_task(tenant_id, task_id)}
class PluginUninstallApi(Resource):
@setup_required
@login_required
@ -277,4 +291,5 @@ api.add_resource(PluginInstallFromMarketplaceApi, "/workspaces/current/plugin/in
api.add_resource(PluginFetchManifestApi, "/workspaces/current/plugin/fetch-manifest")
api.add_resource(PluginFetchInstallTasksApi, "/workspaces/current/plugin/tasks")
api.add_resource(PluginFetchInstallTaskApi, "/workspaces/current/plugin/tasks/<task_id>")
api.add_resource(PluginDeleteInstallTaskApi, "/workspaces/current/plugin/tasks/<task_id>/delete")
api.add_resource(PluginUninstallApi, "/workspaces/current/plugin/uninstall")