ClaudeandClaude
dbcce79536
重构前保存现有 apps/ 结构基线，供后续各阶段回滚。 Co-Authored-By: Claude <noreply@anthropic.com>
153 lines
4.9 KiB
Python
153 lines
4.9 KiB
Python
from datetime import datetime, timezone
|
||
|
||
from fastapi.logger import logger
|
||
from sqlalchemy import and_
|
||
|
||
from apps.constants.message import PAGE_SIZE
|
||
from apps.forms.position import PositionForm, PositionStatusForm
|
||
from apps.models.position import Position
|
||
from extensions import get_db_session
|
||
from utils import R
|
||
from utils.security import get_user_id
|
||
from utils.request import like_contains
|
||
|
||
|
||
# 查询岗位分页数据
|
||
|
||
def get_position_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(Position).filter(Position.is_delete == 0)
|
||
# 岗位名称
|
||
name = request.query_params.get('name')
|
||
if name:
|
||
query = query.filter(Position.name.like(like_contains(name), escape='\\'))
|
||
# 岗位状态筛选
|
||
status = request.query_params.get('status')
|
||
if status:
|
||
query = query.filter(Position.status == status)
|
||
# 排序
|
||
query = query.order_by(Position.sort.asc())
|
||
# 记录总数
|
||
count = query.count()
|
||
# 分页查询
|
||
position_list = query.limit(limit).offset((page - 1) * limit).all()
|
||
# 若当前页超出末页（数据被并发删除），自动回退到第一页
|
||
if page > 1 and not position_list:
|
||
page = 1
|
||
count = query.count()
|
||
position_list = query.limit(limit).offset(0).all()
|
||
# 实例化结果
|
||
result = []
|
||
# 遍历数据源
|
||
if len(position_list) > 0:
|
||
for item in position_list:
|
||
# 对象转字典
|
||
data = item.to_dict()
|
||
# 加入列表
|
||
result.append(data)
|
||
# 返回结果
|
||
return R.ok(data=result, count=count)
|
||
# 根据ID查询岗位详情
|
||
|
||
def get_position_detail(position_id):
|
||
db = get_db_session()
|
||
# 根据ID查询岗位
|
||
position = db.query(Position).filter(and_(Position.id == position_id, Position.is_delete == 0)).first()
|
||
# 查询结果判空
|
||
if not position:
|
||
return None
|
||
# 对象转字典
|
||
data = position.to_dict()
|
||
# 返回结果
|
||
return data
|
||
# 添加岗位
|
||
|
||
def add_position(request, data: PositionForm):
|
||
db = get_db_session()
|
||
# 唯一性校验
|
||
if db.query(Position).filter(and_(Position.name == data.name, Position.is_delete == 0)).first():
|
||
return R.failed("岗位名称不能重复")
|
||
# 表单数据赋值给对象
|
||
position = Position(**data.model_dump(exclude={'id'}))
|
||
position.create_user = get_user_id(request)
|
||
# 插入数据
|
||
position.save()
|
||
# 返回结果
|
||
return R.ok(msg="添加成功")
|
||
|
||
|
||
# 更新岗位
|
||
|
||
def update_position(request, data: PositionForm):
|
||
db = get_db_session()
|
||
# 记录ID判空
|
||
if not data.id or data.id <= 0:
|
||
return R.failed("记录ID不能为空")
|
||
# 唯一性校验（排除自身）
|
||
if db.query(Position).filter(and_(Position.name == data.name, Position.id != data.id, Position.is_delete == 0)).first():
|
||
return R.failed("岗位名称不能重复")
|
||
|
||
# 根据ID查询记录
|
||
position = db.query(Position).filter(and_(Position.id == data.id, Position.is_delete == 0)).first()
|
||
# 查询结果判空
|
||
if not position:
|
||
return R.failed("记录不存在")
|
||
# 构造更新数据（排除 id）
|
||
updData = data.model_dump(exclude={'id'})
|
||
updData['update_user'] = get_user_id(request)
|
||
updData['update_time'] = datetime.now(tz=timezone.utc)
|
||
result = db.query(Position).filter_by(id=data.id).update(updData)
|
||
# 提交数据
|
||
if not result:
|
||
return R.failed("更新失败")
|
||
# 返回结果
|
||
return R.ok(msg="更新成功")
|
||
# 删除岗位
|
||
|
||
def delete_position(position_id):
|
||
return Position.batch_delete_r(position_id)
|
||
# 设置状态
|
||
|
||
def update_position_status(request, data: PositionStatusForm):
|
||
db = get_db_session()
|
||
# 根据ID查询记录
|
||
position = db.query(Position).filter(and_(Position.id == data.id, Position.is_delete == 0)).first()
|
||
# 查询结果判空
|
||
if not position:
|
||
return R.failed("记录不存在")
|
||
|
||
# 更新记录
|
||
result = db.query(Position).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="设置成功")
|
||
# 获取岗位数据列表
|
||
|
||
def get_position_list():
|
||
db = get_db_session()
|
||
# 查询岗位数据列表
|
||
position_rows = db.query(Position).filter(and_(Position.is_delete == 0, Position.status == 1)).order_by(
|
||
Position.sort.asc()).all()
|
||
# 实例化列表
|
||
position_list = []
|
||
# 遍历数据源
|
||
if position_rows:
|
||
for v in position_rows:
|
||
# 对象转字典
|
||
item = v.to_dict()
|
||
# 加入数组
|
||
position_list.append(item)
|
||
# 返回结果
|
||
return position_list
|