from database import get_db, init_db
from config import IS_STAGING

def seed():
    """Seed staging DB with test data. Never runs on prod."""
    if not IS_STAGING:
        print("Refusing to seed prod database")
        return
    
    init_db()
    conn = get_db()
    # Insert test documents from TEST_FAMILY_CONTEXT
    conn.close()
    print("Staging database seeded")

if __name__ == "__main__":
    seed()