|
1 from mod_pywebsocket import msgutil |
|
2 |
|
3 def web_socket_do_extra_handshake(request): |
|
4 # must set request.ws_protocol to the selected version from ws_requested_protocols |
|
5 request.ws_protocol = request.ws_requested_protocols[0] |
|
6 |
|
7 if (request.ws_protocol == 'error'): |
|
8 raise ValueError('Error') |
|
9 pass |
|
10 |
|
11 def web_socket_transfer_data(request): |
|
12 while True: |
|
13 line = msgutil.receive_message(request) |
|
14 if line == 'protocol': |
|
15 msgutil.send_message(request, request.ws_protocol) |
|
16 continue |
|
17 |
|
18 if line == 'resource': |
|
19 msgutil.send_message(request, request.ws_resource) |
|
20 continue |
|
21 |
|
22 if line == 'origin': |
|
23 msgutil.send_message(request, request.ws_origin) |
|
24 continue |
|
25 |
|
26 msgutil.send_message(request, line) |
|
27 |
|
28 if line == 'end': |
|
29 return |