from Teachers.model.tea_model import Tea
from datetime import datetime

#增加
def add_tea_dao(t,db):
    try:
        t1 = Tea(**t)
        db.add(t1)
    except:
        db.rollback()
        return False
    else:
        db.commit()
        return True

#更新
def update_tea_dao(id,update_data,db):
    try:
        rows = db.query(Tea).filter(Tea.tea_id == id
                                    ,Tea.is_delete == False).update(update_data)
    except:
        db.rollback()
        return False
    else:
        db.commit()
        return rows

#查询
def get_tea_dao(t,db):
    query = db.query(Tea).filter(Tea.is_delete == False)
    if t.tea_id:
        query = query.filter(Tea.tea_id == t.tea_id)
    if t.tea_name:
        query = query.filter(Tea.tea_name == t.tea_name)
    if t.tea_sex:
        query = query.filter(Tea.tea_sex == t.tea_sex)
    if t.claa_id:
        query = query.filter(Tea.class_id == t.claa_id)
    if t.class_sub:
        query = query.filter(Tea.class_sub == t.class_sub)
    r = query.all()
    return [{'tea_id':i.tea_id,'tea_name':i.tea_name,'class_id':i.class_id,'class_sub':i.class_sub,'tea.sex':i.tea_sex} for i in r]

#删除
def delete_tea_dao(id,db):
    try:
        row = db.query(Tea).filter(Tea.tea_id == id
                                   ,Tea.is_delete == False
                                   ).update({'is_delete':True,'delete_time':datetime.now()})
    except:
        db.rollback()
        row = 0
        return row
    else:
        db.commit()
        return row