brotherton-erpnext/erpnext/hr/report/employee_analytics/employee_analytics.py

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

from __future__ import unicode_literals
import frappe
from frappe import _

def execute(filters=None):
	if not filters: filters = {}

	if not filters["company"]:
		frappe.throw(_('{0} is mandatory').format(_('Company')))

	columns = get_columns()
	employees = get_employees(filters)
	parameters_result = get_parameters(filters)
	parameters = []
	if parameters_result:
		for department in parameters_result:
			parameters.append(department)

	chart = get_chart_data(parameters,employees, filters)
	return columns, employees, None, chart

def get_columns():
	return [
		_("Employee") + ":Link/Employee:120", _("Name") + ":Data:200", _("Date of Birth")+ ":Date:100",
		_("Branch") + ":Link/Branch:120", _("Department") + ":Link/Department:120",
		_("Designation") + ":Link/Designation:120", _("Gender") + "::100", _("Company") + ":Link/Company:120"
	]

def get_conditions(filters):
	conditions = " and "+filters.get("parameter").lower().replace(" ","_")+" IS NOT NULL "

	if filters.get("company"): conditions += " and company = '%s'" % \
		filters["company"].replace("'", "\\'")
	return conditions

def get_employees(filters):
	conditions = get_conditions(filters)
	return frappe.db.sql("""select name, employee_name, date_of_birth,
	branch, department, designation,
	gender, company from `tabEmployee` where status = 'Active' %s""" % conditions, as_list=1)

def get_parameters(filters):
	if filters.get("parameter") == "Grade":
		parameter = "Employee Grade"
	else:
		parameter = filters.get("parameter")

	return frappe.db.sql("""select name from `tab"""+ parameter +"""` """, as_list=1)

def get_chart_data(parameters,employees, filters):
	if not parameters:
		parameters = []
	datasets = []
	parameter_field_name = filters.get("parameter").lower().replace(" ","_")
	label = []
	for parameter in parameters:
		if parameter:
			total_employee = frappe.db.sql("""select count(*) from
				`tabEmployee` where """+
				parameter_field_name + """ = %s and  company = %s""" ,( parameter[0], filters.get("company")), as_list=1)
			if total_employee[0][0]:
				label.append(parameter)
			datasets.append(total_employee[0][0])

	values = [ value for value in datasets if value !=0]

	total_employee = frappe.db.count('Employee', {'status':'Active'})
	others = total_employee - sum(values)

	label.append(["Not Set"])
	values.append(others)

	chart = {
		"data": {
			'labels': label,
			'datasets': [{'name': 'Employees','values': values}]
		}
	}
	chart["type"] = "donut"
	return chart