brotherton-erpnext/erpnext/shopping_cart/utils.py

40 lines
1.2 KiB
Python
Raw Normal View History

# Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors
2014-10-21 10:46:30 +00:00
# License: GNU General Public License v3. See license.txt
import frappe
from erpnext.e_commerce.doctype.e_commerce_settings.e_commerce_settings import is_cart_enabled
2014-10-21 10:46:30 +00:00
def show_cart_count():
if (is_cart_enabled() and
2014-10-21 10:46:30 +00:00
frappe.db.get_value("User", frappe.session.user, "user_type") == "Website User"):
return True
return False
def set_cart_count(login_manager):
2016-03-03 08:30:35 +00:00
role, parties = check_customer_or_supplier()
if role == 'Supplier': return
2014-10-21 10:46:30 +00:00
if show_cart_count():
2015-02-23 16:44:12 +00:00
from erpnext.shopping_cart.cart import set_cart_count
2014-10-21 10:46:30 +00:00
set_cart_count()
def clear_cart_count(login_manager):
if show_cart_count():
frappe.local.cookie_manager.delete_cookie("cart_count")
2015-02-24 12:20:44 +00:00
def update_website_context(context):
2015-02-23 16:44:12 +00:00
cart_enabled = is_cart_enabled()
2014-10-21 10:46:30 +00:00
context["shopping_cart_enabled"] = cart_enabled
2016-03-03 08:30:35 +00:00
def check_customer_or_supplier():
if frappe.session.user:
2017-01-13 18:55:22 +00:00
contact_name = frappe.get_value("Contact", {"email_id": frappe.session.user})
if contact_name:
contact = frappe.get_doc('Contact', contact_name)
for link in contact.links:
if link.link_doctype in ('Customer', 'Supplier'):
return link.link_doctype, link.link_name
2016-03-03 08:30:35 +00:00
return 'Customer', None