Merge pull request 'main,schemas,dao,api,model consultant.py已更新完毕' (#19) from bx into main
Reviewed-on: #19
This commit is contained in:
@@ -0,0 +1,82 @@
|
||||
from models.consultant import Consultant
|
||||
from schemas.consultant import ConsultantCreate, ConsultantUpdate
|
||||
from sqlalchemy.orm import Session
|
||||
def create_consultant(db:Session, obj_in: ConsultantCreate):
|
||||
try:
|
||||
db_obj = Consultant(**obj_in.model_dump())
|
||||
db.add(db_obj)
|
||||
except:
|
||||
db.rollback()
|
||||
return False
|
||||
else:
|
||||
db.commit()
|
||||
return db_obj
|
||||
# 查询全部可分页(过滤已删除)
|
||||
def get_consultant_list(
|
||||
db: Session,
|
||||
page: int = 1,
|
||||
page_size: int = 20,
|
||||
name: str | None = None,
|
||||
phone: str | None = None
|
||||
):
|
||||
page_size = max(1, min(page_size, 100))
|
||||
skip = (page - 1) * page_size
|
||||
|
||||
query = db.query(Consultant).filter(Consultant.is_deleted == 0)
|
||||
# 有传参数才追加过滤条件
|
||||
if name:
|
||||
query = query.filter(Consultant.consultant_name.like(f"%{name}%"))
|
||||
if phone:
|
||||
query = query.filter(Consultant.phone.like(f"%{phone}%"))
|
||||
|
||||
total = query.count()
|
||||
items = query.offset(skip).limit(page_size).all()
|
||||
|
||||
return {
|
||||
"items": items,
|
||||
"total": total,
|
||||
"page": page,
|
||||
"page_size": page_size
|
||||
}
|
||||
|
||||
|
||||
|
||||
# 根据id查单条
|
||||
def get_consultant_by_id(db:Session, cid: int):
|
||||
return db.query(Consultant).filter(Consultant.id == cid, Consultant.is_deleted == 0 ).first()
|
||||
|
||||
|
||||
# 修改
|
||||
def update_consultant(db: Session, cid: int, update_data: dict):
|
||||
try:
|
||||
db_obj = db.query(Consultant).filter(
|
||||
Consultant.id == cid,
|
||||
Consultant.is_deleted == 0
|
||||
).first()
|
||||
if not db_obj:
|
||||
return None
|
||||
# 循环赋值
|
||||
for k, v in update_data.items():
|
||||
setattr(db_obj, k, v)
|
||||
db.commit()
|
||||
db.refresh(db_obj) # 刷新从数据库拿最新数据(updated_at)
|
||||
return db_obj
|
||||
except Exception:
|
||||
db.rollback()
|
||||
return None
|
||||
|
||||
|
||||
|
||||
# 逻辑删除(只改is_deleted=1)
|
||||
def delete_consultant(db:Session, cid: int):
|
||||
try:
|
||||
db_obj = get_consultant_by_id(db, cid)
|
||||
if not db_obj:
|
||||
return None
|
||||
db_obj.is_deleted = 1
|
||||
except:
|
||||
db.rollback()
|
||||
return None
|
||||
else:
|
||||
db.commit()
|
||||
return db_obj
|
||||
|
||||
Reference in New Issue
Block a user