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()
	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()