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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
###############################################################################
#
# The MIT License (MIT)
#
# Copyright (c) typedef int GmbH
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
###############################################################################
import wx
import json
from pprint import pprint
from autobahn.twisted.websocket import WebSocketClientProtocol, \
WebSocketClientFactory
class MyFrame(wx.Frame):
"""
Our UI frame to show.
This is taken from http://www.wxpython.org/test7.py.html and modified
for sending events via WebSocket.
"""
def __init__(self, app):
# First, call the base class' __init__ method to create the frame
wx.Frame.__init__(self, None, -1, "wxPython/Autobahn WebSocket Demo")
self._app = app
# Associate some events with methods of this class
self.Bind(wx.EVT_MOVE, self.OnMove)
# Add a panel and some controls to display the size and position
panel = wx.Panel(self, -1)
label1 = wx.StaticText(panel, -1, "WebSocket messages received:")
label2 = wx.StaticText(panel, -1, "Window position:")
self.sizeCtrl = wx.TextCtrl(panel, -1, "", style=wx.TE_READONLY)
self.posCtrl = wx.TextCtrl(panel, -1, "", style=wx.TE_READONLY)
self.panel = panel
# Use some sizers for layout of the widgets
sizer = wx.FlexGridSizer(2, 2, 5, 5)
sizer.Add(label1)
sizer.Add(self.sizeCtrl)
sizer.Add(label2)
sizer.Add(self.posCtrl)
border = wx.BoxSizer()
border.Add(sizer, 0, wx.ALL, 15)
panel.SetSizerAndFit(border)
self.Fit()
# This method is called by the System when the window is moved,
# because of the association above.
def OnMove(self, event):
pos = event.GetPosition()
self.posCtrl.SetValue("%s, %s" % (pos.x, pos.y))
if self._app._factory:
proto = self._app._factory._proto
if proto:
evt = {'x': pos.x, 'y': pos.y}
msg = json.dumps(evt).encode('utf8')
proto.sendMessage(msg)
class MyClientProtocol(WebSocketClientProtocol):
"""
Our protocol for WebSocket client connections.
"""
def onOpen(self):
print("WebSocket connection open.")
# the WebSocket connection is open. we store ourselves on the
# factory object, so that we can access this protocol instance
# from wxPython, e.g. to use sendMessage() for sending WS msgs
##
self.factory._proto = self
self._received = 0
def onMessage(self, payload, isBinary):
# a WebSocket message was received. now interpret it, possibly
# accessing the wxPython app `self.factory._app` or our
# single UI frame `self.factory._app._frame`
##
if isBinary:
print("Binary message received: {0} bytes".format(len(payload)))
else:
print("Text message received: {0}".format(payload.decode('utf8')))
self._received += 1
frame = self.factory._app._frame
frame.sizeCtrl.SetValue("{}".format(self._received))
def onClose(self, wasClean, code, reason):
print("WebSocket connection closed: {0}".format(reason))
# the WebSocket connection is gone. clear the reference to ourselves
# on the factory object. when accessing this protocol instance from
# wxPython, always check if the ref is None. only use it when it's
# not None (which means, we are actually connected)
##
self.factory._proto = None
class MyClientFactory(WebSocketClientFactory):
"""
Our factory for WebSocket client connections.
"""
protocol = MyClientProtocol
def __init__(self, url, app):
WebSocketClientFactory.__init__(self, url)
self._app = app
self._proto = None
if __name__ == '__main__':
import sys
from twisted.internet import wxreactor
wxreactor.install()
from twisted.internet import reactor
from twisted.python import log
log.startLogging(sys.stdout)
app = wx.App(False)
app._factory = None
app._frame = MyFrame(app)
app._frame.Show()
reactor.registerWxApp(app)
app._factory = MyClientFactory("ws://127.0.0.1:9000", app)
reactor.connectTCP("127.0.0.1", 9000, app._factory)
reactor.run()
|
