from model.class_management_model import ClassInfo
from sqlalchemy.orm import sessionmaker,Session
from schema.class_management_request import ClassCreate, ClassUpdate

#查询
def get_class_dao(c,db):
    q = db.query(ClassInfo)
    if c.get('class_id'):
        q = q.filter(ClassInfo.class_id == c.get('class_id'))
    if c.get('class_name'):
        q = q.filter(ClassInfo.class_name == c.get('class_name'))
    if c.get('class_teacher'):
        q = q.filter(ClassInfo.class_teacher == c.get('class_teacher'))
    if c.get('class_lecturer'):
        q = q.filter(ClassInfo.class_lecturer == c.get('class_lecturer'))
    if c.get('class_start_time'):
        q = q.filter(ClassInfo.class_start_time == c.get('class_start_time'))
    if c.get('class_end_time'):
        q = q.filter(ClassInfo.class_end_time == c.get('class_end_time'))
    c1 = q.all()
    if c1:
        return[{'班级编号':i.class_id
                ,'班级名称':i.class_name
                ,'班主任':i.class_teacher
                ,'授课老师':i.class_lecturer
                ,'开课时间':i.class_start_time
                ,'结课时间':i.class_end_time
                } for i in c1
               ]


#新增
def add_class_dao(c,db):
    try:
        c1 = ClassInfo(**c)
        db.add(c1)
    except:
        db.rollback()
        return False
    else:
        db.commit()
        return True

#修改
def update_class_dao(cname,update_data,db):
    try:
        rows = db.query(ClassInfo).filter(ClassInfo.class_id == cname).update(update_data)
    except:
        db.rollback()
        return False
    else:
        db.commit()
        return rows
#删除
def delete_class_dao(cid,db):
    try:
        rows = db.query(ClassInfo).filter(ClassInfo.class_id == cid).delete()
    except:
        db.rollback()
        rows = 0
    else:
        db.commit()
    finally:
        return rows