[2682] | 1 | from zope.interface import implements |
---|
| 2 | from zope.component import adapts, queryUtility |
---|
| 3 | |
---|
| 4 | from gdata.spreadsheet.service import SpreadsheetsService |
---|
| 5 | from gdata.spreadsheet.service import DocumentQuery as shDocumentQuery |
---|
| 6 | |
---|
[2686] | 7 | from quintagroup.gauth.interfaces import IGAuthUtility |
---|
[2682] | 8 | from quintagroup.gdocs.spreadsheet import logException, logger |
---|
| 9 | from quintagroup.gdocs.spreadsheet.interfaces import IGSpreadsheet |
---|
| 10 | from quintagroup.gdocs.spreadsheet.interfaces import IGSpreadsheetDataProvider |
---|
| 11 | |
---|
| 12 | |
---|
| 13 | class GSpreadsheetDataProvider(object): |
---|
| 14 | |
---|
| 15 | adapts(IGSpreadsheet) |
---|
| 16 | implements(IGSpreadsheetDataProvider) |
---|
| 17 | |
---|
| 18 | def __init__(self, context): |
---|
[2698] | 19 | gauth = queryUtility(IGAuthUtility) or queryUtility(IGAuthUtility, context=context) |
---|
[2682] | 20 | self.context = context |
---|
| 21 | self.shcl = SpreadsheetsService(gauth.email, gauth.password) |
---|
| 22 | self.shcl.ProgrammaticLogin() |
---|
| 23 | |
---|
[2686] | 24 | def getListFeed(self, query=None): |
---|
[2682] | 25 | """ Get SpreadsheetsListFeed |
---|
| 26 | """ |
---|
| 27 | return self._safeQuery(self.shcl.GetListFeed, |
---|
| 28 | self.context.spreadsheet_id, |
---|
| 29 | wksht_id=self.context.worksheet_id, |
---|
| 30 | query=query) |
---|
| 31 | |
---|
[2689] | 32 | def getWorksheetColumnsInfo(self): |
---|
| 33 | title_idxs = set() |
---|
| 34 | query = shDocumentQuery() |
---|
| 35 | query.max_results = '1' |
---|
| 36 | feed = self.getListFeed(query=query) |
---|
| 37 | if len(feed.entry): |
---|
| 38 | title_idxs.update(feed.entry[0].custom.keys()) |
---|
[2707] | 39 | return sorted(title_idxs) |
---|
[2682] | 40 | |
---|
| 41 | def _safeQuery(self, meth, *margs, **mkwargs): |
---|
| 42 | # Make safe method call with logging information about exception |
---|
| 43 | try: |
---|
| 44 | return meth(*margs, **mkwargs) |
---|
| 45 | except Exception: |
---|
| 46 | logException('%s function call: key=%s, wksht_id=%s' % ( |
---|
| 47 | meth.__name__, self.context.spreadsheet_id, |
---|
| 48 | self.context.worksheet_id), self.context) |
---|
| 49 | return None |
---|