Index: gm/rebaseline_server/column.py |
diff --git a/gm/rebaseline_server/column.py b/gm/rebaseline_server/column.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1b9d0bf1c5c242a376b89f8b1879aea2bf00871b |
--- /dev/null |
+++ b/gm/rebaseline_server/column.py |
@@ -0,0 +1,70 @@ |
+#!/usr/bin/python |
+ |
+""" |
+Copyright 2014 Google Inc. |
+ |
+Use of this source code is governed by a BSD-style license that can be |
+found in the LICENSE file. |
+ |
+ColumnHeaderFactory class (see class docstring for details) |
+""" |
+ |
+# Keys used within dictionary representation of each column header. |
+# NOTE: Keep these in sync with static/constants.js |
+KEY__EXTRACOLUMNHEADERS__HEADER_TEXT = 'headerText' |
+KEY__EXTRACOLUMNHEADERS__HEADER_URL = 'headerUrl' |
+KEY__EXTRACOLUMNHEADERS__IS_FILTERABLE = 'isFilterable' |
+KEY__EXTRACOLUMNHEADERS__IS_SORTABLE = 'isSortable' |
+KEY__EXTRACOLUMNHEADERS__USE_FREEFORM_FILTER = 'useFreeformFilter' |
+KEY__EXTRACOLUMNHEADERS__VALUES_AND_COUNTS = 'valuesAndCounts' |
+ |
+ |
+class ColumnHeaderFactory(object): |
+ """Factory which assembles the header for a single column of data.""" |
+ |
+ def __init__(self, header_text, header_url=None, |
+ is_filterable=True, is_sortable=True, |
+ use_freeform_filter=False): |
+ """ |
+ Args: |
+ header_text: string; text the client should display within column header. |
+ header_url: string; target URL if user clicks on column header. |
+ If None, nothing to click on. |
+ is_filterable: boolean; whether client should allow filtering on this |
+ column. |
+ is_sortable: boolean; whether client should allow sorting on this column. |
+ use_freeform_filter: boolean; *recommendation* to the client indicating |
+ whether to allow freeform text matching, as opposed to listing all |
+ values alongside checkboxes. If is_filterable==false, this is |
+ meaningless. |
+ """ |
+ self._header_text = header_text |
+ self._header_url = header_url |
+ self._is_filterable = is_filterable |
+ self._is_sortable = is_sortable |
+ self._use_freeform_filter = use_freeform_filter |
+ |
+ def create_as_dict(self, values_and_counts_dict=None): |
+ """Creates the header for this column, in dictionary form. |
+ |
+ Creates the header for this column in dictionary form, as needed when |
+ constructing the JSON representation. Uses the KEY__EXTRACOLUMNHEADERS__* |
+ constants as keys. |
+ |
+ Args: |
+ values_and_counts_dict: dictionary mapping each possible column value |
+ to its count (how many entries in the column have this value), or |
+ None if this information is not available. |
+ """ |
+ asdict = { |
+ KEY__EXTRACOLUMNHEADERS__HEADER_TEXT: self._header_text, |
+ KEY__EXTRACOLUMNHEADERS__IS_FILTERABLE: self._is_filterable, |
+ KEY__EXTRACOLUMNHEADERS__IS_SORTABLE: self._is_sortable, |
+ KEY__EXTRACOLUMNHEADERS__USE_FREEFORM_FILTER: self._use_freeform_filter, |
+ } |
+ if self._header_url: |
+ asdict[KEY__EXTRACOLUMNHEADERS__HEADER_URL] = self._header_url |
+ if values_and_counts_dict: |
+ asdict[KEY__EXTRACOLUMNHEADERS__VALUES_AND_COUNTS] = sorted( |
+ values_and_counts_dict.items()) |
+ return asdict |