Added: Ability to change store name and overwrite protection

2026-01-08 19:16:59 +00:00
parent 1fb786c262
commit 6b363b0788
7 changed files with 445 additions and 14 deletions
--- a/migrate_field_names.py
+++ b/migrate_field_names.py
@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+"""
+Migration script to rename NL43 measurement field names to match actual device output.
+
+Changes:
+- lp -> laeq (A-weighted equivalent continuous sound level)
+- leq -> lae (A-weighted sound exposure level)
+- lmax -> lasmax (A-weighted slow maximum)
+- lmin -> lasmin (A-weighted slow minimum)
+- lpeak -> lapeak (A-weighted peak)
+"""
+
+import sqlite3
+import sys
+from pathlib import Path
+
+def migrate_database(db_path: str):
+    """Migrate the database schema to use correct field names."""
+
+    print(f"Migrating database: {db_path}")
+
+    # Connect to database
+    conn = sqlite3.connect(db_path)
+    cur = conn.cursor()
+
+    try:
+        # Check if migration is needed
+        cur.execute("PRAGMA table_info(nl43_status)")
+        columns = [row[1] for row in cur.fetchall()]
+
+        if 'laeq' in columns:
+            print("✓ Database already migrated")
+            return
+
+        if 'lp' not in columns:
+            print("✗ Database schema does not match expected format")
+            sys.exit(1)
+
+        print("Starting migration...")
+
+        # SQLite doesn't support column renaming directly, so we need to:
+        # 1. Create new table with correct column names
+        # 2. Copy data from old table
+        # 3. Drop old table
+        # 4. Rename new table
+
+        # Create new table with correct column names
+        cur.execute("""
+            CREATE TABLE nl43_status_new (
+                unit_id VARCHAR PRIMARY KEY,
+                last_seen DATETIME,
+                measurement_state VARCHAR,
+                laeq VARCHAR,
+                lae VARCHAR,
+                lasmax VARCHAR,
+                lasmin VARCHAR,
+                lapeak VARCHAR,
+                battery_level VARCHAR,
+                power_source VARCHAR,
+                sd_remaining_mb VARCHAR,
+                sd_free_ratio VARCHAR,
+                raw_payload TEXT
+            )
+        """)
+        print("✓ Created new table with correct column names")
+
+        # Copy data from old table to new table
+        cur.execute("""
+            INSERT INTO nl43_status_new
+            (unit_id, last_seen, measurement_state, laeq, lae, lasmax, lasmin, lapeak,
+             battery_level, power_source, sd_remaining_mb, sd_free_ratio, raw_payload)
+            SELECT
+                unit_id, last_seen, measurement_state, lp, leq, lmax, lmin, lpeak,
+                battery_level, power_source, sd_remaining_mb, sd_free_ratio, raw_payload
+            FROM nl43_status
+        """)
+        rows_copied = cur.rowcount
+        print(f"✓ Copied {rows_copied} rows from old table")
+
+        # Drop old table
+        cur.execute("DROP TABLE nl43_status")
+        print("✓ Dropped old table")
+
+        # Rename new table
+        cur.execute("ALTER TABLE nl43_status_new RENAME TO nl43_status")
+        print("✓ Renamed new table to nl43_status")
+
+        # Commit changes
+        conn.commit()
+        print("✓ Migration completed successfully")
+
+    except Exception as e:
+        conn.rollback()
+        print(f"✗ Migration failed: {e}")
+        sys.exit(1)
+    finally:
+        conn.close()
+
+if __name__ == "__main__":
+    # Default database path
+    db_path = Path(__file__).parent / "data" / "slmm.db"
+
+    # Allow custom path as command line argument
+    if len(sys.argv) > 1:
+        db_path = Path(sys.argv[1])
+
+    if not db_path.exists():
+        print(f"✗ Database not found: {db_path}")
+        sys.exit(1)
+
+    migrate_database(str(db_path))