From c1820026f97eaf671c29ab30f02879de0ac4df89 Mon Sep 17 00:00:00 2001 From: Cody Logan Date: Fri, 20 Oct 2023 16:36:14 -0700 Subject: Add type annotations to source files --- src/wikiget/dl.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'src/wikiget/dl.py') diff --git a/src/wikiget/dl.py b/src/wikiget/dl.py index 5491378..5b5b43b 100644 --- a/src/wikiget/dl.py +++ b/src/wikiget/dl.py @@ -18,9 +18,11 @@ import logging import os import sys +from argparse import Namespace from concurrent.futures import ThreadPoolExecutor from mwclient import APIError, InvalidResponse, LoginError, Site +from mwclient.image import Image from requests import ConnectionError, HTTPError from tqdm import tqdm @@ -32,7 +34,7 @@ from wikiget.parse import get_dest from wikiget.validations import verify_hash -def query_api(filename, site_name, args): +def query_api(filename: str, site_name: str, args: Namespace) -> Image: # connect to site and identify ourselves logging.info(f"Connecting to {site_name}") try: @@ -75,13 +77,13 @@ def query_api(filename, site_name, args): return image -def prep_download(dl, args): +def prep_download(dl: str, args: Namespace) -> File: file = get_dest(dl, args) file.image = query_api(file.name, file.site, args) return file -def batch_download(args): +def batch_download(args: Namespace) -> int: input_file = args.FILE dl_list = {} errors = 0 @@ -129,7 +131,7 @@ def batch_download(args): return errors -def download(f, args): +def download(f: File, args: Namespace) -> int: file = f.image filename = f.name dest = f.dest -- cgit v1.2.3