mgqpy 0.1__py2.py3-none-any.whl

mgqpy/__init__.py

@@ -0,0 +1,90 @@
+"""mongo query as a python filter predicate"""
+
+__version__ = "0.1"
+
+# Terminology
+#
+# Query refers to the document that is passed to the compiler / mongodb find that holds
+#   the query conditions (can be multiple),
+#   e.g. { "fruits.type": { "$eq": "berry", "$ne": "aggregate" }, "fruits": { "$size": 3 }}
+#
+# Cond (conditions) refers to a single path-expression pair,
+#   e.g. { "fruits.type": { "$eq": "berry", "$ne": "aggregate" }
+#
+# Path refers to dot-separated fields,
+#   e.g. "fruits.type"
+#
+# Exp (expression) refers to the object that holds operator and value pairs (can be multiple),
+#   e.g. { "$eq": "berry", "$ne": "aggregate" }
+#
+# Op (operator) refers to the logical operator that is matched against the value,
+#   e.g. "$eq"
+#
+# Ov (operator value) refers to the value that you are matching against in
+#   the context of a single operator e.g. "berry"
+#
+# Doc refers to the object that is passed to the compiled filter function
+
+from typing import List
+
+cmp_ops = {
+    "$eq",
+    "$gt",
+    "$gte",
+    "$in",
+    "$lt",
+    "$lte",
+    "$ne",
+    "$nin",
+}
+
+
+class Query:
+    def __init__(self, query):
+        self._query = query
+        pass
+
+    def match(self, doc) -> bool:
+        results: List[bool] = []
+
+        for path in self._query:
+            exp_or_ov = self._query[path]
+            is_all_ops = (
+                exp_or_ov
+                and isinstance(exp_or_ov, dict)
+                and all(key in cmp_ops for key in exp_or_ov.keys())
+            )
+
+            path_parts = path.split(".")
+
+            if is_all_ops:
+                if "$eq" in exp_or_ov:
+                    results.append(_match_eq(doc, path_parts, exp_or_ov["$eq"]))
+            else:
+                results.append(_match_eq(doc, path_parts, exp_or_ov))
+
+        return all(results)
+
+
+def _match_eq(doc: dict, path: List[str], ov) -> bool:
+    if len(path) == 0:
+        if isinstance(doc, list) and any([_match_eq(d, path, ov) for d in doc]):
+            return True
+
+        return doc == ov
+
+    key = path[0]
+    rest = path[1:]
+
+    if isinstance(doc, dict) and key in doc:
+        return _match_eq(doc[key], rest, ov)
+
+    if isinstance(doc, list) and key.isdigit():
+        idx = int(key)
+        if idx < len(doc):
+            return _match_eq(doc[idx], rest, ov)
+
+    if isinstance(doc, list):
+        return any([_match_eq(d, path, ov) for d in doc])
+
+    return ov is None

mgqpy-0.1.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Chiawei Ong
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mgqpy-0.1.dist-info/METADATA

@@ -0,0 +1,7 @@
+Metadata-Version: 2.1
+Name: mgqpy
+Version: 0.1
+Summary: mongo query as a python filter predicate
+Author-email: Chiawei Ong <ongchiawei@gmail.com>
+Classifier: License :: OSI Approved :: MIT License
+Project-URL: Home, https://github.com/weiliddat/mgqpy

mgqpy-0.1.dist-info/RECORD

@@ -0,0 +1,5 @@
+mgqpy/__init__.py,sha256=cEALfA6yFzVkT6R4oVkH-Yxr77a9spqG9hTxh7blm7g,2401
+mgqpy-0.1.dist-info/LICENSE,sha256=pEMbA1CSJ6STVH6OvDwoswM9--WxyerjT44X0-scvCU,1068
+mgqpy-0.1.dist-info/WHEEL,sha256=Sgu64hAMa6g5FdzHxXv9Xdse9yxpGGMeagVtPMWpJQY,99
+mgqpy-0.1.dist-info/METADATA,sha256=4ghhCFUHmOdT6GRDK-UST79XimkuFYyFYSRyGM0CjWA,251
+mgqpy-0.1.dist-info/RECORD,,

mgqpy-0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: flit 3.9.0
+Root-Is-Purelib: true
+Tag: py2-none-any
+Tag: py3-none-any