From 231bd80cbc2f5f0931bb63b45b2879e2f3e6ba94 Mon Sep 17 00:00:00 2001
From: wurstsalat <mailtrash@posteo.de>
Date: Thu, 26 May 2022 22:22:02 +0200
Subject: [PATCH] chore: EntityTime: Add type annotations
---
gajim/common/modules/entity_time.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/gajim/common/modules/entity_time.py b/gajim/common/modules/entity_time.py
index 30cf09e4c9..bcc575b174 100644
--- a/gajim/common/modules/entity_time.py
+++ b/gajim/common/modules/entity_time.py
@@ -14,7 +14,12 @@
# XEP-0202: Entity Time
+from __future__ import annotations
+
+from nbxmpp.protocol import JID
+
from gajim.common import app
+from gajim.common import types
from gajim.common.modules.base import BaseModule
@@ -27,12 +32,12 @@ class EntityTime(BaseModule):
'disable',
]
- def __init__(self, con):
+ def __init__(self, con: types.Client) -> None:
BaseModule.__init__(self, con)
self.handlers = []
- def set_enabled(self, enabled):
+ def set_enabled(self, enabled: bool) -> None:
if not enabled:
self._nbxmpp('EntityTime').disable()
return
@@ -44,7 +49,7 @@ def set_enabled(self, enabled):
self._nbxmpp('EntityTime').enable()
self._nbxmpp('EntityTime').set_allow_reply_func(self._allow_reply)
- def _allow_reply(self, jid):
+ def _allow_reply(self, jid: JID) -> bool:
item = self._con.get_module('Roster').get_item(jid.bare)
if item is None:
return False
--
GitLab