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

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 quintagroup.gdocs.spreadsheet import spreadsheetMessageFactory as _
from quintagroup.gdocs.spreadsheet import logException, logger
from quintagroup.gdocs.spreadsheet.config import ORDER_COLUMNS


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):
        """
        """

        # 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()

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