在远程 shell 中使用 Fabric 进行 run() 调用时，我可以捕获错误代码吗？

Question

通常，一旦 run() 调用返回非零退出代码，Fabric 就会退出。然而，对于某些呼叫来说，这是预期的。例如，当 PNGOut 无法压缩文件时，它会返回错误代码 2。

目前，我只能通过使用 shell 逻辑（do_something_that_fails || true 或 do_something_that_fails || do_something_else）来规避此限制，但我宁愿能够将我的逻辑保留在普通 Python（正如 Fabric 的承诺）。

有没有办法检查错误代码并对其做出反应，而不是让 Fabric 恐慌并死掉？我仍然想要其他调用的默认行为，因此通过修改环境来改变它的行为似乎不是一个好的选择（据我记得，你只能用它来告诉它警告而不是死亡）。

Answer 1

您可以使用 settings 上下文管理器和 warn_only 设置来防止非零退出代码中止：

from fabric.api import settings

with settings(warn_only=True):
    result = run('pngout old.png new.png')
    if result.return_code == 0: 
        do something
    elif result.return_code == 2: 
        do something else 
    else: #print error to user
        print result
        raise SystemExit()

更新： 我的答案已过时。请参阅下面的评论。

Answer 2

是的，你可以。只需更改环境的 abort_exception 即可。例如：

from fabric.api import settings

class FabricException(Exception):
    pass

with settings(abort_exception = FabricException):
    try:
        run(<something that might fail>)
    except FabricException:
        <handle the exception>

关于 abort_exception 的文档是 在这里。

Answer 3

显然，破坏环境就是答案。

fabric.api.settings 可以用作上下文管理器（使用 with）将其应用到各个语句。 run()、local() 和 sudo() 调用的返回值不仅仅是 shell 命令的输出，还包括具有允许对错误做出反应的特殊属性（return_code 和 failed）。

我想我正在寻找更接近 subprocess.Popen 或 Python 通常的异常处理行为的东西。

Answer 4

试试这个

from fabric.api import run, env
env.warn_only = True # if you want to ignore exceptions and handle them yurself

command = "your command"
x = run(command, capture=True) # run or local or sudo
if(x.stderr != ""):
    error = "On %s: %s" %(command, x.stderr)
    print error
    print x.return_code # which may be 1 or 2
    # do what you want or
    raise Exception(error) #optional
else:
    print "the output of %s is: %s" %(command, x)
    print x.return_code # which is 0