fdfcaa00b9
forward query parameters https://github.com/mikf/gallery-dl/issues/8061#issuecomment-3266326296
185 lines
6.4 KiB
Python
185 lines
6.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
from gallery_dl.extractor import tungsten
|
|
|
|
|
|
__tests__ = (
|
|
{
|
|
"#url" : "https://tungsten.run/post/VuXCdyw9vohCAFyN",
|
|
"#class" : tungsten.TungstenPostExtractor,
|
|
"#results" : "https://api.tungsten.run/v1/upload/aPg2RYeezA9wZ8uQ",
|
|
|
|
"comment_count" : int,
|
|
"created_at" : "2024-12-25T00:05:39Z",
|
|
"date" : "dt:2024-12-25 00:05:39",
|
|
"description" : None,
|
|
"extension" : "webp",
|
|
"filename" : "aPg2RYeezA9wZ8uQ",
|
|
"generation_postprocessing": None,
|
|
"generation_type": "flux1d",
|
|
"is_tweakable" : True,
|
|
"like_count" : range(50, 200),
|
|
"model_versions" : [{
|
|
"cover_url" : "https://api.tungsten.run/v1/model_version_cover/48M6hN3HsQ/1726596449",
|
|
"created_at" : "2024-09-17T18:05:05Z",
|
|
"fullname" : "Flux.1 [Dev] - v1",
|
|
"model_base" : "flux1d",
|
|
"model_name" : "Flux.1 [Dev]",
|
|
"model_resolved_nsfw_score": 0,
|
|
"model_type" : "checkpoint",
|
|
"model_uuid" : "9fThn2WaVP",
|
|
"name" : "v1",
|
|
"sm_cover_url": "https://api.tungsten.run/v1/model_version_cover/48M6hN3HsQ/1726596449",
|
|
"status" : "approved",
|
|
"uuid" : "48M6hN3HsQ",
|
|
"extra" : {"dtype": "fp16"},
|
|
}],
|
|
"nsfw" : False,
|
|
"original_url" : "https://api.tungsten.run/v1/upload/aPg2RYeezA9wZ8uQ",
|
|
"original_width" : 832,
|
|
"original_height": 1216,
|
|
"rating" : None,
|
|
"resized_url" : "https://api.tungsten.run/v1/upload/jRJMBAD5zG3NFar6",
|
|
"resized_width" : 832,
|
|
"resized_height" : 1216,
|
|
"title" : "Moon",
|
|
"uuid" : "VuXCdyw9vohCAFyN",
|
|
"view_count" : int,
|
|
"tags" : [
|
|
"art",
|
|
"moon",
|
|
"surreal",
|
|
],
|
|
"generation_data": {
|
|
"cfg" : 4,
|
|
"height" : 1216,
|
|
"img2img" : None,
|
|
"loras" : [],
|
|
"model_version_uuid": "48M6hN3HsQ",
|
|
"num_images" : 2,
|
|
"postprocess": [],
|
|
"prompt" : "low angle photograph of a woman, standing on top of a small mountain, hands in the air holding up the full moon late at night, long brunette hair, loose flowing spaghetti strap dress, hair and dress blowing in the wind, cler sky",
|
|
"sampler" : "euler",
|
|
"seed" : 7147106691,
|
|
"steps" : 20,
|
|
"width" : 832,
|
|
},
|
|
"user" : {
|
|
"avatar_url" : "https://api.tungsten.run/v1/avatar/512x512/HDiix5cCtzg2VrZU/1726485265",
|
|
"badges" : list,
|
|
"bio" : "Hello, world! 😋",
|
|
"created_at" : "2024-09-16T17:18:58Z",
|
|
"display_name" : "Survivor",
|
|
"sm_avatar_url": "https://api.tungsten.run/v1/avatar/64x64/HDiix5cCtzg2VrZU/1726485265",
|
|
"type" : "admin",
|
|
"username" : "survivor",
|
|
"uuid" : "HDiix5cCtzg2VrZU",
|
|
},
|
|
},
|
|
|
|
{
|
|
"#url" : "https://tungsten.run/model/9vHB2hNUdg/chroma",
|
|
"#class" : tungsten.TungstenModelExtractor,
|
|
"#pattern" : r"https://api\.tungsten\.run/v1/upload/\w+",
|
|
"#count" : 22,
|
|
|
|
"comment_count" : int,
|
|
"created_at" : "iso:8601",
|
|
"date" : "type:datetime",
|
|
"extension" : "webp",
|
|
"filename" : str,
|
|
"like_count" : int,
|
|
"nsfw" : bool,
|
|
"original_url" : str,
|
|
"original_width" : int,
|
|
"original_height": int,
|
|
"rating" : None,
|
|
"resized_url" : str,
|
|
"resized_width" : int,
|
|
"resized_height" : int,
|
|
"title" : str,
|
|
"uuid" : "len:str:16",
|
|
"view_count" : int,
|
|
"user" : {
|
|
"avatar_url" : str,
|
|
"badges" : list,
|
|
"bio" : {str, None},
|
|
"created_at" : str,
|
|
"display_name" : str,
|
|
"sm_avatar_url": str,
|
|
"type" : "regular",
|
|
"username" : str,
|
|
"uuid" : "len:str:16",
|
|
},
|
|
},
|
|
|
|
{
|
|
"#url" : "https://tungsten.run/model/9vHB2hNUdg?model_version=CxvEH6esrG",
|
|
"#class" : tungsten.TungstenModelExtractor,
|
|
"#range" : "1-50",
|
|
"#pattern" : r"https://api\.tungsten\.run/v1/upload/\w+",
|
|
"#count" : 50,
|
|
},
|
|
|
|
{
|
|
"#url" : "https://tungsten.run/user/lynodie",
|
|
"#class" : tungsten.TungstenUserExtractor,
|
|
"#range" : "1-50",
|
|
"#pattern" : r"https://api\.tungsten\.run/v1/upload/\w+",
|
|
"#count" : 50,
|
|
|
|
"user" : {
|
|
"avatar_url" : "https://api.tungsten.run/v1/avatar/512x512/EYJsXQDD69u26NYT/1751499148",
|
|
"bio" : "Just a normal lesbian girl from Germany.",
|
|
"created_at" : "2024-09-16T18:01:20Z",
|
|
"display_name" : "Sarah",
|
|
"sm_avatar_url": "https://api.tungsten.run/v1/avatar/64x64/EYJsXQDD69u26NYT/1751499148",
|
|
"type" : "regular",
|
|
"username" : "lynodie",
|
|
"uuid" : "EYJsXQDD69u26NYT",
|
|
},
|
|
},
|
|
|
|
{
|
|
"#url" : "https://tungsten.run/user/RD.Tungsten?tag=thanks&sort=top_month",
|
|
"#comment" : "query parameters",
|
|
"#class" : tungsten.TungstenUserExtractor,
|
|
"#results" : "https://api.tungsten.run/v1/upload/TYLShY6EDVyiV8ye",
|
|
|
|
"comment_count" : int,
|
|
"created_at" : "2025-08-10T11:13:55Z",
|
|
"date" : "dt:2025-08-10 11:13:55",
|
|
"extension" : "webp",
|
|
"filename" : "TYLShY6EDVyiV8ye",
|
|
"like_count" : range(15, 50),
|
|
"nsfw" : False,
|
|
"original_height": 1216,
|
|
"original_url" : "https://api.tungsten.run/v1/upload/TYLShY6EDVyiV8ye",
|
|
"original_width" : 832,
|
|
"rating" : None,
|
|
"resized_height" : 936,
|
|
"resized_url" : "https://api.tungsten.run/v1/upload/bMfhipdctg79TqCo",
|
|
"resized_width" : 640,
|
|
"search_tags" : "thanks",
|
|
"title" : "Thank you !!",
|
|
"uuid" : "FcRnVM6btz7vbM9Z",
|
|
"view_count" : int,
|
|
"user" : {
|
|
"avatar_url" : "https://api.tungsten.run/v1/avatar/512x512/dTqnGJmASg9WSjfL/1746094098",
|
|
"badges" : [],
|
|
"bio" : None,
|
|
"created_at" : "2025-04-27T14:48:27Z",
|
|
"display_name" : "RD.Tungsten",
|
|
"sm_avatar_url": "https://api.tungsten.run/v1/avatar/64x64/dTqnGJmASg9WSjfL/1746094098",
|
|
"type" : "regular",
|
|
"username" : "RD.Tungsten",
|
|
"uuid" : "dTqnGJmASg9WSjfL",
|
|
},
|
|
},
|
|
|
|
)
|