Question

我正在python中使用 telethon 库。我正在尝试使用类型提示来使 PyCharm 自动完成功能正常运行。在下面的代码段中，函数filter_open_dialogs将函数get_dialogs()的返回值用作输入。在阅读telethon文档时，我发现get_dialogs()的返回类型为TotalList，因此将类型提示添加到dialogs输入参数中。然后我尝试调用函数filter_open_dialogs：

from telethon.tl.types import User
from telethon.helpers import TotalList
from telethon import TelegramClient, sync

class Crawler:

    def __init__(self, fetch: bool):

        self._client = TelegramClient('some_name', my_api_id, 'my_secret_api_hash')
        self._me = self._client.start(phone='my_phone_number', password='my_2fa_password')
        if fetch:
            self.get_open_dialogs()

    def get_open_dialogs(self):
        if self._me:
            Crawler.filter_open_dialogs(self._me.get_dialogs(), [])
            return self._me.get_dialogs()

    @staticmethod
    def filter_open_dialogs(dialogs: TotalList, filter_list: list):
        result = []
        if dialogs and dialogs.total:
            for dialog in dialogs:
                entity = dialog.entity
                if not isinstance(entity, User) and entity.id not in filter_list:
                    result.append(entity)
        return result

但是在filter_open_dialogs(self._me.get_dialogs(), [])行中，PyCharm显示此警告：

预期类型为TotalList'，取而代之的是“协程” ...

有人认为出了什么问题吗？

Answer 1

TotalList只是一个便捷类，可以帮助我返回带有.total字段的列表。您可能只想添加以下行：

from telethon.tl.custom import Dialog

def filter_open_dialogs(dialogs, filter_list):
    dialog: Dialog
    ...  # rest of code in the method

那应该告诉PyCharm正确键入提示。我认为您无法指定自定义类的内部类型。

Telethon异步类型提示

1 个答案: