From 2a9abe92b0636369c1af632b1f0a0a36225f8e30 Mon Sep 17 00:00:00 2001 From: Alex Hutko Date: Mon, 15 Apr 2024 17:03:20 -0700 Subject: [PATCH] Update load_from_fdsn.py Add command line argument --net. --- app/nslc/management/commands/load_from_fdsn.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/app/nslc/management/commands/load_from_fdsn.py b/app/nslc/management/commands/load_from_fdsn.py index 2e67d50..9fba178 100644 --- a/app/nslc/management/commands/load_from_fdsn.py +++ b/app/nslc/management/commands/load_from_fdsn.py @@ -24,6 +24,7 @@ [optional args] --path='.' --datacenter='IRISDMC,...' + --net='UO,UW...' --sta='BEER,...' --cha='HN?,ENN,...' --loc=* @@ -63,6 +64,12 @@ def add_arguments(self, parser): default="IRISDMC,NCEDC,SCEDC,NRCAN", help="Comma separated list of datacenters" ) + parser.add_argument( + '--net', + default="*", + help="Comma separated regex for stations, default \ + is pre-defined list of stations" + ) parser.add_argument( '--sta', default="*", @@ -119,7 +126,7 @@ def build_url(self, params, level): f"&starttime={params['starttime']}" "&format=text" ) - if (level != "network"): + if (level != 'network'): url += ( f"&sta={params['sta']}" f"&cha={params['cha']}" @@ -143,7 +150,8 @@ def handle(self, *args, **options): "TD", "TR", "TX", "UM", "UO", "US", "UU", "UW", "WR", "WU", "WW", "WY" ] - options["net"] = ','.join(ALLOWED_NETWORKS) + if 'net' not in options or options['net'] == '*': + options['net'] = ','.join(ALLOWED_NETWORKS) LOADER_EMAIL = os.environ.get('LOADER_EMAIL') if not LOADER_EMAIL: print( @@ -165,7 +173,7 @@ def handle(self, *args, **options): ) sys.exit(1) - network_url = self.build_url(options, "network") + network_url = self.build_url(options, 'network') with requests.Session() as s: download = s.get(network_url) decoded_content = download.content.decode('utf-8')