def connect(self):
factory = WebSocketClientFactory(self.endpoint)
factory.protocol = self.protocol
factory.client = self
factory.kwargs = self.kwargs
factory.headers["X-Host-ID"] = self.host_id
factory.headers["X-Client-Type"] = self.type
if self.persistent:
factory.headers["X-Persistence-Policy"] = "Persistent"
else:
factory.headers["X-Persistence-Policy"] = "DeleteOnDisconnect"
if self.username:
authorization = base64.b64encode(
b"%s:%s" % (self.username, self.password)).decode("ascii")
factory.headers["Authorization"] = 'Basic %s' % authorization
if self.filter:
factory.headers["X-Gremlin-Filter"] = self.filter
self.loop = asyncio.get_event_loop()
u = urlparse(self.endpoint)
coro = self.loop.create_connection(factory, u.hostname, u.port)
self.loop.run_until_complete(coro)
try:
self.loop.run_forever()
except KeyboardInterrupt:
self.loop.close()
finally:
pass
def connect(self):
factory = WebSocketClientFactory(self.endpoint)
factory.protocol = self.protocol
factory.client = self
factory.kwargs = self.kwargs
factory.headers["X-Host-ID"] = self.host_id
factory.headers["X-Client-Type"] = self.type
factory.headers["X-Client-Protocol"] = "json"
if self.persistent:
factory.headers["X-Persistence-Policy"] = "Persistent"
else:
factory.headers["X-Persistence-Policy"] = "DeleteOnDisconnect"
if self.username:
if self.auth.login():
cookie = "authtoken={}".format(self.auth.token)
if self.cookies:
self.cookies.append(cookie)
else:
self.cookies = [
cookie,
]
if self.filter:
factory.headers["X-Gremlin-Filter"] = self.filter
if self.cookies:
factory.headers["Cookie"] = ";".join(self.cookies)
coro = self.loop.create_connection(factory,
self.url.hostname,
self.url.port,
ssl=self.ssl_context)
(transport, protocol) = self.loop.run_until_complete(coro)
LOG.debug("transport, protocol: %r, %r", transport, protocol)
def connect(self):
factory = WebSocketClientFactory(self.endpoint)
factory.protocol = self.protocol
factory.client = self
factory.kwargs = self.kwargs
factory.headers["X-Host-ID"] = self.host_id
factory.headers["X-Client-Type"] = self.type
if self.persistent:
factory.headers["X-Persistence-Policy"] = "Persistent"
else:
factory.headers["X-Persistence-Policy"] = "DeleteOnDisconnect"
if self.username:
authorization = base64.b64encode(
b"%s:%s" % (self.username, self.password)).decode("ascii")
factory.headers["Authorization"] = 'Basic %s' % authorization
if self.filter:
factory.headers["X-Gremlin-Filter"] = self.filter
if self.cookies:
factory.headers['Cookie'] = ';'.join(self.cookies)
self.loop = asyncio.get_event_loop()
u = urlparse(self.endpoint)
coro = self.loop.create_connection(factory, u.hostname, u.port)
(transport, protocol) = self.loop.run_until_complete(coro)
LOG.debug('transport, protocol: %r, %r', transport, protocol)
def connect(self):
factory = WebSocketClientFactory(self.endpoint)
factory.protocol = self.protocol
factory.client = self
factory.kwargs = self.kwargs
factory.headers["X-Host-ID"] = self.host_id
factory.headers["X-Client-Type"] = self.type
factory.headers["X-Client-Protocol"] = "json"
if self.persistent:
factory.headers["X-Persistence-Policy"] = "Persistent"
else:
factory.headers["X-Persistence-Policy"] = "DeleteOnDisconnect"
if self.username:
if self.auth.login():
cookie = 'authtok={}'.format(self.auth.authtok)
if self.cookies:
self.cookies.append(cookie)
else:
self.cookies = [
cookie,
]
if self.filter:
factory.headers["X-Gremlin-Filter"] = self.filter
if self.cookies:
factory.headers['Cookie'] = ';'.join(self.cookies)
context = None
if self.url.scheme == "wss":
if self.insecure:
context = ssl._create_unverified_context()
else:
context = ssl._create_default_context()
coro = self.loop.create_connection(factory,
self.url.hostname,
self.url.port,
ssl=context)
(transport, protocol) = self.loop.run_until_complete(coro)
LOG.debug('transport, protocol: %r, %r', transport, protocol)
def worker(queue, send_queue, config):
host = config.get('host', DEFAULT_HOST)
port = config.get('port', DEFAULT_PORT)
encoding = config.get('encoding', DEFAULT_ENCODING)
if 0:
# debug asyncio
logging.basicConfig(level=logging.DEBUG)
else:
formatter = logging.Formatter('%(asctime)s %(name)s [%(levelname)s] %(message)s')
handler = logging.StreamHandler()
handler.setFormatter(formatter)
handler.setLevel(logging.DEBUG)
logger.setLevel(logging.DEBUG)
logger.addHandler(handler)
logger.info("Start main")
import asyncio
factory = WebSocketClientFactory('ws://{}:{}'.format(host, port))
factory.nick = config.get('nick', DEFAULT_NICK)
factory.client = None
factory.protocol = WSChatClientProtocol
factory.queue = queue
factory.encoding = encoding
factory.send_queue = send_queue
def ws_send_worker(queue, factory):
for payload in queue_loop(queue):
if __debug__:
logger.debug("payload-length: {}".format(len(payload)))
if factory.client:
factory.client.sendMessage(payload)
thread = Thread(target=ws_send_worker, args=(send_queue, factory), daemon=True)
thread.start()
loop = asyncio.get_event_loop()
coro = loop.create_connection(factory, host, port)
loop.run_until_complete(coro)
loop.run_forever()
loop.close()
def connect(self):
factory = WebSocketClientFactory(self.endpoint)
factory.protocol = self.protocol
factory.client = self
factory.kwargs = self.kwargs
factory.headers["X-Host-ID"] = self.host_id
factory.headers["X-Client-Type"] = self.type
factory.headers["X-Client-Protocol"] = "json"
if self.persistent:
factory.headers["X-Persistence-Policy"] = "Persistent"
else:
factory.headers["X-Persistence-Policy"] = "DeleteOnDisconnect"
if self.username:
if self.auth.login():
cookie = 'authtok={}'.format(self.auth.authtok)
if self.cookies:
self.cookies.append(cookie)
else:
self.cookies = [cookie, ]
if self.filter:
factory.headers["X-Gremlin-Filter"] = self.filter
if self.cookies:
factory.headers['Cookie'] = ';'.join(self.cookies)
context = None
if self.url.scheme == "wss":
if self.insecure:
context = ssl._create_unverified_context()
else:
context = ssl._create_default_context()
coro = self.loop.create_connection(factory,
self.url.hostname, self.url.port,
ssl=context)
(transport, protocol) = self.loop.run_until_complete(coro)
LOG.debug('transport, protocol: %r, %r', transport, protocol)
