fix: arg parser patterns

Bearer · May 13, 2024 · f52cb55 · f52cb55
1 parent b2239e6
commit f52cb55
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/rules/python/shared/lang/dynamic_input.yml b/rules/python/shared/lang/dynamic_input.yml
@@ -11,10 +11,23 @@ patterns:
 auxiliary:
   - id: python_shared_lang_dynamic_input_parser
     patterns:
-      - pattern: $<ARG_PARSER>
+      - pattern: $<ARG_PARSER>()
         filters:
           - variable: ARG_PARSER
-            regex: \A(argparse\.)ArgumentParser?\z
+            detection: python_shared_lang_dynamic_input_argparser_init
+  - id: python_shared_lang_dynamic_input_argparser_init
+    patterns:
+      - ArgumentParser
+      - from argparse import ArgumentParser as $<!>$<_>
+      - pattern: $<ARGPARSE>.ArgumentParser
+        filters:
+          - variable: ARGPARSE
+            detection: python_shared_lang_dynamic_input_argparse_import
+  - id: python_shared_lang_dynamic_input_argparse_import
+    patterns:
+      - argparse
+      - import $<!>argparse
+      - import argparse as $<!>$<_>
 metadata:
   description: "Python dynamic input."
   id: python_shared_lang_dynamic_input