fix: save site setting not work (#14700)

pull/14710/head
非法操作 1 year ago committed by GitHub
parent 7259c0d69f
commit 4125e575af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -2,7 +2,6 @@ from datetime import UTC, datetime
from flask_login import current_user # type: ignore from flask_login import current_user # type: ignore
from flask_restful import Resource, marshal_with, reqparse # type: ignore from flask_restful import Resource, marshal_with, reqparse # type: ignore
from sqlalchemy.orm import Session
from werkzeug.exceptions import Forbidden, NotFound from werkzeug.exceptions import Forbidden, NotFound
from constants.languages import supported_language from constants.languages import supported_language
@ -51,37 +50,35 @@ class AppSite(Resource):
if not current_user.is_editor: if not current_user.is_editor:
raise Forbidden() raise Forbidden()
with Session(db.engine) as session: site = db.session.query(Site).filter(Site.app_id == app_model.id).first()
site = session.query(Site).filter(Site.app_id == app_model.id).first() if not site:
raise NotFound
if not site:
raise NotFound for attr_name in [
"title",
for attr_name in [ "icon_type",
"title", "icon",
"icon_type", "icon_background",
"icon", "description",
"icon_background", "default_language",
"description", "chat_color_theme",
"default_language", "chat_color_theme_inverted",
"chat_color_theme", "customize_domain",
"chat_color_theme_inverted", "copyright",
"customize_domain", "privacy_policy",
"copyright", "custom_disclaimer",
"privacy_policy", "customize_token_strategy",
"custom_disclaimer", "prompt_public",
"customize_token_strategy", "show_workflow_steps",
"prompt_public", "use_icon_as_answer_icon",
"show_workflow_steps", ]:
"use_icon_as_answer_icon", value = args.get(attr_name)
]: if value is not None:
value = args.get(attr_name) setattr(site, attr_name, value)
if value is not None:
setattr(site, attr_name, value) site.updated_by = current_user.id
site.updated_at = datetime.now(UTC).replace(tzinfo=None)
site.updated_by = current_user.id db.session.commit()
site.updated_at = datetime.now(UTC).replace(tzinfo=None)
session.commit()
return site return site

Loading…
Cancel
Save