Use endswith instead of in to look for execute aliases

Author: Evell Lam

py_find_injection/__init__.py

@@ -5,7 +5,7 @@
 import sys
 
 version_info = (0, 1, 1)
-__version__ = '0.1.1+dd.3'
+__version__ = '0.1.1+dd.4'
 
 
 def stringify(node):
@@ -77,7 +77,8 @@ def check_execute(self, node):
 
     def visit_Call(self, node):
         function_name = stringify(node.func)
-        if '.execute' in function_name.lower():
+        # catch and check aliases of session.execute and cursor.execute
+        if function_name.lower().endswith('.execute'):
             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.3",
+    version="0.1.1+dd.4",
     author="James Brown",
     author_email="jbrown@uber.com",
     url="https://github.com/uber/py-find-injection",