27 lines
1.0 KiB
Python
27 lines
1.0 KiB
Python
import random
|
|
from django.core.management.base import BaseCommand
|
|
from weirdlittleempire.models import Commodity, CommodityInventory, SmartCard, SmartCardInventory, SmartPhoneVariant, SmartPhoneInventory
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Create Inventories for all products using random values."
|
|
|
|
def handle(self, verbosity, *args, **options):
|
|
self.verbosity = verbosity
|
|
for commodity in Commodity.objects.all():
|
|
CommodityInventory.objects.create(
|
|
product=commodity,
|
|
quantity=random.randint(0, 15)
|
|
)
|
|
for smart_card in SmartCard.objects.all():
|
|
SmartCardInventory.objects.create(
|
|
product=smart_card,
|
|
quantity=random.randint(0, 55)
|
|
)
|
|
for smart_phone in SmartPhoneVariant.objects.all():
|
|
SmartPhoneInventory.objects.create(
|
|
product=smart_phone,
|
|
quantity=random.randint(0, 8)
|
|
)
|
|
self.stdout.write("Created inventories with random quantities.")
|