From ff2f141b75ceaab51340c771f18a91d4debda30e Mon Sep 17 00:00:00 2001
From: Viktor Barzin <vbarzin@gmail.com>
Date: Sat, 14 Jun 2025 13:39:56 +0000
Subject: [PATCH] allow limiting the number of listings fetched

---
 crawler/repositories/listing_repository.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/crawler/repositories/listing_repository.py b/crawler/repositories/listing_repository.py
index e628e2f..c34bc3a 100644
--- a/crawler/repositories/listing_repository.py
+++ b/crawler/repositories/listing_repository.py
@@ -25,6 +25,7 @@ class ListingRepository:
         self,
         query_parameters: QueryParameters | None = None,
         only_ids: list[int] | None = None,
+        limit: int | None = None,
     ) -> list[modelListing]:
         """
         Get all listings from the database.
@@ -37,6 +38,8 @@ class ListingRepository:
         if only_ids:
             query = query.where(RentListing.id.in_(only_ids))  # type: ignore
         query = self._add_where_from_query_parameters(query, query_parameters)
+        if limit:
+            query = query.limit(limit)
 
         with Session(self.engine) as session:
             # query = select(modelListing)