chore: Add more invoices

This commit is contained in:
Deepesh Garg
2023-06-19 09:44:57 +05:30
parent 5b6a9fcca9
commit 86744b6fbb
8 changed files with 265 additions and 17 deletions

View File

@@ -3,8 +3,10 @@
import json
import os
from random import randint
import frappe
from frappe.utils import add_days
import erpnext
@@ -18,7 +20,7 @@ def setup_demo_data():
@frappe.whitelist()
def clear_demo_data():
company = erpnext.get_default_company()
company = frappe.db.get_single_value("Global Defaults", "demo_company")
create_transaction_deletion_record(company)
clear_masters()
delete_company(company)
@@ -39,10 +41,10 @@ def create_demo_company():
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)
frappe.db.set_single_value("Global Defaults", "demo_company", new_company.name)
frappe.db.set_default("company", new_company.name)
return new_company.name
@@ -59,15 +61,24 @@ def create_demo_record(doctype):
def make_transactions(company):
fiscal_year = frappe.db.get_single_value("Global Defaults", "current_fiscal_year")
start_date = frappe.db.get_value("Fiscal Year", fiscal_year, "year_start_date")
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)
create_transaction(item, company, start_date)
def create_transaction(doctype, company):
doctype.update({"company": company})
def create_transaction(doctype, company, start_date):
doctype.update(
{
"company": company,
"set_posting_time": 1,
"posting_date": get_random_date(start_date),
}
)
income_account, expense_account = frappe.db.get_value(
"Company", company, ["default_income_account", "default_expense_account"]
@@ -87,6 +98,10 @@ def create_transaction(doctype, company):
doc.submit()
def get_random_date(start_date):
return add_days(start_date, randint(1, 365))
def create_transaction_deletion_record(company):
transaction_deletion_record = frappe.new_doc("Transaction Deletion Record")
transaction_deletion_record.company = company
@@ -110,8 +125,6 @@ def clear_demo_record(doctype):
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)