proof-of-concept client

This commit is contained in:
meejah 2019-03-20 18:14:56 -06:00
parent b734c893df
commit 38ed6a094a

56
ws_client.py Normal file
View File

@ -0,0 +1,56 @@
from autobahn.twisted.websocket import WebSocketClientProtocol, \
WebSocketClientFactory
class MyClientProtocol(WebSocketClientProtocol):
def onConnect(self, response):
print("Server connected: {0}".format(response.peer))
def onOpen(self):
print("WebSocket connection open.")
def hello():
self.sendMessage(u"Hello, world!".encode('utf8'))
self.sendMessage(b"\x00\x01\x03\x04", isBinary=True)
self.factory.reactor.callLater(1, hello)
# start sending messages every second ..
hello()
def onMessage(self, payload, isBinary):
if isBinary:
print("Binary message received: {0} bytes".format(len(payload)))
else:
print("Text message received: {0}".format(payload.decode('utf8')))
def onClose(self, wasClean, code, reason):
print("WebSocket connection closed: {0}".format(reason))
if __name__ == '__main__':
import sys
from twisted.python import log
from twisted.internet import reactor
log.startLogging(sys.stdout)
with open("alice/private/api_auth_token", "r") as f:
token = f.read().strip()
factory = WebSocketClientFactory(
url=u"ws://127.0.0.1:6301/logs_v1",
headers={
"Authorization": token,
}
)
factory.protocol = MyClientProtocol
reactor.connectTCP("127.0.0.1", 6301, factory)
reactor.run()