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

[civitai] improve '_extract_version_id()'

Browse Source 
also check for 'modelVersionIdsManual'
This commit is contained in:
Mike Fährmann
2025-07-19 10:54:23 +02:00
parent 272df323f3
commit 0a245e8d80
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
gallery_dl/extractor/civitai.py
View File

@@ -235,12 +235,11 @@ class CivitaiExtractor(Extractor):
return None, None return None, None
def _extract_version_id(self, item, is_post=True): def _extract_version_id(self, item, is_post=True):
version_id = item.get("modelVersionId") if version_id := item.get("modelVersionId"):
if version_id:
return version_id return version_id
if version_ids := item.get("modelVersionIds"):
version_ids = item.get("modelVersionIds") return version_ids[0]
if version_ids: if version_ids := item.get("modelVersionIdsManual"):
return version_ids[0] return version_ids[0]
if is_post: if is_post: