mirror of https://github.com/langgenius/dify.git
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from flask import request
|
|
from flask_restx import Resource
|
|
|
|
from controllers.console import api
|
|
from controllers.console.explore.wraps import explore_banner_enabled
|
|
from extensions.ext_database import db
|
|
from models.model import ExporleBanner
|
|
|
|
|
|
class BannerApi(Resource):
|
|
"""Resource for banner list."""
|
|
|
|
@explore_banner_enabled
|
|
def get(self):
|
|
"""Get banner list."""
|
|
language = request.args.get("language", "en-US")
|
|
|
|
# Build base query for enabled banners
|
|
base_query = db.session.query(ExporleBanner).where(ExporleBanner.status == "enabled")
|
|
|
|
# Try to get banners in the requested language
|
|
banners = base_query.where(ExporleBanner.language == language).order_by(ExporleBanner.sort).all()
|
|
|
|
# Fallback to en-US if no banners found and language is not en-US
|
|
if not banners and language != "en-US":
|
|
banners = base_query.where(ExporleBanner.language == "en-US").order_by(ExporleBanner.sort).all()
|
|
# Convert banners to serializable format
|
|
result = []
|
|
for banner in banners:
|
|
banner_data = {
|
|
"id": banner.id,
|
|
"content": banner.content, # Already parsed as JSON by SQLAlchemy
|
|
"link": banner.link,
|
|
"sort": banner.sort,
|
|
"status": banner.status,
|
|
"created_at": banner.created_at.isoformat() if banner.created_at else None,
|
|
}
|
|
result.append(banner_data)
|
|
|
|
return result
|
|
|
|
|
|
api.add_resource(BannerApi, "/explore/banners")
|