setting return value of adding link to db

This commit is contained in:
Thomas Peetz
2025-01-27 15:20:35 +01:00
parent fe89cc6e0f
commit 93c7498a83
@@ -263,7 +263,7 @@ class KontorDB:
return existing_ids return existing_ids
def add_entry(self, table_name: str, update_item: dict): def add_entry(self, table_name: str, update_item: dict):
# self.log.info("add entry to table %s with %s", table_name, update_item) self.log.debug("add entry to table %s with %s", table_name, update_item)
__session__ = sessionmaker(self.engine) __session__ = sessionmaker(self.engine)
with __session__() as session: with __session__() as session:
add_item = self.registry[table_name]() add_item = self.registry[table_name]()
@@ -307,7 +307,7 @@ class KontorDB:
try: try:
session.add(media_file) session.add(media_file)
session.commit() session.commit()
result['added'] = media_file result['added'] = {'url': media_file.url, 'title': media_file.title, 'review': media_file.review, 'download': media_file.should_download}
except IntegrityError as error: except IntegrityError as error:
session.rollback() session.rollback()
result['error'] = error.orig result['error'] = error.orig