brotherton-erpnext/utilities/make_demo.py

183 lines
4.7 KiB
Python

# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd.
# License: GNU General Public License v3. See license.txt
import webnotes, os, datetime
import webnotes.utils
import random
webnotes.session = webnotes._dict({"user":"Administrator"})
from core.page.data_import_tool.data_import_tool import upload
company = "Wind Power LLC"
start_date = '2010-01-01'
runs_for = 100
prob = {
"Quotation": { "make": 0.5, "qty": (1,3) },
"Sales Order": { "make": 0.5, "qty": (1,2) }
}
def make():
webnotes.connect()
webnotes.print_messages = True
webnotes.mute_emails = True
# setup()
simulate()
def setup():
install()
complete_setup()
make_items()
make_customers_suppliers_contacts()
make_users_and_employees()
# make_opening_stock()
# make_opening_accounts()
def simulate():
current_date = None
for i in xrange(runs_for):
print i
if not current_date:
current_date = webnotes.utils.getdate(start_date)
else:
current_date = webnotes.utils.add_days(current_date, 1)
if current_date.weekday() in (5, 6):
continue
run_sales(current_date)
run_purchase(current_date)
run_manufacturing(current_date)
webnotes.conn.commit()
def run_sales(current_date):
if random.random() < prob["Quotation"]["make"]:
for i in xrange(random.randrange(*prob["Quotation"]["qty"])):
make_quotation(current_date)
if random.random() < prob["Sales Order"]["make"]:
for i in xrange(random.randrange(*prob["Sales Order"]["qty"])):
make_sales_order(current_date)
def run_purchase(current_date):
pass
def run_manufacturing(current_date):
ppt = webnotes.bean("Production Planning Tool", "Production Planning Tool")
ppt.doc.company = company
ppt.doc.use_multi_level_bom = 1
ppt.doc.purchase_request_for_warehouse = "Stores - WP"
ppt.run_method("get_open_sales_orders")
ppt.run_method("get_items_from_so")
ppt.run_method("get_items_from_so")
ppt.run_method("raise_production_order")
ppt.run_method("raise_purchase_request")
def make_quotation(current_date):
b = webnotes.bean([{
"creation": current_date,
"doctype": "Quotation",
"quotation_to": "Customer",
"customer": get_random("Customer"),
"order_type": "Sales",
"price_list_name": "Standard Selling",
"transaction_date": current_date,
"fiscal_year": "2010"
}])
add_random_children(b, {
"doctype": "Quotation Item",
"parentfield": "quotation_details",
}, rows=3, randomize = {
"qty": (1, 5),
"item_code": ("Item", {"is_sales_item": "Yes"})
}, unique="item_code")
b.insert()
b.submit()
def make_sales_order(current_date):
q = get_random("Quotation", {"status": "Submitted"})
from selling.doctype.quotation.quotation import make_sales_order
so = webnotes.bean(make_sales_order(q))
so.doc.transaction_date = current_date
so.doc.delivery_date = webnotes.utils.add_days(current_date, 10)
so.insert()
so.submit()
def add_random_children(bean, template, rows, randomize, unique=None):
for i in xrange(random.randrange(1, rows)):
d = template.copy()
for key, val in randomize.items():
if isinstance(val[0], basestring):
d[key] = get_random(*val)
else:
d[key] = random.randrange(*val)
if unique:
if not bean.doclist.get({"doctype": d["doctype"], unique:d[unique]}):
bean.doclist.append(d)
else:
bean.doclist.append(d)
def get_random(doctype, filters=None):
condition = []
if filters:
for key, val in filters.items():
condition.append("%s='%s'" % (key, val))
if condition:
condition = " where " + " and ".join(condition)
else:
condition = ""
out = webnotes.conn.sql("""select name from `tab%s` %s
order by RAND() limit 0,1""" % (doctype, condition))[0][0]
return out
def install():
print "Creating Fresh Database..."
from webnotes.install_lib.install import Installer
inst = Installer('root')
inst.import_from_db("demo", verbose = 1)
def complete_setup():
print "Complete Setup..."
webnotes.get_obj("Setup Control").setup_account({
"first_name": "Test",
"last_name": "User",
"fy_start": "1st Jan",
"industry": "Manufacturing",
"company_name": company,
"company_abbr": "WP",
"currency": "USD",
"timezone": "America/New York",
"country": "United States"
})
import_data("Fiscal_Year")
def make_items():
import_data(["Item", "Item_Price", "BOM"])
def make_customers_suppliers_contacts():
import_data(["Customer", "Supplier", "Contact", "Address", "Lead"])
def make_users_and_employees():
webnotes.conn.set_value("HR Settings", None, "emp_created_by", "Naming Series")
webnotes.conn.commit()
import_data(["Profile", "Employee", "Salary_Structure"])
def import_data(dt):
if not isinstance(dt, (tuple, list)):
dt = [dt]
for doctype in dt:
print "Importing", doctype.replace("_", " "), "..."
webnotes.uploaded_file = os.path.join(os.path.dirname(__file__), "demo_docs", doctype+".csv")
upload()
if __name__=="__main__":
make()