[civitai] add 'search-images' extractor (#7609)

This commit is contained in:
Mike Fährmann
2025-06-01 09:57:12 +02:00
parent 55f26a558a
commit dc467c7790
2 changed files with 21 additions and 0 deletions

View File

@@ -389,6 +389,19 @@ class CivitaiSearchExtractor(CivitaiExtractor):
return self.api.models(params)
class CivitaiSearchImagesExtractor(CivitaiExtractor):
subcategory = "search-images"
pattern = BASE_PATTERN + r"/search/images\?([^#]+)"
example = "https://civitai.com/search/images?query=QUERY"
def images(self):
params = self._parse_query(self.groups[0])
return CivitaiSearchAPI(self).search(
params.get("query"),
params.get("sortBy") or "images_v6",
self.api.nsfw)
class CivitaiModelsExtractor(CivitaiExtractor):
subcategory = "models"
pattern = BASE_PATTERN + r"/models(?:/?\?([^#]+))?(?:$|#)"

View File

@@ -361,6 +361,14 @@ __tests__ = (
"#class": civitai.CivitaiSearchExtractor,
},
{
"#url" : "https://civitai.com/search/images?sortBy=images_v6&query=Voynich",
"#class" : civitai.CivitaiSearchImagesExtractor,
"#options": {"nsfw": False},
"#count" : range(150, 200),
"#archive": False,
},
{
"#url" : "https://civitai.com/user/waomodder",
"#class": civitai.CivitaiUserExtractor,