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 @@ ...@@ -14,7 +14,12 @@
# XEP-0202: Entity Time # XEP-0202: Entity Time
from __future__ import annotations
from nbxmpp.protocol import JID
from gajim.common import app from gajim.common import app
from gajim.common import types
from gajim.common.modules.base import BaseModule from gajim.common.modules.base import BaseModule
...@@ -27,12 +32,12 @@ class EntityTime(BaseModule): ...@@ -27,12 +32,12 @@ class EntityTime(BaseModule):
'disable', 'disable',
] ]
def __init__(self, con): def __init__(self, con: types.Client) -> None:
BaseModule.__init__(self, con) BaseModule.__init__(self, con)
self.handlers = [] self.handlers = []
def set_enabled(self, enabled): def set_enabled(self, enabled: bool) -> None:
if not enabled: if not enabled:
self._nbxmpp('EntityTime').disable() self._nbxmpp('EntityTime').disable()
return return
...@@ -44,7 +49,7 @@ def set_enabled(self, enabled): ...@@ -44,7 +49,7 @@ def set_enabled(self, enabled):
self._nbxmpp('EntityTime').enable() self._nbxmpp('EntityTime').enable()
self._nbxmpp('EntityTime').set_allow_reply_func(self._allow_reply) 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) item = self._con.get_module('Roster').get_item(jid.bare)
if item is None: if item is None:
return False 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