lordwelch/comictagger
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix page handling

Browse Source
This commit is contained in:
Timmy Welch 2024-06-09 13:40:42 -07:00
parent 44b4857fc3
commit d0312e050b
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comicapi/genericmetadata.py
View File

@ -293,22 +293,20 @@ class GenericMetadata:
# generate a default page list, with the first page marked as the cover
# Create a dictionary of all pages in the metadata
pages = {p["image_index"]: p for p in self.pages}
pages = self.pages
cover_set = False
# Go through each page in the archive
# The indexes should always match up
# It might be a good idea to validate that each page in `pages` is found
for i, filename in enumerate(page_list):
if i not in pages:
pages[i] = ImageMetadata(image_index=i, filename=filename)
else:
if i < len(pages):
pages[i]["filename"] = filename
else:
pages.append(ImageMetadata(image_index=i, filename=filename))
# Check if we know what the cover is
cover_set = pages[i].get("type", None) == PageType.FrontCover or cover_set
self.pages = [p[1] for p in sorted(pages.items())]
# Set the cover to the first image if we don't know what the cover is
if not cover_set:
self.pages[0]["type"] = PageType.FrontCover
@ -322,13 +320,15 @@ class GenericMetadata:
def get_cover_page_index_list(self) -> list[int]:
# return a list of archive page indices of cover pages
if not self.pages:
return [0]
coverlist = []
for p in self.pages:
if "type" in p and p["type"] == PageType.FrontCover:
coverlist.append(int(p["image_index"]))
if len(coverlist) == 0:
coverlist.append(0)
coverlist.append(self.pages[0].get("image_index", 0))
return coverlist

2
comictaggerlib/pagelisteditor.py
View File

@ -275,6 +275,8 @@ class PageListEditor(QtWidgets.QWidget):
def get_first_front_cover(self) -> int:
front_cover = 0
if self.listWidget.count() > 0:
front_cover = int(self.listWidget.item(0).data(QtCore.Qt.ItemDataRole.UserRole).get("image_index", 0))
for i in range(self.listWidget.count()):
item = self.listWidget.item(i)
page_dict: ImageMetadata = item.data(QtCore.Qt.ItemDataRole.UserRole)