move to model based code layout

app/main/views/organisations.py

@@ -70,17 +70,13 @@ def add_organisation():
 @login_required
 @user_has_permissions()
 def organisation_dashboard(org_id):
-    organisation_services = [
-        service for service in organisations_client.get_organisation_services(org_id)
-        if service['active'] and not service['restricted']
-    ]
-    for service in organisation_services:
+    for service in current_organisation.live_services:
         has_permission = current_user.has_permission_for_service(service['id'], 'view_activity')
         service.update({'has_permission_to_view': has_permission})
 
     return render_template(
         'views/organisations/organisation/index.html',
-        organisation_services=organisation_services
+        organisation_services=current_organisation.live_services
     )
 
 
@@ -88,12 +84,10 @@ def organisation_dashboard(org_id):
 @login_required
 @user_is_platform_admin
 def organisation_trial_mode_services(org_id):
-    organisation_services = organisations_client.get_organisation_services(org_id)
-
     return render_template(
         'views/organisations/organisation/trial-mode-services.html',
         search_form=SearchByNameForm(),
-        services=[service for service in organisation_services if not service['active'] or service['restricted']]
+        services=current_organisation.trial_services
     )
 
 
@@ -101,18 +95,10 @@ def organisation_trial_mode_services(org_id):
 @login_required
 @user_has_permissions()
 def manage_org_users(org_id):
-    users = sorted(
-        user_api_client.get_users_for_organisation(org_id=org_id) + [
-            invite for invite in org_invite_api_client.get_invites_for_organisation(org_id=org_id)
-            if invite.status != 'accepted'
-        ],
-        key=lambda user: user.email_address,
-    )
-
     return render_template(
         'views/organisations/organisation/users/index.html',
-        users=users,
-        show_search_box=(len(users) > 7),
+        users=current_organisation.team_members,
+        show_search_box=(len(current_organisation.team_members) > 7),
         form=SearchUsersForm(),
     )
 
@@ -199,16 +185,16 @@ def organisation_settings(org_id):
 
     email_branding = 'GOV.UK'
 
-    if current_organisation['email_branding_id']:
+    if current_organisation.email_branding_id:
         email_branding = email_branding_client.get_email_branding(
-            current_organisation['email_branding_id']
+            current_organisation.email_branding_id
         )['email_branding']['name']
 
     letter_branding = None
 
-    if current_organisation['letter_branding_id']:
+    if current_organisation.letter_branding_id:
         letter_branding = letter_branding_client.get_letter_branding(
-            current_organisation['letter_branding_id']
+            current_organisation.letter_branding_id
         )['name']
 
     return render_template(
@@ -225,7 +211,7 @@ def edit_organisation_name(org_id):
     form = RenameOrganisationForm()
 
     if request.method == 'GET':
-        form.name.data = current_organisation.get('name')
+        form.name.data = current_organisation.name
 
     if form.validate_on_submit():
         unique_name = organisations_client.is_organisation_name_unique(org_id, form.name.data)
@@ -248,12 +234,12 @@ def edit_organisation_name(org_id):
 def edit_organisation_type(org_id):
 
     form = OrganisationOrganisationTypeForm(
-        organisation_type=current_organisation['organisation_type']
+        organisation_type=current_organisation.organisation_type
     )
 
     if form.validate_on_submit():
         organisations_client.update_organisation(
-            current_organisation['id'],
+            current_organisation.id,
             organisation_type=form.organisation_type.data,
         )
         return redirect(url_for('.organisation_settings', org_id=org_id))
@@ -275,12 +261,12 @@ def edit_organisation_crown_status(org_id):
             True: 'crown',
             False: 'non-crown',
             None: 'unknown',
-        }.get(current_organisation['crown'])
+        }.get(current_organisation.crown)
     )
 
     if form.validate_on_submit():
         organisations_client.update_organisation(
-            current_organisation['id'],
+            current_organisation.id,
             crown={
                 'crown': True,
                 'non-crown': False,
@@ -306,12 +292,12 @@ def edit_organisation_agreement(org_id):
             True: 'yes',
             False: 'no',
             None: 'unknown',
-        }.get(current_organisation['agreement_signed'])
+        }.get(current_organisation.agreement_signed)
     )
 
     if form.validate_on_submit():
         organisations_client.update_organisation(
-            current_organisation['id'],
+            current_organisation.id,
             agreement_signed={
                 'yes': True,
                 'no': False,
@@ -336,7 +322,7 @@ def edit_organisation_email_branding(org_id):
 
     form = SetEmailBranding(
         all_branding_options=get_branding_as_value_and_label(email_branding),
-        current_branding=current_organisation['email_branding_id'],
+        current_branding=current_organisation.email_branding_id,
     )
 
     if form.validate_on_submit():
@@ -385,7 +371,7 @@ def edit_organisation_letter_branding(org_id):
 
     form = SetLetterBranding(
         all_branding_options=get_branding_as_value_and_label(letter_branding),
-        current_branding=current_organisation['letter_branding_id'],
+        current_branding=current_organisation.letter_branding_id,
     )
 
     if form.validate_on_submit():
@@ -442,7 +428,7 @@ def edit_organisation_domains(org_id):
         )
         return redirect(url_for('.organisation_settings', org_id=org_id))
 
-    form.populate(current_organisation.get('domains', []))
+    form.populate(current_organisation.domains)
 
     return render_template(
         'views/organisations/organisation/settings/edit-domains.html',
@@ -463,7 +449,7 @@ def confirm_edit_organisation_name(org_id):
     if form.validate_on_submit():
         try:
             organisations_client.update_organisation_name(
-                current_organisation['id'],
+                current_organisation.id,
                 name=session['organisation_name_change'],
             )
         except HTTPError as e:

app/main/views/service_settings.py

@@ -911,16 +911,16 @@ def request_letter_branding(service_id):
 @user_is_platform_admin
 def link_service_to_organisation(service_id):
 
-    organisations = organisations_client.get_organisations()
-    current_organisation = organisations_client.get_service_organisation(service_id).get('id', None)
+    all_organisations = organisations_client.get_organisations()
+    current_linked_organisation = organisations_client.get_service_organisation(service_id).get('id', None)
 
     form = LinkOrganisationsForm(
-        choices=convert_dictionary_to_wtforms_choices_format(organisations, 'id', 'name'),
-        organisations=current_organisation
+        choices=convert_dictionary_to_wtforms_choices_format(all_organisations, 'id', 'name'),
+        organisations=current_linked_organisation
     )
 
     if form.validate_on_submit():
-        if form.organisations.data != current_organisation:
+        if form.organisations.data != current_linked_organisation:
             organisations_client.update_service_organisation(
                 service_id,
                 form.organisations.data
@@ -929,7 +929,7 @@ def link_service_to_organisation(service_id):
 
     return render_template(
         'views/service-settings/link-service-to-organisation.html',
-        has_organisations=organisations,
+        has_organisations=all_organisations,
         form=form,
     )