From 382784a92f03bcd63de1813939d1c41b48066559 Mon Sep 17 00:00:00 2001 From: wurstsalat <mailtrash@posteo.de> Date: Sat, 28 May 2022 21:17:06 +0200 Subject: [PATCH] chore: VCardTemp: Add type annotations --- gajim/common/modules/vcard_temp.py | 8 ++++++-- pyrightconfig.json | 1 + 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/gajim/common/modules/vcard_temp.py b/gajim/common/modules/vcard_temp.py index 17d5adccd8..4606f85f0b 100644 --- a/gajim/common/modules/vcard_temp.py +++ b/gajim/common/modules/vcard_temp.py @@ -14,8 +14,12 @@ # XEP-0054: vcard-temp +from __future__ import annotations + from nbxmpp.namespaces import Namespace +from nbxmpp.structs import DiscoInfo +from gajim.common import types from gajim.common.modules.base import BaseModule @@ -27,13 +31,13 @@ class VCardTemp(BaseModule): 'set_vcard', ] - def __init__(self, con): + def __init__(self, con: types.Client) -> None: BaseModule.__init__(self, con) self._own_vcard = None self.supported = False - def pass_disco(self, info): + def pass_disco(self, info: DiscoInfo) -> None: if Namespace.VCARD not in info.features: return diff --git a/pyrightconfig.json b/pyrightconfig.json index e2bbb49119..cdd8838816 100644 --- a/pyrightconfig.json +++ b/pyrightconfig.json @@ -32,6 +32,7 @@ "gajim/common/modules/chat_markers.py", "gajim/common/modules/pep.py", "gajim/common/modules/register.py", + "gajim/common/modules/vcard_temp.py", "gajim/common/passwords.py", "gajim/common/regex.py", "gajim/common/setting_values.py", -- GitLab