Optimize multibuy database queries

stregsystem/models.py

@@ -128,15 +128,15 @@ def execute(self):
         self.member = Member.objects.select_for_update().get(id=self.member.id)
         self.member.fulfill(transaction)
 
+        # Collect all the sales of the order
+        Sales = []
         for item in self.items:
-            # @HACK Since we want to use the old database layout, we need to
-            # add a sale for every item and every instance of that item
             for i in range(item.count):
                 s = Sale(member=self.member, product=item.product, room=self.room, price=item.product.price)
-                s.save()
+                Sales.append(s)
+        # Save all the sales
+        Sale.objects.bulk_create(Sales)
 
-            # Bought (used above) is automatically calculated, so we don't need
-            # to update it
         # We changed the user balance, so save that
         self.member.save()
 

stregsystem/views.py

@@ -24,7 +24,6 @@
 from django_select2 import forms as s2forms
 import urllib.parse
 
-
 from stregsystem import parser
 from stregsystem.models import (
     Member,
@@ -140,15 +139,14 @@ def _multibuy_hint(now, member):
     # get the sales with this timestamp
     recent_purchases = Sale.objects.filter(member=member, timestamp__gt=earliest_recent_purchase)
     number_of_recent_distinct_purchases = recent_purchases.values('timestamp').distinct().count()
+    recent_unique_purchases = recent_purchases.values('product').distinct().annotate(total=Count('product'))
 
     # add hint for multibuy
     if number_of_recent_distinct_purchases > 1:
         sale_dict = {}
-        for sale in recent_purchases:
-            if not str(sale.product.id) in sale_dict:
-                sale_dict[str(sale.product.id)] = 1
-            else:
-                sale_dict[str(sale.product.id)] = sale_dict[str(sale.product.id)] + 1
+        for unique_sale in recent_unique_purchases:
+            sale_dict[str(unique_sale['product'])] = unique_sale['total']
+
         sale_hints = ["<span class=\"username\">{}</span>".format(member.username)]
         if all(sale_count == 1 for sale_count in sale_dict.values()):
             return (False, None)
@@ -715,14 +713,23 @@ def api_quicksale(request, room, member: Member, bought_ids):
 
 def __append_bought_ids_to_product_list(products, bought_ids, time_now, room):
     try:
-        for i in bought_ids:
+        # Get the amount of unique items bought
+        unique_product_dict = {}
+        for unique_id in bought_ids:
+            if str(unique_id) not in unique_product_dict:
+                unique_product_dict[str(unique_id)] = 1
+            else:
+                unique_product_dict[str(unique_id)] += 1
+
+        # Add the given amount of different products
+        for i in unique_product_dict:
             product = Product.objects.get(
                 Q(pk=i),
                 Q(active=True),
                 Q(deactivate_date__gte=time_now) | Q(deactivate_date__isnull=True),
                 Q(rooms__id=room.id) | Q(rooms=None),
             )
-            products.append(product)
+            products.extend([product for x in range(unique_product_dict[str(i)])])
     except Product.DoesNotExist:
         return "Invalid product id", 400, i
     return "OK", 200, None