Set an actual operation for pymysql

AikidoSec · Jul 30, 2024 · 5446ac5 · 5446ac5
1 parent 6e8c3f6
commit 5446ac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aikido_firewall/sinks/pymysql.py b/aikido_firewall/sinks/pymysql.py
@@ -33,7 +33,7 @@ def aikido_new_query(_self, sql, unbuffered=False):
         logger.debug("Wrapper - `pymysql` version : %s", version("pymysql"))
 
         context = get_current_context()
-        result = check_context_for_sql_injection(sql, "Test_op", context, MySQL())
+        result = check_context_for_sql_injection(sql, "pymysql.connections.query", context, MySQL())
 
         logger.info("sql_injection results : %s", json.dumps(result))
         if result: