124 lines
3.4 KiB
Python
124 lines
3.4 KiB
Python
# Copyright (c) 2023, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: GNU General Public License v3. See license.txt
|
|
|
|
import json
|
|
import os
|
|
|
|
import frappe
|
|
|
|
import erpnext
|
|
|
|
|
|
@frappe.whitelist()
|
|
def setup_demo_data():
|
|
company = create_demo_company()
|
|
process_masters()
|
|
make_transactions(company)
|
|
|
|
|
|
@frappe.whitelist()
|
|
def clear_demo_data():
|
|
company = erpnext.get_default_company()
|
|
create_transaction_deletion_record(company)
|
|
clear_masters()
|
|
delete_company(company)
|
|
|
|
|
|
def create_demo_company():
|
|
company = erpnext.get_default_company()
|
|
company_doc = frappe.get_doc("Company", company)
|
|
|
|
# Make a dummy company
|
|
new_company = frappe.new_doc("Company")
|
|
new_company.company_name = company_doc.company_name + " (Demo)"
|
|
new_company.abbr = company_doc.abbr + "D"
|
|
new_company.enable_perpetual_inventory = 1
|
|
new_company.default_currency = company_doc.default_currency
|
|
new_company.country = company_doc.country
|
|
new_company.chart_of_accounts_based_on = "Standard Template"
|
|
new_company.chart_of_accounts = company_doc.chart_of_accounts
|
|
new_company.insert()
|
|
|
|
frappe.db.set_single_value("Global Defaults", "original_default_company", company)
|
|
|
|
# Set Demo Company as default to
|
|
frappe.db.set_single_value("Global Defaults", "default_company", new_company.name)
|
|
return new_company.name
|
|
|
|
|
|
def process_masters():
|
|
for doctype in frappe.get_hooks("demo_master_doctypes"):
|
|
data = read_data_file_using_hooks(doctype)
|
|
if data:
|
|
for item in json.loads(data):
|
|
create_demo_record(item)
|
|
|
|
|
|
def create_demo_record(doctype):
|
|
frappe.get_doc(doctype).insert(ignore_permissions=True)
|
|
|
|
|
|
def make_transactions(company):
|
|
for doctype in frappe.get_hooks("demo_transaction_doctypes"):
|
|
data = read_data_file_using_hooks(doctype)
|
|
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()
|
|
|
|
|
|
def clear_masters():
|
|
for doctype in frappe.get_hooks("demo_master_doctypes")[::-1]:
|
|
data = read_data_file_using_hooks(doctype)
|
|
if data:
|
|
for item in json.loads(data):
|
|
clear_demo_record(item)
|
|
|
|
|
|
def clear_demo_record(doctype):
|
|
doc_type = doctype.get("doctype")
|
|
del doctype["doctype"]
|
|
doc = frappe.get_doc(doc_type, doctype)
|
|
frappe.delete_doc(doc.doctype, doc.name, ignore_permissions=True)
|
|
|
|
|
|
def delete_company(company):
|
|
original_company = frappe.db.get_single_value("Global Defaults", "original_default_company")
|
|
frappe.db.set_single_value("Global Defaults", "default_company", original_company)
|
|
frappe.delete_doc("Company", company, ignore_permissions=True)
|
|
|
|
|
|
def read_data_file_using_hooks(doctype):
|
|
path = os.path.join(os.path.dirname(__file__), "demo_data")
|
|
with open(os.path.join(path, doctype + ".json"), "r") as f:
|
|
data = f.read()
|
|
|
|
return data
|