Update rules/python/lang/eval_using_user_input.yml

Co-authored-by: David Roe <[email protected]>
Bearer · May 14, 2024 · 7be2b14 · 7be2b14
1 parent 5b0e130
commit 7be2b14
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/rules/python/lang/eval_using_user_input.yml b/rules/python/lang/eval_using_user_input.yml
@@ -24,10 +24,17 @@ patterns:
 auxiliary:
   - id: python_lang_eval_using_user_input_literal_eval
     patterns:
-      - pattern: $<AST_LITERAL_EVAL>
+      - pattern: $<AST_MODULE>.literal_eval
         filters:
-          - variable: AST_LITERAL_EVAL
-            regex: \A(ast\.)?literal_eval\z
+          - variable: AST_MODULE
+            detection: python_lang_eval_using_user_input_ast_module
+            scope: cursor
+      - from ast import $<!>literal_eval
+      - from ast import literal_eval as $<!>$<_>
+  - id: python_lang_eval_using_user_input_ast_module
+    patterns:
+      - import $<!>ast
+      - import ast as $<!>$<_>
   - id: python_lang_eval_using_user_input_subinterpreters
     patterns:
       - import $<!>_xxsubinterpreters