131 lines
3.8 KiB
Python
131 lines
3.8 KiB
Python
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",
|
|
)
|