python - 如果从命令行运行，我如何退出 Python，否则打印堆栈跟踪

Question

如果我的函数是从命令行执行的，我想在我的函数中遇到异常时退出 Python，但如果我的函数不是从命令行运行的，我想引发异常并打印堆栈跟踪。

现在我有

    try:
        #...
    except Exception as e:
        print('ERROR: Some useful message')
        if __name__ == '__main__':
            raise SystemExit
        else:
            raise e 

但我觉得我在这里要么做得太多，要么做得太少。

当我的函数从命令行运行时，是否有一种惯用的方法来获取带有原始异常的堆栈跟踪？但如果它是从命令行运行的，只需退出？

Answer 1

更好的方法是这样做:

import sys

def func():
    do_actual_processing()
    if not successful:
        raise Exception("Yadayada")

if __name__ == '__main__'
    try:
        func()
    except Exception as e:
        sys.exit(1)

也就是说，函数本身不需要关心它是否从命令行运行。