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)