returned to ListResponse

Author: art-vasilyev

toggl_python/repository.py

@@ -19,6 +19,7 @@
     ReportTimeEntry,
 )
 from .exceptions import MethodNotAllowed, NotSupported
+from .response import ListResponse
 
 
 class BaseRepository(Api):
@@ -110,24 +111,14 @@ def _list(self, _url, entity_class, data_key: str = None, **kwargs):
 
         response = self.get(_url, params=params)
         response_body = response.json()
-        response_parameters = (
-            "total_count",
-            "per_page",
-            "total_grand",
-            "total_billable",
-            "total_currencies",
-        )
 
         data = response_body
         data_key = data_key or self.DATA_CONTAINER.get("list", None)
         if data_key:
             data = data[data_key]
         if data:
-            result = [entity_class(**entity) for entity in data]
-            for parameter in response_parameters:
-                if parameter in response_body:
-                    setattr(result, parameter, response_body[parameter])
-            return result
+            value = [entity_class(**entity) for entity in data]
+            return ListResponse(value, response_body)
 
     def list(self, **kwargs):
         if "list" in self.EXCLUDED_METHODS:

toggl_python/response.py

@@ -0,0 +1,15 @@
+class ListResponse(list):
+    def __init__(self, value, response_body):
+        super(ListResponse, self).__init__(value)
+
+        response_parameters = (
+            "total_count",
+            "per_page",
+            "total_grand",
+            "total_billable",
+            "total_currencies",
+        )
+
+        for parameter in response_parameters:
+            if parameter in response_body:
+                setattr(self, parameter, response_body[parameter])