2

我在 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

但是在 line 中filter_open_dialogs(self._me.get_dialogs(), []),PyCharm 显示了这个警告:

预期类型 TotalList',取而代之的是 'Coroutine' ...

有没有想过出了什么问题?

4

1 回答 1

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 正确输入提示。我认为您不能指定自定义类的内部类型。

于 2018-09-19T11:21:04.883 回答