49a1a219ae
This script is for bootstrapping the initial database. It can be run via a container, but requires that a Postgres superuser's credentials be provided via our normal config. That way the superuser can provision a less-privileged user for the application's database connection.
40 lines
854 B
Python
40 lines
854 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)
|
|
meta.reflect()
|
|
|
|
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()
|