brotherton-erpnext/erpnext/e_commerce/variant_selector/item_variants_cache.py

import frappe


class ItemVariantsCacheManager:
	def __init__(self, item_code):
		self.item_code = item_code

	def get_item_variants_data(self):
		val = frappe.cache().hget("item_variants_data", self.item_code)

		if not val:
			self.build_cache()

		return frappe.cache().hget("item_variants_data", self.item_code)

	def get_attribute_value_item_map(self):
		val = frappe.cache().hget("attribute_value_item_map", self.item_code)

		if not val:
			self.build_cache()

		return frappe.cache().hget("attribute_value_item_map", self.item_code)

	def get_item_attribute_value_map(self):
		val = frappe.cache().hget("item_attribute_value_map", self.item_code)

		if not val:
			self.build_cache()

		return frappe.cache().hget("item_attribute_value_map", self.item_code)

	def get_optional_attributes(self):
		val = frappe.cache().hget("optional_attributes", self.item_code)

		if not val:
			self.build_cache()

		return frappe.cache().hget("optional_attributes", self.item_code)

	def get_ordered_attribute_values(self):
		val = frappe.cache().get_value("ordered_attribute_values_map")
		if val:
			return val

		all_attribute_values = frappe.get_all(
			"Item Attribute Value", ["attribute_value", "idx", "parent"], order_by="idx asc"
		)

		ordered_attribute_values_map = frappe._dict({})
		for d in all_attribute_values:
			ordered_attribute_values_map.setdefault(d.parent, []).append(d.attribute_value)

		frappe.cache().set_value("ordered_attribute_values_map", ordered_attribute_values_map)
		return ordered_attribute_values_map

	def build_cache(self):
		parent_item_code = self.item_code

		attributes = [
			a.attribute
			for a in frappe.get_all(
				"Item Variant Attribute", {"parent": parent_item_code}, ["attribute"], order_by="idx asc"
			)
		]

		# Get Variants and tehir Attributes that are not disabled
		iva = frappe.qb.DocType("Item Variant Attribute")
		item = frappe.qb.DocType("Item")
		query = (
			frappe.qb.from_(iva)
			.join(item)
			.on(item.name == iva.parent)
			.select(iva.parent, iva.attribute, iva.attribute_value)
			.where((iva.variant_of == parent_item_code) & (item.disabled == 0))
			.orderby(iva.name)
		)
		item_variants_data = query.run()

		attribute_value_item_map = frappe._dict()
		item_attribute_value_map = frappe._dict()

		for row in item_variants_data:
			item_code, attribute, attribute_value = row
			# (attr, value) => [item1, item2]
			attribute_value_item_map.setdefault((attribute, attribute_value), []).append(item_code)
			# item => {attr1: value1, attr2: value2}
			item_attribute_value_map.setdefault(item_code, {})[attribute] = attribute_value

		optional_attributes = set()
		for item_code, attr_dict in item_attribute_value_map.items():
			for attribute in attributes:
				if attribute not in attr_dict:
					optional_attributes.add(attribute)

		frappe.cache().hset("attribute_value_item_map", parent_item_code, attribute_value_item_map)
		frappe.cache().hset("item_attribute_value_map", parent_item_code, item_attribute_value_map)
		frappe.cache().hset("item_variants_data", parent_item_code, item_variants_data)
		frappe.cache().hset("optional_attributes", parent_item_code, optional_attributes)

	def clear_cache(self):
		keys = [
			"attribute_value_item_map",
			"item_attribute_value_map",
			"item_variants_data",
			"optional_attributes",
		]

		for key in keys:
			frappe.cache().hdel(key, self.item_code)

	def rebuild_cache(self):
		self.clear_cache()
		enqueue_build_cache(self.item_code)


def build_cache(item_code):
	frappe.cache().hset("item_cache_build_in_progress", item_code, 1)
	i = ItemVariantsCacheManager(item_code)
	i.build_cache()
	frappe.cache().hset("item_cache_build_in_progress", item_code, 0)


def enqueue_build_cache(item_code):
	if frappe.cache().hget("item_cache_build_in_progress", item_code):
		return
	frappe.enqueue(
		"erpnext.e_commerce.variant_selector.item_variants_cache.build_cache",
		item_code=item_code,
		queue="long",
	)