mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-14 20:41:21 +08:00
FamousMai
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
from flask_login import current_user
|
|
from flask_restful import Resource, marshal_with
|
|
from werkzeug.exceptions import Forbidden
|
|
|
|
from controllers.console import api
|
|
from controllers.console.app.wraps import get_app_model
|
|
from controllers.console.wraps import (
|
|
account_initialization_required,
|
|
setup_required,
|
|
)
|
|
from fields.app_fields_extend import (
|
|
recommended_app_list_fields,
|
|
)
|
|
from libs.login import login_required
|
|
from services.account_service_extend import TenantExtendService
|
|
from services.recommended_app_service_extend import RecommendedAppService
|
|
|
|
|
|
# ---------------- start sync app to
|
|
class InstalledSyncAppApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@marshal_with(recommended_app_list_fields)
|
|
def get(self):
|
|
"""Installed app"""
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
return app_service.installed_app_list(current_user.current_tenant_id)
|
|
|
|
|
|
class AppSyncApi(Resource):
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@get_app_model
|
|
def put(self, app_model):
|
|
"""Sync app"""
|
|
|
|
# The role of the current user in the ta table must be admin or owner
|
|
tenant_extend_service = TenantExtendService
|
|
super_admin_id = tenant_extend_service.get_super_admin_id().id
|
|
if super_admin_id != current_user.id:
|
|
raise Forbidden()
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
appId = app_service.sync_recommended_app(app_model.id)
|
|
|
|
return appId, 200
|
|
|
|
@setup_required
|
|
@login_required
|
|
@account_initialization_required
|
|
@get_app_model
|
|
def delete(self, app_model):
|
|
"""Delete sync app"""
|
|
# The role of the current user in the ta table must be admin or owner
|
|
tenant_extend_service = TenantExtendService
|
|
super_admin_id = tenant_extend_service.get_super_admin_id().id
|
|
if super_admin_id != current_user.id:
|
|
raise Forbidden()
|
|
|
|
app_service = RecommendedAppService()
|
|
|
|
app_service.delete_sync_recommended_app(app_model.id)
|
|
|
|
return "", 200
|
|
|
|
|
|
# ----------------start sync app------------------------
|
|
api.add_resource(AppSyncApi, "/apps/<uuid:app_id>/sync")
|
|
api.add_resource(InstalledSyncAppApi, "/installed/apps")
|
|
# ---------------- stop sync app ------------------------
|