如何在Python類中組織導入？

Question

假設我有一個Python類ABC； 我想將一些非默認模塊導入到我的項目中，但是我不確定運行我的代碼的用戶是否已安裝它們。 為了進行檢查，我將導入內容封裝在try and catch塊內的類中，如下所示：

class ABC:
    _canRun = True
    try:
        import XYZ
    except Exception:
        _canRun = False


    def test_function(self):
        if self._canRun:
            import XYZ
            #do stuff using XYZ module
        else:
            print("Cannot do stuff")
            return None

我出於某種原因感覺這是不好的設計。 我可以使用更好的模式嗎？

Answer 1

導入通常放置在py文件的開頭：

try:
    import XYZ
except ImportError:
    XYZ = None

class ABC:
    def test_function(self):
        if XYZ is None:
            raise Exception("Cannot do stuff")

但是，當您可以選擇其他方法時，通常會執行try / except ImportError技巧：

try:
    import XYZ
except ImportError:
    import ZZTop as XYZ # if ZZTop fails, all fails. And that's fine.

class ABC:
    def test_function(self):
        XYZ.something()

否則，建議盡可能輕松地失敗：

import XYZ

class ABC:
    def test_function(self):
        XYZ.something()