108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import hashlib
|
|
import logging
|
|
import six
|
|
|
|
from flask import Blueprint
|
|
from werkzeug.utils import import_string
|
|
|
|
import ckan.logic as logic
|
|
import ckan.plugins.toolkit as tk
|
|
import ckan.views.api as api
|
|
import ckan.views.resource as resource
|
|
|
|
from ckan.common import g
|
|
|
|
CONFIG_HANDLER_PATH = 'googleanalytics.download_handler'
|
|
|
|
log = logging.getLogger(__name__)
|
|
ga = Blueprint("google_analytics", "google_analytics")
|
|
|
|
|
|
def action(logic_function, ver=api.API_MAX_VERSION):
|
|
try:
|
|
function = logic.get_action(logic_function)
|
|
side_effect_free = getattr(function, "side_effect_free", False)
|
|
request_data = api._get_request_data(try_url_params=side_effect_free)
|
|
if isinstance(request_data, dict):
|
|
id = request_data.get("id", "")
|
|
if "q" in request_data:
|
|
id = request_data["q"]
|
|
if "query" in request_data:
|
|
id = request_data[u"query"]
|
|
_post_analytics(g.user, "CKAN API Request", logic_function, "", id)
|
|
except Exception as e:
|
|
log.debug(e)
|
|
pass
|
|
|
|
return api.action(logic_function, ver)
|
|
|
|
|
|
ga.add_url_rule(
|
|
"/api/action/<logic_function>", methods=["GET", "POST"], view_func=action,
|
|
)
|
|
ga.add_url_rule(
|
|
u"/<int(min=3, max={0}):ver>/action/<logic_function>".format(
|
|
api.API_MAX_VERSION
|
|
),
|
|
methods=["GET", "POST"],
|
|
view_func=action,
|
|
)
|
|
|
|
|
|
def download(id, resource_id, filename=None, package_type="dataset"):
|
|
handler_path = tk.config.get(CONFIG_HANDLER_PATH)
|
|
if handler_path:
|
|
handler = import_string(handler_path, silent=True)
|
|
else:
|
|
handler = None
|
|
log.warning((
|
|
'Missing {} config option.'
|
|
).format(CONFIG_HANDLER_PATH))
|
|
if not handler:
|
|
log.debug('Use default CKAN callback for resource.download')
|
|
handler = resource.download
|
|
_post_analytics(
|
|
g.user,
|
|
"CKAN Resource Download Request",
|
|
"Resource",
|
|
"Download",
|
|
resource_id,
|
|
)
|
|
return handler(
|
|
package_type=package_type, id=id,
|
|
resource_id=resource_id, filename=filename)
|
|
|
|
|
|
ga.add_url_rule(
|
|
"/dataset/<id>/resource/<resource_id>/download", view_func=download
|
|
)
|
|
ga.add_url_rule(
|
|
"/dataset/<id>/resource/<resource_id>/download/<filename>",
|
|
view_func=download,
|
|
)
|
|
|
|
|
|
def _post_analytics(
|
|
user, event_type, request_obj_type, request_function, request_id
|
|
):
|
|
|
|
from ckanext.googleanalytics.plugin import GoogleAnalyticsPlugin
|
|
|
|
if tk.config.get("googleanalytics.id"):
|
|
data_dict = {
|
|
"v": 1,
|
|
"tid": tk.config.get("googleanalytics.id"),
|
|
"cid": hashlib.md5(six.ensure_binary(tk.c.user)).hexdigest(),
|
|
# customer id should be obfuscated
|
|
"t": "event",
|
|
"dh": tk.request.environ["HTTP_HOST"],
|
|
"dp": tk.request.environ["PATH_INFO"],
|
|
"dr": tk.request.environ.get("HTTP_REFERER", ""),
|
|
"ec": event_type,
|
|
"ea": request_obj_type + request_function,
|
|
"el": request_id,
|
|
}
|
|
GoogleAnalyticsPlugin.analytics_queue.put(data_dict)
|