diff --git a/atst/routes/dev.py b/atst/routes/dev.py index cbd02cea..181ddc3d 100644 --- a/atst/routes/dev.py +++ b/atst/routes/dev.py @@ -9,37 +9,37 @@ _DEV_USERS = { "dod_id": "1234567890", "first_name": "Sam", "last_name": "Seeceepio", - "atat_role": "ccpo", + "atat_role_name": "ccpo", }, "amanda": { "dod_id": "2345678901", "first_name": "Amanda", "last_name": "Adamson", - "atat_role": "default", + "atat_role_name": "default", }, "brandon": { "dod_id": "3456789012", "first_name": "Brandon", "last_name": "Buchannan", - "atat_role": "default", + "atat_role_name": "default", }, "christina": { "dod_id": "4567890123", "first_name": "Christina", "last_name": "Collins", - "atat_role": "default", + "atat_role_name": "default", }, "dominick": { "dod_id": "5678901234", "first_name": "Dominick", "last_name": "Domingo", - "atat_role": "default", + "atat_role_name": "default", }, "erica": { "dod_id": "6789012345", "first_name": "Erica", "last_name": "Eichner", - "atat_role": "default", + "atat_role_name": "default", }, } @@ -48,8 +48,12 @@ _DEV_USERS = { def login_dev(): role = request.args.get("username", "amanda") user_data = _DEV_USERS[role] - basic_data = {k:v for k,v in user_data.items() if k not in ["dod_id", "atat_role"]} - user = _set_user_permissions(user_data["dod_id"], user_data["atat_role"], basic_data) + basic_data = { + k: v for k, v in user_data.items() if k not in ["dod_id", "atat_role"] + } + user = _set_user_permissions( + user_data["dod_id"], user_data["atat_role_name"], basic_data + ) session["user_id"] = user.id return redirect(url_for("atst.home")) diff --git a/script/seed.py b/script/seed.py new file mode 100644 index 00000000..81be936f --- /dev/null +++ b/script/seed.py @@ -0,0 +1,25 @@ +# Add root project 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) + +from atst.app import make_config, make_app +from atst.domain.users import Users +from atst.domain.requests import Requests +from atst.routes.dev import _DEV_USERS as DEV_USERS + + +def seed_db(): + users = [Users.create(**dev_user) for (_, dev_user) in DEV_USERS.items()] + + for user in users: + [Requests.create(user, {}) for _ in range(5)] + + +if __name__ == "__main__": + config = make_config() + app = make_app(config) + with app.app_context(): + seed_db()