From df7b04cb21fb9ffbcc4c8e54722f88a4f9620fe6 Mon Sep 17 00:00:00 2001
From: pcain-usgs <pcain@usgs.gov>
Date: Tue, 8 Dec 2020 13:15:15 -0700
Subject: [PATCH] Add entry point and main method

---
 geomagio/processing/obsrio.py | 29 ++++++-----------------------
 setup.py                      |  1 +
 2 files changed, 7 insertions(+), 23 deletions(-)

diff --git a/geomagio/processing/obsrio.py b/geomagio/processing/obsrio.py
index 7d29f1fd5..d3d07b3cc 100644
--- a/geomagio/processing/obsrio.py
+++ b/geomagio/processing/obsrio.py
@@ -1,6 +1,8 @@
 import os
 from typing import Optional
 
+import typer
+
 from ..algorithm import Algorithm, FilterAlgorithm
 from ..edge import EdgeFactory, MiniSeedFactory
 from ..Controller import Controller, get_realtime_interval
@@ -8,6 +10,10 @@ from ..TimeseriesFactory import TimeseriesFactory
 from .factory import get_edge_factory, get_miniseed_factory
 
 
+def main():
+    typer.run(filter_realtime)
+
+
 def obsrio_minute(
     observatory: str,
     input_factory: Optional[TimeseriesFactory] = None,
@@ -202,26 +208,3 @@ def obsrio_day(
             rename_output_channel=((input_channel, output_channel),),
             update_limit=update_limit,
         )
-
-
-def filter_realtime(
-    observatory: str,
-    input_factory: Optional[TimeseriesFactory] = None,
-    output_factory: Optional[TimeseriesFactory] = None,
-    realtime_interval: int = 600,
-    update_limit: int = 10,
-):
-    """Filter 10Hz miniseed, 1 second and one minute data.
-    Defaults set for realtime processing; can also be implemented to update legacy data"""
-    obsrio_tenhertz(
-        observatory, realtime_interval, input_factory, output_factory, update_limit
-    )
-    obsrio_second(
-        observatory, realtime_interval, input_factory, output_factory, update_limit
-    )
-    obsrio_minute(
-        observatory, realtime_interval, input_factory, output_factory, update_limit
-    )
-    obsrio_temperatures(
-        observatory, realtime_interval, input_factory, output_factory, update_limit
-    )
diff --git a/setup.py b/setup.py
index 23146078a..b1dc547f0 100644
--- a/setup.py
+++ b/setup.py
@@ -26,6 +26,7 @@ setuptools.setup(
     entry_points={
         "console_scripts": [
             "magproc-prepfiles=geomagio.processing.magproc:main",
+            "update-legacy=geomagio.processing.obsrio:main",
         ],
     },
 )
-- 
GitLab