在异常中获取python函数调用的行号

Question

I have a function like this

def try_strip(s):
    try:
        return s.strip()
    except Exception as e:
        print(e)
        # (I've tried inspect, traceback, logging, sys)

and if I call it somewhere like this

try_strip('could be a string or not')

then the exception line number would be the one in which the try_strip is defined. Is there a way to get info about where was it called? Thank you in advance.

Answer 1

The Traceback module which is included with Python provides this functionality. According to its documentation it:

provides a standard interface to extract, format and print stack traces of Python programs. It exactly mimics the behaviour of the Python interpreter when it prints a stack trace.

The function traceback.format_stack() will return the stack trace information you need as a list of strings while the function traceback.print_stack() will print the stack trace information to the console. Below I have included some code which shows how you might use this in the example you provided:

import traceback


def try_strip(s):
  try:
    return s.strip()
  except Exception as e:
    traceback.print_stack()
    stack_trace_info = traceback.format_stack()
    #  Code that write stack_trace_info to a log could go here

try_strip(5)  # This will cause an error at runtime

For additional info on the Traceback module, please see https://docs.python.org/3/library/traceback.html .