兼容变量直接赋值的情况

wukan1986 · Mar 31, 2024 · a4310e9 · a4310e9
1 parent bdeb1dc
commit a4310e9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/expr_codegen/_version.py b/expr_codegen/_version.py
@@ -1 +1 @@
-__version__ = "0.6.0"
+__version__ = "0.6.1"
diff --git a/expr_codegen/codes.py b/expr_codegen/codes.py
@@ -56,6 +56,12 @@ def visit_Assign(self, node):
                 target.id = self.targets_map.get(target.id, target.id)
                 self.targets_new.add(target.id)
 
+        # 处理 alpha=close 这种情况
+        if isinstance(node.value, ast.Name):
+            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