61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from flask import Blueprint, request, session, redirect, url_for
|
|
from flask import current_app as app
|
|
|
|
from atst.domain.users import Users
|
|
|
|
bp = Blueprint("dev", __name__)
|
|
|
|
_DEV_USERS = {
|
|
"sam": {
|
|
"dod_id": "1234567890",
|
|
"first_name": "Sam",
|
|
"last_name": "Seeceepio",
|
|
"atat_role": "ccpo",
|
|
},
|
|
"amanda": {
|
|
"dod_id": "2345678901",
|
|
"first_name": "Amanda",
|
|
"last_name": "Adamson",
|
|
"atat_role": "default",
|
|
},
|
|
"brandon": {
|
|
"dod_id": "3456789012",
|
|
"first_name": "Brandon",
|
|
"last_name": "Buchannan",
|
|
"atat_role": "default",
|
|
},
|
|
"christina": {
|
|
"dod_id": "4567890123",
|
|
"first_name": "Christina",
|
|
"last_name": "Collins",
|
|
"atat_role": "default",
|
|
},
|
|
"dominick": {
|
|
"dod_id": "5678901234",
|
|
"first_name": "Dominick",
|
|
"last_name": "Domingo",
|
|
"atat_role": "default",
|
|
},
|
|
"erica": {
|
|
"dod_id": "6789012345",
|
|
"first_name": "Erica",
|
|
"last_name": "Eichner",
|
|
"atat_role": "default",
|
|
},
|
|
}
|
|
|
|
|
|
@bp.route("/login-dev")
|
|
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)
|
|
session["user_id"] = user.id
|
|
app.logger.warning(session)
|
|
return redirect(url_for("atst.home"))
|
|
|
|
|
|
def _set_user_permissions(dod_id, role, user_data):
|
|
return Users.get_or_create_by_dod_id(dod_id, atat_role_name=role, **user_data)
|