blob: 1f448b47d457b793fe2df930ab85ff333eda4d68 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
"""ZAP Authenticator integrated with the tornado IOLoop.
.. versionadded:: 14.1
"""
# Copyright (C) PyZMQ Developers
# Distributed under the terms of the Modified BSD License.
from zmq.eventloop import ioloop, zmqstream
from .base import Authenticator
class IOLoopAuthenticator(Authenticator):
"""ZAP authentication for use in the tornado IOLoop"""
def __init__(self, context=None, encoding='utf-8', log=None, io_loop=None):
super(IOLoopAuthenticator, self).__init__(context)
self.zap_stream = None
self.io_loop = io_loop or ioloop.IOLoop.instance()
def start(self):
"""Start ZAP authentication"""
super(IOLoopAuthenticator, self).start()
self.zap_stream = zmqstream.ZMQStream(self.zap_socket, self.io_loop)
self.zap_stream.on_recv(self.handle_zap_message)
def stop(self):
"""Stop ZAP authentication"""
if self.zap_stream:
self.zap_stream.close()
self.zap_stream = None
super(IOLoopAuthenticator, self).stop()
__all__ = ['IOLoopAuthenticator']
|
