source: products/quintagroup.gdocs.spreadsheet/trunk/quintagroup/gdocs/spreadsheet/browser/viewworksheetview.py @ 2668

from zope.component import queryUtility
from zope.interface import implements, Interface

from Products.Five import BrowserView
from Products.CMFCore.utils import getToolByName

from plone.memoize.view import memoize_contextless

from quintagroup.gauth.interfaces import IGAuthUtility
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


class IViewWorksheetView(Interface):
    """
    ViewWorksheet view interface
    """

    def renderWorksheet(ssh_id, wsh_idx, startrow_idx):
        """
            ssh_id - Id of the spreadsheet
            wsh_id - Id of the worksheet
            startrow_idx - index of row, from which table should be rendered
        """


class ViewWorksheetView(BrowserView):
    """
    ViewWorksheet browser view
    """
    implements(IViewWorksheetView)

    def __init__(self, context, request):
        self.context = context
        self.request = request
        self.query = shDocumentQuery()

    @property
    def portal_catalog(self):
        return getToolByName(self.context, 'portal_catalog')

    @property
    def portal(self):
        return getToolByName(self.context, 'portal_url').getPortalObject()

    @memoize_contextless
    def renderWorksheet(self, ssh_id="", wsh_id='', startrow_idx=0):
        """
        """
        table = ''
        feed = self.getFeed(ssh_id, wsh_id, startrow_idx)
        if isinstance(feed, SpreadsheetsListFeed):
            # akc is a list of keys of all columns. The context is GSpreadsheet content type
            akc = self.context.all_keys_columns
            # key_title_columns is list of tuples with two elements (column_key, column_title or column_key)
            key_title_columns = [(e['column_key'], e['column_title'] or e['column_key'])
                for e in self.context.order_columns if e['column_key'] in akc]
            if key_title_columns:
                # Prepare table
                table = "<table id=\"sshwsh\">"
                table += "<tr>"
                table += ''.join(["<th>%s</th>" % el[1] for el in key_title_columns])
                table += "</tr>"
                for i, entry in enumerate(feed.entry):
                    if i >= startrow_idx:
                        td_row = "<tr>"
                        for key in zip(*key_title_columns)[0]:
                            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

    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)
        self.sh_client.ProgrammaticLogin()
        try:
            feed = self.sh_client.GetListFeed(ssh_id, wksht_id=wsh_id, query=self.query)
            self.context.all_keys_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), self.context)
            feed = None
        return feed