Fix operations for psycopg2

AikidoSec · Aug 23, 2024 · 47a6335 · 47a6335
1 parent 813fae3
commit 47a6335
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aikido_firewall/sinks/psycopg2.py b/aikido_firewall/sinks/psycopg2.py
@@ -44,7 +44,7 @@ def __getattr__(self, name):
         def execute(*args, **kwargs):
             run_vulnerability_scan(
                 kind="sql_injection",
-                op="pymysql.connection.cursor.execute",
+                op="psycopg2.Connection.Cursor.execute",
                 args=(args[0], Postgres()),  #  args[0] : sql
             )
             return self._execute_func_copy(*args, **kwargs)
@@ -53,7 +53,7 @@ def executemany(*args, **kwargs):
             for sql in args[0]:
                 run_vulnerability_scan(
                     kind="sql_injection",
-                    op="pymysql.connection.cursor.executemany",
+                    op="psycopg2.Connection.Cursor.executemany",
                     args=(sql, Postgres()),
                 )
             return self._executemany_func_copy(*args, **kwargs)