python - 一种包装另一个脚本以在python中进行错误处理并保持相同的命令行API的优雅方法

Question

我有一个脚本，该脚本要包装一个错误处理程序，如果脚本失败，该处理程序将发送电子邮件。

简便的方法是:

try:
  <script imports>
  <script argument parse>
  <script contents>
except Exception as e:
  send_email(e)
  raise

但是我觉得这个解决方案很麻烦。在保持原始脚本相同的命令行API的同时，还有哪些更优雅的方法呢？

编辑:拥有一个像 python email_wrap.py script_to_run.py <arguments>这样的API的解决方案将是很棒的。但是，我不确定如何操作命令行参数。

Answer 1

(此处为OP)我发现了如何做:
email_wrap.py

import runpy

try:
    scriptname_to_wrap = sys.argv[1]
    sys.argv = [sys.argv[0]] + sys.argv[2:]

    runpy.run_path(scriptname_to_wrap, run_name="__main__")

except BaseException as e:
    send_email(e)
    raise