From 302428e10a1d0d892db87ee62894189baea0ab7f Mon Sep 17 00:00:00 2001 From: Eric Duminil <eric.duminil@gmail.com> Date: Sun, 23 Mar 2025 17:10:38 +0100 Subject: [PATCH] Split main --- download_files_from_LGL_BW.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/download_files_from_LGL_BW.py b/download_files_from_LGL_BW.py index 9e4724c..a9fcead 100644 --- a/download_files_from_LGL_BW.py +++ b/download_files_from_LGL_BW.py @@ -241,14 +241,8 @@ Examples: return parser.parse_args() -def main(): +def main(location_name: str, wkt_or_zipcode: str, download_only: bool = False, simstadt_folder: Path | None = None): """Main function to process arguments and run the download/extraction""" - args = parse_arguments() - - location_name = args.name - wkt_or_zipcode = args.region - download_only = args.download_only - simstadt_folder = args.simstadt_folder # Validate location name if ' ' in location_name: @@ -283,4 +277,6 @@ def main(): if __name__ == '__main__': - main() + args = parse_arguments() + main(args.name, args.region, args.download_only, args.simstadt_folder) + -- GitLab