Mirrors/itch-dl
2
0
Fork 0
mirror of https://github.com/DragoonAethis/itch-dl.git synced 2024-12-21 02:21:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Move argparse out of main

Browse Source
This commit is contained in:
Ryszard Knop 2021-10-03 13:31:08 +02:00
parent 8159244f8d
commit a552bb2393
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
downloader.py
View File

@ -286,17 +286,23 @@ def download_jam(path_to_json: str, download_to: str, api_key: str, continue_fro
print(f"{title} ({game_id}) - {url} - {message}") print(f"{title} ({game_id}) - {url} - {message}")
if __name__ == "__main__": def get_parser():
parser = argparse.ArgumentParser(description="Downloads games from public Itch.io game jams.") parser = argparse.ArgumentParser(description="Downloads games from public Itch.io game jams.")
parser.add_argument("entries", help="path to the game jam entries.json file") parser.add_argument("entries", help="path to the game jam entries.json file")
parser.add_argument("--api-key", metavar="key", required=True, help="itch.io API key from https://itch.io/user/settings/api-keys") parser.add_argument("--api-key", metavar="key", required=True, help="itch.io API key from https://itch.io/user/settings/api-keys")
parser.add_argument("--download-to", metavar="path", help="directory to save results into (default: current dir)") parser.add_argument("--download-to", metavar="path", help="directory to save results into (default: current dir)")
parser.add_argument("--continue-from", metavar="id", type=int, help="skip all entries until the provided entry ID is found") parser.add_argument("--continue-from", metavar="id", type=int, help="skip all entries until the provided entry ID is found")
args = parser.parse_args() return parser
def get_download_dir(args: argparse.Namespace) -> str:
download_to = os.getcwd() download_to = os.getcwd()
if args.download_to is not None: if args.download_to is not None:
download_to = os.path.normpath(args.download_to) download_to = os.path.normpath(args.download_to)
os.makedirs(download_to) os.makedirs(download_to)
if __name__ == "__main__":
args = get_parser().parse_args()
download_to = get_download_dir(args)
download_jam(args.entries, download_to, args.api_key, continue_from=args.continue_from) download_jam(args.entries, download_to, args.api_key, continue_from=args.continue_from)