checkout/src/purchase/management/commands/generate_dummy_data.py

import random
from datetime import timedelta

import freezegun
from django.core.management.base import BaseCommand
from django.utils.timezone import now

from purchase.models import Basket, BasketItem, PaymentMethod, Product


class Command(BaseCommand):
    help = "Generates dummy data"  # noqa: A003

    def handle(self, *args, **options):
        products = [
            Product(name="Clou", unit_price_cents=134),
            Product(name="Villard'Ain", unit_price_cents=290),
            Product(name="Herbier", unit_price_cents=330),
            Product(name="Blanc vache", unit_price_cents=650),
        ]
        products = Product.objects.bulk_create(products)
        self.stdout.write(
            self.style.SUCCESS(f"Successfully created {len(products)} products.")
        )

        payment_methods = [
            PaymentMethod(name="Espèces"),
            PaymentMethod(name="CB"),
            PaymentMethod(name="Chèque"),
        ]
        payment_methods = PaymentMethod.objects.bulk_create(payment_methods)
        self.stdout.write(
            self.style.SUCCESS(
                f"Successfully created {len(payment_methods)} payment methods."
            )
        )

        count = 0
        hours = list(range(-29, -20))
        hours += list(range(-10, -2))
        for hour in hours:
            with freezegun.freeze_time(now() + timedelta(hours=hour)):
                count += self.generate_baskets(payment_methods, products)

        self.stdout.write(self.style.SUCCESS(f"Successfully created {count} baskets."))

    def delete(self, cls):
        _, count = cls.objects.all().delete()
        self.stdout.write(self.style.WARNING(f"Successfully deleted {count} {cls}."))

    def generate_baskets(self, payment_methods, products):
        count = int(random.normalvariate(20, 10))
        for _ in range(count):
            method = random.choice(payment_methods)
            basket = Basket.objects.create(payment_method=method)
            items = []
            item_count = int(random.normalvariate(3, 2))
            for _ in range(item_count):
                product = random.choice(products)
                items.append(
                    BasketItem(
                        product=product, basket=basket, quantity=random.randint(1, 3)
                    )
                )
            BasketItem.objects.bulk_create(items)
        return count