46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from model.classinfo import ClassInfo
|
|
from scheme.classinfo_scheme import ClassCreate, ClassUpdate
|
|
|
|
# 增删改查
|
|
def create_class(db,obj:ClassCreate):
|
|
new_class=ClassInfo(class_name=obj.class_name,start_time=obj.start_time,statu=1)
|
|
db.add(new_class)
|
|
db.commit()
|
|
db.refresh(new_class)
|
|
return new_class
|
|
# 查询单个班级
|
|
def get_class(db,class_id=None,class_name=None):
|
|
u1=db.query(ClassInfo).filter(ClassInfo.statu==1)
|
|
if class_id is not None:
|
|
u1=u1.filter(ClassInfo.id==class_id)
|
|
if class_name is not None:
|
|
u1=u1.filter(ClassInfo.class_name.like(f"%{class_name}%"))
|
|
if class_id is not None:
|
|
return u1.first()
|
|
else:
|
|
return u1.all()
|
|
# 查询所有班级
|
|
def get_all_class(db):
|
|
return db.query(ClassInfo).filter(ClassInfo.statu==1).all()
|
|
# 修改班级`
|
|
def update_class(db,class_id:int,obj:ClassUpdate):
|
|
u2=get_class(db,class_id)
|
|
if u2:
|
|
u2.class_name=obj.class_name
|
|
u2.start_time=obj.start_time
|
|
db.commit()
|
|
db.refresh(u2)
|
|
return u2
|
|
else:
|
|
return None
|
|
# 删除班级
|
|
def delete_class(db,class_id:int):
|
|
u3=get_class(db,class_id)
|
|
if u3:
|
|
u3.statu=0
|
|
db.commit()
|
|
db.refresh(u3)
|
|
return u3
|
|
else:
|
|
return None
|