pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add script/seed.py for convenience

Browse Source
This commit is contained in:
richard-dds 2018-08-07 21:32:02 -04:00
parent dcb45c64e8
commit f80668c638
2 changed files with 37 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
atst/routes/dev.py
View File

@ -9,37 +9,37 @@ _DEV_USERS = {
"dod_id": "1234567890", "dod_id": "1234567890",
"first_name": "Sam", "first_name": "Sam",
"last_name": "Seeceepio", "last_name": "Seeceepio",
"atat_role": "ccpo", "atat_role_name": "ccpo",
}, },
"amanda": { "amanda": {
"dod_id": "2345678901", "dod_id": "2345678901",
"first_name": "Amanda", "first_name": "Amanda",
"last_name": "Adamson", "last_name": "Adamson",
"atat_role": "default", "atat_role_name": "default",
}, },
"brandon": { "brandon": {
"dod_id": "3456789012", "dod_id": "3456789012",
"first_name": "Brandon", "first_name": "Brandon",
"last_name": "Buchannan", "last_name": "Buchannan",
"atat_role": "default", "atat_role_name": "default",
}, },
"christina": { "christina": {
"dod_id": "4567890123", "dod_id": "4567890123",
"first_name": "Christina", "first_name": "Christina",
"last_name": "Collins", "last_name": "Collins",
"atat_role": "default", "atat_role_name": "default",
}, },
"dominick": { "dominick": {
"dod_id": "5678901234", "dod_id": "5678901234",
"first_name": "Dominick", "first_name": "Dominick",
"last_name": "Domingo", "last_name": "Domingo",
"atat_role": "default", "atat_role_name": "default",
}, },
"erica": { "erica": {
"dod_id": "6789012345", "dod_id": "6789012345",
"first_name": "Erica", "first_name": "Erica",
"last_name": "Eichner", "last_name": "Eichner",
"atat_role": "default", "atat_role_name": "default",
}, },
} }
@ -48,8 +48,12 @@ _DEV_USERS = {
def login_dev(): def login_dev():
role = request.args.get("username", "amanda") role = request.args.get("username", "amanda")
user_data = _DEV_USERS[role] user_data = _DEV_USERS[role]
basic_data = {k:v for k,v in user_data.items() if k not in ["dod_id", "atat_role"]} basic_data = {
user = _set_user_permissions(user_data["dod_id"], user_data["atat_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_name"], basic_data
)
session["user_id"] = user.id session["user_id"] = user.id
return redirect(url_for("atst.home")) return redirect(url_for("atst.home"))

25
script/seed.py Normal file
View File

@ -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()