mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-09-08 20:53:11 -04:00
notify-api-412 use black to enforce python coding style
This commit is contained in:
Kenneth Kehl
@@ -38,156 +38,156 @@ from app.utils.csv import Spreadsheet
|
||||
from app.utils.user import user_has_permissions, user_is_platform_admin
|
||||
|
||||
|
||||
@main.route("/organizations", methods=['GET'])
|
||||
@main.route("/organizations", methods=["GET"])
|
||||
@user_is_platform_admin
|
||||
def organizations():
|
||||
return render_template(
|
||||
'views/organizations/index.html',
|
||||
"views/organizations/index.html",
|
||||
organizations=AllOrganizations(),
|
||||
search_form=SearchByNameForm(),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/add", methods=['GET', 'POST'])
|
||||
@main.route("/organizations/add", methods=["GET", "POST"])
|
||||
@user_is_platform_admin
|
||||
def add_organization():
|
||||
form = AdminNewOrganizationForm()
|
||||
|
||||
if form.validate_on_submit():
|
||||
try:
|
||||
return redirect(url_for(
|
||||
'.organization_settings',
|
||||
org_id=Organization.create_from_form(form).id,
|
||||
))
|
||||
return redirect(
|
||||
url_for(
|
||||
".organization_settings",
|
||||
org_id=Organization.create_from_form(form).id,
|
||||
)
|
||||
)
|
||||
except HTTPError as e:
|
||||
msg = 'Organization name already exists'
|
||||
msg = "Organization name already exists"
|
||||
if e.status_code == 400 and msg in e.message:
|
||||
form.name.errors.append("This organization name is already in use")
|
||||
else:
|
||||
raise e
|
||||
|
||||
return render_template(
|
||||
'views/organizations/add-organization.html',
|
||||
form=form
|
||||
)
|
||||
return render_template("views/organizations/add-organization.html", form=form)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>", methods=["GET"])
|
||||
@user_has_permissions()
|
||||
def organization_dashboard(org_id):
|
||||
year, current_financial_year = requested_and_current_financial_year(request)
|
||||
services = current_organization.services_and_usage(
|
||||
financial_year=year
|
||||
)['services']
|
||||
services = current_organization.services_and_usage(financial_year=year)["services"]
|
||||
return render_template(
|
||||
'views/organizations/organization/index.html',
|
||||
"views/organizations/organization/index.html",
|
||||
services=services,
|
||||
years=get_tuples_of_financial_years(
|
||||
partial(url_for, '.organization_dashboard', org_id=current_organization.id),
|
||||
partial(url_for, ".organization_dashboard", org_id=current_organization.id),
|
||||
start=current_financial_year - 2,
|
||||
end=current_financial_year,
|
||||
),
|
||||
selected_year=year,
|
||||
search_form=SearchByNameForm() if len(services) > 7 else None,
|
||||
**{
|
||||
f'total_{key}': sum(service[key] for service in services)
|
||||
for key in ('emails_sent', 'sms_cost')
|
||||
f"total_{key}": sum(service[key] for service in services)
|
||||
for key in ("emails_sent", "sms_cost")
|
||||
},
|
||||
download_link=url_for(
|
||||
'.download_organization_usage_report',
|
||||
org_id=org_id,
|
||||
selected_year=year
|
||||
)
|
||||
".download_organization_usage_report", org_id=org_id, selected_year=year
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/download-usage-report.csv", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>/download-usage-report.csv", methods=["GET"])
|
||||
@user_has_permissions()
|
||||
def download_organization_usage_report(org_id):
|
||||
selected_year = request.args.get('selected_year')
|
||||
selected_year = request.args.get("selected_year")
|
||||
services_usage = current_organization.services_and_usage(
|
||||
financial_year=selected_year
|
||||
)['services']
|
||||
)["services"]
|
||||
|
||||
unit_column_names = OrderedDict([
|
||||
('service_id', 'Service ID'),
|
||||
('service_name', 'Service Name'),
|
||||
('emails_sent', 'Emails sent'),
|
||||
('sms_remainder', 'Free text message allowance remaining'),
|
||||
])
|
||||
unit_column_names = OrderedDict(
|
||||
[
|
||||
("service_id", "Service ID"),
|
||||
("service_name", "Service Name"),
|
||||
("emails_sent", "Emails sent"),
|
||||
("sms_remainder", "Free text message allowance remaining"),
|
||||
]
|
||||
)
|
||||
|
||||
monetary_column_names = OrderedDict([
|
||||
('sms_cost', 'Spent on text messages ($)'),
|
||||
])
|
||||
monetary_column_names = OrderedDict(
|
||||
[
|
||||
("sms_cost", "Spent on text messages ($)"),
|
||||
]
|
||||
)
|
||||
|
||||
org_usage_data = [
|
||||
list(unit_column_names.values()) + list(monetary_column_names.values())
|
||||
] + [
|
||||
[
|
||||
service[attribute] for attribute in unit_column_names.keys()
|
||||
] + [
|
||||
'{:,.2f}'.format(service[attribute]) for attribute in monetary_column_names.keys()
|
||||
[service[attribute] for attribute in unit_column_names.keys()]
|
||||
+ [
|
||||
"{:,.2f}".format(service[attribute])
|
||||
for attribute in monetary_column_names.keys()
|
||||
]
|
||||
for service in services_usage
|
||||
]
|
||||
|
||||
return Spreadsheet.from_rows(org_usage_data).as_csv_data, 200, {
|
||||
'Content-Type': 'text/csv; charset=utf-8',
|
||||
'Content-Disposition': (
|
||||
'inline;'
|
||||
'filename="{} organization usage report for year {}'
|
||||
' - generated on {}.csv"'.format(
|
||||
current_organization.name,
|
||||
selected_year,
|
||||
datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%fZ")
|
||||
))
|
||||
}
|
||||
return (
|
||||
Spreadsheet.from_rows(org_usage_data).as_csv_data,
|
||||
200,
|
||||
{
|
||||
"Content-Type": "text/csv; charset=utf-8",
|
||||
"Content-Disposition": (
|
||||
"inline;"
|
||||
'filename="{} organization usage report for year {}'
|
||||
' - generated on {}.csv"'.format(
|
||||
current_organization.name,
|
||||
selected_year,
|
||||
datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
|
||||
)
|
||||
),
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/trial-services", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>/trial-services", methods=["GET"])
|
||||
@user_is_platform_admin
|
||||
def organization_trial_mode_services(org_id):
|
||||
return render_template(
|
||||
'views/organizations/organization/trial-mode-services.html',
|
||||
"views/organizations/organization/trial-mode-services.html",
|
||||
search_form=SearchByNameForm(),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/users", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>/users", methods=["GET"])
|
||||
@user_has_permissions()
|
||||
def manage_org_users(org_id):
|
||||
return render_template(
|
||||
'views/organizations/organization/users/index.html',
|
||||
"views/organizations/organization/users/index.html",
|
||||
users=current_organization.team_members,
|
||||
show_search_box=(len(current_organization.team_members) > 7),
|
||||
form=SearchUsersForm(),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/users/invite", methods=['GET', 'POST'])
|
||||
@main.route("/organizations/<uuid:org_id>/users/invite", methods=["GET", "POST"])
|
||||
@user_has_permissions()
|
||||
def invite_org_user(org_id):
|
||||
form = InviteOrgUserForm(
|
||||
inviter_email_address=current_user.email_address
|
||||
)
|
||||
form = InviteOrgUserForm(inviter_email_address=current_user.email_address)
|
||||
if form.validate_on_submit():
|
||||
email_address = form.email_address.data
|
||||
invited_org_user = InvitedOrgUser.create(
|
||||
current_user.id,
|
||||
org_id,
|
||||
email_address
|
||||
)
|
||||
invited_org_user = InvitedOrgUser.create(current_user.id, org_id, email_address)
|
||||
|
||||
flash('Invite sent to {}'.format(invited_org_user.email_address), 'default_with_tick')
|
||||
return redirect(url_for('.manage_org_users', org_id=org_id))
|
||||
flash(
|
||||
"Invite sent to {}".format(invited_org_user.email_address),
|
||||
"default_with_tick",
|
||||
)
|
||||
return redirect(url_for(".manage_org_users", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/users/invite-org-user.html',
|
||||
form=form
|
||||
"views/organizations/organization/users/invite-org-user.html", form=form
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/users/<uuid:user_id>", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>/users/<uuid:user_id>", methods=["GET"])
|
||||
@user_has_permissions()
|
||||
def edit_organization_user(org_id, user_id):
|
||||
# The only action that can be done to an org user is to remove them from the org.
|
||||
@@ -195,69 +195,77 @@ def edit_organization_user(org_id, user_id):
|
||||
# query string, but it uses the template for all org team members in order to avoid
|
||||
# having a page containing a single link.
|
||||
return render_template(
|
||||
'views/organizations/organization/users/index.html',
|
||||
"views/organizations/organization/users/index.html",
|
||||
users=current_organization.team_members,
|
||||
show_search_box=(len(current_organization.team_members) > 7),
|
||||
form=SearchUsersForm(),
|
||||
user_to_remove=User.from_id(user_id)
|
||||
user_to_remove=User.from_id(user_id),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/users/<uuid:user_id>/delete", methods=['POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/users/<uuid:user_id>/delete", methods=["POST"]
|
||||
)
|
||||
@user_has_permissions()
|
||||
def remove_user_from_organization(org_id, user_id):
|
||||
organizations_client.remove_user_from_organization(org_id, user_id)
|
||||
|
||||
return redirect(url_for('.show_accounts_or_dashboard'))
|
||||
return redirect(url_for(".show_accounts_or_dashboard"))
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/cancel-invited-user/<uuid:invited_user_id>", methods=['GET'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/cancel-invited-user/<uuid:invited_user_id>",
|
||||
methods=["GET"],
|
||||
)
|
||||
@user_has_permissions()
|
||||
def cancel_invited_org_user(org_id, invited_user_id):
|
||||
org_invite_api_client.cancel_invited_user(org_id=org_id, invited_user_id=invited_user_id)
|
||||
org_invite_api_client.cancel_invited_user(
|
||||
org_id=org_id, invited_user_id=invited_user_id
|
||||
)
|
||||
|
||||
invited_org_user = InvitedOrgUser.by_id_and_org_id(org_id, invited_user_id)
|
||||
|
||||
flash(f'Invitation cancelled for {invited_org_user.email_address}', 'default_with_tick')
|
||||
return redirect(url_for('main.manage_org_users', org_id=org_id))
|
||||
flash(
|
||||
f"Invitation cancelled for {invited_org_user.email_address}",
|
||||
"default_with_tick",
|
||||
)
|
||||
return redirect(url_for("main.manage_org_users", org_id=org_id))
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/", methods=['GET'])
|
||||
@main.route("/organizations/<uuid:org_id>/settings/", methods=["GET"])
|
||||
@user_is_platform_admin
|
||||
def organization_settings(org_id):
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/index.html',
|
||||
"views/organizations/organization/settings/index.html",
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-name", methods=['GET', 'POST'])
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-name", methods=["GET", "POST"])
|
||||
@user_is_platform_admin
|
||||
def edit_organization_name(org_id):
|
||||
form = RenameOrganizationForm(name=current_organization.name)
|
||||
|
||||
if form.validate_on_submit():
|
||||
|
||||
try:
|
||||
current_organization.update(name=form.name.data)
|
||||
except HTTPError as http_error:
|
||||
error_msg = 'Organization name already exists'
|
||||
error_msg = "Organization name already exists"
|
||||
if http_error.status_code == 400 and error_msg in http_error.message:
|
||||
form.name.errors.append('This organization name is already in use')
|
||||
form.name.errors.append("This organization name is already in use")
|
||||
else:
|
||||
raise http_error
|
||||
else:
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-name.html',
|
||||
"views/organizations/organization/settings/edit-name.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-type", methods=['GET', 'POST'])
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-type", methods=["GET", "POST"])
|
||||
@user_is_platform_admin
|
||||
def edit_organization_type(org_id):
|
||||
|
||||
form = OrganizationOrganizationTypeForm(
|
||||
organization_type=current_organization.organization_type
|
||||
)
|
||||
@@ -267,18 +275,19 @@ def edit_organization_type(org_id):
|
||||
organization_type=form.organization_type.data,
|
||||
delete_services_cache=True,
|
||||
)
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-type.html',
|
||||
"views/organizations/organization/settings/edit-type.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/set-email-branding", methods=['GET', 'POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/settings/set-email-branding", methods=["GET", "POST"]
|
||||
)
|
||||
@user_is_platform_admin
|
||||
def edit_organization_email_branding(org_id):
|
||||
|
||||
email_branding = email_branding_client.get_all_email_branding()
|
||||
|
||||
form = AdminSetEmailBrandingForm(
|
||||
@@ -287,24 +296,28 @@ def edit_organization_email_branding(org_id):
|
||||
)
|
||||
|
||||
if form.validate_on_submit():
|
||||
return redirect(url_for(
|
||||
'.organization_preview_email_branding',
|
||||
org_id=org_id,
|
||||
branding_style=form.branding_style.data,
|
||||
))
|
||||
return redirect(
|
||||
url_for(
|
||||
".organization_preview_email_branding",
|
||||
org_id=org_id,
|
||||
branding_style=form.branding_style.data,
|
||||
)
|
||||
)
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/set-email-branding.html',
|
||||
"views/organizations/organization/settings/set-email-branding.html",
|
||||
form=form,
|
||||
search_form=SearchByNameForm()
|
||||
search_form=SearchByNameForm(),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/preview-email-branding", methods=['GET', 'POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/settings/preview-email-branding",
|
||||
methods=["GET", "POST"],
|
||||
)
|
||||
@user_is_platform_admin
|
||||
def organization_preview_email_branding(org_id):
|
||||
|
||||
branding_style = request.args.get('branding_style', None)
|
||||
branding_style = request.args.get("branding_style", None)
|
||||
|
||||
form = AdminPreviewBrandingForm(branding_style=branding_style)
|
||||
|
||||
@@ -313,94 +326,97 @@ def organization_preview_email_branding(org_id):
|
||||
email_branding_id=form.branding_style.data,
|
||||
delete_services_cache=True,
|
||||
)
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/preview-email-branding.html',
|
||||
"views/organizations/organization/settings/preview-email-branding.html",
|
||||
form=form,
|
||||
action=url_for('main.organization_preview_email_branding', org_id=org_id),
|
||||
action=url_for("main.organization_preview_email_branding", org_id=org_id),
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-organization-domains", methods=['GET', 'POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/settings/edit-organization-domains",
|
||||
methods=["GET", "POST"],
|
||||
)
|
||||
@user_is_platform_admin
|
||||
def edit_organization_domains(org_id):
|
||||
|
||||
form = AdminOrganizationDomainsForm()
|
||||
|
||||
if form.validate_on_submit():
|
||||
try:
|
||||
organizations_client.update_organization(
|
||||
org_id,
|
||||
domains=list(OrderedDict.fromkeys(
|
||||
domain.lower()
|
||||
for domain in filter(None, form.domains.data)
|
||||
)),
|
||||
domains=list(
|
||||
OrderedDict.fromkeys(
|
||||
domain.lower() for domain in filter(None, form.domains.data)
|
||||
)
|
||||
),
|
||||
)
|
||||
except HTTPError as e:
|
||||
error_message = "Domain already exists"
|
||||
if e.status_code == 400 and error_message in e.message:
|
||||
flash("This domain is already in use", "error")
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-domains.html',
|
||||
"views/organizations/organization/settings/edit-domains.html",
|
||||
form=form,
|
||||
)
|
||||
else:
|
||||
raise e
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
form.populate(current_organization.domains)
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-domains.html',
|
||||
"views/organizations/organization/settings/edit-domains.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-go-live-notes", methods=['GET', 'POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/settings/edit-go-live-notes", methods=["GET", "POST"]
|
||||
)
|
||||
@user_is_platform_admin
|
||||
def edit_organization_go_live_notes(org_id):
|
||||
|
||||
form = AdminOrganizationGoLiveNotesForm()
|
||||
|
||||
if form.validate_on_submit():
|
||||
organizations_client.update_organization(
|
||||
org_id,
|
||||
request_to_go_live_notes=form.request_to_go_live_notes.data
|
||||
org_id, request_to_go_live_notes=form.request_to_go_live_notes.data
|
||||
)
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
org = organizations_client.get_organization(org_id)
|
||||
form.request_to_go_live_notes.data = org['request_to_go_live_notes']
|
||||
form.request_to_go_live_notes.data = org["request_to_go_live_notes"]
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-go-live-notes.html',
|
||||
"views/organizations/organization/settings/edit-go-live-notes.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/notes", methods=['GET', 'POST'])
|
||||
@main.route("/organizations/<uuid:org_id>/settings/notes", methods=["GET", "POST"])
|
||||
@user_is_platform_admin
|
||||
def edit_organization_notes(org_id):
|
||||
form = AdminNotesForm(notes=current_organization.notes)
|
||||
|
||||
if form.validate_on_submit():
|
||||
|
||||
if form.notes.data == current_organization.notes:
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
current_organization.update(
|
||||
notes=form.notes.data
|
||||
)
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
current_organization.update(notes=form.notes.data)
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-organization-notes.html',
|
||||
"views/organizations/organization/settings/edit-organization-notes.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
|
||||
@main.route("/organizations/<uuid:org_id>/settings/edit-billing-details", methods=['GET', 'POST'])
|
||||
@main.route(
|
||||
"/organizations/<uuid:org_id>/settings/edit-billing-details",
|
||||
methods=["GET", "POST"],
|
||||
)
|
||||
@user_is_platform_admin
|
||||
def edit_organization_billing_details(org_id):
|
||||
form = AdminBillingDetailsForm(
|
||||
@@ -419,10 +435,10 @@ def edit_organization_billing_details(org_id):
|
||||
purchase_order_number=form.purchase_order_number.data,
|
||||
notes=form.notes.data,
|
||||
)
|
||||
return redirect(url_for('.organization_settings', org_id=org_id))
|
||||
return redirect(url_for(".organization_settings", org_id=org_id))
|
||||
|
||||
return render_template(
|
||||
'views/organizations/organization/settings/edit-organization-billing-details.html',
|
||||
"views/organizations/organization/settings/edit-organization-billing-details.html",
|
||||
form=form,
|
||||
)
|
||||
|
||||
@@ -430,6 +446,4 @@ def edit_organization_billing_details(org_id):
|
||||
@main.route("/organizations/<uuid:org_id>/billing")
|
||||
@user_is_platform_admin
|
||||
def organization_billing(org_id):
|
||||
return render_template(
|
||||
'views/organizations/organization/billing.html'
|
||||
)
|
||||
return render_template("views/organizations/organization/billing.html")
|
||||
|
||||
Reference in New Issue
Block a user