import os
|
import shutil
|
import uuid
|
from datetime import datetime
|
|
import yaml
|
from fastapi import UploadFile
|
|
from Log import logger
|
from app.api import pwd_context
|
from app.config.const import SYSTEM_ID, ENV_CONF_PATH, APP_STATIC_PATH
|
from app.models.system import SystemDataModel
|
|
|
async def services_get_system_data(db):
|
system = db.query(SystemDataModel).filter_by(id=SYSTEM_ID).first()
|
if not system:
|
with open(os.path.join(ENV_CONF_PATH, "system.yaml"), 'r', encoding='utf-8') as file:
|
# 加载JSON数据
|
config = yaml.safe_load(file)
|
system = SystemDataModel(id=SYSTEM_ID, title=config["smart_system"]["title"], desc=config["smart_system"]["desc"])
|
db.add(system)
|
db.commit()
|
db.refresh(system)
|
return system.to_dict()
|
|
|
async def services_update_system_data(db, title, desc, logo):
|
try:
|
shutil.move(os.path.join(APP_STATIC_PATH, logo), os.path.join(APP_STATIC_PATH, "logo.png"))
|
system = db.query(SystemDataModel).filter_by(id=SYSTEM_ID).first()
|
system.title = title
|
system.desc = desc
|
system.updated_at = datetime.now()
|
db.commit()
|
db.refresh(system)
|
return system.to_dict()
|
except Exception as e:
|
logger.error(e)
|
return {}
|
|
|
async def service_upload_logo_image(file: UploadFile):
|
file_name = str(uuid.uuid4())
|
try:
|
save_path = os.path.join(APP_STATIC_PATH, file_name)
|
# 将上传的文件保存到本地
|
with open(save_path, "wb") as buffer:
|
shutil.copyfileobj(file.file, buffer)
|
file.file.close()
|
return file_name
|
except Exception as e:
|
logger.error(f"保存失败: {str(e)}")
|
return ""
|
