calvin.erfmann/TagStudio
1
0
Fork 0
mirror of https://github.com/TagStudioDev/TagStudio.git synced 2026-02-01 15:49:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add success return flag to add_tags_to_entry()

Browse Source
This commit is contained in:
Travis Abendshien
2024-12-31 16:26:01 -08:00
parent 82e06ecc2b
commit a306d68f10
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tagstudio/src/core/library/alchemy/library.py
View File

@@ -934,7 +934,7 @@ class Library:
session.rollback()
return None
def add_tags_to_entry(self, entry_id: int, tag_ids: int | list[int] | set[int]):
def add_tags_to_entry(self, entry_id: int, tag_ids: int | list[int] | set[int]) -> bool:
tag_ids_ = [tag_ids] if isinstance(tag_ids, int) else tag_ids
with Session(self.engine, expire_on_commit=False) as session:
try:
@@ -942,10 +942,11 @@ class Library:
session.add(TagEntry(tag_id=tag_id, entry_id=entry_id))
session.flush()
session.commit()
return True
except IntegrityError as e:
logger.exception(e)
session.rollback()
return None
return False
def save_library_backup_to_disk(self) -> Path:
assert isinstance(self.library_dir, Path)