OLD | NEW |
1 # Copyright 2011, Google Inc. | 1 # Copyright 2015-2016, Google Inc. |
2 # All rights reserved. | 2 # All rights reserved. |
3 # | 3 # |
4 # Redistribution and use in source and binary forms, with or without | 4 # Redistribution and use in source and binary forms, with or without |
5 # modification, are permitted provided that the following conditions are | 5 # modification, are permitted provided that the following conditions are |
6 # met: | 6 # met: |
7 # | 7 # |
8 # * Redistributions of source code must retain the above copyright | 8 # * Redistributions of source code must retain the above copyright |
9 # notice, this list of conditions and the following disclaimer. | 9 # notice, this list of conditions and the following disclaimer. |
10 # * Redistributions in binary form must reproduce the above | 10 # * Redistributions in binary form must reproduce the above |
11 # copyright notice, this list of conditions and the following disclaimer | 11 # copyright notice, this list of conditions and the following disclaimer |
12 # in the documentation and/or other materials provided with the | 12 # in the documentation and/or other materials provided with the |
13 # distribution. | 13 # distribution. |
14 # * Neither the name of Google Inc. nor the names of its | 14 # * Neither the name of Google Inc. nor the names of its |
15 # contributors may be used to endorse or promote products derived from | 15 # contributors may be used to endorse or promote products derived from |
16 # this software without specific prior written permission. | 16 # this software without specific prior written permission. |
17 # | 17 # |
18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | 18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | 19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | 20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | 21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | 22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
29 | 29 |
30 | 30 """Allows dot-accessible dictionaries.""" |
31 import logging | |
32 _GOODBYE_MESSAGE = u'Goodbye' | |
33 | 31 |
34 | 32 |
35 def web_socket_do_extra_handshake(request): | 33 class Bunch(dict): |
36 pass # Always accept. | 34 |
| 35 def __init__(self, d): |
| 36 dict.__init__(self, d) |
| 37 self.__dict__.update(d) |
37 | 38 |
38 | 39 |
39 def web_socket_transfer_data(request): | 40 # Converts any kind of variable to a Bunch |
40 while True: | 41 def to_bunch(var): |
41 line = request.ws_stream.receive_message() | 42 if isinstance(var, list): |
42 if line is None: | 43 return [to_bunch(i) for i in var] |
43 return | 44 if isinstance(var, dict): |
44 if isinstance(line, unicode): | 45 ret = {} |
45 request.ws_stream.send_message(line, binary=False) | 46 for k, v in var.items(): |
46 if line == _GOODBYE_MESSAGE: | 47 if isinstance(v, (list, dict)): |
47 return | 48 v = to_bunch(v) |
48 else: | 49 ret[k] = v |
49 request.ws_stream.send_message(line, binary=True) | 50 return Bunch(ret) |
| 51 else: |
| 52 return var |
50 | 53 |
51 | 54 |
52 def web_socket_passive_closing_handshake(request): | 55 # Merges JSON 'add' into JSON 'dst' |
53 return request.ws_close_code, request.ws_close_reason | 56 def merge_json(dst, add): |
| 57 if isinstance(dst, dict) and isinstance(add, dict): |
| 58 for k, v in add.items(): |
| 59 if k in dst: |
| 60 if k == '#': continue |
| 61 merge_json(dst[k], v) |
| 62 else: |
| 63 dst[k] = v |
| 64 elif isinstance(dst, list) and isinstance(add, list): |
| 65 dst.extend(add) |
| 66 else: |
| 67 raise Exception('Tried to merge incompatible objects %s %s\n\n%r\n\n%r' % (t
ype(dst).__name__, type(add).__name__, dst, add)) |
| 68 |
OLD | NEW |