32 lines
979 B
Python
32 lines
979 B
Python
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) |