mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
from flask import (render_template, redirect, url_for, session)
|
|
from flask_login import login_required, current_user
|
|
from app.main import main
|
|
from app import service_api_client
|
|
from app.notify_client.api_client import ServicesBrowsableItem
|
|
|
|
|
|
@main.route("/services")
|
|
@login_required
|
|
def choose_service():
|
|
return render_template(
|
|
'views/choose-service.html',
|
|
services=[ServicesBrowsableItem(x) for x in
|
|
service_api_client.get_services({'user_id': current_user.id})['data']]
|
|
)
|
|
|
|
|
|
@main.route("/services-or-dashboard")
|
|
@login_required
|
|
def show_all_services_or_dashboard():
|
|
services = service_api_client.get_services({'user_id': current_user.id})['data']
|
|
|
|
if 1 == len(services):
|
|
return redirect(url_for('.service_dashboard', service_id=services[0]['id']))
|
|
else:
|
|
service_id = session.get('service_id', None)
|
|
if any([service_id == x['id'] for x in services]):
|
|
return redirect(url_for('.service_dashboard', service_id=service_id))
|
|
return redirect(url_for('.choose_service'))
|