316cfa84f8
- Implement migration script to add ftp_username and ftp_password columns to nl43_config table. - Create set_ftp_credentials.py script for updating FTP credentials in the database. - Update requirements.txt to include aioftp for FTP functionality. - Enhance index.html with FTP controls including enable, disable, check status, and list files features. - Add JavaScript functions for handling FTP operations and displaying file lists.
60 lines
1.8 KiB
Python
Executable File
60 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Migration script to add FTP username and password columns to nl43_config table.
|
|
Run this once to update existing database schema.
|
|
"""
|
|
|
|
import sqlite3
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
DB_PATH = Path(__file__).parent / "data" / "slmm.db"
|
|
|
|
|
|
def migrate():
|
|
"""Add ftp_username and ftp_password columns to nl43_config table."""
|
|
|
|
if not DB_PATH.exists():
|
|
print(f"Database not found at {DB_PATH}")
|
|
print("No migration needed - database will be created with new schema")
|
|
return
|
|
|
|
conn = sqlite3.connect(DB_PATH)
|
|
cursor = conn.cursor()
|
|
|
|
try:
|
|
# Check if columns already exist
|
|
cursor.execute("PRAGMA table_info(nl43_config)")
|
|
columns = [row[1] for row in cursor.fetchall()]
|
|
|
|
if "ftp_username" in columns and "ftp_password" in columns:
|
|
print("✓ FTP credential columns already exist, no migration needed")
|
|
return
|
|
|
|
# Add ftp_username column if it doesn't exist
|
|
if "ftp_username" not in columns:
|
|
print("Adding ftp_username column...")
|
|
cursor.execute("ALTER TABLE nl43_config ADD COLUMN ftp_username TEXT")
|
|
print("✓ Added ftp_username column")
|
|
|
|
# Add ftp_password column if it doesn't exist
|
|
if "ftp_password" not in columns:
|
|
print("Adding ftp_password column...")
|
|
cursor.execute("ALTER TABLE nl43_config ADD COLUMN ftp_password TEXT")
|
|
print("✓ Added ftp_password column")
|
|
|
|
conn.commit()
|
|
print("\n✓ Migration completed successfully!")
|
|
print("\nYou can now set FTP credentials via the web UI or database.")
|
|
|
|
except Exception as e:
|
|
conn.rollback()
|
|
print(f"✗ Migration failed: {e}", file=sys.stderr)
|
|
sys.exit(1)
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
migrate()
|