def handle(self, *args, **options):
# Clear all existing ProductCrossSell objects
ProductCrossSell.objects.filter(type=ProductCrossSellType.BOUGHT_WITH).delete()
# Handle all ordered products
ordered_product_ids = OrderLine.objects.filter(type=OrderLineType.PRODUCT).values_list("product_id", flat=True)
seen_product_ids = set()
for product_id in ordered_product_ids:
if product_id in seen_product_ids:
continue
seen_product_ids.add(product_id)
add_bought_with_relations_for_product(product_id)
def test_product_relations_max_quantity(rf):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product("simple-test-product", shop)
quantities = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
for i, quantity in enumerate(quantities):
order = create_order_with_product(product, supplier, quantity=1, taxless_base_unit_price=6, shop=shop)
add_product_to_order(
order,
supplier,
create_product("product-%s" % i, shop),
quantity=quantity,
taxless_base_unit_price=6
)
assert ProductCrossSell.objects.count() == 0
add_bought_with_relations_for_product(product.pk, max_quantity=5)
assert ProductCrossSell.objects.count() == 5
# Test that ordering is ok
assert not ProductCrossSell.objects.filter(weight=1).exists()
assert ProductCrossSell.objects.filter(weight=11).exists()
def test_computing_simple_product_relations(rf):
shop = get_default_shop()
supplier = get_default_supplier()
product = create_product("simple-test-product", shop)
related_product = create_product("simple-related-product", shop)
quantities = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
for quantity in quantities:
order = create_order_with_product(product, supplier, quantity=1, taxless_base_unit_price=6, shop=shop)
add_product_to_order(order, supplier, related_product, quantity=quantity, taxless_base_unit_price=6)
assert ProductCrossSell.objects.count() == 0
add_bought_with_relations_for_product(product.pk)
assert ProductCrossSell.objects.count() == 1
cross_sell_product = ProductCrossSell.objects.filter(product1=product).first()
assert cross_sell_product.product2 == related_product
assert cross_sell_product.weight == sum(quantities)
add_bought_with_relations_for_product(related_product.id)
assert ProductCrossSell.objects.count() == 2
cross_sell_product = ProductCrossSell.objects.filter(product1=related_product).first()
assert cross_sell_product.product2 == product
assert cross_sell_product.weight == len(quantities)
