brotherton-erpnext/erpnext/utilities/report/youtube_interactions/youtube_interactions.py

# Copyright (c) 2013, Frappe Technologies Pvt. Ltd. and contributors
# For license information, please see license.txt


import frappe
from frappe import _
from frappe.utils import flt


def execute(filters=None):
	if not frappe.db.get_single_value("Video Settings", "enable_youtube_tracking") or not filters:
		return [], []

	columns = get_columns()
	data = get_data(filters)
	chart_data, summary = get_chart_summary_data(data)
	return columns, data, None, chart_data, summary


def get_columns():
	return [
		{"label": _("Published Date"), "fieldname": "publish_date", "fieldtype": "Date", "width": 100},
		{"label": _("Title"), "fieldname": "title", "fieldtype": "Data", "width": 200},
		{"label": _("Duration"), "fieldname": "duration", "fieldtype": "Duration", "width": 100},
		{"label": _("Views"), "fieldname": "view_count", "fieldtype": "Float", "width": 200},
		{"label": _("Likes"), "fieldname": "like_count", "fieldtype": "Float", "width": 200},
		{"label": _("Dislikes"), "fieldname": "dislike_count", "fieldtype": "Float", "width": 100},
		{"label": _("Comments"), "fieldname": "comment_count", "fieldtype": "Float", "width": 100},
	]


def get_data(filters):
	return frappe.db.sql(
		"""
		SELECT
			publish_date, title, provider, duration,
			view_count, like_count, dislike_count, comment_count
		FROM `tabVideo`
		WHERE view_count is not null
			and publish_date between %(from_date)s and %(to_date)s
		ORDER BY view_count desc""",
		filters,
		as_dict=1,
	)


def get_chart_summary_data(data):
	labels, likes, views = [], [], []
	total_views = 0

	for row in data:
		labels.append(row.get("title"))
		likes.append(row.get("like_count"))
		views.append(row.get("view_count"))
		total_views += flt(row.get("view_count"))

	chart_data = {
		"data": {
			"labels": labels,
			"datasets": [{"name": "Likes", "values": likes}, {"name": "Views", "values": views}],
		},
		"type": "bar",
		"barOptions": {"stacked": 1},
	}

	summary = [
		{
			"value": total_views,
			"indicator": "Blue",
			"label": "Total Views",
			"datatype": "Float",
		}
	]
	return chart_data, summary