mirror of
https://github.com/kittywitch/konawall-py.git
synced 2026-02-09 12:29:19 -08:00
65 lines
No EOL
2.5 KiB
Python
65 lines
No EOL
2.5 KiB
Python
import requests
|
|
import logging
|
|
from konawall.custom_print import kv_print
|
|
from konawall.custom_errors import RequestFailed
|
|
from konawall.module_loader import add_source
|
|
from konawall.downloader import download_files
|
|
|
|
"""
|
|
Turn a list of tags and a count into a list of URLs to download from
|
|
|
|
:param count: The number of images to provide download URLs for
|
|
:param user_tags: A list of tags to search for
|
|
:returns: A list of URLs to download from
|
|
"""
|
|
def request_posts(count: int, tags: list) -> list:
|
|
logging.debug(f"request_posts() called with count={count}, tags=[{', '.join(tags)}]")
|
|
# Make sure we get a different result every time by using "order:random" as a tag
|
|
if "order:random" not in tags:
|
|
tags.append("order:random")
|
|
# Tags are separated by a plus sign for this API
|
|
tag_string: str = "+".join(tags)
|
|
# Request URL for getting posts from the API
|
|
url: str = f"https://konachan.com/post.json?limit={str(count)}&tags={tag_string}"
|
|
logging.debug(f"Request URL: {url}")
|
|
response = requests.get(url, headers={"User-Agent": "konachan-py/alpha"})
|
|
# Check if the request was successful
|
|
logging.debug("Status code: " + str(response.status_code))
|
|
# List of URLs to download
|
|
posts: list = []
|
|
if response.status_code == 200:
|
|
# Get the JSON data from the response
|
|
json = response.json()
|
|
for post in json:
|
|
# Give the user data about the post retrieved
|
|
kv_print("Post ID", post["id"])
|
|
kv_print("Author", post["author"])
|
|
kv_print("Rating", post["rating"])
|
|
kv_print("Resolution", f"{post['width']}x{post['height']}")
|
|
kv_print("Tags", post["tags"])
|
|
kv_print("URL", post["file_url"])
|
|
# Append the URL to the list
|
|
posts.append(post)
|
|
else:
|
|
# Raise an exception if the request failed
|
|
RequestFailed(response.status_code)
|
|
return posts
|
|
|
|
"""
|
|
Download a number of images from Konachan given a list of tags and a count
|
|
|
|
:param count: The number of images to download
|
|
:param tags: A list of tags to search for
|
|
"""
|
|
@add_source("konachan")
|
|
def handle(count: int, tags: list) -> list:
|
|
logging.debug(f"handle_konachan() called with count={count}, tags=[{', '.join(tags)}]")
|
|
# Get a list of URLs to download
|
|
posts: list = request_posts(count, tags)
|
|
urls: list = []
|
|
# Download the images
|
|
for post in posts:
|
|
urls.append(post["file_url"])
|
|
files = download_files(urls)
|
|
# Return the downloaded files
|
|
return files, posts |