From df98a896f5334815f12992d869f11598ed645a2f Mon Sep 17 00:00:00 2001
From: Pratiksha <128999446+Prab-27@users.noreply.github.com>
Date: Wed, 14 Jan 2026 21:40:52 +0530
Subject: [PATCH] [v3-1-test] checking session.query() (#60510) (cherry picked
 from commit 62f28049c760809d62628a831747c7a2af92394a)

Co-authored-by: Pratiksha <128999446+Prab-27@users.noreply.github.com>
---
 scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py b/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py
index 7dd9ca9dd6cfb..6ba23384922b3 100755
--- a/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py
+++ b/scripts/ci/prek/prevent_deprecated_sqlalchemy_usage.py
@@ -36,7 +36,11 @@ def check_session_query(mod: ast.Module, file_path: str) -> bool:
     errors = False
     for node in ast.walk(mod):
         if isinstance(node, ast.Call) and isinstance(node.func, ast.Attribute):
-            if node.func.attr == "query":
+            if (
+                node.func.attr == "query"
+                and isinstance(node.func.value, ast.Name)
+                and node.func.value.id == "session"
+            ):
                 console.print(f"Deprecated query-obj found at line {node.lineno} in {file_path}.")
                 errors = True
     return errors