import frappe
from frappe.model.utils.rename_field import rename_field
from frappe.modules import get_doctype_module, scrub

field_rename_map = {
	"Healthcare Settings": [
		["patient_master_name", "patient_name_by"],
		["max_visit", "max_visits"],
		["reg_sms", "send_registration_msg"],
		["reg_msg", "registration_msg"],
		["app_con", "send_appointment_confirmation"],
		["app_con_msg", "appointment_confirmation_msg"],
		["no_con", "avoid_confirmation"],
		["app_rem", "send_appointment_reminder"],
		["app_rem_msg", "appointment_reminder_msg"],
		["rem_before", "remind_before"],
		["manage_customer", "link_customer_to_patient"],
		["create_test_on_si_submit", "create_lab_test_on_si_submit"],
		["require_sample_collection", "create_sample_collection_for_lab_test"],
		["require_test_result_approval", "lab_test_approval_required"],
		["manage_appointment_invoice_automatically", "automate_appointment_invoicing"],
	],
	"Drug Prescription": [["use_interval", "usage_interval"], ["in_every", "interval_uom"]],
	"Lab Test Template": [
		["sample_quantity", "sample_qty"],
		["sample_collection_details", "sample_details"],
	],
	"Sample Collection": [
		["sample_quantity", "sample_qty"],
		["sample_collection_details", "sample_details"],
	],
	"Fee Validity": [["max_visit", "max_visits"]],
}


def execute():
	for dn in field_rename_map:
		if frappe.db.exists("DocType", dn):
			if dn == "Healthcare Settings":
				frappe.reload_doctype("Healthcare Settings")
			else:
				frappe.reload_doc(get_doctype_module(dn), "doctype", scrub(dn))

	for dt, field_list in field_rename_map.items():
		if frappe.db.exists("DocType", dt):
			for field in field_list:
				if dt == "Healthcare Settings":
					rename_field(dt, field[0], field[1])
				elif frappe.db.has_column(dt, field[0]):
					rename_field(dt, field[0], field[1])

	# first name mandatory in Patient
	if frappe.db.exists("DocType", "Patient"):
		patients = frappe.db.sql("select name, patient_name from `tabPatient`", as_dict=1)
		frappe.reload_doc("healthcare", "doctype", "patient")
		for entry in patients:
			name = entry.patient_name.split(" ")
			frappe.db.set_value("Patient", entry.name, "first_name", name[0])

	# mark Healthcare Practitioner status as Disabled
	if frappe.db.exists("DocType", "Healthcare Practitioner"):
		practitioners = frappe.db.sql(
			"select name from `tabHealthcare Practitioner` where 'active'= 0", as_dict=1
		)
		practitioners_lst = [p.name for p in practitioners]
		frappe.reload_doc("healthcare", "doctype", "healthcare_practitioner")
		if practitioners_lst:
			frappe.db.sql(
				"update `tabHealthcare Practitioner` set status = 'Disabled' where name IN %(practitioners)s"
				"",
				{"practitioners": practitioners_lst},
			)

	# set Clinical Procedure status
	if frappe.db.exists("DocType", "Clinical Procedure"):
		frappe.reload_doc("healthcare", "doctype", "clinical_procedure")
		frappe.db.sql(
			"""
			UPDATE
				`tabClinical Procedure`
			SET
				docstatus = (CASE WHEN status = 'Cancelled' THEN 2
								WHEN status = 'Draft' THEN 0
								ELSE 1
							END)
		"""
		)

	# set complaints and diagnosis in table multiselect in Patient Encounter
	if frappe.db.exists("DocType", "Patient Encounter"):
		field_list = [["visit_department", "medical_department"], ["type", "appointment_type"]]
		encounter_details = frappe.db.sql(
			"""select symptoms, diagnosis, name from `tabPatient Encounter`""", as_dict=True
		)
		frappe.reload_doc("healthcare", "doctype", "patient_encounter")
		frappe.reload_doc("healthcare", "doctype", "patient_encounter_symptom")
		frappe.reload_doc("healthcare", "doctype", "patient_encounter_diagnosis")

		for field in field_list:
			if frappe.db.has_column(dt, field[0]):
				rename_field(dt, field[0], field[1])

		for entry in encounter_details:
			doc = frappe.get_doc("Patient Encounter", entry.name)
			symptoms = entry.symptoms.split("\n") if entry.symptoms else []
			for symptom in symptoms:
				if not frappe.db.exists("Complaint", symptom):
					frappe.get_doc({"doctype": "Complaint", "complaints": symptom}).insert()
				row = doc.append("symptoms", {"complaint": symptom})
				row.db_update()

			diagnosis = entry.diagnosis.split("\n") if entry.diagnosis else []
			for d in diagnosis:
				if not frappe.db.exists("Diagnosis", d):
					frappe.get_doc({"doctype": "Diagnosis", "diagnosis": d}).insert()
				row = doc.append("diagnosis", {"diagnosis": d})
				row.db_update()
			doc.db_update()

	if frappe.db.exists("DocType", "Fee Validity"):
		# update fee validity status
		frappe.db.sql(
			"""
			UPDATE
				`tabFee Validity`
			SET
				status = (CASE WHEN visited >= max_visits THEN 'Completed'
								ELSE 'Pending'
							END)
		"""
		)