valitydev/redash
14
0
Fork 0
mirror of https://github.com/valitydev/redash.git synced 2024-11-06 17:15:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

google_analytics: added accounts and properties as schema tables

Browse Source
This commit is contained in:
Vladislav Denisov 2016-11-23 17:17:05 +03:00
parent 8eefad290b
commit 92dee61bcd
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
redash/query_runner/google_analytics.py
View File

@ -1,3 +1,5 @@
# -*- coding: utf-8 -*-
from base64 import b64decode
import json
import logging
@ -32,7 +34,7 @@ types_conv = dict(
)
class GoogleAnalytics(BaseQueryRunner):
class GoogleAnalytics(BaseSQLQueryRunner):
@classmethod
def annotate_query(cls):
return False
@ -62,6 +64,21 @@ class GoogleAnalytics(BaseQueryRunner):
def __init__(self, configuration):
super(GoogleAnalytics, self).__init__(configuration)
def _get_tables(self, schema):
accounts = self._get_analytics_service().management().accounts().list().execute().get('items')
if accounts is None:
raise Exception("Failed getting accounts.")
else:
for account in accounts:
schema[account['name']] = {'name': account['name'], 'columns': []}
properties = self._get_analytics_service().management().webproperties().list(
accountId=account['id']).execute().get('items', [])
for property_ in properties:
schema[account['name']]['columns'].append(
u'{0} (ga:{1})'.format(property_['name'], property_['defaultProfileId'])
)
return schema.values()
def _get_analytics_service(self):
scope = ['https://www.googleapis.com/auth/analytics.readonly']
key = json.loads(b64decode(self.configuration['jsonKeyFile']))