From e97263bac2c59c3b80c5a6ffbe8e6cab8c9b04be Mon Sep 17 00:00:00 2001
From: Sanket322 <shahsanket322003.com>
Date: Fri, 24 Jan 2025 13:11:58 +0530
Subject: [PATCH] fix: refresh fields after update

---
 .../gst_india/client_scripts/address.js       | 27 ++++++++++---------
 1 file changed, 14 insertions(+), 13 deletions(-)

diff --git a/india_compliance/gst_india/client_scripts/address.js b/india_compliance/gst_india/client_scripts/address.js
index 915713b8e..2e6eff11a 100644
--- a/india_compliance/gst_india/client_scripts/address.js
+++ b/india_compliance/gst_india/client_scripts/address.js
@@ -24,7 +24,7 @@ frappe.ui.form.on(DOCTYPE, {
     async refresh(frm) {
         india_compliance.set_state_options(frm);
 
-        update_address_info(frm);
+        frm.add_custom_button(__("Update Address"), () => update_address_fields(frm));
 
         // set default values for GST fields
         if (!frm.is_new() || !frm.doc.links || !frm.doc.links.length || frm.doc.gstin) return;
@@ -55,17 +55,18 @@ frappe.ui.form.on(DOCTYPE, {
     },
 });
 
-function update_address_info(frm) {
-    frm.add_custom_button(__("Update Address"), function () {
-        const doc = frappe.get_doc(DOCTYPE, frm.doc.name);
-        doc._gstin = frm.doc.gstin;
-        doc._pincode = frm.doc.pincode;
+function update_address_fields(frm) {
+    const doc = frappe.get_doc(DOCTYPE, frm.doc.name);
+    doc._gstin = frm.doc.gstin;
+    doc._pincode = frm.doc.pincode;
 
-        const Address = frappe.ui.form.AddressQuickEntryForm;
-        frappe.ui.form.AddressQuickEntryForm = class extends frappe.ui.form.AddressQuickEntryForm {
-            get_dynamic_link_fields() { return []; }
-        }
-        frappe.ui.form.make_quick_entry(DOCTYPE,null,null,doc);
-        frappe.ui.form.AddressQuickEntryForm = Address;
-    });
+    const original_quick_entry_form = frappe.ui.form.AddressQuickEntryForm;
+
+    frappe.ui.form.AddressQuickEntryForm = class extends frappe.ui.form.AddressQuickEntryForm {
+        get_dynamic_link_fields() { return []; }
+        update_doc() { frm.refresh(); }
+    }
+    frappe.ui.form.make_quick_entry(DOCTYPE, null, null, doc);
+
+    frappe.ui.form.AddressQuickEntryForm = original_quick_entry_form;
 }
\ No newline at end of file