# Copyright (c) 2017, Frappe and Contributors
# License: GNU General Public License v3. See license.txt

from __future__ import unicode_literals
import frappe

def execute():
	if frappe.db.a_row_exists("Leave Application"):
		frappe.reload_doc("hr", "doctype", "leave_application")
		frappe.reload_doc("workflow", "doctype", "workflow")
		states = {"Approved": "Success", "Rejected": "Danger", "Open": "Warning"}

		for state, style in states.items():
			if not frappe.db.exists("Workflow State", state):
				frappe.get_doc({
					"doctype": "Workflow State",
					"workflow_state_name": state,
					"style": style
				}).insert(ignore_permissions=True)

		for action in ["Approve", "Reject"]:
			if not frappe.db.exists("Workflow Action", action):
				frappe.get_doc({
					"doctype": "Workflow Action",
					"workflow_action_name": action
				}).insert(ignore_permissions=True)

		if not frappe.db.exists("Workflow", "Leave Approval"):
			frappe.get_doc({
				"doctype": "Workflow",
				"workflow_name": "Leave Approval",
				"document_type": "Leave Application",
				"is_active": 1,
				"workflow_state_field": "workflow_state",
				"states": [{
					"state": "Open",
					"doc_status": 0,
					"update_field": "status",
					"update_value": "Open",
					"allow_edit": "Employee"
				}, {
					"state": "Approved",
					"doc_status": 1,
					"update_field": "status",
					"update_value": "Approved",
					"allow_edit": "Leave Approver"
				}, {
					"state": "Rejected",
					"doc_status": 0,
					"update_field": "status",
					"update_value": "Rejected",
					"allow_edit": "Leave Approver"
				}],
				"transitions": [{
					"state": "Open",
					"action": "Approve",
					"next_state": "Approved",
					"allowed": "Leave Approver"
				},
				{
					"state": "Open",
					"action": "Reject",
					"next_state": "Rejected",
					"allowed": "Leave Approver"
				}]
			}).insert(ignore_permissions=True)

		if frappe.db.has_column("Leave Application", "status"):
			frappe.db.sql("""update `tabLeave Application` set workflow_state = status""")

		if frappe.db.has_column("Leave Application", "workflow_state"):
			frappe.db.sql("""update `tabWorkflow Document State` set doc_status = 0 where parent = "Leave Approval" \
			and state = "Rejected" and doc_status = 1""")
			frappe.db.sql("""update `tabLeave Application` set docstatus = 0  where workflow_state = "Rejected" and docstatus = 1""")