Skip to content
Snippets Groups Projects
Commit ff400625 authored by Wernle, Alexandra Nicole's avatar Wernle, Alexandra Nicole Committed by Wilbur, Spencer Franklin
Browse files

Working on Jeremy's comments

parent 22c4a0ac
No related branches found
No related tags found
2 merge requests!223Attempt StringIO after failed BytesIO file read,!212Draft: Reattempt to merge old Iris-Factory Branch from Swilbur's local repo
...@@ -7,7 +7,8 @@ Created on Thu Jul 7 11:05:19 2022 ...@@ -7,7 +7,8 @@ Created on Thu Jul 7 11:05:19 2022
import json import json
import os import os
from datetime import date, datetime, time, timedelta from datetime import date, datetime, time, timedelta
from typing import List from typing import List, Optional
from pathlib import Path
import typer import typer
from enum import Enum from enum import Enum
...@@ -19,9 +20,9 @@ from ..residual import Reading, WebAbsolutesFactory, SpreadsheetAbsolutesFactory ...@@ -19,9 +20,9 @@ from ..residual import Reading, WebAbsolutesFactory, SpreadsheetAbsolutesFactory
TODAY = datetime.combine(date.today(), time(0, 0, 0)) TODAY = datetime.combine(date.today(), time(0, 0, 0))
class factories(str, Enum): class ResidualFactory(str, Enum):
web_absolutes_factory = "web_absolutes_factory" SPREADSHEET = "spreadsheet"
spreadsheet_factory = "spreadsheet_factory" WEB_ABSOLUTES = "webabsolutes"
def copy_absolutes( def copy_absolutes(
...@@ -46,19 +47,23 @@ def copy_absolutes( ...@@ -46,19 +47,23 @@ def copy_absolutes(
help="URL to web absolutes service", help="URL to web absolutes service",
metavar="URL", metavar="URL",
), ),
spreadsheets_dir: str = typer.Option( directory: Optional[Path] = typer.Option(
default="---", help="Directory to residual spreadsheets", metavar="DIR" None,
exists=True,
file_okay=False,
dir_okay=True,
writable=False,
readable=False,
resolve_path=True,
help="Residual spreadsheet directory",
), ),
# I am not sure what to specify as the default directory, I was using a local directory as a test factory: ResidualFactory = ResidualFactory.WEB_ABSOLUTES,
factory: factories = factories.web_absolutes_factory,
): ):
"""Copy absolutes from the web absolutes service OR residual spreadsheets into the metadata service.""" """Copy absolutes from the web absolutes service OR residual spreadsheets into the metadata service."""
print(f"Using factory: {factory.value}") if factory.value == ResidualFactory.WEB_ABSOLUTES:
if factory.value == factories.web_absolutes_factory:
factory = WebAbsolutesFactory(url=web_absolutes_url) factory = WebAbsolutesFactory(url=web_absolutes_url)
else: else:
factory = SpreadsheetAbsolutesFactory(base_directory=spreadsheets_dir) factory = SpreadsheetAbsolutesFactory(base_directory=directory)
readings = get_readings( readings = get_readings(
factory=factory, factory=factory,
...@@ -104,7 +109,10 @@ def create_reading_metadata(reading: Reading) -> Metadata: ...@@ -104,7 +109,10 @@ def create_reading_metadata(reading: Reading) -> Metadata:
def get_readings( def get_readings(
factory: factories, observatory: str, starttime: UTCDateTime, endtime: UTCDateTime factory: ResidualFactory,
observatory: str,
starttime: UTCDateTime,
endtime: UTCDateTime,
) -> List[Reading]: ) -> List[Reading]:
"""Get readings from web absolutes or residual spreadsheets. """Get readings from web absolutes or residual spreadsheets.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment