Add script/seed.py for convenience

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

View File

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