Skip to content
Snippets Groups Projects
Commit 231bd80c authored by Daniel Brötzmann's avatar Daniel Brötzmann
Browse files

chore: EntityTime: Add type annotations

parent 085cba6b
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment