mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-24 23:28:32 +08:00
31 lines
831 B
Python
31 lines
831 B
Python
from flask_restful import Resource, marshal_with
|
|
from werkzeug.exceptions import Forbidden
|
|
|
|
from controllers.common import fields
|
|
from controllers.service_api import api
|
|
from controllers.service_api.wraps import validate_app_token
|
|
from extensions.ext_database import db
|
|
from models.account import TenantStatus
|
|
from models.model import App, Site
|
|
|
|
|
|
class AppSiteApi(Resource):
|
|
"""Resource for app sites."""
|
|
|
|
@validate_app_token
|
|
@marshal_with(fields.site_fields)
|
|
def get(self, app_model: App):
|
|
"""Retrieve app site info."""
|
|
site = db.session.query(Site).filter(Site.app_id == app_model.id).first()
|
|
|
|
if not site:
|
|
raise Forbidden()
|
|
|
|
if app_model.tenant.status == TenantStatus.ARCHIVE:
|
|
raise Forbidden()
|
|
|
|
return site
|
|
|
|
|
|
api.add_resource(AppSiteApi, "/site")
|