From eef9e380dc3f76956155fc3cdad80d48a2a80cbc Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Mon, 6 Oct 2025 09:45:21 -0300
Subject: [PATCH] Added opensearch queries

---
 src/utils/opensearch_queries.py | 55 +++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 src/utils/opensearch_queries.py

diff --git a/src/utils/opensearch_queries.py b/src/utils/opensearch_queries.py
new file mode 100644
index 00000000..f29c6283
--- /dev/null
+++ b/src/utils/opensearch_queries.py
@@ -0,0 +1,55 @@
+"""
+Utility functions for constructing OpenSearch queries consistently.
+"""
+from typing import Union, List
+
+
+def build_filename_query(filename: str) -> dict:
+    """
+    Build a standardized query for finding documents by filename.
+
+    Args:
+        filename: The exact filename to search for
+
+    Returns:
+        A dict containing the OpenSearch query body
+    """
+    return {
+        "term": {
+            "filename": filename
+        }
+    }
+
+
+def build_filename_search_body(filename: str, size: int = 1, source: Union[bool, List[str]] = False) -> dict:
+    """
+    Build a complete search body for checking if a filename exists.
+
+    Args:
+        filename: The exact filename to search for
+        size: Number of results to return (default: 1)
+        source: Whether to include source fields, or list of specific fields to include (default: False)
+
+    Returns:
+        A dict containing the complete OpenSearch search body
+    """
+    return {
+        "query": build_filename_query(filename),
+        "size": size,
+        "_source": source
+    }
+
+
+def build_filename_delete_body(filename: str) -> dict:
+    """
+    Build a delete-by-query body for removing all documents with a filename.
+
+    Args:
+        filename: The exact filename to delete
+
+    Returns:
+        A dict containing the OpenSearch delete-by-query body
+    """
+    return {
+        "query": build_filename_query(filename)
+    }