# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd.
# License: GNU General Public License v3. See license.txt

import webnotes
def execute():
	webnotes.reload_doc("setup", "doctype", "company")
	webnotes.reload_doc("accounts", "doctype", "cost_center")
	create_chart_of_accounts_if_not_exists()
	add_group_accounts()
	add_ledger_accounts()
	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 = [
		["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 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 = '', default_bank_account = '', default_cash_account = '',
				default_expense_account = '', default_income_account = '', cost_center = '', 
				stock_received_but_not_billed = '', stock_adjustment_account = '', 
				expenses_included_in_valuation = '' where name = %s""", company[0])
				
			webnotes.conn.sql("""update `tabCompany` set domain = 'Services' 
				where name = %s and ifnull(domain, '') = ''""", company[0])
				
			webnotes.bean("Company", company[0]).save()