Backend/device_hub/catalog_seed.py

from .models import DeviceCatalog


SENSOR_CATALOG_ITEMS = [
    {
        "code": "sensor_7_soil_moisture_sensor_v1_2",
        "name": "Sensor 7 - Soil Moisture Sensor v1.2",
        "description": (
            "This sensor is typically the YL-69 or FC-28 soil moisture sensor. "
            "It measures only soil moisture and provides analog and digital outputs. "
            "It does not report soil temperature, pH, or nutrients."
        ),
        "device_communication_type": "output_only",
        "customizable_fields": [],
        "supported_power_sources": ["solar", "direct_power"],
        "returned_data_fields": ["soil_moisture", "analog_output", "digital_output"],
        "sample_payload": {
            "soil_moisture": 42,
            "analog_output": 610,
            "digital_output": 1,
        },
        "is_active": True,
    }
]


def seed_sensor_catalog():
    created_count = 0
    updated_count = 0
    results = []

    for item in SENSOR_CATALOG_ITEMS:
        sensor, created = DeviceCatalog.objects.update_or_create(
            code=item["code"],
            defaults={
                "name": item["name"],
                "description": item["description"],
                "device_communication_type": item.get("device_communication_type", "output_only"),
                "customizable_fields": item["customizable_fields"],
                "supported_power_sources": item["supported_power_sources"],
                "returned_data_fields": item["returned_data_fields"],
                "sample_payload": item["sample_payload"],
                "is_active": item["is_active"],
            },
        )
        results.append((sensor, created))
        if created:
            created_count += 1
        else:
            updated_count += 1

    return results, created_count, updated_count