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