expr-codegen 0.13.0__tar.gz → 0.13.2__tar.gz

{expr_codegen-0.13.0 → expr_codegen-0.13.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: expr_codegen
-Version: 0.13.0
+Version: 0.13.2
 Summary: symbol expression to polars expression tool
 Author-email: wukan <wu-kan@163.com>
 License: BSD 3-Clause License
@@ -94,6 +94,13 @@ def _code_block_2():
     # talib在模板中没有默认导入。这种写法可实现在生成的代码中导入
     from polars_ta.prefix.talib import ts_LINEARREG_SLOPE  # noqa
 
+    # 还支持def和class的原样导入
+    def cs_rank_if(condition, factor):
+        return cs_rank(if_else(condition, factor, None))
+
+    class Clazz:
+        pass
+
     # 1. 下划线开头的变量只是中间变量,会被自动更名，最终输出时会被剔除
     # 2. 下划线开头的变量可以重复使用。多个复杂因子多行书写时有重复中间变时不再冲突
     _avg = ts_mean(corr, 20)
@@ -268,6 +275,7 @@ X3 = (ts_returns(CLOSE, 3)).over(_ASSET_, order_by=_DATE_),
    C = unpack(_x_0, 2)
    ```
 11. 单行注释支持参数输入，如：`# --over_null`、`# --over_null=order_by`、`# --over_null=partition_by`
+12. 代码块中，对`import`、`def`、`class`三种语法，自动原样插入到生成的代码中
 
 ## 下划线开头的变量
 

{expr_codegen-0.13.0 → expr_codegen-0.13.2}/README.md

@@ -44,6 +44,13 @@ def _code_block_2():
     # talib在模板中没有默认导入。这种写法可实现在生成的代码中导入
     from polars_ta.prefix.talib import ts_LINEARREG_SLOPE  # noqa
 
+    # 还支持def和class的原样导入
+    def cs_rank_if(condition, factor):
+        return cs_rank(if_else(condition, factor, None))
+
+    class Clazz:
+        pass
+
     # 1. 下划线开头的变量只是中间变量,会被自动更名，最终输出时会被剔除
     # 2. 下划线开头的变量可以重复使用。多个复杂因子多行书写时有重复中间变时不再冲突
     _avg = ts_mean(corr, 20)
@@ -218,6 +225,7 @@ X3 = (ts_returns(CLOSE, 3)).over(_ASSET_, order_by=_DATE_),
    C = unpack(_x_0, 2)
    ```
 11. 单行注释支持参数输入，如：`# --over_null`、`# --over_null=order_by`、`# --over_null=partition_by`
+12. 代码块中，对`import`、`def`、`class`三种语法，自动原样插入到生成的代码中
 
 ## 下划线开头的变量
 

expr_codegen-0.13.2/expr_codegen/_version.py

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

{expr_codegen-0.13.0 → expr_codegen-0.13.2}/expr_codegen/codes.py

@@ -121,6 +121,18 @@ class SyntaxTransformer(ast.NodeTransformer):
         self.generic_visit(node)
         return node
 
+    def visit_FunctionDef(self, node):
+        return node
+
+    def visit_ClassDef(self, node):
+        return node
+
+    def visit_Import(self, node):
+        return node
+
+    def visit_ImportFrom(self, node):
+        return node
+
 
 class RenameTransformer(ast.NodeTransformer):
     """改名处理。改名前需要语法规范"""
@@ -308,6 +320,18 @@ class RenameTransformer(ast.NodeTransformer):
         self.generic_visit(node)
         return node
 
+    def visit_FunctionDef(self, node):
+        return node
+
+    def visit_ClassDef(self, node):
+        return node
+
+    def visit_Import(self, node):
+        return node
+
+    def visit_ImportFrom(self, node):
+        return node
+
 
 def source_replace(source: str) -> str:
     # 三元表达式转换成 错误版if( )else，一定得在Transformer中修正
@@ -382,7 +406,7 @@ def sources_to_asts(*sources, convert_xor: bool):
             assigns.append(node)
             continue
         # TODO 是否要把其它语句也加入？是否有安全问题？
-        if isinstance(node, (ast.Import, ast.ImportFrom)):
+        if isinstance(node, (ast.Import, ast.ImportFrom, ast.FunctionDef, ast.ClassDef)):
             raw.append(node)
             continue
         if isinstance(node, ast_comments.Comment):

{expr_codegen-0.13.0 → expr_codegen-0.13.2}/expr_codegen/polars/template.py.j2

@@ -18,6 +18,7 @@ from polars_ta.prefix.tdx import *  # noqa
 from polars_ta.prefix.ta import *  # noqa
 from polars_ta.prefix.wq import *  # noqa
 from polars_ta.prefix.cdl import *  # noqa
+from polars_ta.prefix.vec import *  # noqa
 
 DataFrame = TypeVar('DataFrame', _pl_LazyFrame, _pl_DataFrame)
 # ===================================

expr_codegen-0.13.0/expr_codegen/_version.py

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