Merge pull request #18 from mustafakemalgilor/enhancement/run_search_…

…tidbits SearchTask::_run_search: small refactor
dosaboy · May 13, 2024 · c58410b · c58410b
2 parents 18d375a + a0a69f7
commit c58410b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/searchkit/search.py b/searchkit/search.py
@@ -1177,14 +1177,14 @@ def execute(self):
             with gzip.open(path, 'rb') as fd:
                 try:
                     # test if file is gzip
-                    fd.read(1)
-                    fd.seek(0)
-                    stats = self._run_search(fd)
-                    self._flush_results_buffer()
+                    fd.peek(1)
                 except OSError:
                     with open(path, 'rb') as fd:
                         stats = self._run_search(fd)
-                        self._flush_results_buffer()
+                else:
+                    stats = self._run_search(fd)
+                finally:
+                    self._flush_results_buffer()
         except UnicodeDecodeError:
             log.exception("caught UnicodeDecodeError while searching %s", path)
             raise