python - sleekxmpp线程认证

所以...我有一个简单的聊天客户端，如下所示：

class ChatClient(sleekxmpp.ClientXMPP):
    def __init__(self, jid, password, server):
        sleekxmpp.ClientXMPP.__init__(self, jid, password, ssl=True)

        self.add_event_handler("session_start", self.start)

        self.register_plugin('xep_0030')
        self.register_plugin('xep_0004')
        self.register_plugin('xep_0060')
        self.register_plugin('xep_0199')

        self.ssl_version = ssl.PROTOCOL_SSLv3
        self.connected = self.connect()
        if self.connected:
            self.process(threaded=True)


    def start(self, event):
        self.send_presence(priority = "-9001")
        self.get_roster(blocking = True, timeout = 3)

    def message(self, targets, msg):
        for target in targets:
            self.send_message(target, msg)

并且我有一个“验证”功能，以确保您输入了用户名/密码：

def authenticate(username, password, server):
    xmppuser = username + '@' + server
    passTester = ChatClient(xmppuser, password)

    passTester.disconnect(wait = True)
    authenticated = passTester.authenticated
    return authenticated

现在，问题出在我的聊天客户端带有线程的地方，我遇到了这样一种情况，即我尝试在服务器有机会实际连接之前检查ChatClient.authenticated。如您所见，我试图在断开连接上“等待”，但是发送队列中没有任何内容，因此它立即断开连接。

我尝试过的替代方法是这样的：

def authenticate(username, password, server):
    xmppuser = username + '@' + server
    passTester = ChatClient(xmppuser, password)

    passTester.message('bogusName', 'ladfhkjdglkhjdfg')
    passTester.disconnect(wait = True)
    authenticated = passTester.authenticated
    return authenticated

现在，我发送了一个虚假消息，断开连接有一些等待的内容。当我使用此代码输入正确的用户名/密码时，断开连接等待消息发送（涉及等待实际连接），不发送任何消息，并且ChatClient.authenticated设置为True！

不幸的是，当我输入了错误的用户名/密码时，消息永远不会发送，并且disconnect（wait = True）永远不会断开，因为消息永远不会发送。

还有其他人有更正确的方法来“验证”吗？

最佳答案

这是将.authenticated和相关字段更改为threading.Event对象的好理由，这样您就可以在这种情况下使用wait()，但是我不确定会破坏现有用户代码多少。

但是除了修改SleekXMPP，您需要做的就是等待某些事件触发。例如，如果您的客户端成功通过身份验证，则将出现一个session_start事件（我稍后可能会添加一个auth_success或类似事件）。同样，如果单一机制的身份验证失败，则会发生failed_auth事件。如果根本没有成功通过身份验证的方法（这是您感兴趣的），则将出现一个no_auth事件。

因此，您可以为这些事件添加处理程序，并使这些处理程序将令牌放入队列中，然后等待所需的令牌到达。

class ChatClient(ClientXMPP):
    def __init__(self, ...):
        ...
        self.auth_queue = queue.Queue()
        self.add_event_handler('no_auth', self.failed)

    def start(self, event):
        self.auth_queue.put('success')
        ...

    def failed(self, event):
        self.auth_queue.put('failed')


def authenticate(username, password, server):
    xmppuser = username + '@' + server
    passTester = ChatClient(xmppuser, password)
    try:
        result = passTester.auth_queue.get(timeout=10)
    except queue.Empty:
        result = 'failed'
    passTester.disconnect()
    return result == 'success'

别忘了我们还有[email protected]的聊天室。

-兰斯

Authenticated

python - sleekxmpp线程认证