[civitai] do not restrict file types by default (#8053)

This commit is contained in:
Mike Fährmann
2025-08-16 10:17:51 +02:00
parent d1a9fec2a8
commit c2b11dd475
2 changed files with 12 additions and 1 deletions

View File

@@ -459,6 +459,7 @@ class CivitaiImagesExtractor(CivitaiExtractor):
def images(self):
params = self._parse_query(self.groups[0])
params["types"] = ("image",)
return self.api.images(params)
@@ -521,6 +522,7 @@ class CivitaiUserImagesExtractor(CivitaiExtractor):
def __init__(self, match):
user, query = match.groups()
self.params = self._parse_query(query)
self.params["types"] = ("image",)
if self.params.get("section") == "reactions":
self.subcategory = "reactions-images"
self.images = self._image_reactions
@@ -708,7 +710,6 @@ class CivitaiTrpcAPI():
"useIndex" : True,
"period" : "AllTime",
"sort" : "Newest",
"types" : ("image",),
"withMeta" : False, # Metadata Only
"fromPlatform" : False, # Made On-Site
"browsingLevel": self.nsfw,

View File

@@ -363,6 +363,16 @@ __tests__ = (
"version": None,
},
{
"#url" : "https://civitai.com/posts/20403514",
"#comment" : "mixed image & video (#8053)",
"#class" : civitai.CivitaiPostExtractor,
"#results" : (
"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/62dbebbe-48e9-4232-b4da-33c70d19683d/original=true/91967659.png",
"https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/e5786ddc-29b3-4a69-aec9-fba4dc2c78b5/quality=100/91967639.webm",
),
},
{
"#url" : "https://civitai.com/tag/mecha",
"#class": civitai.CivitaiTagExtractor,