From 3f2ee8509fada26c9bb43aeceb76e121980b68e9 Mon Sep 17 00:00:00 2001
From: venkataakhil <support@hnsonline.com>
Date: Tue, 7 Nov 2023 16:53:47 +0530
Subject: [PATCH] Update vehicle publish flags

---
 .../transactions/doctype/publish/publish.py         | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/smart_service/transactions/doctype/publish/publish.py b/smart_service/transactions/doctype/publish/publish.py
index 6b357b5..e0bf6df 100644
--- a/smart_service/transactions/doctype/publish/publish.py
+++ b/smart_service/transactions/doctype/publish/publish.py
@@ -6,12 +6,16 @@ from html import unescape
 import frappe
 from frappe.model.document import Document
 import json
+from datetime import date
+
 current_db_name = frappe.conf.get("db_name")
 sys.tracebacklimit = 0
 
 
 class Publish(Document):
     def validate(self):
+        self.published_date = str(date.today())
+        self.published_by = frappe.session.user
         # Selected variant and kilometer mapping details list
         variant = []
         for vm in self.get('publish_documents'):
@@ -83,6 +87,15 @@ class Publish(Document):
                         "Publish", self.name, "published_by", str(name))
                     frappe.db.commit()
 
+    def on_submit(self):
+        if self.publish_type == 'Internal':
+            frappe.db.sql(
+                '''update tabVehicle set internal_publish = 1 where vehicle= '{0}'; '''.format(self.vehicle))
+
+        else:
+            frappe.db.sql(
+                '''update tabVehicle set global_publish = 1 where vehicle= '{0}'; '''.format(self.vehicle))
+            
     def on_cancel(self):
         # Published document should not allow to cancel
         if self.publish_status == "Published":