From a87138b9f90b96e2dae017659d82cf916da45249 Mon Sep 17 00:00:00 2001
From: Ben Thorner <ben.thorner@digital.cabinet-office.gov.uk>
Date: Thu, 26 May 2022 13:59:16 +0100
Subject: [PATCH] Refactor loop over TemplateList items

This is a technique we'll use in later commits.
---
 app/models/template_list.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/app/models/template_list.py b/app/models/template_list.py
index e0e8b4563..4f48c0863 100644
--- a/app/models/template_list.py
+++ b/app/models/template_list.py
@@ -76,12 +76,11 @@ class TemplateLists():
         self.user = user
 
     def __iter__(self):
-
         if len(self.services) == 1:
-
-            for template_or_folder in TemplateList(self.services[0], user=self.user):
-                yield template_or_folder
-
+            yield from TemplateList(
+                service=self.services[0],
+                user=self.user,
+            )
             return
 
         for service in self.services: