Files
DoubaoTeam/dao/class_dao.py
T
2026-09-21 17:29:32 +08:00

57 lines
1.6 KiB
Python

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