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()