From 5be46b84bfac7b293da828c426406208ff5f8547 Mon Sep 17 00:00:00 2001 From: mgleed Date: Tue, 28 Nov 2023 15:37:40 -0500 Subject: [PATCH] api key logic --- src/mods.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/mods.py b/src/mods.py index c266200..b90b475 100644 --- a/src/mods.py +++ b/src/mods.py @@ -15,8 +15,6 @@ load_dotenv() # Load environment variables from .env file -API_KEY = os.getenv("CURSEFORGE_API_KEY") - @dataclass class Mod: @@ -25,6 +23,15 @@ class Mod: latest_dt: datetime | None +def _get_api_key() -> str: + key = os.getenv("CURSEFORGE_API_KEY") + if key: + logger.debug("CURSEFORGE_API_KEY found in environment variables") + else: + logger.warning("CURSEFORGE_API_KEY not found in environment variables") + return key + + @cache def _local_mod_file() -> dict: file_path = os.path.join( @@ -72,7 +79,7 @@ def _get_latest_mods_timestamps(mod_ids: list[int]) -> dict[int, tuple[str, date headers = { "Content-Type": "application/json", "Accept": "application/json", - "x-api-key": API_KEY, + "x-api-key": _get_api_key(), } payload = {"modIds": mod_ids, "filterPcOnly": True}