Show More
@@ -0,0 +1,1 b'' | |||||
|
1 | from widget import Widget No newline at end of file |
@@ -0,0 +1,148 b'' | |||||
|
1 | from IPython.kernel.comm import Comm | |||
|
2 | from IPython.config import LoggingConfigurable | |||
|
3 | from IPython.utils.traitlets import Unicode, Float, Bool, Dict | |||
|
4 | from IPython.display import clear_output | |||
|
5 | ||||
|
6 | from copy import copy | |||
|
7 | import uuid | |||
|
8 | import sys | |||
|
9 | ||||
|
10 | ||||
|
11 | class Widget(Comm): | |||
|
12 | ||||
|
13 | ### Public declarations | |||
|
14 | target_name = Unicode('widget') | |||
|
15 | default_view_name = Unicode() | |||
|
16 | ||||
|
17 | ||||
|
18 | ### Private/protected declarations | |||
|
19 | _keys = [] | |||
|
20 | _property_lock = False | |||
|
21 | _parent = None | |||
|
22 | _children = [] | |||
|
23 | _css = Dict() | |||
|
24 | ||||
|
25 | ||||
|
26 | ### Public constructor | |||
|
27 | def __init__(self, parent=None, **kwargs): | |||
|
28 | super(Widget, self).__init__(**kwargs) | |||
|
29 | ||||
|
30 | self._children = [] | |||
|
31 | if parent is not None: | |||
|
32 | parent._children.append(self) | |||
|
33 | self._parent = parent | |||
|
34 | ||||
|
35 | self.comm = Comm(target_name=self.target_name) | |||
|
36 | self.comm.on_msg(self._handle_msg) | |||
|
37 | ||||
|
38 | # Register after init to allow default values to be specified | |||
|
39 | self.on_trait_change(self._handle_property_changed, self.keys) | |||
|
40 | ||||
|
41 | # Set initial properties on client model. | |||
|
42 | self.send_state() | |||
|
43 | ||||
|
44 | ||||
|
45 | def __del__(self): | |||
|
46 | self.close() | |||
|
47 | ||||
|
48 | def close(self): | |||
|
49 | self.comm.close() | |||
|
50 | del self.comm | |||
|
51 | ||||
|
52 | ||||
|
53 | ### Properties | |||
|
54 | def _get_parent(self): | |||
|
55 | return self._parent | |||
|
56 | parent = property(_get_parent) | |||
|
57 | ||||
|
58 | ||||
|
59 | def _get_children(self): | |||
|
60 | return copy(self._children) | |||
|
61 | children = property(_get_children) | |||
|
62 | ||||
|
63 | ||||
|
64 | def _get_keys(self): | |||
|
65 | keys = ['_css'] | |||
|
66 | keys.extend(self._keys) | |||
|
67 | return keys | |||
|
68 | keys = property(_get_keys) | |||
|
69 | ||||
|
70 | def _get_css(self, key, selector=""): | |||
|
71 | if selector in self._css and key in self._css[selector]: | |||
|
72 | return self._css[selector][key] | |||
|
73 | else: | |||
|
74 | return None | |||
|
75 | def _set_css(self, value, key, selector=""): | |||
|
76 | if selector not in self._css: | |||
|
77 | self._css[selector] = {} | |||
|
78 | ||||
|
79 | # Only update the property if it has changed. | |||
|
80 | if not (key in self._css[selector] and value in self._css[selector][key]): | |||
|
81 | self._css[selector][key] = value | |||
|
82 | self.send_state() # Send new state to client. | |||
|
83 | ||||
|
84 | css = property(_get_css, _set_css) | |||
|
85 | ||||
|
86 | ||||
|
87 | ### Event handlers | |||
|
88 | def _handle_msg(self, msg): | |||
|
89 | ||||
|
90 | # Handle backbone sync methods | |||
|
91 | sync_method = msg['content']['data']['sync_method'] | |||
|
92 | sync_data = msg['content']['data']['sync_data'] | |||
|
93 | if sync_method.lower() in ['create', 'update']: | |||
|
94 | self._handle_recieve_state(sync_data) | |||
|
95 | ||||
|
96 | ||||
|
97 | def _handle_recieve_state(self, sync_data): | |||
|
98 | self._property_lock = True | |||
|
99 | try: | |||
|
100 | ||||
|
101 | # Use _keys instead of keys - Don't get retrieve the css from the client side. | |||
|
102 | for name in self._keys: | |||
|
103 | if name in sync_data: | |||
|
104 | setattr(self, name, sync_data[name]) | |||
|
105 | finally: | |||
|
106 | self._property_lock = False | |||
|
107 | ||||
|
108 | ||||
|
109 | def _handle_property_changed(self, name, old, new): | |||
|
110 | if not self._property_lock: | |||
|
111 | # TODO: Validate properties. | |||
|
112 | # Send new state to frontend | |||
|
113 | self.send_state() | |||
|
114 | ||||
|
115 | ||||
|
116 | ### Public methods | |||
|
117 | def show(self, view_name=None): | |||
|
118 | if not view_name: | |||
|
119 | view_name = self.default_view_name | |||
|
120 | if not view_name: | |||
|
121 | view_name = self.target_name | |||
|
122 | ||||
|
123 | # Make sure model is syncronized | |||
|
124 | self.send_state() | |||
|
125 | ||||
|
126 | # Show view. | |||
|
127 | if self.parent is None: | |||
|
128 | self.comm.send({"method": "show", "view_name": view_name}) | |||
|
129 | else: | |||
|
130 | self.comm.send({"method": "show", | |||
|
131 | "view_name": view_name, | |||
|
132 | "parent": self.parent.comm.comm_id}) | |||
|
133 | ||||
|
134 | # Now show children if any. | |||
|
135 | for child in self.children: | |||
|
136 | child.show() | |||
|
137 | ||||
|
138 | ||||
|
139 | def send_state(self): | |||
|
140 | state = {} | |||
|
141 | for key in self.keys: | |||
|
142 | try: | |||
|
143 | state[key] = getattr(self, key) | |||
|
144 | except Exception as e: | |||
|
145 | pass # Eat errors, nom nom nom | |||
|
146 | self.comm.send({"method": "update", | |||
|
147 | "state": state}) | |||
|
148 | No newline at end of file |
General Comments 0
You need to be logged in to leave comments.
Login now