diff --git a/app/main/views/choose_service.py b/app/main/views/choose_service.py index 39b0b9f8e..5ddaf61b0 100644 --- a/app/main/views/choose_service.py +++ b/app/main/views/choose_service.py @@ -1,13 +1,29 @@ from flask import (render_template, redirect, url_for, session) from flask_login import login_required, current_user -from app.main.dao import services_dao +from app.main.dao.services_dao import ServicesBrowsableItem +from app import service_api_client from app.main import main @main.route("/services") @login_required def choose_service(): - services = services_dao.get_services(current_user.id) return render_template( 'views/choose-service.html', - services=[services_dao.ServicesBrowsableItem(x) for x in services['data']]) + services=[ServicesBrowsableItem(x) for x in service_api_client.get_services()['data']] + ) + + +@main.route("/services-or-dashboard") +def show_all_services_or_dashboard(): + + if current_user.is_authenticated(): + + services = service_api_client.get_services()['data'] + + if 1 == len(services): + return redirect(url_for('.service_dashboard', service_id=services[0]['id'])) + else: + return redirect(url_for('.choose_service')) + + return redirect(url_for('main.index')) diff --git a/app/templates/admin_template.html b/app/templates/admin_template.html index 573b3f1b2..a371476db 100644 --- a/app/templates/admin_template.html +++ b/app/templates/admin_template.html @@ -59,12 +59,7 @@ {% set global_header_text = "GOV.UK Notify" %} - -{% if not current_user.is_authenticated() %} - {% set homepage_url = url_for('main.index') %} -{% else %} - {% set homepage_url = url_for('main.choose_service') %} -{% endif %} +{% set homepage_url = url_for('main.show_all_services_or_dashboard') %} {% block content %}