DoubaoTeam/dao/class_dao.py

from models.class_model import Classes


def post_classes_dao( c , db ):
    try :
        cc = Classes( **c )
        db.add( cc )
    except :
        db.rollback()
        return False
    else :
        db.commit()
        return True

def update_classes_dao( class_id , db , updates ):
    try :
        db.query( Classes ).filter( Classes.id == class_id , Classes.is_deleted == 0 ) .update( updates )
    except :
        db.rollback()
        return False
    else :
        db.commit()
        return True

def get_classes_dao( class_id, class_no , class_name , page , page_size , db ):
    q = db.query( Classes ).filter( Classes.is_deleted == 0 )
    if class_id :
        q = q.filter( Classes.id == class_id )
    if class_no :
        q = q.filter( Classes.class_no == class_no )
    if class_name :
        q = q.filter( Classes.class_name == class_name )
    total = q.count()
    r = q.offset( ( page-1 ) * page_size  ).limit( page_size ).all()
    return total,[ { 'class_id':i.id
                   , 'class_no':i.class_no
                   , 'class_name':i.class_name
                   , 'start_date':i.start_date
                   , 'end_date':i.end_date
                   , 'created_at':i.created_at
                   , 'update_at':i.update_at
                   }
                      for i in r
                 ]



def delete_classes_dao( class_id , db  ):
    try :
        db.query( Classes ).filter( Classes.id == class_id , Classes.is_deleted == 0 ) .update( Classes.is_deleted == 1 )
    except :
        db.rollback()
        return False
    else :
        db.commit()
        return True