[civitai] add 'search-images' extractor (#7609)
This commit is contained in:
@@ -389,6 +389,19 @@ class CivitaiSearchExtractor(CivitaiExtractor):
|
|||||||
return self.api.models(params)
|
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):
|
class CivitaiModelsExtractor(CivitaiExtractor):
|
||||||
subcategory = "models"
|
subcategory = "models"
|
||||||
pattern = BASE_PATTERN + r"/models(?:/?\?([^#]+))?(?:$|#)"
|
pattern = BASE_PATTERN + r"/models(?:/?\?([^#]+))?(?:$|#)"
|
||||||
|
|||||||
@@ -361,6 +361,14 @@ __tests__ = (
|
|||||||
"#class": civitai.CivitaiSearchExtractor,
|
"#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",
|
"#url" : "https://civitai.com/user/waomodder",
|
||||||
"#class": civitai.CivitaiUserExtractor,
|
"#class": civitai.CivitaiUserExtractor,
|
||||||
|
|||||||
Reference in New Issue
Block a user