summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authormathieui <mathieui@mathieui.net>2021-07-05 22:41:19 +0200
committermathieui <mathieui@mathieui.net>2021-07-05 22:42:18 +0200
commit524c352da33e39d2d42a955af6bfe4f9fef38562 (patch)
tree65e755d382383bc34c737d042c84bad08ecdde00
parent35eafadb443d4e7276e39d4f177e68a3c8577566 (diff)
downloadslixmpp-524c352da33e39d2d42a955af6bfe4f9fef38562.tar.gz
slixmpp-524c352da33e39d2d42a955af6bfe4f9fef38562.tar.bz2
slixmpp-524c352da33e39d2d42a955af6bfe4f9fef38562.tar.xz
slixmpp-524c352da33e39d2d42a955af6bfe4f9fef38562.zip
clientxmpp: cleanup some types
-rw-r--r--slixmpp/basexmpp.py2
-rw-r--r--slixmpp/clientxmpp.py25
2 files changed, 16 insertions, 11 deletions
diff --git a/slixmpp/basexmpp.py b/slixmpp/basexmpp.py
index a74db5ac..cd228312 100644
--- a/slixmpp/basexmpp.py
+++ b/slixmpp/basexmpp.py
@@ -415,7 +415,7 @@ class BaseXMPP(XMLStream):
if not iq:
iq = self.Iq()
iq['type'] = 'set'
- if sub != None:
+ if sub is not None:
iq.append(sub)
if ito:
iq['to'] = ito
diff --git a/slixmpp/clientxmpp.py b/slixmpp/clientxmpp.py
index f0a7d83e..754db100 100644
--- a/slixmpp/clientxmpp.py
+++ b/slixmpp/clientxmpp.py
@@ -8,7 +8,7 @@
# :license: MIT, see LICENSE for more details
import asyncio
import logging
-from typing import Optional, Any, Callable, Tuple
+from typing import Optional, Any, Callable, Tuple, Dict, Set, List
from slixmpp.jid import JID
from slixmpp.stanza import StreamFeatures, Iq
@@ -16,6 +16,7 @@ from slixmpp.basexmpp import BaseXMPP
from slixmpp.exceptions import XMPPError
from slixmpp.types import JidStr
from slixmpp.xmlstream import XMLStream
+from slixmpp.xmlstream.stanzabase import StanzaBase
from slixmpp.xmlstream.matcher import StanzaPath, MatchXPath
from slixmpp.xmlstream.handler import Callback, CoroutineCallback
@@ -63,7 +64,7 @@ class ClientXMPP(BaseXMPP):
self.default_port = 5222
self.default_lang = lang
- self.credentials = {}
+ self.credentials: Dict[str, str] = {}
self.password = password
@@ -75,9 +76,9 @@ class ClientXMPP(BaseXMPP):
"version='1.0'")
self.stream_footer = "</stream:stream>"
- self.features = set()
- self._stream_feature_handlers = {}
- self._stream_feature_order = []
+ self.features: Set[str] = set()
+ self._stream_feature_handlers: Dict[str, Tuple[Callable, bool]] = {}
+ self._stream_feature_order: List[Tuple[int, str]] = []
self.dns_service = 'xmpp-client'
@@ -94,11 +95,14 @@ class ClientXMPP(BaseXMPP):
self.register_stanza(StreamFeatures)
self.register_handler(
- CoroutineCallback('Stream Features',
- MatchXPath('{%s}features' % self.stream_ns),
- self._handle_stream_features))
-
- def roster_push_filter(iq: Iq) -> None:
+ CoroutineCallback(
+ 'Stream Features',
+ MatchXPath('{%s}features' % self.stream_ns),
+ self._handle_stream_features, # type: ignore
+ )
+ )
+
+ def roster_push_filter(iq: StanzaBase) -> None:
from_ = iq['from']
if from_ and from_ != JID('') and from_ != self.boundjid.bare:
reply = iq.reply()
@@ -274,6 +278,7 @@ class ClientXMPP(BaseXMPP):
return True
log.debug('Finished processing stream features.')
self.event('stream_negotiated')
+ return None
def _handle_roster(self, iq: Iq) -> None:
"""Update the roster after receiving a roster stanza.