124 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			124 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| import frappe
 | |
| from frappe.utils.nestedset import get_root_of
 | |
| 
 | |
| from erpnext.e_commerce.doctype.e_commerce_settings.e_commerce_settings import (
 | |
| 	get_shopping_cart_settings,
 | |
| )
 | |
| from erpnext.e_commerce.shopping_cart.cart import get_debtors_account
 | |
| 
 | |
| 
 | |
| def set_default_role(doc, method):
 | |
| 	'''Set customer, supplier, student, guardian based on email'''
 | |
| 	if frappe.flags.setting_role or frappe.flags.in_migrate:
 | |
| 		return
 | |
| 
 | |
| 	roles = frappe.get_roles(doc.name)
 | |
| 
 | |
| 	contact_name = frappe.get_value('Contact', dict(email_id=doc.email))
 | |
| 	if contact_name:
 | |
| 		contact = frappe.get_doc('Contact', contact_name)
 | |
| 		for link in contact.links:
 | |
| 			frappe.flags.setting_role = True
 | |
| 			if link.link_doctype=='Customer' and 'Customer' not in roles:
 | |
| 				doc.add_roles('Customer')
 | |
| 			elif link.link_doctype=='Supplier' and 'Supplier' not in roles:
 | |
| 				doc.add_roles('Supplier')
 | |
| 	elif frappe.get_value('Student', dict(student_email_id=doc.email)) and 'Student' not in roles:
 | |
| 		doc.add_roles('Student')
 | |
| 	elif frappe.get_value('Guardian', dict(email_address=doc.email)) and 'Guardian' not in roles:
 | |
| 		doc.add_roles('Guardian')
 | |
| 
 | |
| def create_customer_or_supplier():
 | |
| 	'''Based on the default Role (Customer, Supplier), create a Customer / Supplier.
 | |
| 	Called on_session_creation hook.
 | |
| 	'''
 | |
| 	user = frappe.session.user
 | |
| 
 | |
| 	if frappe.db.get_value('User', user, 'user_type') != 'Website User':
 | |
| 		return
 | |
| 
 | |
| 	user_roles = frappe.get_roles()
 | |
| 	portal_settings = frappe.get_single('Portal Settings')
 | |
| 	default_role = portal_settings.default_role
 | |
| 
 | |
| 	if default_role not in ['Customer', 'Supplier']:
 | |
| 		return
 | |
| 
 | |
| 	# create customer / supplier if the user has that role
 | |
| 	if portal_settings.default_role and portal_settings.default_role in user_roles:
 | |
| 		doctype = portal_settings.default_role
 | |
| 	else:
 | |
| 		doctype = None
 | |
| 
 | |
| 	if not doctype:
 | |
| 		return
 | |
| 
 | |
| 	if party_exists(doctype, user):
 | |
| 		return
 | |
| 
 | |
| 	party = frappe.new_doc(doctype)
 | |
| 	fullname = frappe.utils.get_fullname(user)
 | |
| 
 | |
| 	if doctype == 'Customer':
 | |
| 		cart_settings = get_shopping_cart_settings()
 | |
| 
 | |
| 		if cart_settings.enable_checkout:
 | |
| 			debtors_account = get_debtors_account(cart_settings)
 | |
| 		else:
 | |
| 			debtors_account = ''
 | |
| 
 | |
| 		party.update({
 | |
| 			"customer_name": fullname,
 | |
| 			"customer_type": "Individual",
 | |
| 			"customer_group": cart_settings.default_customer_group,
 | |
| 			"territory": get_root_of("Territory")
 | |
| 		})
 | |
| 
 | |
| 		if debtors_account:
 | |
| 			party.update({
 | |
| 				"accounts": [{
 | |
| 					"company": cart_settings.company,
 | |
| 					"account": debtors_account
 | |
| 				}]
 | |
| 			})
 | |
| 	else:
 | |
| 		party.update({
 | |
| 			"supplier_name": fullname,
 | |
| 			"supplier_group": "All Supplier Groups",
 | |
| 			"supplier_type": "Individual"
 | |
| 		})
 | |
| 
 | |
| 	party.flags.ignore_mandatory = True
 | |
| 	party.insert(ignore_permissions=True)
 | |
| 
 | |
| 	alternate_doctype = "Customer" if doctype == "Supplier" else "Supplier"
 | |
| 
 | |
| 	if party_exists(alternate_doctype, user):
 | |
| 		# if user is both customer and supplier, alter fullname to avoid contact name duplication
 | |
| 		fullname +=  "-" + doctype
 | |
| 
 | |
| 	create_party_contact(doctype, fullname, user, party.name)
 | |
| 
 | |
| 	return party
 | |
| 
 | |
| def create_party_contact(doctype, fullname, user, party_name):
 | |
| 	contact = frappe.new_doc("Contact")
 | |
| 	contact.update({
 | |
| 		"first_name": fullname,
 | |
| 		"email_id": user
 | |
| 	})
 | |
| 	contact.append('links', dict(link_doctype=doctype, link_name=party_name))
 | |
| 	contact.append('email_ids', dict(email_id=user))
 | |
| 	contact.flags.ignore_mandatory = True
 | |
| 	contact.insert(ignore_permissions=True)
 | |
| 
 | |
| def party_exists(doctype, user):
 | |
| 	# check if contact exists against party and if it is linked to the doctype
 | |
| 	contact_name = frappe.db.get_value("Contact", {"email_id": user})
 | |
| 	if contact_name:
 | |
| 		contact = frappe.get_doc('Contact', contact_name)
 | |
| 		doctypes = [d.link_doctype for d in contact.links]
 | |
| 		return doctype in doctypes
 | |
| 
 | |
| 	return False
 |