Changeset 2667 in products

Timestamp:

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

Author:

liebster

Message:

Added choose the titles of columns by contentype

Location:

quintagroup.gdocs.spreadsheet/trunk

Files:

• quintagroup/gdocs/spreadsheet/__init__.py (modified) (1 diff)
• quintagroup/gdocs/spreadsheet/browser/viewworksheetview.py (modified) (2 diffs)
• quintagroup/gdocs/spreadsheet/content/gspreadsheet.py (modified) (4 diffs)
• setup.py (modified) (2 diffs)

quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/__init__.py

@@ -1 +1 @@
 """Main product initializer
 """
+import sys
 import logging
 

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

quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/content/gspreadsheet.py

@@ -7 +7 @@
 from Products.ATContentTypes.content import base
 from Products.ATContentTypes.content import schemata
+
+from Products.DataGridField import DataGridField, DataGridWidget
+from Products.DataGridField.SelectColumn import SelectColumn
 
 from quintagroup.gdocs.spreadsheet import spreadsheetMessageFactory as _
@@ -48 +51 @@
     ),
 
+    DataGridField(
+        name='title_column',
+        searchable = True,
+        languageIndependent=True,
+        storage=atapi.AnnotationStorage(),
+        columns=("select_title",),
+        widget = DataGridWidget(
+            label = _(
+                u'label_title_column',
+                default=u'Titles of columns'),
+            description=_(
+                u'help_title_column',
+                default=u"Choose titles of columns"),
+            columns={
+                'select_title' : SelectColumn("Titles of columns", vocabulary="getTitleColumnVocabulary"),
+            },
+        ),
+    ),
+
     # -*- Your Archetypes field definitions here ... -*-
 
@@ -61 +83 @@
 
 class GSpreadsheet(base.ATCTContent):
-    """Lets you select google spreadsheet and worksheet id"""
+    """ Lets you select google spreadsheet worksheet id and choose title of columns """
     implements(IGSpreadsheet)
 
@@ -67 +89 @@
     schema = GSpreadsheetSchema
 
+    order_columns = []
     title = atapi.ATFieldProperty('title')
     description = atapi.ATFieldProperty('description')
     spreadsheet_id = atapi.ATFieldProperty('spreadsheet_id')
     worksheet_id = atapi.ATFieldProperty('worksheet_id')
+    title_column = atapi.ATFieldProperty('title_column')
 
     # -*- Your ATSchema to Python Property Bridges Here ... -*-
 
+    def getTitleColumnVocabulary(self):
+        """ Get a list of titles of columns """
+        return atapi.DisplayList(
+            ([('%s'%i, t) for i,t in enumerate(self.order_columns)])
+        )
+
 atapi.registerType(GSpreadsheet, PROJECTNAME)

quintagroup.gdocs.spreadsheet/trunk/setup.py

@@ -58 +58 @@
                         'gdata',
                         'quintagroup.gauth',
+                        'Products.DataGridField',
                         # -*- Extra requirements: -*-
                         ],
@@ -67 +68 @@
       [z3c.autoinclude.plugin]
       target = plone
+
+      [distutils.setup_keywords]
+      paster_plugins = setuptools.dist:assert_string_list
+
+      [egg_info.writers]
+      paster_plugins.txt = setuptools.command.egg_info:write_arg
       """,
+      #paster_plugins = ["ZopeSkel"],
       )