brotherton-erpnext/erpnext/patches/v7_0/create_warehouse_nestedset.py

128 lines
4.6 KiB
Python
Raw Normal View History

2016-08-31 07:02:10 +00:00
from __future__ import unicode_literals
import frappe, erpnext
2016-05-24 02:19:00 +00:00
from frappe import _
from frappe.utils import cint
2016-06-28 10:13:02 +00:00
from frappe.utils.nestedset import rebuild_tree
2016-05-24 02:19:00 +00:00
def execute():
2016-06-29 13:33:40 +00:00
"""
Patch Reference:
1. check whether warehouse is associated to company or not
2. if warehouse is associated with company
a. create warehouse group for company
b. set warehouse group as parent to other warehouses and set is_group as 0
3. if warehouses is not associated with company
a. get distinct companies from stock ledger entries
b. if sle have only company,
i. set default company to all warehouse
ii. repeat 2.a and 2.b
c. if have multiple companies,
i. create group warehouse without company
ii. repeat 2.b
"""
frappe.reload_doc("stock", "doctype", "warehouse")
2016-06-29 13:33:40 +00:00
if check_is_warehouse_associated_with_company():
for company in frappe.get_all("Company", fields=["name", "abbr"]):
make_warehouse_nestedset(company)
else:
sle_against_companies = frappe.db.sql_list("""select distinct company from `tabStock Ledger Entry`""")
2016-06-29 13:33:40 +00:00
if len(sle_against_companies) == 1:
2016-09-26 12:30:00 +00:00
company = frappe.db.get_value("Company", sle_against_companies[0],
fieldname=["name", "abbr"], as_dict=1)
set_company_to_warehouse(company.name)
2016-06-29 13:33:40 +00:00
make_warehouse_nestedset(company)
2016-06-29 13:33:40 +00:00
elif len(sle_against_companies) > 1:
make_warehouse_nestedset()
def check_is_warehouse_associated_with_company():
warehouse_associcated_with_company = False
for warehouse in frappe.get_all("Warehouse", fields=["name", "company"]):
if warehouse.company:
warehouse_associcated_with_company = True
return warehouse_associcated_with_company
def make_warehouse_nestedset(company=None):
validate_parent_account_for_warehouse(company)
2016-07-04 12:42:08 +00:00
stock_account_group = get_stock_account_group(company.name)
enable_perpetual_inventory = cint(erpnext.is_perpetual_inventory_enabled(company)) or 0
if not stock_account_group and enable_perpetual_inventory:
2016-07-04 12:42:08 +00:00
return
2016-06-29 13:33:40 +00:00
if company:
warehouse_group = "{0} - {1}".format(_("All Warehouses"), company.abbr)
ignore_mandatory = False
else:
warehouse_group = _("All Warehouses")
ignore_mandatory = True
if not frappe.db.get_value("Warehouse", warehouse_group):
2016-07-04 12:42:08 +00:00
create_default_warehouse_group(company, stock_account_group, ignore_mandatory)
2016-06-29 13:33:40 +00:00
set_parent_to_warehouse(warehouse_group, company)
if enable_perpetual_inventory:
2016-07-04 12:42:08 +00:00
set_parent_to_warehouse_account(company)
2016-06-29 13:33:40 +00:00
def validate_parent_account_for_warehouse(company=None):
if not company:
return
2016-06-28 10:13:02 +00:00
if cint(erpnext.is_perpetual_inventory_enabled(company.name)):
parent_account = frappe.db.sql("""select name from tabAccount
where account_type='Stock' and company=%s and is_group=1
and (warehouse is null or warehouse = '')""", company.name)
2016-06-28 10:13:02 +00:00
if not parent_account:
current_parent_accounts_for_warehouse = frappe.db.sql("""select parent_account from tabAccount
where account_type='Warehouse' and (warehouse is not null or warehouse != '') """)
2016-06-28 10:13:02 +00:00
if current_parent_accounts_for_warehouse:
2016-06-28 10:13:02 +00:00
frappe.db.set_value("Account", current_parent_accounts_for_warehouse[0][0], "account_type", "Stock")
2016-06-29 13:33:40 +00:00
2016-07-04 12:42:08 +00:00
def create_default_warehouse_group(company=None, stock_account_group=None, ignore_mandatory=False):
2016-06-29 13:33:40 +00:00
wh = frappe.get_doc({
"doctype": "Warehouse",
"warehouse_name": _("All Warehouses"),
"is_group": 1,
"company": company.name if company else "",
"parent_warehouse": ""
2016-06-29 13:33:40 +00:00
})
if ignore_mandatory:
wh.flags.ignore_mandatory = ignore_mandatory
wh.insert(ignore_permissions=True)
def set_parent_to_warehouse(warehouse_group, company=None):
frappe.db.sql(""" update tabWarehouse set parent_warehouse = %s, is_group = 0
where (is_group = 0 or is_group is null or is_group = '') and ifnull(company, '') = %s
""",(warehouse_group, company.name if company else ""))
rebuild_tree("Warehouse", "parent_warehouse")
2016-07-04 12:42:08 +00:00
def set_parent_to_warehouse_account(company):
2016-06-29 13:33:40 +00:00
frappe.db.sql(""" update tabAccount set parent_account = %s
where is_group = 0 and account_type = "Warehouse"
and (warehouse is not null or warehouse != '') and company = %s
""",("{0} - {1}".format(_("All Warehouses"), company.abbr), company.name))
rebuild_tree("Account", "parent_account")
def set_company_to_warehouse(company):
frappe.db.sql("update tabWahouse set company=%s", company)
2016-07-04 12:42:08 +00:00
def get_stock_account_group(company):
stock_account_group = frappe.db.get_all('Account', filters = {'company': company, 'is_group': 1,
'account_type': 'Stock', 'root_type': 'Asset'}, limit=1)
if not stock_account_group:
stock_account_group = frappe.db.get_all('Account', filters = {'company': company, 'is_group': 1,
'parent_account': '', 'root_type': 'Asset'}, limit=1)
return stock_account_group[0].name if stock_account_group else None