cd3cb7b614
This updates the script for resetting the database so that it drops and recreates all the tables, instead of disabling Postgres triggers and truncating most of the tables. The latter strategy requires superuser permissions in Postgres that the db user we manage in Azure does not have. The script now: - drops the tables - reruns the alembic migrations - reseeds the permission sets
38 lines
848 B
Python
38 lines
848 B
Python
# Add root application dir to the python path
|
|
import os
|
|
import sys
|
|
|
|
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
|
sys.path.append(parent_dir)
|
|
|
|
import sqlalchemy
|
|
from alembic import config as alembic_config
|
|
|
|
from seed_roles import seed_roles
|
|
from atst.database import db
|
|
from atst.app import make_config, make_app
|
|
|
|
|
|
def reset_database():
|
|
conn = db.engine.connect()
|
|
|
|
meta = sqlalchemy.MetaData(bind=conn, reflect=True)
|
|
trans = conn.begin()
|
|
|
|
# drop all tables
|
|
meta.drop_all()
|
|
trans.commit()
|
|
|
|
# rerun the migrations
|
|
alembic_config.main(argv=["upgrade", "head"])
|
|
|
|
# seed the permission sets
|
|
seed_roles()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
config = make_config({"DISABLE_CRL_CHECK": True, "DEBUG": False})
|
|
app = make_app(config)
|
|
with app.app_context():
|
|
reset_database()
|