96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
from __future__ import absolute_import
|
|
|
|
import ckan.plugins.toolkit as tk
|
|
from ckan.logic import validate
|
|
|
|
from . import schema
|
|
from .. import config
|
|
from ..model import PackageStats, ResourceStats
|
|
from ..ga_auth import init_service, get_profile_id
|
|
|
|
|
|
def get_actions():
|
|
return dict(
|
|
googleanalytics_package_stats_show=package_stats_show,
|
|
googleanalytics_resource_stats_show=resource_stats_show,
|
|
googleanalytics_event_report=event_report,
|
|
)
|
|
|
|
|
|
@validate(schema.package_stats_show)
|
|
@tk.side_effect_free
|
|
def package_stats_show(context, data_dict):
|
|
tk.check_access("googleanalytics_package_stats_show", context, data_dict)
|
|
rec = (
|
|
context["session"]
|
|
.query(PackageStats)
|
|
.filter(PackageStats.package_id == data_dict["id"])
|
|
.one_or_none()
|
|
)
|
|
|
|
if not rec:
|
|
raise tk.ObjectNotFound()
|
|
|
|
return rec.for_json(context)
|
|
|
|
|
|
@validate(schema.resource_stats_show)
|
|
@tk.side_effect_free
|
|
def resource_stats_show(context, data_dict):
|
|
tk.check_access("googleanalytics_resource_stats_show", context, data_dict)
|
|
rec = (
|
|
context["session"]
|
|
.query(ResourceStats)
|
|
.filter(ResourceStats.resource_id == data_dict["id"])
|
|
.one_or_none()
|
|
)
|
|
|
|
if not rec:
|
|
raise tk.ObjectNotFound()
|
|
|
|
return rec.for_json(context)
|
|
|
|
|
|
@validate(schema.event_report)
|
|
@tk.side_effect_free
|
|
def event_report(context, data_dict):
|
|
tk.check_access("sysadmin", context, data_dict)
|
|
|
|
se = init_service(config.credentials())
|
|
filters = []
|
|
if "action" in data_dict:
|
|
filters.append(
|
|
"ga:eventAction=={action}".format(action=data_dict["action"])
|
|
)
|
|
|
|
if "category" in data_dict:
|
|
filters.append(
|
|
"ga:eventCategory=={category}".format(
|
|
category=data_dict["category"]
|
|
)
|
|
)
|
|
|
|
if "label" in data_dict:
|
|
filters.append(
|
|
"ga:eventLabel=={label}".format(label=data_dict["label"])
|
|
)
|
|
|
|
report = (
|
|
se.data()
|
|
.ga()
|
|
.get(
|
|
ids="ga:{id}".format(id=get_profile_id(se)),
|
|
dimensions=",".join(data_dict["dimensions"]),
|
|
metrics=",".join(data_dict["metrics"]),
|
|
start_date=data_dict["start_date"].date().isoformat(),
|
|
end_date=data_dict["end_date"].date().isoformat(),
|
|
filters=";".join(filters) or None,
|
|
)
|
|
.execute()
|
|
)
|
|
|
|
return {
|
|
"headers": [h["name"] for h in report["columnHeaders"]],
|
|
"rows": report.get("rows", []),
|
|
}
|