fix: Create transactions

erpnext/setup/demo.py

@@ -6,12 +6,20 @@ import os
 
 import frappe
 
+import erpnext
+
 
 @frappe.whitelist()
 def setup_demo_data():
-	create_demo_company()
-	process_demo_data()
-	make_transactions()
+	company = create_demo_company()
+	process_masters()
+	make_transactions(company)
+
+
+@frappe.whitelist()
+def clear_demo_data():
+	company = frappe.db.get_single_value("Global Defaults", "demo_company")
+	create_transaction_deletion_record(company)
 
 
 def create_demo_company():
@@ -29,9 +37,12 @@ def create_demo_company():
 	new_company.chart_of_accounts = company_doc.chart_of_accounts
 	new_company.insert()
 
+	frappe.db.set_single_value("Global Defaults", "demo_company", new_company.name)
+	return new_company.name
 
-def process_demo_data():
-	demo_doctypes = frappe.get_hooks("demo_doctypes") or []
+
+def process_masters():
+	demo_doctypes = frappe.get_hooks("demo_master_doctypes") or []
 	path = os.path.join(os.path.dirname(__file__), "demo_data")
 	for doctype in demo_doctypes:
 		with open(os.path.join(path, doctype + ".json"), "r") as f:
@@ -45,5 +56,40 @@ def create_demo_record(doctype):
 	frappe.get_doc(doctype).insert(ignore_permissions=True)
 
 
-def make_transactions():
-	pass
+def make_transactions(company):
+	transaction_doctypes = frappe.get_hooks("demo_transaction_doctypes") or []
+	path = os.path.join(os.path.dirname(__file__), "demo_data")
+	for transaction in transaction_doctypes:
+		with open(os.path.join(path, transaction + ".json"), "r") as f:
+			data = f.read()
+			if data:
+				for item in json.loads(data):
+					create_transaction(item, company)
+
+
+def create_transaction(doctype, company):
+	doctype.update({"company": company})
+
+	income_account, expense_account = frappe.db.get_value(
+		"Company", company, ["default_income_account", "default_expense_account"]
+	)
+
+	for item in doctype.get("items"):
+		item.update(
+			{
+				"cost_center": erpnext.get_default_cost_center(company),
+				"income_account": income_account,
+				"expense_account": expense_account,
+			}
+		)
+
+	doc = frappe.get_doc(doctype)
+	doc.save(ignore_permissions=True)
+	doc.submit()
+
+
+def create_transaction_deletion_record(company):
+	transaction_deletion_record = frappe.new_doc("Transaction Deletion Record")
+	transaction_deletion_record.company = company
+	transaction_deletion_record.save(ignore_permissions=True)
+	transaction_deletion_record.submit()