expr-codegen 0.6.3__tar.gz → 0.6.5__tar.gz

{expr_codegen-0.6.3 → expr_codegen-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: expr_codegen
-Version: 0.6.3
+Version: 0.6.5
 Summary: symbol expression to polars expression tool
 Author-email: wukan <wu-kan@163.com>
 License: BSD 3-Clause License
@@ -75,9 +75,35 @@ https://exprcodegen.streamlit.app
 
 更完整示例访问[alpha_examples](https://github.com/wukan1986/alpha_examples)
 
-## 使用方法
+## 使用示例
 
-运行`demo_cn.py`生成`output.py`，将此文件复制到其它项目中直接`import`使用即可。一般生成的文件不需要再修改。
+```python
+from expr_codegen.tool import codegen_exec
+
+
+def _code_block_():
+    # 因子编辑区，可利用IDE的智能提示在此区域编辑因子
+
+    # 会在生成的代码中自动导入
+    from polars_ta.wq import cs_mad_zscore_resid
+
+    # 1. 下划线开头的变量只是中间变量，最终输出时会被剔除
+    _a = ts_returns(CLOSE, 1)
+    _b = ts_sum(min_(_a, 0) ** 2, 20)
+    _c = ts_sum(max_(_a, 0) ** 2, 20)
+    _d = ts_sum(_a ** 2, 20)
+    _e = (_b - _c) / _d
+    # 2. 下划线开头的变量可以重复使用。 多个复杂因子多行书写时有重复中间变时不再冲突
+    # 3. 下划线开头的变量循环赋值。 在调试时可快速用注释进行切换了
+    _e = cs_mad_zscore_resid(_e, LOG_MC_ZS, ONE)
+    RSJ = _e
+
+
+df = None  # 替换成真实的polars数据
+df = codegen_exec(_code_block_, df, output_file="output.py")
+
+
+```
 
 ## 目录结构
 
@@ -88,6 +114,7 @@ https://exprcodegen.streamlit.app
 ├─examples
 │      alpha101.txt # WorldQuant Alpha101示例，可复制到`streamlit`应用
 │      demo_cn.py # 中文注释示例。演示如何将表达式转换成代码
+│      demo_express.py # 速成示例
 │      demo_exec_pl.py # 演示调用转换后代码并绘图
 │      demo_transformer.py # 演示将第三方表达式转成内部表达式
 │      output.py # 结果输出。可不修改代码，直接被其它项目导入

{expr_codegen-0.6.3 → expr_codegen-0.6.5}/README.md

@@ -25,9 +25,35 @@ https://exprcodegen.streamlit.app
 
 更完整示例访问[alpha_examples](https://github.com/wukan1986/alpha_examples)
 
-## 使用方法
+## 使用示例
 
-运行`demo_cn.py`生成`output.py`，将此文件复制到其它项目中直接`import`使用即可。一般生成的文件不需要再修改。
+```python
+from expr_codegen.tool import codegen_exec
+
+
+def _code_block_():
+    # 因子编辑区，可利用IDE的智能提示在此区域编辑因子
+
+    # 会在生成的代码中自动导入
+    from polars_ta.wq import cs_mad_zscore_resid
+
+    # 1. 下划线开头的变量只是中间变量，最终输出时会被剔除
+    _a = ts_returns(CLOSE, 1)
+    _b = ts_sum(min_(_a, 0) ** 2, 20)
+    _c = ts_sum(max_(_a, 0) ** 2, 20)
+    _d = ts_sum(_a ** 2, 20)
+    _e = (_b - _c) / _d
+    # 2. 下划线开头的变量可以重复使用。 多个复杂因子多行书写时有重复中间变时不再冲突
+    # 3. 下划线开头的变量循环赋值。 在调试时可快速用注释进行切换了
+    _e = cs_mad_zscore_resid(_e, LOG_MC_ZS, ONE)
+    RSJ = _e
+
+
+df = None  # 替换成真实的polars数据
+df = codegen_exec(_code_block_, df, output_file="output.py")
+
+
+```
 
 ## 目录结构
 
@@ -38,6 +64,7 @@ https://exprcodegen.streamlit.app
 ├─examples
 │      alpha101.txt # WorldQuant Alpha101示例，可复制到`streamlit`应用
 │      demo_cn.py # 中文注释示例。演示如何将表达式转换成代码
+│      demo_express.py # 速成示例
 │      demo_exec_pl.py # 演示调用转换后代码并绘图
 │      demo_transformer.py # 演示将第三方表达式转成内部表达式
 │      output.py # 结果输出。可不修改代码，直接被其它项目导入

expr_codegen-0.6.5/expr_codegen/_version.py

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

{expr_codegen-0.6.3 → expr_codegen-0.6.5}/expr_codegen/tool.py

@@ -247,11 +247,14 @@ class ExprTool:
 _TOOL_ = ExprTool()
 
 
-def codegen_exec(globals_, code_block, df_input,
+def codegen_exec(code_block, df_input,
                  extra_codes: str = r'CS_SW_L1 = pl.col(r"^sw_l1_\d+$")',
                  output_file: Optional[str] = None):
     """快速转换源代码并执行"""
-    _TOOL_.globals_ = globals_
+    # 此代码来自于sympy.var
+    frame = inspect.currentframe().f_back
+    _TOOL_.globals_ = frame.f_globals.copy()
+    del frame
 
     if isinstance(code_block, str):
         source = code_block
@@ -260,4 +263,7 @@ def codegen_exec(globals_, code_block, df_input,
 
     codes = _TOOL_._get_codes(source, extra_codes, output_file)
 
-    return _TOOL_.exec(codes, df_input)
+    if df_input is None:
+        return df_input
+    else:
+        return _TOOL_.exec(codes, df_input)

{expr_codegen-0.6.3 → expr_codegen-0.6.5}/expr_codegen.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: expr_codegen
-Version: 0.6.3
+Version: 0.6.5
 Summary: symbol expression to polars expression tool
 Author-email: wukan <wu-kan@163.com>
 License: BSD 3-Clause License
@@ -75,9 +75,35 @@ https://exprcodegen.streamlit.app
 
 更完整示例访问[alpha_examples](https://github.com/wukan1986/alpha_examples)
 
-## 使用方法
+## 使用示例
 
-运行`demo_cn.py`生成`output.py`，将此文件复制到其它项目中直接`import`使用即可。一般生成的文件不需要再修改。
+```python
+from expr_codegen.tool import codegen_exec
+
+
+def _code_block_():
+    # 因子编辑区，可利用IDE的智能提示在此区域编辑因子
+
+    # 会在生成的代码中自动导入
+    from polars_ta.wq import cs_mad_zscore_resid
+
+    # 1. 下划线开头的变量只是中间变量，最终输出时会被剔除
+    _a = ts_returns(CLOSE, 1)
+    _b = ts_sum(min_(_a, 0) ** 2, 20)
+    _c = ts_sum(max_(_a, 0) ** 2, 20)
+    _d = ts_sum(_a ** 2, 20)
+    _e = (_b - _c) / _d
+    # 2. 下划线开头的变量可以重复使用。 多个复杂因子多行书写时有重复中间变时不再冲突
+    # 3. 下划线开头的变量循环赋值。 在调试时可快速用注释进行切换了
+    _e = cs_mad_zscore_resid(_e, LOG_MC_ZS, ONE)
+    RSJ = _e
+
+
+df = None  # 替换成真实的polars数据
+df = codegen_exec(_code_block_, df, output_file="output.py")
+
+
+```
 
 ## 目录结构
 
@@ -88,6 +114,7 @@ https://exprcodegen.streamlit.app
 ├─examples
 │      alpha101.txt # WorldQuant Alpha101示例，可复制到`streamlit`应用
 │      demo_cn.py # 中文注释示例。演示如何将表达式转换成代码
+│      demo_express.py # 速成示例
 │      demo_exec_pl.py # 演示调用转换后代码并绘图
 │      demo_transformer.py # 演示将第三方表达式转成内部表达式
 │      output.py # 结果输出。可不修改代码，直接被其它项目导入

expr_codegen-0.6.3/expr_codegen/_version.py

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