feat: 🚧 add model to marshall response #192

Author: Angeluz-07

time_tracker_api/time_entries/time_entries_model.py

@@ -461,7 +461,7 @@ def get_all_paginated(self, conditions: dict = None, **kwargs) -> list:
         start = conditions.get("start", None)
         conditions.pop("start", None)
 
-        return self.repository.find_all(
+        time_entries = self.repository.find_all(
             event_ctx,
             conditions=conditions,
             custom_sql_conditions=custom_query,
@@ -470,6 +470,12 @@ def get_all_paginated(self, conditions: dict = None, **kwargs) -> list:
             offset=start,
         )
 
+        return {
+            'records_total': 0,
+            'records_filtered': 0,
+            'data': time_entries,
+        }
+
     def get(self, id):
         event_ctx = self.create_event_context("read")
 

time_tracker_api/time_entries/time_entries_namespace.py

@@ -333,6 +333,20 @@ def get(self):
         return time_entries_dao.get_worked_time(conditions)
 
 
+time_entry_paginated = ns.model(
+    'TimeEntryPaginated',
+    {
+        'records_total': fields.Integer(
+            title='Records total', description='Total number of entries.',
+        ),
+        'records_filtered': fields.Integer(
+            title='Records filtered',
+            description='Number of entries returned by the endpoint.',
+        ),
+        'data': fields.List(fields.Nested(time_entry)),
+    },
+)
+
 paginated_attribs_parser = ns.parser()
 paginated_attribs_parser.add_argument(
     'length',
@@ -357,7 +371,7 @@ def get(self):
 class PaginatedTimeEntry(Resource):
     @ns.expect(paginated_attribs_parser)
     @ns.doc('list_time_entries_paginated')
-    @ns.marshal_list_with(time_entry)
+    @ns.marshal_list_with(time_entry_paginated)
     def get(self):
         """List all time entries paginated"""
         conditions = paginated_attribs_parser.parse_args()