mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-24 16:24:08 -04:00
b620b677d3
Rather than force us to write the decorators in a specific order let’s just have one decorator call the other. This should make fewer lines of code, and fewer annoying test failures. It also means that the same way of raising a `401` (through the `current_app` method) is used everywhere.
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from flask import redirect, render_template, session, url_for
|
|
from flask_login import current_user
|
|
|
|
from app.main import main
|
|
from app.utils import PermanentRedirect, user_is_logged_in
|
|
|
|
|
|
@main.route("/services")
|
|
def choose_service():
|
|
raise PermanentRedirect(url_for('.choose_account'))
|
|
|
|
|
|
@main.route("/services-or-dashboard")
|
|
def services_or_dashboard():
|
|
raise PermanentRedirect(url_for('.show_accounts_or_dashboard'))
|
|
|
|
|
|
@main.route("/accounts")
|
|
@user_is_logged_in
|
|
def choose_account():
|
|
return render_template(
|
|
'views/choose-account.html',
|
|
can_add_service=current_user.is_gov_user,
|
|
)
|
|
|
|
|
|
@main.route("/accounts-or-dashboard")
|
|
def show_accounts_or_dashboard():
|
|
|
|
if not current_user.is_authenticated:
|
|
return redirect(url_for('.index'))
|
|
|
|
service_id = session.get('service_id')
|
|
if service_id and (current_user.belongs_to_service(service_id) or current_user.platform_admin):
|
|
return redirect(url_for('.service_dashboard', service_id=service_id))
|
|
|
|
organisation_id = session.get('organisation_id')
|
|
if organisation_id and (current_user.belongs_to_organisation(organisation_id) or current_user.platform_admin):
|
|
return redirect(url_for('.organisation_dashboard', org_id=organisation_id))
|
|
|
|
if len(current_user.service_ids) == 1 and not current_user.organisation_ids:
|
|
return redirect(url_for('.service_dashboard', service_id=current_user.service_ids[0]))
|
|
|
|
if len(current_user.organisation_ids) == 1 and not current_user.trial_mode_services:
|
|
return redirect(url_for('.organisation_dashboard', org_id=current_user.organisation_ids[0]))
|
|
|
|
return redirect(url_for('.choose_account'))
|