Python上的“ QXcbConnection：无法连接到显示本地主机：16.0中止”的异常处理

Question

我需要创建一个异常处理（“ try / except”），如果在代码中发生QXcbConnection: Could not connect to display localhost:XX.X Aborted，则会导致异常。该代码有时在设置了$DISPLAY但不可用的位置（没有x服务器）上运行。

导出$DISPLAY变量可能无法正常工作，因为我需要代码在正常的env和无法访问x服务器的群集中运行。这就是为什么我需要能够捕获此异常的原因，以便在x服务器不可用的情况下可以设置plt.switch_backend('agg')。

在某些未设置$DISPLAY变量的情况下，以下方法适用。

try: 
    print("Passed try!")
    plt.figure()
except:
    plt.switch_backend('agg')

但是，如果设置了$DISPLAY但无法访问，则第一个try不会失败，并导致紧随其后的ABORTED。

当出现以下错误时，我需要找到一种方法来使第一次尝试失败： QXcbConnection: Could not connect to display localhost:XX.X Aborted。

有什么想法吗？