From 22c4a0ac2ab007c82663d41107a0c4bb6f1ff7d3 Mon Sep 17 00:00:00 2001
From: Alex Wernle <awernle@usgs.gov>
Date: Tue, 18 Oct 2022 15:36:50 -0600
Subject: [PATCH] Imported Enum to define a CLI parameter with a predefined set
of values to choose from. Created a spreadsheets_dir option and factory
option.
---
geomagio/processing/copy_absolutes.py | 37 ++++++++++++++++++---------
1 file changed, 25 insertions(+), 12 deletions(-)
diff --git a/geomagio/processing/copy_absolutes.py b/geomagio/processing/copy_absolutes.py
index f60c214f4..6d45442f4 100644
--- a/geomagio/processing/copy_absolutes.py
+++ b/geomagio/processing/copy_absolutes.py
@@ -10,14 +10,20 @@ from datetime import date, datetime, time, timedelta
from typing import List
import typer
+from enum import Enum
from obspy import UTCDateTime
-from ..metadata import Metadata, MetadataCategory, MetadataFactory
-from ..residual import Reading, WebAbsolutesFactory
+from ..metadata import Metadata, MetadataFactory, MetadataCategory
+from ..residual import Reading, WebAbsolutesFactory, SpreadsheetAbsolutesFactory
TODAY = datetime.combine(date.today(), time(0, 0, 0))
+class factories(str, Enum):
+ web_absolutes_factory = "web_absolutes_factory"
+ spreadsheet_factory = "spreadsheet_factory"
+
+
def copy_absolutes(
observatory: str = typer.Option(..., help="Observatory code"),
starttime: datetime = typer.Option(
@@ -40,12 +46,22 @@ def copy_absolutes(
help="URL to web absolutes service",
metavar="URL",
),
+ spreadsheets_dir: str = typer.Option(
+ default="---", help="Directory to residual spreadsheets", metavar="DIR"
+ ),
+ # I am not sure what to specify as the default directory, I was using a local directory as a test
+ factory: factories = factories.web_absolutes_factory,
):
- """Copy absolutes from the web absolutes service into the metadata service."""
- # read readings from web absolutes
- web_absolutes_factory = WebAbsolutesFactory(url=web_absolutes_url)
+ """Copy absolutes from the web absolutes service OR residual spreadsheets into the metadata service."""
+ print(f"Using factory: {factory.value}")
+
+ if factory.value == factories.web_absolutes_factory:
+ factory = WebAbsolutesFactory(url=web_absolutes_url)
+ else:
+ factory = SpreadsheetAbsolutesFactory(base_directory=spreadsheets_dir)
+
readings = get_readings(
- factory=web_absolutes_factory,
+ factory=factory,
observatory=observatory,
starttime=UTCDateTime(starttime),
endtime=UTCDateTime(endtime),
@@ -88,17 +104,14 @@ def create_reading_metadata(reading: Reading) -> Metadata:
def get_readings(
- factory: WebAbsolutesFactory,
- observatory: str,
- starttime: UTCDateTime,
- endtime: UTCDateTime,
+ factory: factories, observatory: str, starttime: UTCDateTime, endtime: UTCDateTime
) -> List[Reading]:
- """Get readings from web absolutes.
+ """Get readings from web absolutes or residual spreadsheets.
Parameters
----------
factory
- configured WebAbsolutesFactory
+ configured WebAbsolutesFactory or SpreadsheetAbsolutesFactory
observatory
search observatory
starttime
--
GitLab