121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
from __future__ import unicode_literals
|
|
import frappe
|
|
from erpnext.shopping_cart.doctype.shopping_cart_settings.shopping_cart_settings import get_shopping_cart_settings
|
|
from erpnext.shopping_cart.cart import get_debtors_account
|
|
from frappe.utils.nestedset import get_root_of
|
|
|
|
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
|