compass/backend/script/reset_demo.py
pmoharana-cmd 6d477678a9 Demo ready
2024-04-24 21:23:43 -04:00

28 lines
905 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
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)
resource_test_data.insert_test_data(session)
session.commit()