chore: 基线提交(重构前快照)
重构前保存现有 apps/ 结构基线,供后续各阶段回滚。 Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,184 @@
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from fastapi.logger import logger
|
||||
from sqlalchemy import and_
|
||||
|
||||
|
||||
from apps.constants.constants import AD_TYPE_LIST
|
||||
from apps.constants.message import PAGE_SIZE
|
||||
from apps.forms.ad import AdForm, AdStatusForm
|
||||
from apps.models.ad import Ad
|
||||
from apps.models.ad_sort import AdSort
|
||||
from config import FASTAPI_IMAGE_URL
|
||||
from extensions import get_db_session
|
||||
from utils import R
|
||||
from utils.rich_text import get_file_url, save_file, save_content
|
||||
from utils.request import like_contains
|
||||
from utils.security import get_user_id
|
||||
|
||||
|
||||
# 查询广告分页数据
|
||||
|
||||
def get_ad_page(request):
|
||||
db = get_db_session()
|
||||
# 页码
|
||||
page = int(request.query_params.get("pageNo", 1))
|
||||
# 每页数
|
||||
limit = int(request.query_params.get("pageSize", PAGE_SIZE))
|
||||
# 实例化查询对象
|
||||
query = db.query(Ad).filter(Ad.is_delete == 0)
|
||||
# 广告标题
|
||||
title = request.query_params.get('title')
|
||||
if title:
|
||||
query = query.filter(Ad.title.like(like_contains(title), escape='\\'))
|
||||
# 广告类型:1-图片 2-文字 3-视频 4-推荐
|
||||
type = request.query_params.get('type')
|
||||
if type:
|
||||
query = query.filter(Ad.type == type)
|
||||
# 广告状态
|
||||
status = request.query_params.get('status')
|
||||
if status:
|
||||
query = query.filter(Ad.status == status)
|
||||
# 排序
|
||||
query = query.order_by(Ad.sort.desc())
|
||||
# 记录总数
|
||||
count = query.count()
|
||||
# 分页查询
|
||||
ad_list = query.limit(limit).offset((page - 1) * limit).all()
|
||||
# 若当前页超出末页(数据被并发删除),自动回退到第一页
|
||||
if page > 1 and not ad_list:
|
||||
page = 1
|
||||
count = query.count()
|
||||
ad_list = query.limit(limit).offset(0).all()
|
||||
# 实例化结果
|
||||
result = []
|
||||
# 遍历数据源
|
||||
if len(ad_list) > 0:
|
||||
for item in ad_list:
|
||||
# 对象转字典
|
||||
data = item.to_dict()
|
||||
# 广告类型
|
||||
data['type_name'] = AD_TYPE_LIST.get(item.type)
|
||||
# 广告封面
|
||||
data['cover'] = get_file_url(item.cover) if item.cover else None
|
||||
# 查询广告位信息
|
||||
ad_sort = db.query(AdSort).filter(and_(AdSort.id == item.sort_id, AdSort.is_delete == 0)).first()
|
||||
# 广告位描述
|
||||
data['sort_desc'] = ad_sort.name + "=>" + str(ad_sort.loc_id) if ad_sort else None
|
||||
# 加入列表
|
||||
result.append(data)
|
||||
# 返回结果
|
||||
return R.ok(data=result, count=count)
|
||||
# 根据广告ID查询详情
|
||||
|
||||
def get_ad_detail(ad_id):
|
||||
db = get_db_session()
|
||||
# 根据广告ID查询信息
|
||||
ad = db.query(Ad).filter(and_(Ad.id == ad_id, Ad.is_delete == 0)).first()
|
||||
# 查询结果为空判断
|
||||
if not ad:
|
||||
return None
|
||||
# 对象转字典
|
||||
data = ad.to_dict()
|
||||
# 广告类型描述
|
||||
data['type_name'] = AD_TYPE_LIST.get(ad.type)
|
||||
# 广告封面
|
||||
data['cover'] = get_file_url(ad.cover) if ad.cover else ""
|
||||
# # 富文本内容
|
||||
# content = ad.content.replace("[IMG_URL]", FASTAPI_IMAGE_URL)
|
||||
# data['content'] = content
|
||||
# 返回结果
|
||||
return data
|
||||
# 添加广告
|
||||
|
||||
def add_ad(request, data: AdForm):
|
||||
db = get_db_session()
|
||||
# 广告封面
|
||||
cover = data.cover
|
||||
if data.type == 1 and cover:
|
||||
cover = save_file(cover, "ad")
|
||||
else:
|
||||
cover = None
|
||||
|
||||
# # 广告标题
|
||||
# title = data.title
|
||||
# # 广告内容
|
||||
# content = data.content
|
||||
# # 处理富文本内容
|
||||
# content = save_content(content, title, "ad")
|
||||
|
||||
# 表单数据赋值给对象
|
||||
ad = Ad(**data.model_dump(exclude={'id'}))
|
||||
ad.cover = cover
|
||||
ad.create_user = get_user_id(request)
|
||||
# 插入数据
|
||||
ad.save()
|
||||
# 返回结果
|
||||
return R.ok(msg="添加成功")
|
||||
|
||||
|
||||
# 更新广告
|
||||
|
||||
def update_ad(request, data: AdForm):
|
||||
db = get_db_session()
|
||||
# 记录ID判空
|
||||
if not data.id or data.id <= 0:
|
||||
return R.failed("记录ID不能为空")
|
||||
# 根据ID查询记录
|
||||
ad = db.query(Ad).filter(and_(Ad.id == data.id, Ad.is_delete == 0)).first()
|
||||
# 查询结果判空
|
||||
if not ad:
|
||||
return R.failed("记录不存在")
|
||||
|
||||
# 广告封面
|
||||
cover = data.cover
|
||||
if data.type == 1 and cover:
|
||||
cover = save_file(cover, "ad")
|
||||
else:
|
||||
cover = None
|
||||
|
||||
# # 广告标题
|
||||
# title = data.title
|
||||
# # 广告内容
|
||||
# content = data.content
|
||||
# # 处理富文本内容
|
||||
# content = save_content(content, title, "ad")
|
||||
|
||||
# 构造更新数据(排除 id)
|
||||
updData = data.model_dump(exclude={'id'})
|
||||
updData['cover'] = cover
|
||||
updData['update_user'] = get_user_id(request)
|
||||
updData['update_time'] = datetime.now(tz=timezone.utc)
|
||||
# 更新记录
|
||||
result = db.query(Ad).filter_by(id=data.id).update(updData)
|
||||
# 提交数据
|
||||
# 返回结果
|
||||
if not result:
|
||||
return R.failed("更新失败")
|
||||
# 返回结果
|
||||
return R.ok(msg="更新成功")
|
||||
# 删除广告
|
||||
|
||||
def delete_ad(ad_id):
|
||||
return Ad.batch_delete_r(ad_id)
|
||||
# 设置状态
|
||||
|
||||
def update_ad_status(request, data: AdStatusForm):
|
||||
db = get_db_session()
|
||||
# 根据ID查询记录
|
||||
ad = db.query(Ad).filter(and_(Ad.id == data.id, Ad.is_delete == 0)).first()
|
||||
# 查询结果判空
|
||||
if not ad:
|
||||
return R.failed("记录不存在")
|
||||
|
||||
# 更新记录
|
||||
result = db.query(Ad).filter_by(id=data.id).update({
|
||||
"status": data.status,
|
||||
"update_user": get_user_id(request),
|
||||
"update_time": datetime.now(tz=timezone.utc)
|
||||
})
|
||||
# 提交数据
|
||||
if not result:
|
||||
return R.failed("设置失败")
|
||||
# 返回结果
|
||||
return R.ok(msg="设置成功")
|
||||
Reference in New Issue
Block a user