from database.tcontent.model import TContent def create_content(db, content: TContent): db.add(content) db.commit() db.refresh(content) return content def get_content_by_id(db, content_id: int): return db.query(TContent).filter(TContent.id == content_id).first() def update_content(db, content_id: int, updates: dict): content = db.query(TContent).filter(TContent.id == content_id).first() if content: for key, value in updates.items(): setattr(content, key, value) db.commit() db.refresh(content) return content def delete_content(db, content_id: int): content = db.query(TContent).filter(TContent.id == content_id).first() if content: db.delete(content) db.commit() return content def drop_table(db): TContent.__table__.drop(db.get_bind(), checkfirst=True) def create_table(db): TContent.__table__.create(db.get_bind(), checkfirst=True)