pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-04-01 02:56:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

check for print debug statement

Browse source
This commit is contained in:
Aaditya Subedi 2025-11-19 12:57:00 +05:45
parent f4e025486b
commit 7c3f665ba8
2 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
pre_commit_hooks/debug_statement_hook.py
View file

@ -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)