Skip to content

Commit 056d0c9

Browse files
committed
feat: Get data by a limit number
1 parent 82b80aa commit 056d0c9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

time_tracker_api/time_entries/time_entries_model.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,7 @@ def find_all(
163163
conditions: dict = {},
164164
custom_sql_conditions: List[str] = [],
165165
date_range: dict = {},
166+
**kwargs,
166167
):
167168
custom_sql_conditions.append(
168169
self.create_sql_date_range_filter(date_range)
@@ -175,6 +176,7 @@ def find_all(
175176
conditions=conditions,
176177
custom_sql_conditions=custom_sql_conditions,
177178
custom_params=custom_params,
179+
max_count=kwargs.get("max_count", None),
178180
)
179181

180182
if time_entries:
@@ -381,11 +383,14 @@ def get_all(self, conditions: dict = None, **kwargs) -> list:
381383
HTTPStatus.FORBIDDEN, "You don't have enough permissions."
382384
)
383385
date_range = self.handle_date_filter_args(args=conditions)
386+
limit = conditions.get("limit", None)
387+
conditions.pop("limit", None)
384388
return self.repository.find_all(
385389
event_ctx,
386390
conditions=conditions,
387391
custom_sql_conditions=custom_query,
388392
date_range=date_range,
393+
max_count=limit,
389394
)
390395

391396
def get(self, id):

0 commit comments

Comments
 (0)