[civitai] add extractors for global 'models' and 'images' (#6310)

2024-10-16 21:34:11 +02:00
parent a37ac3c748
commit 69a75b1de2
4 changed files with 57 additions and 20 deletions
--- a/test/results/civitai.py
+++ b/test/results/civitai.py
@@ -121,12 +121,22 @@ __tests__ = (

 {
    "#url"  : "https://civitai.com/tag/mecha",
-    "#class": civitai.CivitaiTagModelsExtractor,
+    "#class": civitai.CivitaiTagExtractor,
 },

 {
    "#url"  : "https://civitai.com/images?tags=482",
-    "#class": civitai.CivitaiTagImagesExtractor,
+    "#class": civitai.CivitaiImagesExtractor,
+},
+
+{
+    "#url"  : "https://civitai.com/images?modelVersionId=786644",
+    "#class": civitai.CivitaiImagesExtractor,
+},
+
+{
+    "#url"  : "https://civitai.com/models",
+    "#class": civitai.CivitaiModelsExtractor,
 },

 {