116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Create a fresh test database with the new schema and some sample data.
|
|
"""
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from datetime import datetime, date, timedelta
|
|
from backend.models import Base, RosterUnit, Emitter
|
|
|
|
# Create a new test database
|
|
TEST_DB_PATH = "/tmp/sfm_test.db"
|
|
engine = create_engine(f"sqlite:///{TEST_DB_PATH}", connect_args={"check_same_thread": False})
|
|
|
|
# Drop all tables and recreate them with the new schema
|
|
Base.metadata.drop_all(bind=engine)
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
# Create a session
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# Add some test seismographs
|
|
seismo1 = RosterUnit(
|
|
id="BE9449",
|
|
device_type="seismograph",
|
|
unit_type="series3",
|
|
deployed=True,
|
|
note="Primary field unit",
|
|
project_id="PROJ-001",
|
|
location="Site A",
|
|
last_calibrated=date(2024, 1, 15),
|
|
next_calibration_due=date(2025, 1, 15),
|
|
deployed_with_modem_id="MDM001",
|
|
last_updated=datetime.utcnow(),
|
|
)
|
|
|
|
seismo2 = RosterUnit(
|
|
id="BE9450",
|
|
device_type="seismograph",
|
|
unit_type="series3",
|
|
deployed=False,
|
|
note="Benched for maintenance",
|
|
project_id="PROJ-001",
|
|
location="Warehouse",
|
|
last_calibrated=date(2023, 6, 20),
|
|
next_calibration_due=date(2024, 6, 20), # Past due
|
|
last_updated=datetime.utcnow(),
|
|
)
|
|
|
|
# Add some test modems
|
|
modem1 = RosterUnit(
|
|
id="MDM001",
|
|
device_type="modem",
|
|
unit_type="modem",
|
|
deployed=True,
|
|
note="Paired with BE9449",
|
|
project_id="PROJ-001",
|
|
location="Site A",
|
|
ip_address="192.168.1.100",
|
|
phone_number="+1-555-0123",
|
|
hardware_model="Raven XTV",
|
|
last_updated=datetime.utcnow(),
|
|
)
|
|
|
|
modem2 = RosterUnit(
|
|
id="MDM002",
|
|
device_type="modem",
|
|
unit_type="modem",
|
|
deployed=False,
|
|
note="Spare modem",
|
|
project_id="PROJ-001",
|
|
location="Warehouse",
|
|
ip_address="192.168.1.101",
|
|
phone_number="+1-555-0124",
|
|
hardware_model="Raven XT",
|
|
last_updated=datetime.utcnow(),
|
|
)
|
|
|
|
# Add test emitters (status reports)
|
|
emitter1 = Emitter(
|
|
id="BE9449",
|
|
unit_type="series3",
|
|
last_seen=datetime.utcnow() - timedelta(hours=2),
|
|
last_file="BE9449.2024.336.12.00.mseed",
|
|
status="OK",
|
|
notes="Running normally",
|
|
)
|
|
|
|
emitter2 = Emitter(
|
|
id="BE9450",
|
|
unit_type="series3",
|
|
last_seen=datetime.utcnow() - timedelta(days=30),
|
|
last_file="BE9450.2024.306.08.00.mseed",
|
|
status="Missing",
|
|
notes="No data received",
|
|
)
|
|
|
|
# Add all units
|
|
db.add_all([seismo1, seismo2, modem1, modem2, emitter1, emitter2])
|
|
db.commit()
|
|
|
|
print(f"✓ Test database created at {TEST_DB_PATH}")
|
|
print(f"✓ Added 2 seismographs (BE9449, BE9450)")
|
|
print(f"✓ Added 2 modems (MDM001, MDM002)")
|
|
print(f"✓ Added 2 emitter status reports")
|
|
print(f"\nDatabase is ready for testing!")
|
|
|
|
except Exception as e:
|
|
print(f"Error creating test database: {e}")
|
|
db.rollback()
|
|
raise
|
|
finally:
|
|
db.close()
|