Finish cleaning up stream feature organization.
Fixed missing references that weren't caught due to leftover pyc file allowing tests to keep working when they shouldn't have.
This commit is contained in:
parent
04def6d925
commit
b0297af38d
4 changed files with 9 additions and 11 deletions
|
@ -21,7 +21,6 @@ from sleekxmpp import stanza
|
|||
from sleekxmpp import features
|
||||
from sleekxmpp.basexmpp import BaseXMPP
|
||||
from sleekxmpp.stanza import *
|
||||
from sleekxmpp.stanza.stream import tls, sasl
|
||||
from sleekxmpp.xmlstream import XMLStream, RestartStream
|
||||
from sleekxmpp.xmlstream import StanzaBase, ET, register_stanza_plugin
|
||||
from sleekxmpp.xmlstream.matcher import *
|
||||
|
|
|
@ -8,11 +8,11 @@
|
|||
|
||||
import logging
|
||||
|
||||
from sleekxmpp.stanza import stream
|
||||
from sleekxmpp.xmlstream import RestartStream
|
||||
from sleekxmpp.xmlstream.matcher import *
|
||||
from sleekxmpp.xmlstream.handler import *
|
||||
from sleekxmpp.plugins.base import base_plugin
|
||||
from sleekxmpp.features.feature_mechanisms import stanza
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
@ -24,23 +24,24 @@ class feature_mechanisms(base_plugin):
|
|||
self.name = 'SASL Mechanisms'
|
||||
self.rfc = '6120'
|
||||
self.description = "SASL Stream Feature"
|
||||
self.stanza = stanza
|
||||
|
||||
self.xmpp.register_stanza(stream.sasl.Success)
|
||||
self.xmpp.register_stanza(stream.sasl.Failure)
|
||||
self.xmpp.register_stanza(stream.sasl.Auth)
|
||||
self.xmpp.register_stanza(stanza.Success)
|
||||
self.xmpp.register_stanza(stanza.Failure)
|
||||
self.xmpp.register_stanza(stanza.Auth)
|
||||
|
||||
self._mechanism_handlers = {}
|
||||
self._mechanism_priorities = []
|
||||
|
||||
self.xmpp.register_handler(
|
||||
Callback('SASL Success',
|
||||
MatchXPath(stream.sasl.Success.tag_name()),
|
||||
MatchXPath(stanza.Success.tag_name()),
|
||||
self._handle_success,
|
||||
instream=True,
|
||||
once=True))
|
||||
self.xmpp.register_handler(
|
||||
Callback('SASL Failure',
|
||||
MatchXPath(stream.sasl.Failure.tag_name()),
|
||||
MatchXPath(stanza.Failure.tag_name()),
|
||||
self._handle_fail,
|
||||
instream=True,
|
||||
once=True))
|
||||
|
|
|
@ -2,7 +2,6 @@ import base64
|
|||
import sys
|
||||
import logging
|
||||
|
||||
from sleekxmpp.stanza.stream import sasl
|
||||
from sleekxmpp.plugins.base import base_plugin
|
||||
|
||||
|
||||
|
@ -24,7 +23,7 @@ class sasl_anonymous(base_plugin):
|
|||
if self.xmpp.boundjid.user:
|
||||
return False
|
||||
|
||||
resp = sasl.Auth(self.xmpp)
|
||||
resp = self.xmpp['feature_sasl'].stanza.Auth(self.xmpp)
|
||||
resp['mechanism'] = 'ANONYMOUS'
|
||||
resp.send(now=True)
|
||||
|
||||
|
|
|
@ -2,7 +2,6 @@ import base64
|
|||
import sys
|
||||
import logging
|
||||
|
||||
from sleekxmpp.stanza.stream import sasl
|
||||
from sleekxmpp.plugins.base import base_plugin
|
||||
|
||||
|
||||
|
@ -34,7 +33,7 @@ class sasl_plain(base_plugin):
|
|||
auth = base64.b64encode(b'\x00' + user + \
|
||||
b'\x00' + password).decode('utf-8')
|
||||
|
||||
resp = sasl.Auth(self.xmpp)
|
||||
resp = self.xmpp['feature_mechanisms'].stanza.Auth(self.xmpp)
|
||||
resp['mechanism'] = 'PLAIN'
|
||||
resp['value'] = auth
|
||||
resp.send(now=True)
|
||||
|
|
Loading…
Reference in a new issue