Catch all functions with execute

More generalized than only check session and cursor

Author: Evell Lam

py_find_injection/__init__.py

@@ -5,7 +5,7 @@
 import sys
 
 version_info = (0, 1, 1)
-__version__ = '0.1.1+dd.2'
+__version__ = '0.1.1+dd.3'
 
 
 def stringify(node):
@@ -77,7 +77,7 @@ def check_execute(self, node):
 
     def visit_Call(self, node):
         function_name = stringify(node.func)
-        if function_name.lower() in ('session.execute', 'cursor.execute', 'conn.execute', 'trans.execute', 'pg.execute', 'db.execute'):
+        if '.execute' in function_name.lower():
             try:
                 node.args[0].parent = node
                 node_error = self.check_execute(node.args[0])

setup.py

@@ -3,7 +3,7 @@
 
 setup(
     name="py-find-injection",
-    version="0.1.1+dd.2",
+    version="0.1.1+dd.3",
     author="James Brown",
     author_email="jbrown@uber.com",
     url="https://github.com/uber/py-find-injection",