105 lines
3.9 KiB
Python
105 lines
3.9 KiB
Python
import webnotes
|
|
def execute():
|
|
webnotes.reload_doc("setup", "doctype", "company")
|
|
create_chart_of_accounts_if_not_exists()
|
|
add_group_accounts()
|
|
add_ledger_accounts()
|
|
add_aii_cost_center()
|
|
set_default_accounts()
|
|
|
|
def set_default_accounts():
|
|
for company in webnotes.conn.sql_list("select name from `tabCompany`"):
|
|
webnotes.get_obj("Company", company).set_default_accounts()
|
|
|
|
def _check(parent_account, company):
|
|
def _get_root(is_pl_account, debit_or_credit):
|
|
res = webnotes.conn.sql("""select name from `tabAccount`
|
|
where company=%s and is_pl_account = %s and debit_or_credit = %s
|
|
and ifnull(parent_account, "") ="" """, (company, is_pl_account, debit_or_credit))
|
|
return res and res[0][0] or None
|
|
|
|
if not webnotes.conn.exists("Account", parent_account):
|
|
if parent_account.startswith("Current Assets"):
|
|
parent_account = _get_root("No", "Debit")
|
|
elif parent_account.startswith("Direct Expenses"):
|
|
parent_account = _get_root("Yes", "Debit")
|
|
elif parent_account.startswith("Current Liabilities"):
|
|
parent_account = _get_root("No", "Credit")
|
|
|
|
return parent_account
|
|
|
|
|
|
def add_group_accounts():
|
|
accounts_to_add = [
|
|
["Stock Assets", "Current Assets", "Group", ""],
|
|
["Stock Expenses", "Direct Expenses", "Group", "Expense Account"],
|
|
["Stock Liabilities", "Current Liabilities", "Group", ""],
|
|
]
|
|
|
|
add_accounts(accounts_to_add, _check)
|
|
|
|
|
|
def add_ledger_accounts():
|
|
accounts_to_add = [
|
|
["Stock In Hand", "Stock Assets", "Ledger", ""],
|
|
["Cost of Goods Sold", "Stock Expenses", "Ledger", "Expense Account"],
|
|
["Stock Adjustment", "Stock Expenses", "Ledger", "Expense Account"],
|
|
["Expenses Included In Valuation", "Stock Expenses", "Ledger", "Expense Account"],
|
|
["Stock Received But Not Billed", "Stock Liabilities", "Ledger", ""],
|
|
]
|
|
add_accounts(accounts_to_add)
|
|
|
|
|
|
def add_accounts(accounts_to_add, check_fn=None):
|
|
for company, abbr in webnotes.conn.sql("""select name, abbr from `tabCompany`"""):
|
|
count = webnotes.conn.sql("""select count(name) from `tabAccount`
|
|
where company=%s and ifnull(parent_account, '')=''""", company)[0][0]
|
|
|
|
if count > 4:
|
|
webnotes.errprint("Company" + company +
|
|
"has more than 4 root accounts. cannot apply patch to this company.")
|
|
continue
|
|
|
|
for account_name, parent_account_name, group_or_ledger, account_type in accounts_to_add:
|
|
if not webnotes.conn.sql("""select name from `tabAccount` where account_name = %s
|
|
and company = %s""", (account_name, company)):
|
|
parent_account = "%s - %s" % (parent_account_name, abbr)
|
|
if check_fn:
|
|
parent_account = check_fn(parent_account, company)
|
|
account = webnotes.bean({
|
|
"doctype": "Account",
|
|
"account_name": account_name,
|
|
"parent_account": parent_account,
|
|
"group_or_ledger": group_or_ledger,
|
|
"account_type": account_type,
|
|
"company": company
|
|
})
|
|
account.insert()
|
|
|
|
def add_aii_cost_center():
|
|
for company, abbr in webnotes.conn.sql("""select name, abbr from `tabCompany`"""):
|
|
if not webnotes.conn.sql("""select name from `tabCost Center` where cost_center_name =
|
|
'Auto Inventory Accounting' and company = %s""", company):
|
|
parent_cost_center = webnotes.conn.get_value("Cost Center",
|
|
{"parent_cost_center['']": '', "company": company})
|
|
|
|
if not parent_cost_center:
|
|
webnotes.errprint("Company " + company + "does not have a root cost center")
|
|
continue
|
|
|
|
cc = webnotes.bean({
|
|
"doctype": "Cost Center",
|
|
"cost_center_name": "Auto Inventory Accounting",
|
|
"parent_cost_center": parent_cost_center,
|
|
"group_or_ledger": "Ledger",
|
|
"company": company
|
|
})
|
|
cc.insert()
|
|
|
|
def create_chart_of_accounts_if_not_exists():
|
|
for company in webnotes.conn.sql("select name from `tabCompany`"):
|
|
if not webnotes.conn.sql("select * from `tabAccount` where company = %s", company[0]):
|
|
webnotes.conn.sql("""update `tabCompany` set receivables_group = '',
|
|
payables_group = '' where name = %s""", company[0])
|
|
webnotes.bean("Company", company[0]).save()
|
|
|