[added] domainify.py

erpnext/setup/setup_wizard/domainify.py

@@ -0,0 +1,53 @@
+# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
+# See license.txt
+
+from __future__ import unicode_literals
+import frappe
+
+domains = {
+	'Manufacturing': {
+		'desktop_icons': ['Item', 'BOM', 'Customer', 'Supplier', 'Sales Order',
+			'Production Order', 'Stock Entry', 'Buying', 'Selling', 'Accounts']
+	},
+	'Retail': {
+		'remove_roles': ['Manufacturing User', 'Manufacturing Manager', 'Maintenance User'],
+		'desktop_icons': ['POS', 'Item', 'Customer', 'Sales Invoice', 'Accounts']
+	},
+	'Distribution': {
+		'remove_roles': ['Manufacturing User', 'Manufacturing Manager', 'Maintenance User'],
+	},
+	'Services': {
+		'desktop_icons': ['Project', 'Time Log', 'Customer', 'Sales Invoice', 'Lead', 'Opportunity',
+			'Expense Claim', 'Employee'],
+		'remove_roles': ['Manufacturing User', 'Manufacturing Manager', 'Maintenance User'],
+		'properties': [
+			{'doctype': 'Item', 'fieldname': 'is_stock_item', 'property': 'default', 'value': 0}
+		]
+	}
+}
+
+def setup_domain(domain):
+	if not domain in domains:
+		return
+
+	from frappe.desk.doctype.desktop_icon.desktop_icon import set_desktop_icons
+	data = frappe._dict(domains[domain])
+
+	if data.remove_roles:
+		for role in data.remove_roles:
+			frappe.db.sql('delete from tabUserRole where role=%s', role)
+
+	if data.desktop_icons:
+		set_desktop_icons(data.desktop_icons)
+
+	if data.properties:
+		for args in data.properties:
+			frappe.make_property_setter(args)
+
+	frappe.clear_cache()
+
+def reset():
+	from frappe.desk.page.setup_wizard.setup_wizard import add_all_roles_to
+	add_all_roles_to('Administrator')
+
+	frappe.db.sql('delete from `tabProperty Setter`')

erpnext/setup/setup_wizard/setup_wizard.py

@@ -12,6 +12,7 @@ import install_fixtures
 from .sample_data import make_sample_data
 from erpnext.accounts.doctype.account.account import RootNotEditable
 from frappe.core.doctype.communication.comment import add_info_comment
+from erpnext.setup.setup_wizard.domainify import setup_domain
 
 def setup_complete(args=None):
 	if frappe.db.sql("select name from tabCompany"):
@@ -32,6 +33,7 @@ def setup_complete(args=None):
 	create_customers(args)
 	create_suppliers(args)
 	frappe.local.message_log = []
+	setup_domain(args.get('domain'))
 
 	website_maker(args.company_name.strip(), args.company_tagline, args.name)
 	create_logo(args)