iframe-proxy | Sunbelt Computer Software

History

76 lines (58 loc) · 2.28 KB

Raw

from __future__ import print_function

from __future__ import unicode_literals

import argparse

import ast

import collections

import traceback

from typing import List

from typing import Optional

from typing import Sequence

DEBUG_STATEMENTS = {'pdb', 'ipdb', 'pudb', 'q', 'rdb'}

Debug = collections.namedtuple('Debug', ('line', 'col', 'name', 'reason'))

class DebugStatementParser(ast.NodeVisitor):

def __init__(self): # type: () -> None

self.breakpoints = [] # type: List[Debug]

def visit_Import(self, node): # type: (ast.Import) -> None

for name in node.names:

if name.name in DEBUG_STATEMENTS:

st = Debug(node.lineno, node.col_offset, name.name, 'imported')

self.breakpoints.append(st)

def visit_ImportFrom(self, node): # type: (ast.ImportFrom) -> None

if node.module in DEBUG_STATEMENTS:

st = Debug(node.lineno, node.col_offset, node.module, 'imported')

self.breakpoints.append(st)

def visit_Call(self, node): # type: (ast.Call) -> None

"""python3.7+ breakpoint()"""

if isinstance(node.func, ast.Name) and node.func.id == 'breakpoint':

st = Debug(node.lineno, node.col_offset, node.func.id, 'called')

self.breakpoints.append(st)

self.generic_visit(node)

def check_file(filename): # type: (str) -> int

try:

with open(filename, 'rb') as f:

ast_obj = ast.parse(f.read(), filename=filename)

except SyntaxError:

print('{} - Could not parse ast'.format(filename))

print()

print('\t' + traceback.format_exc().replace('\n', '\n\t'))

print()

return 1

visitor = DebugStatementParser()

visitor.visit(ast_obj)

for bp in visitor.breakpoints:

print(

'{}:{}:{} - {} {}'.format(

filename, bp.line, bp.col, bp.name, bp.reason,

)

return int(bool(visitor.breakpoints))

def main(argv=None): # type: (Optional[Sequence[str]]) -> int

parser = argparse.ArgumentParser()

parser.add_argument('filenames', nargs='*', help='Filenames to run')

args = parser.parse_args(argv)

retv = 0

for filename in args.filenames:

retv |= check_file(filename)

return retv

if __name__ == '__main__':

exit(main())

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

debug_statement_hook.py

Latest commit

History

debug_statement_hook.py

File metadata and controls