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":