This commit is contained in:
2026-05-05 00:56:05 +03:30
parent 21b734f6a7
commit cfe60f6729
85 changed files with 1786 additions and 3840 deletions
+5
View File
@@ -11,6 +11,8 @@ class PlantSerializer(serializers.ModelSerializer):
fields = [
"id",
"name",
"description",
"metadata",
"light",
"watering",
"soil",
@@ -22,6 +24,9 @@ class PlantSerializer(serializers.ModelSerializer):
"harvest_time",
"spacing",
"fertilizer",
"health_profile",
"irrigation_profile",
"growth_profile",
"created_at",
"updated_at",
]
+55 -72
View File
@@ -1,4 +1,5 @@
from django.db import transaction
from django.conf import settings
from external_api_adapter import request as external_api_request
from external_api_adapter.exceptions import ExternalAPIRequestError
@@ -14,33 +15,13 @@ DEFAULT_GROWTH_STAGES = [
"fruiting",
"maturity",
]
AI_PLANTS_PATH = "/api/plants/"
AI_FARM_DATA_PLANT_SYNC_PATH = "/api/farm-data/plants/sync/"
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
@@ -110,60 +91,62 @@ def ensure_plant_defaults(queryset=None):
return products
@transaction.atomic
def sync_plants_from_ai():
def serialize_products_for_ai(products=None):
products = list(products if products is not None else Product.objects.select_related("farm_type").all().order_by("name"))
ensure_plant_defaults(products)
payload = []
for product in products:
payload.append(
{
"id": product.id,
"name": product.name,
"slug": "",
"icon": product.icon,
"description": product.description,
"metadata": product.metadata if isinstance(product.metadata, dict) else {},
"light": product.light,
"watering": product.watering,
"soil": product.soil,
"temperature": product.temperature,
"growth_stage": product.growth_stage,
"growth_stages": product.growth_stages or [],
"planting_season": product.planting_season,
"harvest_time": product.harvest_time,
"spacing": product.spacing,
"fertilizer": product.fertilizer,
"health_profile": product.health_profile if isinstance(product.health_profile, dict) else {},
"irrigation_profile": product.irrigation_profile if isinstance(product.irrigation_profile, dict) else {},
"growth_profile": product.growth_profile if isinstance(product.growth_profile, dict) else {},
"is_active": True,
"updated_at": product.updated_at.isoformat() if product.updated_at else None,
"farm_type": product.farm_type.name if product.farm_type_id else DEFAULT_FARM_TYPE_NAME,
}
)
return payload
def push_plants_to_ai(products=None):
api_key = getattr(settings, "FARM_DATA_API_KEY", "")
if not api_key:
raise PlantSyncError("FARM_DATA_API_KEY is not configured.")
payload = serialize_products_for_ai(products)
try:
adapter_response = external_api_request("ai", AI_PLANTS_PATH, method="GET")
adapter_response = external_api_request(
"ai",
AI_FARM_DATA_PLANT_SYNC_PATH,
method="POST",
payload=payload,
headers={
"Accept": "application/json",
"Content-Type": "application/json",
"X-API-Key": api_key,
"Authorization": f"Api-Key {api_key}",
},
)
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
return payload
+6 -6
View File
@@ -56,9 +56,9 @@ class PlantApiTests(TestCase):
self.assertIn("flowering", response.data["data"][0]["growth_stages"])
mock_external_api_request.assert_called_once_with("ai", "/api/plants/", method="GET")
@patch("plants.views.sync_plants_from_ai")
def test_names_endpoint_fills_default_icon_and_growth_stages(self, mock_sync_plants_from_ai):
mock_sync_plants_from_ai.return_value = []
@patch("plants.views.push_plants_to_ai")
def test_names_endpoint_fills_default_icon_and_growth_stages(self, mock_push_plants_to_ai):
mock_push_plants_to_ai.return_value = []
product = Product.objects.create(
farm_type=self.farm_type,
name="Pepper",
@@ -79,9 +79,9 @@ class PlantApiTests(TestCase):
self.assertEqual(product.icon, "leaf")
self.assertEqual(product.growth_stages, ["initial", "vegetative", "flowering", "fruiting", "maturity"])
@patch("plants.views.sync_plants_from_ai")
def test_selected_endpoint_returns_farmer_products(self, mock_sync_plants_from_ai):
mock_sync_plants_from_ai.return_value = []
@patch("plants.views.push_plants_to_ai")
def test_selected_endpoint_returns_farmer_products(self, mock_push_plants_to_ai):
mock_push_plants_to_ai.return_value = []
tomato = Product.objects.create(farm_type=self.farm_type, name="Tomato", icon="leaf", growth_stages=["vegetative"])
pepper = Product.objects.create(farm_type=self.farm_type, name="Pepper", icon="leaf", growth_stages=["flowering"])
farm = FarmHub.objects.create(owner=self.user, farm_type=self.farm_type, name="farm-a")
+7 -8
View File
@@ -8,7 +8,7 @@ from drf_spectacular.utils import OpenApiParameter, extend_schema
from config.swagger import code_response, farm_uuid_query_param
from farm_hub.models import FarmHub, Product
from .serializers import PlantNameSerializer, PlantSerializer
from .services import PlantSyncError, ensure_plant_defaults, sync_plants_from_ai
from .services import PlantSyncError, ensure_plant_defaults, push_plants_to_ai
class PlantBaseView(APIView):
@@ -17,7 +17,7 @@ class PlantBaseView(APIView):
@staticmethod
def _sync_plants_if_possible():
try:
sync_plants_from_ai()
push_plants_to_ai()
except PlantSyncError:
return False
return True
@@ -38,13 +38,12 @@ class PlantListView(PlantBaseView):
responses={200: code_response("PlantListResponse", data=PlantSerializer(many=True))},
)
def get(self, request):
try:
sync_plants_from_ai()
except PlantSyncError as exc:
if not Product.objects.exists():
return Response({"code": 503, "msg": str(exc)}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
products = ensure_plant_defaults(Product.objects.order_by("name"))
try:
push_plants_to_ai(products)
except PlantSyncError as exc:
if not products:
return Response({"code": 503, "msg": str(exc)}, status=status.HTTP_503_SERVICE_UNAVAILABLE)
data = PlantSerializer(products, many=True).data
return Response({"code": 200, "msg": "success", "data": data}, status=status.HTTP_200_OK)