UPDATE

plants/services.py

@@ -0,0 +1,169 @@
+from django.db import transaction
+
+from external_api_adapter import request as external_api_request
+from external_api_adapter.exceptions import ExternalAPIRequestError
+from farm_hub.models import FarmType, Product
+
+
+DEFAULT_FARM_TYPE_NAME = "زراعی"
+DEFAULT_ICON = "leaf"
+DEFAULT_GROWTH_STAGES = [
+    "initial",
+    "vegetative",
+    "flowering",
+    "fruiting",
+    "maturity",
+]
+AI_PLANTS_PATH = "/api/plants/"
+
+
+class PlantSyncError(Exception):
+    pass
+
+
+def _extract_plant_items(adapter_data):
+    if isinstance(adapter_data, list):
+        return adapter_data
+    if not isinstance(adapter_data, dict):
+        return []
+
+    data = adapter_data.get("data")
+    if isinstance(data, list):
+        return data
+    if isinstance(data, dict):
+        result = data.get("result")
+        if isinstance(result, list):
+            return result
+
+    result = adapter_data.get("result")
+    if isinstance(result, list):
+        return result
+    return []
+
+
+def _clean_stage_name(value):
+    stage = str(value or "").strip()
+    return stage
+
+
+def _merge_growth_stages(product, supplied_stages=None):
+    stages = []
+    seen = set()
+    has_explicit_stage_data = False
+
+    for stage in supplied_stages or []:
+        normalized = _clean_stage_name(stage)
+        if normalized and normalized not in seen:
+            has_explicit_stage_data = True
+            seen.add(normalized)
+            stages.append(normalized)
+
+    current_stage = _clean_stage_name(getattr(product, "growth_stage", ""))
+    if current_stage and current_stage not in seen:
+        has_explicit_stage_data = True
+        seen.add(current_stage)
+        stages.append(current_stage)
+
+    if not has_explicit_stage_data:
+        for stage in DEFAULT_GROWTH_STAGES:
+            seen.add(stage)
+            stages.append(stage)
+
+    thresholds = product.growth_profile.get("stage_thresholds", {}) if isinstance(product.growth_profile, dict) else {}
+    if isinstance(thresholds, dict):
+        for stage_name in thresholds.keys():
+            normalized = _clean_stage_name(stage_name)
+            if normalized and normalized not in seen:
+                seen.add(normalized)
+                stages.append(normalized)
+
+    return stages
+
+
+@transaction.atomic
+def ensure_plant_defaults(queryset=None):
+    products = list(queryset if queryset is not None else Product.objects.all())
+    updated_products = []
+
+    for product in products:
+        changed = False
+
+        if not product.icon:
+            product.icon = DEFAULT_ICON
+            changed = True
+
+        normalized_stages = _merge_growth_stages(product, product.growth_stages)
+        if normalized_stages != (product.growth_stages or []):
+            product.growth_stages = normalized_stages
+            changed = True
+
+        if not product.growth_stage and product.growth_stages:
+            product.growth_stage = product.growth_stages[0]
+            changed = True
+
+        if changed:
+            updated_products.append(product)
+
+    if updated_products:
+        Product.objects.bulk_update(updated_products, ["icon", "growth_stage", "growth_stages"])
+
+    return products
+
+
+@transaction.atomic
+def sync_plants_from_ai():
+    try:
+        adapter_response = external_api_request("ai", AI_PLANTS_PATH, method="GET")
+    except ExternalAPIRequestError as exc:
+        raise PlantSyncError(str(exc)) from exc
+
+    if adapter_response.status_code >= 400:
+        raise PlantSyncError(f"AI service returned status {adapter_response.status_code}.")
+
+    products = []
+    for item in _extract_plant_items(adapter_response.data):
+        if not isinstance(item, dict):
+            continue
+
+        name = str(item.get("name") or "").strip()
+        if not name:
+            continue
+
+        farm_type_name = str(item.get("farm_type") or DEFAULT_FARM_TYPE_NAME).strip() or DEFAULT_FARM_TYPE_NAME
+        farm_type, _ = FarmType.objects.get_or_create(name=farm_type_name)
+
+        growth_profile = item.get("growth_profile") if isinstance(item.get("growth_profile"), dict) else {}
+        growth_stages = item.get("growth_stages") if isinstance(item.get("growth_stages"), list) else []
+        normalized_growth_stages = []
+        for stage in growth_stages:
+            normalized = _clean_stage_name(stage)
+            if normalized:
+                normalized_growth_stages.append(normalized)
+
+        defaults = {
+            "description": str(item.get("description") or "").strip(),
+            "metadata": item.get("metadata") if isinstance(item.get("metadata"), dict) else {},
+            "light": str(item.get("light") or "").strip(),
+            "watering": str(item.get("watering") or "").strip(),
+            "soil": str(item.get("soil") or "").strip(),
+            "temperature": str(item.get("temperature") or "").strip(),
+            "growth_stage": str(item.get("growth_stage") or "").strip(),
+            "growth_stages": normalized_growth_stages,
+            "planting_season": str(item.get("planting_season") or "").strip(),
+            "harvest_time": str(item.get("harvest_time") or "").strip(),
+            "spacing": str(item.get("spacing") or "").strip(),
+            "fertilizer": str(item.get("fertilizer") or "").strip(),
+            "icon": str(item.get("icon") or "").strip(),
+            "health_profile": item.get("health_profile") if isinstance(item.get("health_profile"), dict) else {},
+            "irrigation_profile": item.get("irrigation_profile") if isinstance(item.get("irrigation_profile"), dict) else {},
+            "growth_profile": growth_profile,
+        }
+        product, _ = Product.objects.update_or_create(
+            farm_type=farm_type,
+            name=name,
+            defaults=defaults,
+        )
+        products.append(product)
+
+    ensure_plant_defaults(products)
+    return products