Refactor download processing code and add some tests

1 files changed, 29 insertions, 0 deletions

diff --git a/src/wikiget/dl.py b/src/wikiget/dl.py
index c0af6e9..20d8a07 100644
--- a/src/wikiget/dl.py
+++ b/src/wikiget/dl.py
@@ -49,6 +49,35 @@ def prep_download(dl: str, args: Namespace) -> File:
     return file
 
 
+def process_download(args: Namespace) -> None:
+    if args.batch:
+        # batch download mode
+        errors = batch_download(args)
+        if errors:
+            # return non-zero exit code if any problems were encountered, even if some
+            # downloads completed successfully
+            logger.warning(
+                f"{errors} problem{'s'[:errors^1]} encountered during batch processing"
+            )
+            sys.exit(1)  # completed with errors
+    else:
+        # single download mode
+        try:
+            file = prep_download(args.FILE, args)
+        except ParseError as e:
+            logger.error(e)
+            sys.exit(1)
+        except FileExistsError as e:
+            logger.warning(e)
+            sys.exit(1)
+        except (ConnectionError, HTTPError, InvalidResponse, LoginError, APIError):
+            sys.exit(1)
+
+        errors = download(file, args)
+        if errors:
+            sys.exit(1)  # completed with errors
+
+
 def batch_download(args: Namespace) -> int:
     errors = 0