Hardening suggestions for codemodder-python / sqlp-fix (#280)

Use Assignment Expression (Walrus) In Conditional Co-authored-by: pixeebot[bot] <[email protected]>
pixee · Feb 19, 2024 · 5b5a66f · 5b5a66f
1 parent fe90431
commit 5b5a66f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/core_codemods/sql_parameterization.py b/src/core_codemods/sql_parameterization.py
@@ -340,8 +340,7 @@ def visit_Attribute(self, node: cst.Attribute) -> Optional[bool]:
 
     def recurse_Name(self, node: cst.Name) -> list[cst.CSTNode]:
         # if the expression is a name, try to find its single assignment
-        resolved = self.resolve_expression(node)
-        if resolved != node:
+        if (resolved := self.resolve_expression(node)) != node:
             visitor = LinearizeQuery(self.context)
             resolved.visit(visitor)
             if len(visitor.leaves) == 1: