fix(Treatment Counselling): Add payment entry on_cancel and paid amount calculation (backport #514)

healthcare/healthcare/custom_doctype/payment_entry.py

@@ -0,0 +1,22 @@
+import frappe
+
+
+@frappe.whitelist()
+def set_paid_amount_in_treatment_counselling(doc, method):
+	if doc.treatment_counselling and doc.paid_amount:
+		on_cancel = True if method == "on_cancel" else False
+		validate_treatment_counselling(doc, on_cancel)
+
+
+def validate_treatment_counselling(doc, on_cancel=False):
+	treatment_counselling_doc = frappe.get_doc("Treatment Counselling", doc.treatment_counselling)
+
+	paid_amount = treatment_counselling_doc.paid_amount + doc.paid_amount
+	if on_cancel:
+		paid_amount = treatment_counselling_doc.paid_amount - doc.paid_amount
+
+	treatment_counselling_doc.paid_amount = paid_amount
+	treatment_counselling_doc.outstanding_amount = (
+		treatment_counselling_doc.amount - treatment_counselling_doc.paid_amount
+	)
+	treatment_counselling_doc.save()

healthcare/healthcare/doctype/treatment_counselling/treatment_counselling.py

@@ -0,0 +1,249 @@
+# Copyright (c) 2023, healthcare and contributors
+# For license information, please see license.txt
+
+import frappe
+from frappe import _
+from frappe.model.document import Document
+from frappe.utils.data import flt
+
+from erpnext.stock.get_item_details import get_item_price
+
+from healthcare.healthcare.doctype.inpatient_record.inpatient_record import create_inpatient_record
+
+
+class TreatmentCounselling(Document):
+	def before_insert(self):
+		if self.customer:
+			self.price_list = frappe.db.get_value("Customer", self.customer, "default_price_list")
+
+		if not self.price_list:
+			self.price_list = frappe.db.get_single_value("Selling Settings", "selling_price_list")
+
+	def validate(self):
+		if self.status == "Active":
+			active_tpc = frappe.db.exists(
+				"Treatment Counselling",
+				{
+					"status": "Active",
+					"docstatus": 1,
+					"inpatient_record": self.inpatient_record,
+					"admission_encounter": self.admission_encounter,
+					"name": ["!=", self.name],
+				},
+			)
+
+			if active_tpc:
+				frappe.throw(
+					_("Active Treatment Counselling {0} already exist for encounter {1}").format(
+						frappe.bold(frappe.utils.get_link_to_form("Treatment Counselling", active_tpc)),
+						frappe.bold(frappe.utils.get_link_to_form("Patient Encounter", self.admission_encounter)),
+					)
+				)
+		set_treatment_plan_template_items(self)
+		set_total_amount(self)
+
+		if self.paid_amount:
+			self.outstanding_amount = self.amount - self.paid_amount
+		else:
+			self.outstanding_amount = self.amount
+
+	def after_insert(self):
+		if self.admission_encounter:  # Update encounter
+			frappe.db.set_value(
+				"Patient Encounter",
+				self.admission_encounter,
+				"inpatient_status",
+				"Treatment Counselling Created",
+			)
+
+	def on_update_after_submit(self):
+		doc_before_save = self.get_doc_before_save()
+		if self.status == "Closed" and doc_before_save.status != "Closed" and self.admission_encounter:
+			frappe.db.set_value("Patient Encounter", self.admission_encounter, "inpatient_status", "")
+
+		if self.paid_amount:
+			self.db_set("outstanding_amount", self.amount - self.paid_amount)
+		else:
+			self.db_set("outstanding_amount", self.amount)
+
+	def on_cancel(self):
+		if self.admission_encounter:
+			frappe.db.set_value("Patient Encounter", self.admission_encounter, "inpatient_status", "")
+
+
+def set_treatment_plan_template_items(doc):
+	if doc.treatment_plan_template and not doc.treatment_plan_template_items:
+		item_list = get_encounter_items(doc.admission_encounter)
+
+		template_item_list = frappe.get_all(
+			"Treatment Plan Template Item",
+			{"parent": doc.treatment_plan_template},
+			["type", "template", "qty", "instructions"],
+		)
+		item_list.extend(template_item_list)
+		medication_list = frappe.get_all(
+			"Drug Prescription", {"parent": doc.treatment_plan_template}, ["*"]
+		)
+		item_list.extend(medication_list)
+		for item in item_list:
+			args = {"price_list": doc.price_list}
+			if item.get("drug_code"):
+				item_details = get_item_price(args, item.get("drug_code"))
+			else:
+				uom = frappe.db.exists("UOM", "Unit") or frappe.db.get_single_value(
+					"Stock Settings", "stock_uom"
+				)
+				args["uom"] = uom
+				item_details = get_item_price(args, item.get("template"))
+			if item_details:
+				item_price = [tupl[element] for tupl in item_details for element in tupl][1]
+				item["amount"] = flt(item_price) if item_price else 0
+
+			if item.get("drug_code"):
+				if not any(
+					d.get("drug_code") == item.get("drug_code") for d in doc.treatment_plan_template_items
+				):
+					doc.append(
+						"treatment_plan_template_items",
+						{
+							"type": "Medication",
+							"template": item.get("medication"),
+							"drug_code": item.get("drug_code"),
+							"drug_name": item.get("drug_name"),
+							"strength": item.get("strength"),
+							"strength_uom": item.get("strength_uom"),
+							"dosage_form": item.get("dosage_form"),
+							"dosage": item.get("dosage"),
+							"period": item.get("period"),
+							"number_of_repeats_allowed": item.get("number_of_repeats_allowed"),
+							"service_request": item.get("medication_request", ""),
+						},
+					)
+			else:
+				if not any(d.get("template") == item["template"] for d in doc.treatment_plan_template_items):
+					doc.append(
+						"treatment_plan_template_items",
+						{
+							"type": item.get("type"),
+							"template": item.get("template"),
+							"qty": item.get("qty", 1),
+							"amount": item.get("amount", 0),
+							"instructions": item.get("instructions", ""),
+							"service_request": item.get("service_request", ""),
+						},
+					)
+		if doc.admission_service_unit_type:
+			su_item, su_rate = frappe.db.get_value(
+				"Healthcare Service Unit Type", doc.admission_service_unit_type, ["item", "rate"]
+			)
+			if su_item and su_rate:
+				doc.append(
+					"treatment_plan_template_items",
+					{
+						"type": "Item",
+						"template": su_item,
+						"qty": (doc.expected_length_of_stay or 1),
+						"amount": su_rate,
+					},
+				)
+
+
+def set_total_amount(doc):
+	total_price = 0
+	for item in doc.treatment_plan_template_items:
+		if item.amount:
+			total_price += item.qty * item.amount
+
+	doc.amount = total_price
+
+
+@frappe.whitelist()
+def create_ip_from_treatment_counselling(admission_order, treatment_counselling):
+	ip_name = create_inpatient_record(admission_order)
+	frappe.db.set_value(
+		"Treatment Counselling",
+		treatment_counselling,
+		{"status": "Completed", "inpatient_record": ip_name},
+	)
+
+
+@frappe.whitelist()
+def create_payment_entry(treatment_counselling):
+	treatment_counselling_doc = frappe.get_doc("Treatment Counselling", treatment_counselling)
+	payment_entry_doc = frappe.new_doc("Payment Entry")
+	payment_entry_doc.update(
+		{
+			"payment_type": "Receive",
+			"party_type": "Customer",
+			"party": treatment_counselling_doc.customer,
+			"treatment_counselling": treatment_counselling,
+			"paid_amount": treatment_counselling_doc.outstanding_amount,
+			"received_amount": treatment_counselling_doc.outstanding_amount,
+			"target_exchange_rate": 1,
+		}
+	)
+
+	payment_entry_doc.insert(ignore_mandatory=True)
+	return payment_entry_doc.name
+
+
+def get_encounter_items(encounter):
+	item_list = []
+	if encounter:
+		encounter_doc = frappe.get_doc("Patient Encounter", encounter)
+		for lab_presc in encounter_doc.lab_test_prescription:
+			if lab_presc.get("observation_template"):
+				item_list.append(
+					{
+						"template": lab_presc.get("observation_template"),
+						"type": "Observation Template",
+						"service_request": lab_presc.get("service_request"),
+					}
+				)
+			elif lab_presc.get("lab_test_code"):
+				item_list.append(
+					{
+						"template": lab_presc.get("lab_test_code"),
+						"type": "Lab Test Template",
+						"service_request": lab_presc.get("service_request"),
+					}
+				)
+			else:
+				continue
+
+		for pro_pres in encounter_doc.procedure_prescription:
+			if pro_pres.get("procedure"):
+				item_list.append(
+					{
+						"template": pro_pres.get("procedure"),
+						"type": "Clinical Procedure Template",
+						"service_request": pro_pres.get("service_request"),
+					}
+				)
+
+		for ther in encounter_doc.therapies:
+			if ther.get("therapy_type"):
+				item_list.append(
+					{
+						"template": ther.get("therapy_type"),
+						"type": "Therapy Type",
+						"service_request": ther.get("service_request"),
+					}
+				)
+
+		for drug in encounter_doc.drug_prescription:
+			item_list.append(
+				{
+					"medication": drug.get("medication"),
+					"drug_code": drug.get("drug_code"),
+					"drug_name": drug.get("drug_name"),
+					"strength": drug.get("strength"),
+					"strength_uom": drug.get("strength_uom"),
+					"dosage_form": drug.get("dosage_form"),
+					"dosage": drug.get("dosage"),
+					"period": drug.get("period"),
+					"number_of_repeats_allowed": drug.get("number_of_repeats_allowed"),
+					"medication_request": drug.get("medication_request"),
+				}
+			)
+	return item_list

healthcare/hooks.py

@@ -125,6 +125,13 @@
 	"Patient": {
 		"after_insert": "healthcare.regional.india.abdm.utils.set_consent_attachment_details"
 	},
+<<<<<<< HEAD
+=======
+	"Payment Entry": {
+		"on_submit": "healthcare.healthcare.custom_doctype.payment_entry.set_paid_amount_in_treatment_counselling",
+		"on_cancel": "healthcare.healthcare.custom_doctype.payment_entry.set_paid_amount_in_treatment_counselling",
+	},
+>>>>>>> 69f4b9d (fix(Treatment Counselling): Add payment entry on_cancel and paid amount calculation)
 }
 
 scheduler_events = {