pipeline-eds 0.2.26__tar.gz → 0.2.27__tar.gz

{pipeline_eds-0.2.26 → pipeline_eds-0.2.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pipeline-eds
-Version: 0.2.26
+Version: 0.2.27
 Summary: The official API pipeline library for mulch-based projects. Key target: Emerson Ovation EDS REST API.
 License: BSD-3
 Author: George Clayton Bennett

{pipeline_eds-0.2.26 → pipeline_eds-0.2.27}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-eds"
-version = "0.2.26"
+version = "0.2.27"
 description = "The official API pipeline library for mulch-based projects. Key target: Emerson Ovation EDS REST API."
 authors = ["George Clayton Bennett <george.bennett@memphistn.gov>"]
 license = "BSD-3"
@@ -11,8 +11,11 @@ packages = [
 ] # This makes `pipeline` available as an importable top-level module
 homepage = "https://github.com/city-of-memphis-wastewater/pipeline"
 repository = "https://github.com/city-of-memphis-wastewater/pipeline"
-
-[tool.poetry.scripts]
+include = [
+    "src/pipeline/data/*"
+]
+	
+[tool.poetry.scripts]	
 #time-manager = "pipeline.time_manager:main"
 #workspace-manager = "pipeline.workspace_manager:main" # included in init
 #collector = "pipeline.collector:main"

{pipeline_eds-0.2.26 → pipeline_eds-0.2.27}/src/pipeline/cli.py

@@ -10,6 +10,10 @@ def list_workspaces(workspaces_dir: Path = Path("workspaces")):
             typer.echo(f"🪴 {path.name}")
 
 '''
+
+import sqlite3
+from rich.table import Table
+from rich.console import Console
 import typer
 import importlib
 from pathlib import Path
@@ -17,6 +21,7 @@ from importlib.metadata import version, PackageNotFoundError
 
 from pipeline.env import SecretConfig
 from pipeline.time_manager import TimeManager
+from pipeline.create_sensors_db import get_user_db_path, ensure_user_db, get_db_connection
 #from pipeline.helpers import setup_logging
 #from pipeline.workspace_manager import WorkspaceManager
 
@@ -49,7 +54,7 @@ except PackageNotFoundError:
 ### Pipeline CLI
 
 app = typer.Typer(help="CLI for running pipeline workspaces.")
-
+console = Console()
 
 @app.callback(invoke_without_command=True)
 def main(
@@ -92,12 +97,60 @@ def run(
         typer.echo(f"💥 Error while running {workspace}: {e}")
         raise typer.Exit(1)
 
+# -----------------------------
+# CLI commands
+# -----------------------------
 @app.command()
-def typical(zd: str):
-    """
-    Print the typical idcs list for an EDS zd.
-    """
-    pass
+def list_sensors():
+    """Print all sensors from the database."""
+    conn = get_db_connection()
+    cur = conn.cursor()
+    cur.execute("SELECT idcs, iess, zd FROM sensors")
+    rows = cur.fetchall()
+    conn.close()
+
+    table = Table(title="Sensor Correlations")
+    table.add_column("IDCS", style="cyan")
+    table.add_column("IESS", style="magenta")
+    table.add_column("ZD", style="green")
+
+    for idcs, iess, zd in rows:
+        table.add_row(idcs, iess, zd)
+
+    console.print(table)
+
+@app.command()
+def reset_db():
+    """Reset the user DB from the packaged default."""
+    user_db = get_user_db_path()
+    if user_db.exists():
+        user_db.unlink()
+    ensure_user_db()
+    typer.echo(f"✅ User DB reset to default at {user_db}")
+
+
+@app.command()
+def sensors(db_path: str = None):
+    # db_path: str = "sensors.db"
+    if db_path is not None:
+        conn = sqlite3.connect(db_path)
+    else:  
+        conn = get_db_connection()
+    cur = conn.cursor()
+    cur.execute("SELECT idcs, iess, zd FROM sensors")
+    rows = cur.fetchall()
+    conn.close()
+
+    table = Table(title="Sensor Correlations")
+    table.add_column("IDCS", style="cyan")
+    table.add_column("IESS", style="magenta")
+    table.add_column("ZD", style="green")
+
+    for idcs, iess, zd in rows:
+        table.add_row(idcs, iess, zd)
+
+    console.print(table)
+
 
 @app.command()
 def trend(

pipeline_eds-0.2.27/src/pipeline/create_sensors_db.py

@@ -0,0 +1,113 @@
+# create_sensors_db.py
+import sqlite3
+from pathlib import Path
+import os
+import sys
+import shutil
+import sqlite3
+from pathlib import Path
+from importlib import resources
+
+# Your sensor data here:
+sensors_data = [
+    # (idcs, iess, zd)
+    ("M310LI", "M310L0I.UNIT0@NET0", "Maxson"),
+    ("M100FI", "M100FI.UNIT0@NET0", "Maxson"),
+    ("FI8001", "FI8001.UNIT0@NET0", "Maxson"),
+    ("I-0201A", "I-0201A.UNIT1@NET1", "WWTF"),
+    # add as many as you like...
+]
+
+
+# -----------------------------
+# Rollout packaged DB
+# -----------------------------
+
+def ensure_user_db() -> Path:
+    """Copy packaged DB to user folder if it doesn't exist."""
+    user_db = get_user_db_path()
+    if not user_db.exists():
+        with resources.as_file(resources.files("pipeline.data") / "sensors.db") as packaged_db:
+            shutil.copy(packaged_db, user_db)
+    return user_db
+
+# -----------------------------
+# DB connection
+# -----------------------------
+def get_db_connection():
+    db_path = ensure_user_db()
+    conn = sqlite3.connect(db_path)
+    return conn
+
+# -----------------------------
+# Packaged DB location
+# -----------------------------
+def get_packaged_db_path() -> Path:
+    db_path = Path(__file__).parent / "data" / "sensors.db"
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+    return db_path
+
+# -----------------------------
+# User DB location
+# -----------------------------
+
+def get_user_db_path() -> Path:
+    if sys.platform == "win32":
+        base = Path(os.getenv("APPDATA", Path.home() / "AppData" / "Roaming"))
+    elif sys.platform == "darwin":
+        base = Path.home() / "Library" / "Application Support"
+    else:
+        base = Path.home() / ".local" / "share"
+
+    user_dir = base / "pipeline"
+    user_dir.mkdir(parents=True, exist_ok=True)
+    return user_dir / "sensors.db"
+
+# -----------------------------
+# Create packaged DB
+# -----------------------------
+def create_packaged_db():
+    db_path = get_packaged_db_path()
+    if db_path.exists():
+        print(f"⚠️ {db_path} already exists. Overwriting...")
+        db_path.unlink()
+
+    conn = sqlite3.connect(db_path)
+    cur = conn.cursor()
+
+    cur.execute("""
+        CREATE TABLE sensors (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            idcs TEXT NOT NULL,
+            iess TEXT NOT NULL,
+            zd TEXT NOT NULL
+        )
+    """)
+
+    cur.executemany(
+        "INSERT INTO sensors (idcs, iess, zd) VALUES (?, ?, ?)",
+        sensors_data
+    )
+
+    conn.commit()
+    conn.close()
+    print(f"✅ Packaged DB created at {db_path}")
+    return db_path
+
+# -----------------------------
+# Reset user DB
+# -----------------------------
+def reset_user_db(packaged_db_path: Path):
+    user_db = get_user_db_path()
+    if user_db.exists():
+        user_db.unlink()
+        print(f"🗑 Old user DB removed: {user_db}")
+    shutil.copy(packaged_db_path, user_db)
+    print(f"✅ User DB reset from packaged DB: {user_db}")
+
+# -----------------------------
+# Main
+# -----------------------------
+if __name__ == "__main__":
+    packaged_db = create_packaged_db()
+    reset_user_db(packaged_db)