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

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):
        """
        """
        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)
        self.sh_client.ProgrammaticLogin()
        try:
            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), self.context)
            feed = None
        return feed