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

[civitai] fix 'extension' for videos without 'name' and 'mimeType'

Browse Source 
use _video_ext instead of _image_ext
This commit is contained in:
Mike Fährmann
2025-07-19 18:20:16 +02:00
parent 0a245e8d80
commit 923e27f4fd
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gallery_dl/extractor/civitai.py
View File

@@ -163,8 +163,11 @@ class CivitaiExtractor(Extractor):
image["uuid"] = url image["uuid"] = url
name = image.get("name") name = image.get("name")
if not name: if not name:
mime = image.get("mimeType") or self._image_ext if mime := image.get("mimeType"):
name = f"{image.get('id')}.{mime.rpartition('/')[2]}" name = f"{image.get('id')}.{mime.rpartition('/')[2]}"
else:
ext = self._video_ext if video else self._image_ext
name = f"{image.get('id')}.{ext}"
return (f"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA" return (f"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA"
f"/{url}/{quality}/{name}") f"/{url}/{quality}/{name}")
@@ -528,7 +531,6 @@ class CivitaiUserVideosExtractor(CivitaiExtractor):
else: else:
self.params["username"] = text.unquote(user) self.params["username"] = text.unquote(user)
CivitaiExtractor.__init__(self, match) CivitaiExtractor.__init__(self, match)
self._image_ext = "mp4"
images = CivitaiUserImagesExtractor.images images = CivitaiUserImagesExtractor.images