|
|
|
|
|
from flask import request
|
|
|
from flask_restful import Resource
|
|
|
|
|
|
from app import db
|
|
|
from app.models import Setting
|
|
|
from app.utils.response import APIResponse
|
|
|
from app.utils.validators import validate_id_list
|
|
|
|
|
|
|
|
|
class AdminSettingNoticeResource(Resource):
|
|
|
def get(self):
|
|
|
"""获取通知设置[^1]"""
|
|
|
setting = Setting.query.filter_by(alias='notice_setting').first()
|
|
|
if not setting:
|
|
|
return APIResponse.success(data={'users': []})
|
|
|
return APIResponse.success(data={'users': eval(setting.value)})
|
|
|
|
|
|
def post(self):
|
|
|
"""更新通知设置[^1]"""
|
|
|
data = request.json
|
|
|
users = validate_id_list(data.get('users'))
|
|
|
|
|
|
setting = Setting.query.filter_by(alias='notice_setting').first()
|
|
|
if not setting:
|
|
|
setting = Setting(alias='notice_setting')
|
|
|
|
|
|
setting.value = str(users)
|
|
|
setting.serialized = True
|
|
|
db.session.add(setting)
|
|
|
db.session.commit()
|
|
|
return APIResponse.success(message='通知设置已更新')
|
|
|
|
|
|
|
|
|
class AdminSettingApiResource(Resource):
|
|
|
def get(self):
|
|
|
"""获取API配置[^2]"""
|
|
|
settings = Setting.query.filter(Setting.group == 'api_setting').all()
|
|
|
data = {
|
|
|
'api_url': settings[0].value,
|
|
|
'api_key': settings[1].value,
|
|
|
'models': settings[2].value,
|
|
|
'default_model': settings[3].value,
|
|
|
'default_backup': settings[4].value
|
|
|
}
|
|
|
return APIResponse.success(data=data)
|
|
|
|
|
|
def post(self):
|
|
|
"""更新API配置[^2]"""
|
|
|
data = request.json
|
|
|
required_fields = ['api_url', 'api_key', 'models', 'default_model', 'default_backup']
|
|
|
if not all(field in data for field in required_fields):
|
|
|
return APIResponse.error('缺少必要参数', 400)
|
|
|
|
|
|
for alias, value in data.items():
|
|
|
setting = Setting.query.filter_by(alias=alias).first()
|
|
|
if not setting:
|
|
|
setting = Setting(alias=alias, group='api_setting')
|
|
|
setting.value = value
|
|
|
db.session.add(setting)
|
|
|
db.session.commit()
|
|
|
return APIResponse.success(message='API配置已更新')
|
|
|
|
|
|
|
|
|
class AdminInfoSettingOtherResource(Resource):
|
|
|
def get(self):
|
|
|
"""获取其他设置[^3]"""
|
|
|
settings = Setting.query.filter(Setting.group == 'other_setting').all()
|
|
|
data = {
|
|
|
'prompt': settings[0].value,
|
|
|
'threads': int(settings[1].value),
|
|
|
'email_limit': settings[2].value
|
|
|
}
|
|
|
return APIResponse.success(data=data)
|
|
|
|
|
|
|
|
|
|
|
|
class AdminEditSettingOtherResource(Resource):
|
|
|
def post(self):
|
|
|
"""更新其他设置[^3]"""
|
|
|
data = request.json
|
|
|
required_fields = ['prompt', 'threads']
|
|
|
if not all(field in data for field in required_fields):
|
|
|
return APIResponse.error('缺少必要参数', 400)
|
|
|
|
|
|
for alias, value in data.items():
|
|
|
setting = Setting.query.filter_by(alias=alias).first()
|
|
|
if not setting:
|
|
|
setting = Setting(alias=alias, group='other_setting')
|
|
|
setting.value = value
|
|
|
db.session.add(setting)
|
|
|
db.session.commit()
|
|
|
return APIResponse.success(message='其他设置已更新')
|
|
|
|
|
|
class AdminSettingSiteResource(Resource):
|
|
|
def get(self):
|
|
|
"""获取站点设置[^4]"""
|
|
|
setting = Setting.query.filter_by(alias='version').first()
|
|
|
if not setting:
|
|
|
return APIResponse.success(data={'version': 'community'})
|
|
|
return APIResponse.success(data={'version': setting.value})
|
|
|
|
|
|
def post(self):
|
|
|
"""更新站点版本[^4]"""
|
|
|
version = request.json.get('version')
|
|
|
if not version or version not in ['business', 'community']:
|
|
|
return APIResponse.error('版本号无效', 400)
|
|
|
|
|
|
setting = Setting.query.filter_by(alias='version').first()
|
|
|
if not setting:
|
|
|
setting = Setting(alias='version', group='site_setting')
|
|
|
setting.value = version
|
|
|
db.session.add(setting)
|
|
|
db.session.commit()
|
|
|
return APIResponse.success(message='站点版本已更新')
|
|
|
|
|
|
|
