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") 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