144 lines
4.1 KiB
Python
144 lines
4.1 KiB
Python
from flask import (
|
|
Blueprint,
|
|
request,
|
|
session,
|
|
redirect,
|
|
render_template,
|
|
url_for,
|
|
current_app as app,
|
|
)
|
|
import pendulum
|
|
|
|
from . import redirect_after_login_url
|
|
from atst.domain.users import Users
|
|
from atst.domain.permission_sets import PermissionSets
|
|
from atst.queue import queue
|
|
from tests.factories import random_service_branch
|
|
from atst.utils import pick
|
|
|
|
bp = Blueprint("dev", __name__)
|
|
|
|
_ALL_PERMS = [
|
|
"ccpo",
|
|
PermissionSets.VIEW_PORTFOLIO,
|
|
PermissionSets.VIEW_PORTFOLIO_APPLICATION_MANAGEMENT,
|
|
PermissionSets.VIEW_PORTFOLIO_FUNDING,
|
|
PermissionSets.VIEW_PORTFOLIO_REPORTS,
|
|
PermissionSets.VIEW_PORTFOLIO_ADMIN,
|
|
PermissionSets.EDIT_PORTFOLIO_APPLICATION_MANAGEMENT,
|
|
PermissionSets.EDIT_PORTFOLIO_FUNDING,
|
|
PermissionSets.EDIT_PORTFOLIO_REPORTS,
|
|
PermissionSets.EDIT_PORTFOLIO_ADMIN,
|
|
PermissionSets.PORTFOLIO_POC,
|
|
]
|
|
|
|
_DEV_USERS = {
|
|
"sam": {
|
|
"dod_id": "6346349876",
|
|
"first_name": "Sam",
|
|
"last_name": "Stevenson",
|
|
"permission_sets": _ALL_PERMS,
|
|
"email": "sam@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
"amanda": {
|
|
"dod_id": "2345678901",
|
|
"first_name": "Amanda",
|
|
"last_name": "Adamson",
|
|
"email": "amanda@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
"brandon": {
|
|
"dod_id": "3456789012",
|
|
"first_name": "Brandon",
|
|
"last_name": "Buchannan",
|
|
"email": "brandon@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
"christina": {
|
|
"dod_id": "4567890123",
|
|
"first_name": "Christina",
|
|
"last_name": "Collins",
|
|
"email": "christina@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
"dominick": {
|
|
"dod_id": "5678901234",
|
|
"first_name": "Dominick",
|
|
"last_name": "Domingo",
|
|
"email": "dominick@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
"erica": {
|
|
"dod_id": "6789012345",
|
|
"first_name": "Erica",
|
|
"last_name": "Eichner",
|
|
"email": "erica@example.com",
|
|
"service_branch": random_service_branch(),
|
|
"phone_number": "1234567890",
|
|
"citizenship": "United States",
|
|
"designation": "Military",
|
|
"date_latest_training": pendulum.date(2018, 1, 1),
|
|
},
|
|
}
|
|
|
|
|
|
@bp.route("/login-dev")
|
|
def login_dev():
|
|
role = request.args.get("username", "amanda")
|
|
user_data = _DEV_USERS[role]
|
|
user = Users.get_or_create_by_dod_id(
|
|
user_data["dod_id"],
|
|
**pick(
|
|
[
|
|
"permission_sets",
|
|
"first_name",
|
|
"last_name",
|
|
"email",
|
|
"service_branch",
|
|
"phone_number",
|
|
"citizenship",
|
|
"designation",
|
|
"date_latest_training",
|
|
],
|
|
user_data,
|
|
),
|
|
)
|
|
session["user_id"] = user.id
|
|
|
|
return redirect(redirect_after_login_url())
|
|
|
|
|
|
@bp.route("/test-email")
|
|
def test_email():
|
|
queue.send_mail(
|
|
[request.args.get("to")], request.args.get("subject"), request.args.get("body")
|
|
)
|
|
|
|
return redirect(url_for("dev.messages"))
|
|
|
|
|
|
@bp.route("/messages")
|
|
def messages():
|
|
return render_template("dev/emails.html", messages=app.mailer.messages)
|