137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Migration script to add polling-related fields to nl43_config and nl43_status tables.
|
|
|
|
Adds to nl43_config:
|
|
- poll_interval_seconds (INTEGER, default 60)
|
|
- poll_enabled (BOOLEAN, default 1/True)
|
|
|
|
Adds to nl43_status:
|
|
- is_reachable (BOOLEAN, default 1/True)
|
|
- consecutive_failures (INTEGER, default 0)
|
|
- last_poll_attempt (DATETIME, nullable)
|
|
- last_success (DATETIME, nullable)
|
|
- last_error (TEXT, nullable)
|
|
|
|
Usage:
|
|
python migrate_add_polling_fields.py
|
|
"""
|
|
|
|
import sqlite3
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def migrate():
|
|
db_path = Path("data/slmm.db")
|
|
|
|
if not db_path.exists():
|
|
print(f"❌ Database not found at {db_path}")
|
|
print(" Run this script from the slmm directory")
|
|
return False
|
|
|
|
try:
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# Check nl43_config columns
|
|
cursor.execute("PRAGMA table_info(nl43_config)")
|
|
config_columns = [row[1] for row in cursor.fetchall()]
|
|
|
|
# Check nl43_status columns
|
|
cursor.execute("PRAGMA table_info(nl43_status)")
|
|
status_columns = [row[1] for row in cursor.fetchall()]
|
|
|
|
changes_made = False
|
|
|
|
# Add nl43_config columns
|
|
if "poll_interval_seconds" not in config_columns:
|
|
print("Adding poll_interval_seconds to nl43_config...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_config
|
|
ADD COLUMN poll_interval_seconds INTEGER DEFAULT 60
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ poll_interval_seconds already exists in nl43_config")
|
|
|
|
if "poll_enabled" not in config_columns:
|
|
print("Adding poll_enabled to nl43_config...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_config
|
|
ADD COLUMN poll_enabled BOOLEAN DEFAULT 1
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ poll_enabled already exists in nl43_config")
|
|
|
|
# Add nl43_status columns
|
|
if "is_reachable" not in status_columns:
|
|
print("Adding is_reachable to nl43_status...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_status
|
|
ADD COLUMN is_reachable BOOLEAN DEFAULT 1
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ is_reachable already exists in nl43_status")
|
|
|
|
if "consecutive_failures" not in status_columns:
|
|
print("Adding consecutive_failures to nl43_status...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_status
|
|
ADD COLUMN consecutive_failures INTEGER DEFAULT 0
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ consecutive_failures already exists in nl43_status")
|
|
|
|
if "last_poll_attempt" not in status_columns:
|
|
print("Adding last_poll_attempt to nl43_status...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_status
|
|
ADD COLUMN last_poll_attempt DATETIME
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ last_poll_attempt already exists in nl43_status")
|
|
|
|
if "last_success" not in status_columns:
|
|
print("Adding last_success to nl43_status...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_status
|
|
ADD COLUMN last_success DATETIME
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ last_success already exists in nl43_status")
|
|
|
|
if "last_error" not in status_columns:
|
|
print("Adding last_error to nl43_status...")
|
|
cursor.execute("""
|
|
ALTER TABLE nl43_status
|
|
ADD COLUMN last_error TEXT
|
|
""")
|
|
changes_made = True
|
|
else:
|
|
print("✓ last_error already exists in nl43_status")
|
|
|
|
if changes_made:
|
|
conn.commit()
|
|
print("\n✓ Migration completed successfully")
|
|
print(" Added polling-related fields to nl43_config and nl43_status")
|
|
else:
|
|
print("\n✓ All polling fields already exist - no changes needed")
|
|
|
|
conn.close()
|
|
return True
|
|
|
|
except Exception as e:
|
|
print(f"❌ Migration failed: {e}")
|
|
return False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = migrate()
|
|
sys.exit(0 if success else 1)
|