github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[civitai] make metadata extraction non-fatal (#7562)

Browse Source
This commit is contained in:
Mike Fährmann
2025-05-22 19:18:53 +02:00
parent f5b8c25559
commit b9f077a1c5
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
gallery_dl/extractor/civitai.py
View File

@@ -181,14 +181,20 @@ class CivitaiExtractor(Extractor):
"types", "fileFormats"})
def _extract_meta_generation(self, image):
return self.api.image_generationdata(image["id"])
try:
return self.api.image_generationdata(image["id"])
except Exception as exc:
return self.log.debug("", exc_info=exc)
def _extract_meta_version(self, item, is_post=True):
version_id = self._extract_version_id(item, is_post)
if version_id is None:
return None, None
version = self.api.model_version(version_id).copy()
return version.pop("model", None), version
try:
version_id = self._extract_version_id(item, is_post)
if version_id:
version = self.api.model_version(version_id).copy()
return version.pop("model", None), version
except Exception as exc:
self.log.debug("", exc_info=exc)
return None, None
def _extract_version_id(self, item, is_post=True):
version_id = item.get("modelVersionId")