UPDATE

2026-05-11 03:27:21 +03:30
parent cf7cbb937c
commit d0e68a1a56
854 changed files with 102985 additions and 76 deletions
@@ -0,0 +1,103 @@
+import uuid
+
+from django.db import transaction
+
+from account.seeds import seed_admin_user
+from device_hub.catalog_seed import seed_sensor_catalog
+from device_hub.models import DeviceCatalog
+
+from .catalog import CATALOG_SEED_DATA
+from .models import FarmHub, FarmType, Product
+from .services import dispatch_farm_zoning
+
+
+ADMIN_FARM_UUID = uuid.UUID("11111111-1111-1111-1111-111111111111")
+ADMIN_FARM_DATA = {
+    "name": "Admin Smart Farm",
+    "is_active": True,
+    "irrigation_method_id": 1,
+    "irrigation_method_name": "آبیاری قطره ای",
+    "sensors": [
+        {
+            "sensor_catalog_code": "sensor_7_soil_moisture_sensor_v1_2",
+            "physical_device_uuid": uuid.UUID("22222222-2222-2222-2222-222222222222"),
+            "name": "Soil Probe 1",
+            "sensor_type": "soil_probe",
+            "is_active": True,
+            "specifications": {
+                "capabilities": ["soil_moisture", "analog_output", "digital_output"],
+            },
+            "power_source": {"type": "solar"},
+        },
+    ],
+}
+
+ADMIN_FARM_AREA_GEOJSON = {
+    "type": "Feature",
+    "properties": {},
+    "geometry": {
+        "type": "Polygon",
+        "coordinates": [
+            [
+                [51.418934, 35.706815],
+                [51.423054, 35.691062],
+                [51.384258, 35.689389],
+                [51.418934, 35.706815],
+            ]
+        ],
+    },
+}
+
+
+def _get_default_catalog():
+    default_farm_type_name = "گلخانه ای"
+    created_products = []
+
+    for farm_type_name, products in CATALOG_SEED_DATA.items():
+        farm_type, _ = FarmType.objects.get_or_create(name=farm_type_name)
+        for product_data in products:
+            product, _ = Product.objects.update_or_create(
+                farm_type=farm_type,
+                name=product_data["name"],
+                defaults={key: value for key, value in product_data.items() if key != "name"},
+            )
+            if farm_type_name == default_farm_type_name:
+                created_products.append(product)
+
+    return FarmType.objects.get(name=default_farm_type_name), created_products[:2]
+
+
+def _get_sensor_catalog_by_code(code):
+    return DeviceCatalog.objects.filter(code=code).first()
+
+
+@transaction.atomic
+def seed_admin_farm():
+    seed_sensor_catalog()
+    owner, _ = seed_admin_user()
+    farm_type, products = _get_default_catalog()
+    farm, created = FarmHub.objects.update_or_create(
+        farm_uuid=ADMIN_FARM_UUID,
+        defaults={
+            "owner": owner,
+            "farm_type": farm_type,
+            "name": ADMIN_FARM_DATA["name"],
+            "is_active": ADMIN_FARM_DATA["is_active"],
+            "irrigation_method_id": ADMIN_FARM_DATA["irrigation_method_id"],
+            "irrigation_method_name": ADMIN_FARM_DATA["irrigation_method_name"],
+        },
+    )
+    farm.products.set(products)
+    farm.sensors.all().delete()
+    sensors = []
+    for sensor_data in ADMIN_FARM_DATA["sensors"]:
+        sensor_data = sensor_data.copy()
+        sensor_catalog_code = sensor_data.pop("sensor_catalog_code", None)
+        sensor_data["sensor_catalog"] = _get_sensor_catalog_by_code(sensor_catalog_code) if sensor_catalog_code else None
+        sensors.append(farm.sensors.model(farm=farm, **sensor_data))
+    farm.sensors.bulk_create(sensors)
+    if created:
+        crop_area, _zoning_payload = dispatch_farm_zoning(ADMIN_FARM_AREA_GEOJSON, farm)
+        farm.current_crop_area = crop_area
+        farm.save(update_fields=["current_crop_area", "updated_at"])
+    return farm, created