compass/backend/script/reset_demo.py

34 lines
991 B
Python

from sqlalchemy import create_engine
from sqlalchemy.orm import Session
import subprocess
from ..database import engine, _engine_str
from ..env import getenv
from .. import entities
from ..test.services import (
user_test_data,
service_test_data,
resource_test_data,
tag_test_data,
)
database = getenv("POSTGRES_DATABASE")
engine = create_engine(_engine_str(), echo=True)
"""Application-level SQLAlchemy database engine."""
# Run Delete and Create Database Scripts
subprocess.run(["python3", "-m", "backend.script.delete_database"])
subprocess.run(["python3", "-m", "backend.script.create_database"])
entities.EntityBase.metadata.drop_all(engine)
entities.EntityBase.metadata.create_all(engine)
with Session(engine) as session:
user_test_data.insert_test_data(session)
service_test_data.insert_test_data(session)
tag_test_data.insert_test_data(session)
resource_test_data.insert_test_data(session)
session.commit()