v4.5 Demo initial commit

2026-06-24 14:53:14 +02:00
commit 3de8ded7c2
110 changed files with 14363112 additions and 0 deletions
@@ -0,0 +1,95 @@
+import os
+import django
+import sys
+from django.db import transaction
+
+# Ensure the correct working directory
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'django_project')))
+
+# Set up Django environment
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "IoTPreTester.settings")
+django.setup()
+
+from IoTAnalyzer.models import Manufacturer, DeviceType, Product, Device
+
+def populate_database():
+    """Populates the database with device types, manufacturers, and a sample device."""
+    print("Populating database...")
+
+    # Define device types
+    device_types = [
+        "Camera",
+        "Smart Plug",
+        "Smart Bulb",
+        "Smart Switch",
+        "Robot Vacuum Cleaner",
+        "Smart Lock",
+        "Smart Controller",
+        "Sensor",
+    ]
+
+    # Define manufacturers
+    manufacturers = [
+        "Dreame",
+        "Roborock",
+        "Shelly",
+        "Sonoff",
+        "Dji",
+        "TP link",
+        "Philips Hue",
+        "Nuki",
+        "Ikea",
+        "Xsense",
+        "Aquara",
+    ]
+
+    # Populate device types
+    with transaction.atomic():
+        for dt in device_types:
+            obj, created = DeviceType.objects.get_or_create(device_type=dt)
+            if created:
+                print(f"Created DeviceType: {obj.device_type}")
+            else:
+                print(f"DeviceType '{dt}' already exists")
+
+    # Populate manufacturers
+    with transaction.atomic():
+        for mf in manufacturers:
+            obj, created = Manufacturer.objects.get_or_create(name=mf)
+            if created:
+                print(f"Created Manufacturer: {obj.name}")
+            else:
+                print(f"Manufacturer '{mf}' already exists")
+
+    # Create a sample product and device
+    with transaction.atomic():
+        sample_manufacturer = Manufacturer.objects.filter(name="Shelly").first()
+        sample_device_type = DeviceType.objects.filter(device_type="Smart Controller").first()
+
+        if sample_manufacturer and sample_device_type:
+            sample_product, created = Product.objects.get_or_create(
+                manufacturer=sample_manufacturer,
+                device_type=sample_device_type,
+                product_name="1PM Gen3"
+            )
+
+            if created:
+                print(f"Created Sample Product: {sample_product.product_name}")
+            else:
+                print(f"Sample Product '{sample_product.product_name}' already exists")
+
+            sample_device, created = Device.objects.get_or_create(
+                product=sample_product,
+                serial="SN123456789",
+                comment="Test device for IoT system"
+            )
+
+            if created:
+                print(f"Created Sample Device with Serial: {sample_device.serial}")
+            else:
+                print(f"Sample Device '{sample_device.serial}' already exists")
+        else:
+            print("Error: Could not find required Manufacturer or DeviceType for sample device.")
+
+if __name__ == "__main__":
+    populate_database()