brotherton-erpnext/erpnext/demo/demo.py

89 lines
2.2 KiB
Python
Raw Normal View History

2016-06-29 13:08:32 +00:00
from __future__ import unicode_literals
import frappe, sys
import erpnext
import frappe.utils
2016-08-22 07:27:09 +00:00
from erpnext.demo.user import hr, sales, purchase, manufacturing, stock, accounts, projects, fixed_asset, schools
from erpnext.demo.setup import education, manufacture, setup_data
2016-07-13 05:59:59 +00:00
"""
Make a demo
1. Start with a fresh account
bench --site demo.erpnext.dev reinstall
2. Install Demo
bench --site demo.erpnext.dev execute erpnext.demo.demo.make
3. If Demo breaks, to continue
bench --site demo.erpnext.dev execute erpnext.demo.demo.simulate
"""
2016-06-29 13:08:32 +00:00
def make(domain='Manufacturing'):
frappe.flags.domain = domain
2016-07-13 12:47:19 +00:00
frappe.flags.mute_emails = True
2016-08-22 07:27:09 +00:00
setup_data.setup(domain)
if domain== 'Manufacturing':
manufacture.setup_data()
elif domain== 'Education':
education.setup_data()
2016-07-13 12:11:03 +00:00
site = frappe.local.site
frappe.destroy()
frappe.init(site)
frappe.connect()
2016-08-22 07:27:09 +00:00
simulate(domain)
2016-06-29 13:08:32 +00:00
2016-08-22 07:27:09 +00:00
def simulate(domain='Manufacturing'):
2016-06-29 13:08:32 +00:00
runs_for = frappe.flags.runs_for or 150
frappe.flags.company = erpnext.get_default_company()
2016-07-13 12:47:19 +00:00
frappe.flags.mute_emails = True
2016-06-29 13:08:32 +00:00
if not frappe.flags.start_date:
# start date = 100 days back
frappe.flags.start_date = frappe.utils.add_days(frappe.utils.nowdate(), -1 * runs_for)
current_date = frappe.utils.getdate(frappe.flags.start_date)
# continue?
demo_last_date = frappe.db.get_global('demo_last_date')
if demo_last_date:
2016-08-22 07:27:09 +00:00
current_date = frappe.utils.add_days(frappe.utils.getdate(demo_last_date), 1)
2016-06-29 13:08:32 +00:00
# run till today
if not runs_for:
runs_for = frappe.utils.date_diff(frappe.utils.nowdate(), current_date)
# runs_for = 100
2016-08-03 10:57:11 +00:00
2016-07-20 10:43:18 +00:00
fixed_asset.work()
2016-06-29 13:08:32 +00:00
for i in xrange(runs_for):
sys.stdout.write("\rSimulating {0}".format(current_date.strftime("%Y-%m-%d")))
sys.stdout.flush()
frappe.flags.current_date = current_date
if current_date.weekday() in (5, 6):
current_date = frappe.utils.add_days(current_date, 1)
continue
2016-08-22 07:27:09 +00:00
try:
hr.work()
purchase.work()
stock.work()
accounts.work()
projects.run_projects(current_date)
#run_messages()
if domain=='Manufacturing':
sales.work()
manufacturing.work()
elif domain=='Education':
schools.work()
except:
frappe.db.set_global('demo_last_date', current_date)
raise
finally:
current_date = frappe.utils.add_days(current_date, 1)
frappe.db.commit()