expr-codegen 0.8.6__tar.gz → 0.8.7__tar.gz

{expr_codegen-0.8.6 → expr_codegen-0.8.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: expr_codegen
-Version: 0.8.6
+Version: 0.8.7
 Summary: symbol expression to polars expression tool
 Author-email: wukan <wu-kan@163.com>
 License: BSD 3-Clause License
@@ -198,11 +198,11 @@ df = codegen_exec(df, _code_block_1, _code_block_2)  # 只执行，不保存代
 
 ## 特别语法
 
-1. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
-2. 支持`C?T:F`三元表达式（只能在字符串中使用），Python只支持`T if C else F`, 而在本项目中会自动转成`if_else(C,T,F)`
+1. 支持`C?T:F`三元表达式（仅可字符串中使用），底层会先转成`C or True if( T )else F`，然后修正成`T if C else F`，最后转成`if_else(C,T,F)`。支持与`if else`混用
+2. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
 3. 为防止`A==B`被`sympy`替换成`False`，底层会换成`Eq(A,B)`
 4. `A^B`的含义与`convert_xor`参数有关，`convert_xor=True`底层会转换成`Pow(A,B)`，反之为`Xor(A,B)`。默认为`False`，用`**`表示乘方
-5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可以替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
+5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可手工替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
 6. 不支持`A<=B<=C`，需手工替换成`(A<=B)&(B<=C)`
 7. 支持`A[0]+B[1]+C[2]`，底层会转成`A+ts_delay(B,1)+ts_delay(C,2)`
 8. 支持`~A`,底层会转换成`Not(A)`

{expr_codegen-0.8.6 → expr_codegen-0.8.7}/README.md

@@ -148,11 +148,11 @@ df = codegen_exec(df, _code_block_1, _code_block_2)  # 只执行，不保存代
 
 ## 特别语法
 
-1. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
-2. 支持`C?T:F`三元表达式（只能在字符串中使用），Python只支持`T if C else F`, 而在本项目中会自动转成`if_else(C,T,F)`
+1. 支持`C?T:F`三元表达式（仅可字符串中使用），底层会先转成`C or True if( T )else F`，然后修正成`T if C else F`，最后转成`if_else(C,T,F)`。支持与`if else`混用
+2. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
 3. 为防止`A==B`被`sympy`替换成`False`，底层会换成`Eq(A,B)`
 4. `A^B`的含义与`convert_xor`参数有关，`convert_xor=True`底层会转换成`Pow(A,B)`，反之为`Xor(A,B)`。默认为`False`，用`**`表示乘方
-5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可以替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
+5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可手工替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
 6. 不支持`A<=B<=C`，需手工替换成`(A<=B)&(B<=C)`
 7. 支持`A[0]+B[1]+C[2]`，底层会转成`A+ts_delay(B,1)+ts_delay(C,2)`
 8. 支持`~A`,底层会转换成`Not(A)`

expr_codegen-0.8.7/expr_codegen/_version.py

@@ -0,0 +1 @@
+__version__ = "0.8.7"

{expr_codegen-0.8.6 → expr_codegen-0.8.7}/expr_codegen/codes.py

@@ -7,8 +7,100 @@ from sympy import Add, Mul, Pow, Eq
 from expr_codegen.expr import register_symbols, dict_to_exprs
 
 
-class SympyTransformer(ast.NodeTransformer):
-    """将ast转换成Sympy要求的格式"""
+class SyntaxTransformer(ast.NodeTransformer):
+    """修改语法。注意：一定要修改语法后才能改名"""
+
+    # ^ 是异或还是乘方呢？
+    convert_xor: bool = True
+
+    def visit_Compare(self, node):
+        assert len(node.comparators) == 1, f"不支持连续等号，请手工添加括号, {ast.unparse(node)}"
+
+        self.generic_visit(node)
+        return node
+
+    def visit_IfExp(self, node):
+        # 三元表达式。需要在外部提前替换成or True if else
+        # 只要body区域，出现了or True，就认为是特殊处理过的
+        if isinstance(node.body, ast.BoolOp) and isinstance(node.body.op, ast.Or):
+            if isinstance(node.body.values[-1], ast.Constant):
+                if node.body.values[-1].value == True:
+                    node.test, node.body = node.body.values[0], node.test
+
+        node = ast.Call(
+            func=ast.Name(id='if_else', ctx=ast.Load()),
+            args=[node.test, node.body, node.orelse],
+            keywords=[],
+        )
+
+        self.generic_visit(node)
+        return node
+
+    def visit_BinOp(self, node):
+        # TypeError: unsupported operand type(s) for *: 'StrictLessThan' and 'int'
+        if isinstance(node.op, (ast.Mult, ast.Add, ast.Div, ast.Sub)):
+            # (OPEN < CLOSE) * -1
+            if isinstance(node.left, ast.Compare):
+                node.left = ast.Call(
+                    func=ast.Name(id='int_', ctx=ast.Load()),
+                    args=[node.left],
+                    keywords=[],
+                )
+            # -1*(OPEN < CLOSE)
+            if isinstance(node.right, ast.Compare):
+                node.right = ast.Call(
+                    func=ast.Name(id='int_', ctx=ast.Load()),
+                    args=[node.right],
+                    keywords=[],
+                )
+            # 这种情况，已经包含
+            # (OPEN < CLOSE)*(OPEN < CLOSE)
+
+        if isinstance(node.op, ast.BitXor):
+            # ^ 运算符，转换为pow还是xor
+            if self.convert_xor:
+                node = ast.Call(
+                    func=ast.Name(id='Pow', ctx=ast.Load()),
+                    args=[node.left, node.right],
+                    keywords=[],
+                )
+            else:
+                node = ast.Call(
+                    func=ast.Name(id='Xor', ctx=ast.Load()),
+                    args=[node.left, node.right],
+                    keywords=[],
+                )
+
+        self.generic_visit(node)
+        return node
+
+    def visit_UnaryOp(self, node):
+        # ~ts_delay 报错，替换成Not(ts_delay)
+        if isinstance(node.op, ast.Invert):
+            node = ast.Call(
+                func=ast.Name(id='Not', ctx=ast.Load()),
+                args=[node.operand],
+                keywords=[],
+            )
+
+        self.generic_visit(node)
+        return node
+
+    def visit_Subscript(self, node):
+        if node.slice.value == 0:
+            node = node.value
+        else:
+            node = ast.Call(
+                func=ast.Name(id='ts_delay', ctx=ast.Load()),
+                args=[node.value, node.slice],
+                keywords=[],
+            )
+        self.generic_visit(node)
+        return node
+
+
+class RenameTransformer(ast.NodeTransformer):
+    """改名处理。改名前需要语法规范"""
 
     # 旧记录
     funcs_old = set()
@@ -25,8 +117,6 @@ class SympyTransformer(ast.NodeTransformer):
     # !!!一定要在drop_symbols时排除
     args_map = {'True': "_TRUE_", 'False': "_FALSE_", 'None': "_NONE_"}
     targets_map = {}  # 只对非下划线开头的生效
-    # ^ 是异或还是乘方呢？
-    convert_xor: bool = True
 
     def config_map(self, funcs_map, args_map, targets_map):
         self.funcs_map = funcs_map
@@ -115,6 +205,7 @@ class SympyTransformer(ast.NodeTransformer):
             self.args_old.add(node.left.id)
             node.left.id = self.args_map.get(node.left.id, node.left.id)
             self.args_new.add(node.left.id)
+
         for i, com in enumerate(node.comparators):
             if isinstance(com, ast.Name):
                 self.args_old.add(com.id)
@@ -128,44 +219,23 @@ class SympyTransformer(ast.NodeTransformer):
                     node.comparators[i] = ast.Name(new_com_value, ctx=ast.Load())
                     self.args_new.add(new_com_value)
 
-        assert len(node.comparators) == 1, f"不支持连续等号，请手工添加括号, {ast.unparse(node)}"
-
         self.generic_visit(node)
         return node
 
     def visit_IfExp(self, node):
-        # 三元表达式。需要在外部提前替换成if else
-        # OPEN>=CLOSE?1:0
-        # OPEN>CLOSE?A==B?3:DE>FG?5:6:0
-        node = ast.Call(
-            func=ast.Name(id='if_else', ctx=ast.Load()),
-            args=[node.body, node.test, node.orelse],
-            keywords=[],
-        )
+        if isinstance(node.body, ast.Name):
+            self.args_old.add(node.body.id)
+            node.body.id = self.args_map.get(node.body.id, node.body.id)
+            self.args_new.add(node.body.id)
+        if isinstance(node.orelse, ast.Name):
+            self.args_old.add(node.orelse.id)
+            node.orelse.id = self.args_map.get(node.orelse.id, node.orelse.id)
+            self.args_new.add(node.orelse.id)
 
         self.generic_visit(node)
         return node
 
     def visit_BinOp(self, node):
-        # TypeError: unsupported operand type(s) for *: 'StrictLessThan' and 'int'
-        if isinstance(node.op, (ast.Mult, ast.Add, ast.Div, ast.Sub)):
-            # (OPEN < CLOSE) * -1
-            if isinstance(node.left, ast.Compare):
-                node.left = ast.Call(
-                    func=ast.Name(id='int_', ctx=ast.Load()),
-                    args=[node.left],
-                    keywords=[],
-                )
-            # -1*(OPEN < CLOSE)
-            if isinstance(node.right, ast.Compare):
-                node.right = ast.Call(
-                    func=ast.Name(id='int_', ctx=ast.Load()),
-                    args=[node.right],
-                    keywords=[],
-                )
-            # 这种情况，已经包含
-            # (OPEN < CLOSE)*(OPEN < CLOSE)
-
         if isinstance(node.left, ast.Name):
             self.args_old.add(node.left.id)
             node.left.id = self.args_map.get(node.left.id, node.left.id)
@@ -189,21 +259,6 @@ class SympyTransformer(ast.NodeTransformer):
                 node.right = ast.Name(new_node_value, ctx=ast.Load())
                 self.args_new.add(new_node_value)
 
-        if isinstance(node.op, ast.BitXor):
-            # ^ 运算符，转换为pow还是xor
-            if self.convert_xor:
-                node = ast.Call(
-                    func=ast.Name(id='Pow', ctx=ast.Load()),
-                    args=[node.left, node.right],
-                    keywords=[],
-                )
-            else:
-                node = ast.Call(
-                    func=ast.Name(id='Xor', ctx=ast.Load()),
-                    args=[node.left, node.right],
-                    keywords=[],
-                )
-
         self.generic_visit(node)
         return node
 
@@ -221,26 +276,14 @@ class SympyTransformer(ast.NodeTransformer):
                 node.operand = ast.Name(new_operand_value, ctx=ast.Load())
                 self.args_new.add(new_operand_value)
 
-        # ~ts_delay 报错，替换成Not(ts_delay)
-        if isinstance(node.op, ast.Invert):
-            node = ast.Call(
-                func=ast.Name(id='Not', ctx=ast.Load()),
-                args=[node.operand],
-                keywords=[],
-            )
-
         self.generic_visit(node)
         return node
 
     def visit_Subscript(self, node):
-        if node.slice.value == 0:
-            node = node.value
-        else:
-            node = ast.Call(
-                func=ast.Name(id='ts_delay', ctx=ast.Load()),
-                args=[node.value, node.slice],
-                keywords=[],
-            )
+        self.args_old.add(node.value.id)
+        node.value.id = self.args_map.get(node.value.id, node.value.id)
+        self.args_new.add(node.value.id)
+
         self.generic_visit(node)
         return node
 
@@ -260,13 +303,13 @@ def sources_to_asts(*sources, convert_xor: bool):
     return '\n'.join(raw), assigns, funcs_new, args_new, targets_new
 
 
-def source_replace(source):
+def source_replace(source: str) -> str:
     # 三元表达式转换成 错误版if( )else，一定得在Transformer中修正
     num = 1
     while num > 0:
-        # A == B?D == E?1: 2:0 + 0
-        # 其实会导致?与:错配，但无所谓，只要多执行几次即可
-        source, num = re.subn(r'\?(.+?):(.+?)', r' if( \1 )else \2', source, flags=re.S)
+        # 利用or 的优先级最低，构造特殊的if else，只要出现，就认为位置要替换
+        # C?T:F --> C or True if( T )else F
+        source, num = re.subn(r'\?(.+?):(.+?)', r' or True if( \1 )else \2', source, flags=re.S)
         # break
     # 或、与
     source = source.replace('||', '|').replace('&&', '&')
@@ -276,8 +319,10 @@ def source_replace(source):
 def _source_to_asts(source, convert_xor: bool):
     """源代码"""
     tree = ast.parse(source_replace(source))
-    t = SympyTransformer()
-    t.convert_xor = convert_xor
+    t1 = SyntaxTransformer()
+    t1.convert_xor = convert_xor
+    t1.visit(tree)
+    t = RenameTransformer()
     t.visit(tree)
 
     raw = []

{expr_codegen-0.8.6 → expr_codegen-0.8.7}/expr_codegen.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: expr_codegen
-Version: 0.8.6
+Version: 0.8.7
 Summary: symbol expression to polars expression tool
 Author-email: wukan <wu-kan@163.com>
 License: BSD 3-Clause License
@@ -198,11 +198,11 @@ df = codegen_exec(df, _code_block_1, _code_block_2)  # 只执行，不保存代
 
 ## 特别语法
 
-1. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
-2. 支持`C?T:F`三元表达式（只能在字符串中使用），Python只支持`T if C else F`, 而在本项目中会自动转成`if_else(C,T,F)`
+1. 支持`C?T:F`三元表达式（仅可字符串中使用），底层会先转成`C or True if( T )else F`，然后修正成`T if C else F`，最后转成`if_else(C,T,F)`。支持与`if else`混用
+2. `(A<B)*-1`,底层将转换成`int_(A<B)*-1`
 3. 为防止`A==B`被`sympy`替换成`False`，底层会换成`Eq(A,B)`
 4. `A^B`的含义与`convert_xor`参数有关，`convert_xor=True`底层会转换成`Pow(A,B)`，反之为`Xor(A,B)`。默认为`False`，用`**`表示乘方
-5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可以替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
+5. 支持`A&B&C`，但不支持`A==B==C`。如果C是布尔，AB是数值，可手工替换成`(A==B)==C`。如果ABC是数值需手工替换成`(A==B)&(B==C)`
 6. 不支持`A<=B<=C`，需手工替换成`(A<=B)&(B<=C)`
 7. 支持`A[0]+B[1]+C[2]`，底层会转成`A+ts_delay(B,1)+ts_delay(C,2)`
 8. 支持`~A`,底层会转换成`Not(A)`

expr_codegen-0.8.6/expr_codegen/_version.py

@@ -1 +0,0 @@
-__version__ = "0.8.6"