mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-23 15:57:23 -04:00
63ba3a6f30
As in other places, putting a model layer between the view and the API client makes the code cleaner and clearer.
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from flask import redirect, render_template, session, url_for
|
|
from flask_login import current_user, login_required
|
|
|
|
from app.main import main
|
|
from app.utils import PermanentRedirect
|
|
|
|
|
|
@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")
|
|
@login_required
|
|
def choose_account():
|
|
orgs_and_services = current_user.orgs_and_services
|
|
|
|
return render_template(
|
|
'views/choose-account.html',
|
|
services_without_organisations=orgs_and_services['services_without_organisations'],
|
|
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 (service_id in current_user.services 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 (organisation_id in current_user.organisation_ids or current_user.platform_admin):
|
|
return redirect(url_for('.organisation_dashboard', org_id=organisation_id))
|
|
|
|
if len(current_user.services) == 1 and not current_user.organisation_ids:
|
|
return redirect(url_for('.service_dashboard', service_id=current_user.services[0]))
|
|
|
|
if len(current_user.organisation_ids) == 1 and not current_user.services:
|
|
return redirect(url_for('.organisation_dashboard', org_id=current_user.organisation_ids[0]))
|
|
|
|
return redirect(url_for('.choose_account'))
|