104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
import uuid
|
|
|
|
from django.db import transaction
|
|
|
|
from account.seeds import seed_admin_user
|
|
from sensor_catalog.management import seed_sensor_catalog
|
|
from sensor_catalog.models import SensorCatalog
|
|
|
|
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 SensorCatalog.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
|