brotherton-erpnext/erpnext/patches/v14_0/update_opportunity_currency_fields.py
2022-09-04 19:03:16 +05:30

47 lines
1.4 KiB
Python

import click
import frappe
from frappe.utils import flt
import erpnext
from erpnext.setup.utils import get_exchange_rate
def execute():
frappe.reload_doctype("Opportunity")
opportunities = frappe.db.get_list(
"Opportunity",
filters={"opportunity_amount": [">", 0]},
fields=["name", "company", "currency", "opportunity_amount"],
)
for opportunity in opportunities:
company_currency = erpnext.get_company_currency(opportunity.company)
if opportunity.currency is None or opportunity.currency == "":
opportunity.currency = company_currency
frappe.db.set_value(
"Opportunity",
opportunity.name,
{"currency": opportunity.currency},
update_modified=False,
)
click.secho(
f' Opportunity `{opportunity.name}` has no currency set. Setting it to company currency as default: `{opportunity.currency}`"\n',
fg="yellow",
)
# base total and total will be 0 only since item table did not have amount field earlier
if opportunity.currency != company_currency:
conversion_rate = get_exchange_rate(opportunity.currency, company_currency)
base_opportunity_amount = flt(conversion_rate) * flt(opportunity.opportunity_amount)
else:
conversion_rate = 1
base_opportunity_amount = flt(opportunity.opportunity_amount)
frappe.db.set_value(
"Opportunity",
opportunity.name,
{"conversion_rate": conversion_rate, "base_opportunity_amount": base_opportunity_amount},
update_modified=False,
)