From 7c3f665ba8beac79656e45ef739f2527a6273399 Mon Sep 17 00:00:00 2001
From: Aaditya Subedi <aaditya.subedi@docsumo.com>
Date: Wed, 19 Nov 2025 12:57:00 +0545
Subject: [PATCH] check for print debug statement

---
 pre_commit_hooks/debug_statement_hook.py | 7 ++++++-
 tests/debug_statement_hook_test.py       | 6 ++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/pre_commit_hooks/debug_statement_hook.py b/pre_commit_hooks/debug_statement_hook.py
index 7e6be95..7f64a23 100644
--- a/pre_commit_hooks/debug_statement_hook.py
+++ b/pre_commit_hooks/debug_statement_hook.py
@@ -20,6 +20,11 @@ DEBUG_STATEMENTS = {
     'wdb',
 }
 
+DEBUG_CALL_STATEMENTS = {
+    'breakpoint',
+    'print'
+}
+
 
 class Debug(NamedTuple):
     line: int
@@ -45,7 +50,7 @@ class DebugStatementParser(ast.NodeVisitor):
 
     def visit_Call(self, node: ast.Call) -> None:
         """python3.7+ breakpoint()"""
-        if isinstance(node.func, ast.Name) and node.func.id == 'breakpoint':
+        if isinstance(node.func, ast.Name) and node.func.id in DEBUG_CALL_STATEMENTS:
             st = Debug(node.lineno, node.col_offset, node.func.id, 'called')
             self.breakpoints.append(st)
         self.generic_visit(node)
diff --git a/tests/debug_statement_hook_test.py b/tests/debug_statement_hook_test.py
index 5a8e0bb..a301fbf 100644
--- a/tests/debug_statement_hook_test.py
+++ b/tests/debug_statement_hook_test.py
@@ -30,6 +30,12 @@ def test_finds_breakpoint():
     visitor = DebugStatementParser()
     visitor.visit(ast.parse('breakpoint()'))
     assert visitor.breakpoints == [Debug(1, 0, 'breakpoint', 'called')]
+    
+    
+def test_finds_print(): 
+    visitor = DebugStatementParser()
+    visitor.visit(ast.parse('print()'))
+    assert visitor.breakpoints == [Debug(1, 0, 'print', 'called')]
 
 
 def test_returns_one_for_failing_file(tmpdir):