ckanext-googleanalytics/ckanext/googleanalytics/logic/action.py

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", []),
    }