Changeset 2667 in products for quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/browser/viewworksheetview.py

Timestamp:

Jul 13, 2010 3:47:05 PM (14 years ago)

Author:

liebster

Message:

Added choose the titles of columns by contentype

File:

• quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/browser/viewworksheetview.py (modified) (2 diffs)

quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/browser/viewworksheetview.py

@@ -10 +10 @@
 from gdata.spreadsheet.service import SpreadsheetsService
 from gdata.spreadsheet.service import DocumentQuery as shDocumentQuery
+from gdata.spreadsheet import SpreadsheetsListFeed
 
 from quintagroup.gdocs.spreadsheet import spreadsheetMessageFactory as _
 from quintagroup.gdocs.spreadsheet import logException, logger
-from quintagroup.gdocs.spreadsheet.config import ORDER_COLUMNS
 
 
@@ -49 +49 @@
 
     @memoize_contextless
-    def renderWorksheet(self, ssh_id="",
-                        wsh_id='', startrow_idx=0):
+    def renderWorksheet(self, ssh_id="", wsh_id='', startrow_idx=0):
         """
         """
+        feed = self.getFeed(ssh_id, wsh_id, startrow_idx)
+        if isinstance(feed, SpreadsheetsListFeed):
+            indx_keys = filter(None, [elem['select_title'] for elem in self.context.title_column])
+            titles_columns = filter(None, [self.context.order_columns[int(i)] for i in indx_keys])
 
+            # Prepare table
+            table = "<table id=\"sshwsh\">"
+            table += "<tr>"
+            table += ''.join(["<th>%s</th>" % key for key in titles_columns])
+            table += "</tr>"
+            for i, entry in enumerate(feed.entry):
+                if i+1 > startrow_idx:
+                    td_row = "<tr>"
+                    for key in titles_columns:
+                        td_row += "<td>%s</td>" \
+                            % (not (entry.custom[key].text == 'None') and entry.custom[key].text or '')
+                    td_row += "</tr>\n"
+                    table += td_row
+            table += "</table>"
+        else:
+            table = ''
+        return table
+
+    def getFeed(self, ssh_id="", wsh_id='', startrow_idx=0):
+        """ Get SpreadsheetsListFeed
+        """
         # Authorization on spreadsheets.google.com
         gauth = queryUtility(IGAuthUtility)
         self.sh_client = SpreadsheetsService(email=gauth.email, password=gauth.password)
-        sh_client = self.sh_client
-        sh_client.ProgrammaticLogin()
-
+        self.sh_client.ProgrammaticLogin()
         try:
-            feed = sh_client.GetListFeed(ssh_id, wksht_id=wsh_id, query=self.query)
+            feed = self.sh_client.GetListFeed(ssh_id, wksht_id=wsh_id, query=self.query)
+            self.context.order_columns = len(feed.entry) and feed.entry[0].custom.keys() or []
         except Exception:
             logException('GetListFeed function call: '
-                         'key=%s, wksht_id=%s' % (ssh_id, wsh_id))
-            return ""
-
-        # Prepare table
-        table = "<table id=\"sshwsh\">"
-        table += "<tr>"
-        table += ''.join(["<th>%s</th>" % key for key in ORDER_COLUMNS])
-        table += "</tr>"
-
-        for i, entry in enumerate(feed.entry):
-            if i+1 > startrow_idx:
-                td_row = "<tr>"
-                for key in ORDER_COLUMNS:
-                    td_row += "<td>%s</td>" \
-                        % (not (entry.custom[key].text == 'None') and entry.custom[key].text or '')
-                td_row += "</tr>\n"
-                table += td_row
-
-        table += "</table>"
-        return table
+                         'key=%s, wksht_id=%s' % (ssh_id, wsh_id), self.context)
+            feed = None
+        return feed