perf: limit number of regex results per file

kantord · Aug 9, 2023 · 13bec2a · 13bec2a
1 parent dacb96d
commit 13bec2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/seagoat/sources/ripgrep.py b/seagoat/sources/ripgrep.py
@@ -14,7 +14,7 @@
 def _fetch(query_text: str, path: str, limit: int):
     query_text = re.sub(r"\s+", "|", query_text)
     files = {}
-    for result in Ripgrepy(query_text, path).json().run().as_dict:
+    for result in Ripgrepy(query_text, path).max_count(limit).json().run().as_dict:
         result_data = result["data"]
         absolute_path = result_data["path"]["text"]
         relative_path = Path(absolute_path).relative_to(path)