重载__getitem__以支持不同的访问参数

Question

我正在尝试简化代码，我认为我可以重载getitem以使其既像列表又像字典。 我有理由不这样做吗？

class Book:
    def __init__(self):
        self.page_id = []
        self.page = []

    def __getitem__(self, ref):
        if type(ref) is int:
            return self.page[ref]
        elif type(ref) is str:
            page_index = self.page_id.index(ref)
            return self.page[page_index]

这样我可以支持这样的事情

book = Book()
page_1 = book[5]
page_8 = book['id_for_page_8']

Answer 1

通过简单地尝试/ except，您可能会更pythonic

class Book:
    def __init__(self):
        self.page_id = []
        self.page = []

    def __getitem__(self, ref):
        try:
            return self.page[ref]   # assume it's an int
        except TypeError:
            return self.page[self.page_id.index(ref)] # assume it's a string if not an int

作为一般提示，使用isinstance是对类型的更好选择，因为isinstance可以理解继承。