kat/konawall-py
1
0
Fork 0
mirror of https://github.com/kittywitch/konawall-py.git synced 2026-02-09 04:19:19 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(e621): add environment variable setter

Browse source
This commit is contained in:
Kat Inskip 2025-02-28 13:12:23 -08:00
parent e426a3ee4e
commit b503cc87a9
Signed by: kat
GPG key ID: 465E64DECEA8CF0F
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
konawall/sources/e621.py
View file

@ -1,5 +1,6 @@
import requests import requests
import logging import logging
import os
from konawall.custom_print import kv_print from konawall.custom_print import kv_print
from konawall.custom_errors import RequestFailed from konawall.custom_errors import RequestFailed
from konawall.module_loader import add_source from konawall.module_loader import add_source
@ -13,7 +14,10 @@ Turn a list of tags and a count into a list of URLs to download from
:returns: A list of URLs to download from :returns: A list of URLs to download from
""" """
def request_posts(count: int, tags: list, config) -> list: def request_posts(count: int, tags: list, config) -> list:
api_key = config["e621_api_key"] if "KONAWALL_E621_API_KEY" in os.environ:
api_key = os.environ["KONAWALL_E621_API_KEY"]
else:
api_key = config["e621_api_key"]
logging.debug(f"request_posts() called with count={count}, tags=[{', '.join(tags)}]") 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 # Make sure we get a different result every time by using "order:random" as a tag
if "order:random" not in tags: if "order:random" not in tags: