Python Get exception nodes with AST

exception_nodes_ast.py

import ast

# define a sample function
def foo():
  try:
    x = 1 / 0
  except ZeroDivisionError:
    pass

# parse the function into an AST tree
tree = ast.parse(foo)

# traverse the AST tree and look for exception nodes
exceptions = []
for node in ast.walk(tree):
  if isinstance(node, ast.ExceptHandler):
    exceptions.append(node.type.id)

# print the list of exceptions
print(exceptions)