def add_model(self, model: BaseModel):
try:
self._session.add(model)
except IntegrityError as e:
raise DBIntegrityException(e)
except DataError as e:
raise DBDataException(e)
def commit_session(self, need_close: bool = False) -> DBEmp:
try:
self._session.commit()
except IntegrityError as e:
raise DBIntegrityException(e)
except DataError as e:
raise DBDataException(e)
if need_close:
self.closeSes()
def add_model(self, model: BaseModel):
try:
# экземпляр класса Session имеет встроенный метод add
# для добавления объекта в базу данных
self._session.add(model)
except IntegrityError as error:
raise DBIntegrityException(error)
except DataError as error:
raise DBDataException(error)
def commit_session(self, need_close: bool = False):
try:
self._session.commit()
except IntegrityError as error:
raise DBIntegrityException(error)
except DataError as error:
raise DBDataException(error)
if need_close:
self.close_session()
def add_model_video(self, model: BaseModel):
try:
self._session.add(model)
self.commit_session()
m = self._session.query(DBVideoQueue).filter(
DBVideoQueue.video_name == model.video_name).first()
return m.id
except IntegrityError as e:
raise DBIntegrityException(e)
except DataError as e:
raise DBDataException(e)
def add_model_progress(self, model: BaseModel):
try:
self._session.add(model)
self.commit_session()
m = self._session.query(DBProgress).filter(
DBProgress.id == model.queue_id).first()
return m.id
except IntegrityError as e:
raise DBIntegrityException(e)
except DataError as e:
raise DBDataException(e)
