34 lines
798 B
Python
34 lines
798 B
Python
from sqlalchemy.orm import Session
|
|
|
|
from Model.PostModel import Post
|
|
|
|
|
|
class BaseCrud:
|
|
def __init__(self, db: Session, model):
|
|
self.db = db
|
|
self.model = model
|
|
|
|
def add(self, item):
|
|
self.db.add(item)
|
|
self.db.commit()
|
|
|
|
def delete_by_id(self, item_id):
|
|
self.db.query(self.model).filter_by(id=item_id).delete()
|
|
self.db.commit()
|
|
|
|
def delete(self, item):
|
|
self.db.delete(item)
|
|
self.db.commit()
|
|
|
|
def change(self, item, new_item_dic):
|
|
self.db.query(item).update(new_item_dic)
|
|
self.db.commit()
|
|
|
|
def get_item_by_id(self, item_id):
|
|
return self.db.query(self.model).filter_by(id=item_id).first()
|
|
|
|
|
|
class test(BaseCrud):
|
|
def __init__(self, db: Session):
|
|
BaseCrud.__init__(self, db, Post)
|