mirror of
https://github.com/DragoonAethis/itch-dl.git
synced 2024-12-21 02:21:52 +01:00
90346f579a
Some of the misc issues were found with mypy. Not adding it to dev deps for now as it complains about missing types in libraries and does not honor noqa where we need it (non-literal TypedDict keys).
32 lines
880 B
Python
32 lines
880 B
Python
import logging
|
|
from typing import Dict
|
|
|
|
from .api import ItchApiClient
|
|
|
|
|
|
def get_download_keys(client: ItchApiClient) -> Dict[int, str]:
|
|
logging.info("Fetching all download keys...")
|
|
download_keys: Dict[int, str] = {}
|
|
page = 1
|
|
|
|
while True:
|
|
logging.info(f"Downloading page {page} (found {len(download_keys)} keys total)")
|
|
r = client.get("/profile/owned-keys", data={"page": page}, timeout=15)
|
|
if not r.ok:
|
|
break
|
|
|
|
data = r.json()
|
|
if 'owned_keys' not in data:
|
|
break # Assuming we're out of keys already...
|
|
|
|
for key in data['owned_keys']:
|
|
download_keys[key['game_id']] = key['id']
|
|
|
|
if len(data['owned_keys']) == data['per_page']:
|
|
page += 1
|
|
else:
|
|
break
|
|
|
|
logging.info(f"Fetched {len(download_keys)} download keys.")
|
|
return download_keys
|