166 def AddHighlight(self, infos, highlight_type): |
166 def AddHighlight(self, infos, highlight_type): |
167 row_highlights = self.Highlights.setdefault(infos[0], {}) |
167 row_highlights = self.Highlights.setdefault(infos[0], {}) |
168 col_highlights = row_highlights.setdefault(infos[1], []) |
168 col_highlights = row_highlights.setdefault(infos[1], []) |
169 col_highlights.append(highlight_type) |
169 col_highlights.append(highlight_type) |
170 |
170 |
|
171 def RemoveHighlight(self, infos, highlight_type): |
|
172 row_highlights = self.Highlights.get(infos[0]) |
|
173 if row_highlights is not None: |
|
174 col_highlights = row_highlights.get(infos[1]) |
|
175 if col_highlights is not None and highlight_type in col_highlights: |
|
176 col_highlights.remove(highlight_type) |
|
177 if len(col_highlights) == 0: |
|
178 row_highlights.pop(infos[1]) |
|
179 |
171 def ClearHighlights(self, highlight_type=None): |
180 def ClearHighlights(self, highlight_type=None): |
172 if highlight_type is None: |
181 if highlight_type is None: |
173 self.Highlights = {} |
182 self.Highlights = {} |
174 else: |
183 else: |
175 for row, row_highlights in self.Highlights.iteritems(): |
184 for row, row_highlights in self.Highlights.iteritems(): |