omlish 0.0.0.dev119__py3-none-any.whl → 0.0.0.dev120__py3-none-any.whl

omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev119'
-__revision__ = 'abf03cc06fe1610ddf750a1392aa9c9de00acfd3'
+__version__ = '0.0.0.dev120'
+__revision__ = 'aada6beb1d685cb095429e9c588d266b8a83aa23'
 
 
 #

omlish/specs/jmespath/__init__.py

@@ -1,6 +1,10 @@
 """
 TODO:
- - @omlish-lite
+ - @omlish-lite ?
+ - 'recipes'
+ - abstract runtime
+  - https://github.com/wrmsr/omnibus/blob/c2ff67b6c5c80aa03fe27a9b6f36212f3212c7ca/omnibus/jmespath/eval.py#L64
+ - pytest-ify
 
 Applied:
  https://github.com/jmespath-community/python-jmespath/compare/bbe7300c60056f52413603cf3e2bcd0b6afeda3d...aef45e9d665de662eee31b06aeb8261e2bc8b90a

omlish/specs/jmespath/ast.py

@@ -20,9 +20,15 @@ class LeafNode(Node, lang.Abstract):
         return []
 
 
+UnaryArithmeticOperator: ta.TypeAlias = ta.Literal[
+    'plus',
+    'minus',
+]
+
+
 @dc.dataclass(frozen=True)
 class ArithmeticUnary(Node, lang.Final):
-    operator: str
+    operator: UnaryArithmeticOperator
     expression: Node
 
     @property
@@ -30,9 +36,19 @@ class ArithmeticUnary(Node, lang.Final):
         return [self.expression]
 
 
+ArithmeticOperator: ta.TypeAlias = ta.Literal[
+    'div',
+    'divide',
+    'minus',
+    'modulo',
+    'multiply',
+    'plus',
+]
+
+
 @dc.dataclass(frozen=True)
 class Arithmetic(Node, lang.Final):
-    operator: str
+    operator: ArithmeticOperator
     left: Node
     right: Node
 
@@ -51,9 +67,19 @@ class Assign(Node, lang.Final):
         return [self.expr]
 
 
+ComparatorName: ta.TypeAlias = ta.Literal[
+    'eq',
+    'ne',
+    'lt',
+    'gt',
+    'lte',
+    'gte',
+]
+
+
 @dc.dataclass(frozen=True)
 class Comparator(Node, lang.Final):
-    name: str
+    name: ComparatorName
     first: Node
     second: Node
 

omlish/specs/jmespath/functions.py

@@ -638,6 +638,14 @@ class ObjectFunctions(FunctionsClass):
     def _func_values(self, arg):
         return list(arg.values())
 
+    @signature({'types': ['expref']}, {'types': ['object']})
+    def _func_filter_keys(self, expref, arg):
+        return {k: v for k, v in arg.items() if expref.visit(expref.expression, k)}
+
+    @signature({'types': ['expref']}, {'types': ['object']})
+    def _func_filter_values(self, expref, arg):
+        return {k: v for k, v in arg.items() if expref.visit(expref.expression, v)}
+
 
 class KeyedFunctions(FunctionsClass):
     def _create_key_func(self, expref, allowed_types, function_name):

omlish/specs/jmespath/parser.py

@@ -27,9 +27,11 @@ import typing as ta
 from ... import check
 from .ast import AndExpression
 from .ast import Arithmetic
+from .ast import ArithmeticOperator
 from .ast import ArithmeticUnary
 from .ast import Assign
 from .ast import Comparator
+from .ast import ComparatorName
 from .ast import CurrentNode
 from .ast import Expref
 from .ast import Field
@@ -52,6 +54,7 @@ from .ast import Projection
 from .ast import RootNode
 from .ast import Slice
 from .ast import Subexpression
+from .ast import UnaryArithmeticOperator
 from .ast import ValueProjection
 from .ast import VariableRef
 from .exceptions import IncompleteExpressionError
@@ -491,15 +494,15 @@ class Parser:
 
     def _parse_comparator(self, left: Node, comparator: str) -> Node:
         right = self._expression(self.BINDING_POWER[comparator])
-        return Comparator(comparator, left, right)
+        return Comparator(ta.cast(ComparatorName, comparator), left, right)
 
     def _parse_arithmetic_unary(self, token: Token) -> Node:
         expression = self._expression(self.BINDING_POWER[token['type']])
-        return ArithmeticUnary(token['type'], expression)
+        return ArithmeticUnary(ta.cast(UnaryArithmeticOperator, token['type']), expression)
 
     def _parse_arithmetic(self, left: Node, operator: str) -> Node:
         right = self._expression(self.BINDING_POWER[operator])
-        return Arithmetic(operator, left, right)
+        return Arithmetic(ta.cast(ArithmeticOperator, operator), left, right)
 
     def _parse_multi_select_list(self) -> Node:
         expressions: list[Node] = []

omlish/specs/jmespath/visitor.py

@@ -6,9 +6,11 @@ from ... import check
 from ... import lang
 from .ast import AndExpression
 from .ast import Arithmetic
+from .ast import ArithmeticOperator
 from .ast import ArithmeticUnary
 from .ast import Assign
 from .ast import Comparator
+from .ast import ComparatorName
 from .ast import CurrentNode
 from .ast import Expref
 from .ast import Field
@@ -31,6 +33,7 @@ from .ast import Projection
 from .ast import RootNode
 from .ast import Slice
 from .ast import Subexpression
+from .ast import UnaryArithmeticOperator
 from .ast import ValueProjection
 from .ast import VariableRef
 from .exceptions import UndefinedVariableError
@@ -164,7 +167,7 @@ class _Expression:
 
 
 class TreeInterpreter(Visitor):
-    COMPARATOR_FUNC: ta.Mapping[str, ta.Callable] = {
+    _COMPARATOR_FUNC: ta.Mapping[ComparatorName, ta.Callable] = {
         'eq': _equals,
         'ne': lambda x, y: not _equals(x, y),
         'lt': operator.lt,
@@ -173,17 +176,12 @@ class TreeInterpreter(Visitor):
         'gte': operator.ge,
     }
 
-    _EQUALITY_OPS: ta.AbstractSet[str] = {
+    _EQUALITY_OPS: ta.AbstractSet[ComparatorName] = {
         'eq',
         'ne',
     }
 
-    _ARITHMETIC_UNARY_FUNC: ta.Mapping[str, ta.Callable] = {
-        'minus': operator.neg,
-        'plus': lambda x: x,
-    }
-
-    _ARITHMETIC_FUNC: ta.Mapping[str, ta.Callable] = {
+    _ARITHMETIC_FUNC: ta.Mapping[ArithmeticOperator, ta.Callable] = {
         'div': operator.floordiv,
         'divide': operator.truediv,
         'minus': operator.sub,
@@ -192,6 +190,11 @@ class TreeInterpreter(Visitor):
         'plus': operator.add,
     }
 
+    _ARITHMETIC_UNARY_FUNC: ta.Mapping[UnaryArithmeticOperator, ta.Callable] = {
+        'minus': operator.neg,
+        'plus': lambda x: x,
+    }
+
     def __init__(self, options: Options | None = None) -> None:
         super().__init__()
 
@@ -235,7 +238,7 @@ class TreeInterpreter(Visitor):
 
     def visit_comparator(self, node: Comparator, value: ta.Any) -> ta.Any:
         # Common case: comparator is == or !=
-        comparator_func = self.COMPARATOR_FUNC[node.name]
+        comparator_func = self._COMPARATOR_FUNC[node.name]
         if node.name in self._EQUALITY_OPS:
             return comparator_func(
                 self.visit(node.first, value),

{omlish-0.0.0.dev119.dist-info → omlish-0.0.0.dev120.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omlish
-Version: 0.0.0.dev119
+Version: 0.0.0.dev120
 Summary: omlish
 Author: wrmsr
 License: BSD-3-Clause

{omlish-0.0.0.dev119.dist-info → omlish-0.0.0.dev120.dist-info}/RECORD

@@ -1,5 +1,5 @@
 omlish/.manifests.json,sha256=CxGnj-UiRPlZgmgWoovDWrOnqpSEmBy_kqA7cdfSA3w,1431
-omlish/__about__.py,sha256=cKBa887pzxYkOLw1HFwEiT4g4dP8AxuJQrGNkTmiju8,3352
+omlish/__about__.py,sha256=JGKddDla0hzssCiRfhxitgG_mY-VKslBVWxhzDUU-Fg,3352
 omlish/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omlish/argparse.py,sha256=cqKGAqcxuxv_s62z0gq29L9KAvg_3-_rFvXKjVpRJjo,8126
 omlish/c3.py,sha256=4vogWgwPb8TbNS2KkZxpoWbwjj7MuHG2lQG-hdtkvjI,8062
@@ -370,16 +370,16 @@ omlish/secrets/secrets.py,sha256=cnDGBoPknVxsCN04_gqcJT_7Ebk3iO3VPkRZ2oMjkMw,786
 omlish/secrets/subprocesses.py,sha256=EcnKlHHtnUMHGrBWXDfu8tv28wlgZx4P4GOiuPW9Vo8,1105
 omlish/specs/__init__.py,sha256=zZwF8yXTEkSstYtORkDhVLK-_hWU8WOJCuBpognb_NY,118
 omlish/specs/jmespath/LICENSE,sha256=IH-ZZlZkS8XMkf_ubNVD1aYHQ2l_wd0tmHtXrCcYpRU,1113
-omlish/specs/jmespath/__init__.py,sha256=Lz8JO0vP-pwrBtq9Y94y6LrOA7o1u1kCdPmMf_4lJBY,1918
+omlish/specs/jmespath/__init__.py,sha256=MfBlUXNSPq5pRyboMgBk8JoIj0lQo-2TQ6Re7IwGQK0,2079
 omlish/specs/jmespath/__main__.py,sha256=wIXm6bs08etNG_GZlN2rBkADPb0rKfL2HSkm8spnpxw,200
-omlish/specs/jmespath/ast.py,sha256=jiIELiQSHU9haOqXJM4eiSgEapCzHIZ1F-fP_RGpRbk,5019
+omlish/specs/jmespath/ast.py,sha256=XhcUGodHIdsY3-hVZEfpeW6LBehRjLbxVFXkMfZhRdk,5386
 omlish/specs/jmespath/cli.py,sha256=Lw57Eq5rmpwTwsvOzNmce_-XyoM84OIx5cuPgjUXNb0,2197
 omlish/specs/jmespath/exceptions.py,sha256=Co1HiUBPFNwFgZY3FV_ayuZoSgZIAmDcImImxauYNxc,4435
-omlish/specs/jmespath/functions.py,sha256=DG6wf4m0EuqD2FZnSKOFb4GvuXyCqlbncQvlBUN1NYY,22209
+omlish/specs/jmespath/functions.py,sha256=lE_MlW5rDQirDCE9HtcAG-17kuHhH36RaPaQfk97xDY,22595
 omlish/specs/jmespath/lexer.py,sha256=hlPGCXPzGhd9ySj-z2cGTbyC9z3e0Io78IMYJZSEwNk,12647
-omlish/specs/jmespath/parser.py,sha256=yidV9KzHBsrO4-HJF53AOOxSvtqq5u0DESFaWY_0llA,24256
+omlish/specs/jmespath/parser.py,sha256=IPJ0fCv1Aj2MhsDx5XuddtM2LnnLrynuGs4g7PYCmf0,24453
 omlish/specs/jmespath/scope.py,sha256=UyDsl9rv_c8DCjJBuVIA2ESu1jrgYvuwEKiaJDQKnT0,1590
-omlish/specs/jmespath/visitor.py,sha256=-A386ebMQAJE_vvbGHpQyESu1h0W-PGvaHE5_jgjiH4,16401
+omlish/specs/jmespath/visitor.py,sha256=yneRMO4qf3k2Mdcm2cPC0ozRgOaudzlxRVRGatztJzs,16569
 omlish/specs/jsonrpc/__init__.py,sha256=E0EogYSKmbj1D-V57tBgPDTyVuD8HosHqVg0Vh1CVwM,416
 omlish/specs/jsonrpc/errors.py,sha256=-Zgmlo6bV6J8w5f8h9axQgLquIFBHDgIwcpufEH5NsE,707
 omlish/specs/jsonrpc/marshal.py,sha256=iXZNR7n0vfL_yiPFFYN-ZyGlzknNXExs2qC1HFChGPU,1913
@@ -471,9 +471,9 @@ omlish/text/glyphsplit.py,sha256=Ug-dPRO7x-OrNNr8g1y6DotSZ2KH0S-VcOmUobwa4B0,329
 omlish/text/indent.py,sha256=6Jj6TFY9unaPa4xPzrnZemJ-fHsV53IamP93XGjSUHs,1274
 omlish/text/parts.py,sha256=7vPF1aTZdvLVYJ4EwBZVzRSy8XB3YqPd7JwEnNGGAOo,6495
 omlish/text/random.py,sha256=jNWpqiaKjKyTdMXC-pWAsSC10AAP-cmRRPVhm59ZWLk,194
-omlish-0.0.0.dev119.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omlish-0.0.0.dev119.dist-info/METADATA,sha256=sMg30UlhW1GLNw8MkOhVqMaaiObB2IXCn22KIqaMpWU,4000
-omlish-0.0.0.dev119.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-omlish-0.0.0.dev119.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
-omlish-0.0.0.dev119.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
-omlish-0.0.0.dev119.dist-info/RECORD,,
+omlish-0.0.0.dev120.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omlish-0.0.0.dev120.dist-info/METADATA,sha256=9KUQ86p-Od1hMU_EibbqCHSW6pojpLGYyvZCxg935BE,4000
+omlish-0.0.0.dev120.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+omlish-0.0.0.dev120.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
+omlish-0.0.0.dev120.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
+omlish-0.0.0.dev120.dist-info/RECORD,,