added additional arguments to download routine to be able to specify storage folder and allow for files to be overwritten

Author: ymahlich

coderdata/cli.py

@@ -1,4 +1,7 @@
 import argparse
+from os import PathLike
+from pathlib import Path
+from typing import Union
 import sys
 from .download.downloader import download as download_datasets
 
@@ -17,6 +20,21 @@ def main():
              'Alternatively "all" will download the full repository of '
              'coderdata datasets. See "coderdata --list" for a complete list '
              'of available datasets. Defaults to "all"'
+        )
+    parser_download.add_argument(
+        '-p', '--local_path',
+        dest="LOCAL_PATH",
+        type=check_folder,
+        default=Path.cwd(),
+        help='Defines the folder the datasets should be stored in. Defaults '
+             'to the current working directory if omitted.'
+    )
+    parser_download.add_argument(
+        '-o', '--overwrite',
+        dest="OVERWRITE",
+        default=False,
+        action='store_true',
+        help='Allow dataset files to be overwritten if they already exist.'
     )
     parser_download.set_defaults(func=download)
     if len(sys.argv) == 1:
@@ -26,7 +44,37 @@ def main():
     args.func(args)
 
 def download(args):
-    download_datasets(name=args.DATASET_NAME)
+    download_datasets(
+        name=args.DATASET_NAME,
+        local_path=args.LOCAL_PATH,
+        exist_ok=args.OVERWRITE,
+        )
+
+
+def check_folder(path: Union[str, PathLike, Path]) -> Path:
+    """
+    Helper function to check if a defined folder exists
+    """
+
+    if not isinstance(path, (str, PathLike, Path)): 
+        raise TypeError(
+            f"'path' must be of type str, PathLike or Path. Supplied argument "
+            f"is of type {type(path)}."
+        )
+    if not isinstance(path, Path):
+        abs_path = Path(path).absolute()
+    else:
+        abs_path = path.absolute()
+    
+    if not abs_path.is_dir():
+        raise OSError(
+            f"The defined folder path '{path}' does not exist or is not a "
+            f"folder."
+            )
+    
+    return abs_path
+
+
 if __name__ == '__main__':
     main()