nsinnovations/erpnext
2
0
Fork 0
mirror of https://github.com/frappe/erpnext.git synced 2026-06-04 20:59:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: code cleanup

Browse Source
This commit is contained in:
Noah Jacob
2021-08-10 18:35:46 +05:30
parent 4ee6e32d74
commit 49b7c7575e
1 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
erpnext/accounts/doctype/promotional_scheme/promotional_scheme.py
View File

@@ -25,12 +25,14 @@ product_discount_fields = ['free_item', 'free_qty', 'free_item_uom',
class PromotionalScheme(Document): class PromotionalScheme(Document):
def validate(self): def validate(self):
if not self.selling and not self.buying:
frappe.throw(_("Atleast one of the Selling or Buying must be selected"))
if not (self.price_discount_slabs if not (self.price_discount_slabs
or self.product_discount_slabs): or self.product_discount_slabs):
frappe.throw(_("Price or product discount slabs are required")) frappe.throw(_("Price or product discount slabs are required"))
def on_update(self): def on_update(self):
data = frappe.get_all( pricing_rules = frappe.get_all(
'Pricing Rule', 'Pricing Rule',
fields = ["promotional_scheme_id", "name", "creation"], fields = ["promotional_scheme_id", "name", "creation"],
filters = { filters = {
@@ -39,15 +41,15 @@ class PromotionalScheme(Document):
}, },
order_by = 'creation asc', order_by = 'creation asc',
) or {} ) or {}
self.update_pricing_rules(data) self.update_pricing_rules(pricing_rules)
def update_pricing_rules(self, data): def update_pricing_rules(self, pricing_rules):
rules = {} rules = {}
count = 0 count = 0
names = [] names = []
for d in data: for rule in pricing_rules:
names.append(d.name) names.append(rule.name)
rules[d.get('promotional_scheme_id')] = names rules[rule.get('promotional_scheme_id')] = names
docs = get_pricing_rules(self, rules) docs = get_pricing_rules(self, rules)
@@ -64,9 +66,9 @@ class PromotionalScheme(Document):
frappe.msgprint(_("New {0} pricing rules are created").format(count)) frappe.msgprint(_("New {0} pricing rules are created").format(count))
def on_trash(self): def on_trash(self):
for d in frappe.get_all('Pricing Rule', for rule in frappe.get_all('Pricing Rule',
{'promotional_scheme': self.name}): {'promotional_scheme': self.name}):
frappe.delete_doc('Pricing Rule', d.name) frappe.delete_doc('Pricing Rule', rule.name)
def get_pricing_rules(doc, rules = {}): def get_pricing_rules(doc, rules = {}):
new_doc = [] new_doc = []
@@ -107,17 +109,20 @@ def _get_pricing_rules(doc, child_doc, discount_fields, rules = {}):
new_doc.append(pr) new_doc.append(pr)
else: else:
for i in range(len(args.get(applicable_for))) : applicable_for_values = args.get(applicable_for) or []
for applicable_for_value in applicable_for_values:
pr = frappe.new_doc("Pricing Rule") pr = frappe.new_doc("Pricing Rule")
pr.title = doc.name pr.title = doc.name
temp_args = args.copy() temp_args = args.copy()
temp_args[applicable_for] = args[applicable_for][i] temp_args[applicable_for] = applicable_for_value
pr = set_args(temp_args, pr, doc, child_doc, discount_fields, d) pr = set_args(temp_args, pr, doc, child_doc, discount_fields, d)
new_doc.append(pr) new_doc.append(pr)
return new_doc return new_doc
def set_args(args, pr, doc, child_doc, discount_fields, child_doc_fields): def set_args(args, pr, doc, child_doc, discount_fields, child_doc_fields):
pr.update(args) pr.update(args)
for field in (other_fields + discount_fields): for field in (other_fields + discount_fields):