From ded33e2c12d09f91d4ba9a7783e201e6a89f6e7d Mon Sep 17 00:00:00 2001 From: James Hiebert Date: Wed, 14 Feb 2024 17:04:50 -0800 Subject: [PATCH] Accept command line args for provinces in pipeline The previous download->process pipeline didn't accept provinces as command line args and had BC/YT hardcoded. This doesn't allow for NT or any other provinces. This small commit permits any other provinces to be accepted by the crmprtd_pipeline script. --- crmprtd/download_cache_process.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/crmprtd/download_cache_process.py b/crmprtd/download_cache_process.py index 4ccbec5..93116a6 100644 --- a/crmprtd/download_cache_process.py +++ b/crmprtd/download_cache_process.py @@ -242,7 +242,9 @@ def dispatch_network( dry_run=dry_run, ) elif network_name == "ec": - for province in ("BC", "YT"): + provinces = kwargs.pop("province") + + for province in provinces: download_and_process( network_name=network_name, log_args=log_args(**kwargs, province=province), @@ -451,6 +453,14 @@ def main(arglist: List[str] = None) -> None: required=True, help="Frequency of download (network ec only)", ) + parser.add_argument( + "-p", + "--province", + action="append", + help="2 letter province code", + default=[], + choices=("AB", "BC", "MB", "NB", "NL", "NS", "NT", "NU", "ON", "PE", "QC", "SK", "YT"), + ) args = parser.parse_args(arglist) # TODO: Add network-dependent time arg here? Currently, it is hardwired in code to