Init Django-SHOP

weirdlittleempire/management/commands/initialize_inventories.py

@@ -0,0 +1,26 @@
+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.")