UPDATE
This commit is contained in:
@@ -2,11 +2,11 @@ from rest_framework import serializers
|
||||
|
||||
|
||||
class EconomicDataItemSerializer(serializers.Serializer):
|
||||
title = serializers.CharField()
|
||||
value = serializers.CharField()
|
||||
subtitle = serializers.CharField()
|
||||
avatarIcon = serializers.CharField()
|
||||
avatarColor = serializers.CharField()
|
||||
title = serializers.CharField(help_text="عنوان شاخص اقتصادی.")
|
||||
value = serializers.CharField(help_text="مقدار شاخص اقتصادی.")
|
||||
subtitle = serializers.CharField(help_text="توضیح تکمیلی شاخص.")
|
||||
avatarIcon = serializers.CharField(help_text="آیکون نمایشی شاخص.")
|
||||
avatarColor = serializers.CharField(help_text="رنگ نمایشی شاخص.")
|
||||
|
||||
|
||||
class ChartSeriesSerializer(serializers.Serializer):
|
||||
@@ -15,6 +15,12 @@ class ChartSeriesSerializer(serializers.Serializer):
|
||||
|
||||
|
||||
class EconomicOverviewSerializer(serializers.Serializer):
|
||||
farm_uuid = serializers.CharField(required=False, allow_blank=True, help_text="UUID مزرعه.")
|
||||
source = serializers.CharField(required=False, allow_blank=True, help_text="منبع داده یا نوع تولید پاسخ.")
|
||||
economicData = EconomicDataItemSerializer(many=True)
|
||||
chartSeries = ChartSeriesSerializer(many=True)
|
||||
chartCategories = serializers.ListField(child=serializers.CharField())
|
||||
chartCategories = serializers.ListField(child=serializers.CharField(), help_text="برچسبهای محور افقی نمودار اقتصادی.")
|
||||
|
||||
|
||||
class EconomicOverviewRequestSerializer(serializers.Serializer):
|
||||
farm_uuid = serializers.UUIDField(required=True, help_text="UUID مزرعه برای دریافت نمای اقتصادی.")
|
||||
|
||||
@@ -0,0 +1,81 @@
|
||||
from unittest.mock import patch
|
||||
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.test import TestCase
|
||||
from django.urls import Resolver404, resolve
|
||||
from rest_framework.test import APIRequestFactory, force_authenticate
|
||||
|
||||
from external_api_adapter.adapter import AdapterResponse
|
||||
from farm_hub.models import FarmHub, FarmType
|
||||
|
||||
from .views import EconomyOverviewView
|
||||
|
||||
|
||||
class EconomyOverviewViewTests(TestCase):
|
||||
def setUp(self):
|
||||
self.factory = APIRequestFactory()
|
||||
self.user = get_user_model().objects.create_user(
|
||||
username="farmer",
|
||||
password="secret123",
|
||||
email="farmer@example.com",
|
||||
phone_number="09120000000",
|
||||
)
|
||||
self.other_user = get_user_model().objects.create_user(
|
||||
username="other-farmer",
|
||||
password="secret123",
|
||||
email="other@example.com",
|
||||
phone_number="09120000001",
|
||||
)
|
||||
self.farm_type = FarmType.objects.create(name="زراعی")
|
||||
self.farm = FarmHub.objects.create(owner=self.user, farm_type=self.farm_type, name="Farm 1")
|
||||
self.other_farm = FarmHub.objects.create(owner=self.other_user, farm_type=self.farm_type, name="Farm 2")
|
||||
|
||||
@patch("economic_overview.views.external_api_request")
|
||||
def test_overview_proxies_to_ai_service(self, mock_external_api_request):
|
||||
mock_external_api_request.return_value = AdapterResponse(
|
||||
status_code=200,
|
||||
data={
|
||||
"data": {
|
||||
"result": {
|
||||
"source": "mock",
|
||||
"economicData": [{"title": "Revenue", "value": "10"}],
|
||||
"chartSeries": [{"name": "Revenue", "data": [1.0, 2.0]}],
|
||||
"chartCategories": ["فروردین", "اردیبهشت"],
|
||||
}
|
||||
}
|
||||
},
|
||||
)
|
||||
|
||||
request = self.factory.post("/api/economy/overview/", {"farm_uuid": str(self.farm.farm_uuid)}, format="json")
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = EconomyOverviewView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(response.data["code"], 200)
|
||||
self.assertEqual(response.data["data"]["farm_uuid"], str(self.farm.farm_uuid))
|
||||
self.assertEqual(response.data["data"]["source"], "mock")
|
||||
mock_external_api_request.assert_called_once_with(
|
||||
"ai",
|
||||
"/api/economy/overview/",
|
||||
method="POST",
|
||||
payload={"farm_uuid": str(self.farm.farm_uuid)},
|
||||
)
|
||||
|
||||
def test_overview_rejects_foreign_farm_uuid(self):
|
||||
request = self.factory.post("/api/economy/overview/", {"farm_uuid": str(self.other_farm.farm_uuid)}, format="json")
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = EconomyOverviewView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 404)
|
||||
self.assertEqual(response.data["code"], 404)
|
||||
|
||||
def test_economy_routes_exist_only_under_economy_prefix(self):
|
||||
self.assertIs(resolve("/api/economy/overview/").func.view_class, EconomyOverviewView)
|
||||
|
||||
with self.assertRaises(Resolver404):
|
||||
resolve("/api/economy/summary/")
|
||||
|
||||
with self.assertRaises(Resolver404):
|
||||
resolve("/api/economic-overview/summary/")
|
||||
@@ -1,7 +1,7 @@
|
||||
from django.urls import path
|
||||
|
||||
from .views import EconomicOverviewView
|
||||
from .views import EconomyOverviewView
|
||||
|
||||
urlpatterns = [
|
||||
path("summary/", EconomicOverviewView.as_view(), name="economic-overview-summary"),
|
||||
path("overview/", EconomyOverviewView.as_view(), name="economy-overview"),
|
||||
]
|
||||
|
||||
@@ -1,11 +1,93 @@
|
||||
from drf_spectacular.utils import extend_schema
|
||||
from rest_framework import status
|
||||
from rest_framework.response import Response
|
||||
from rest_framework.views import APIView
|
||||
|
||||
from .mock_data import ECONOMIC_OVERVIEW
|
||||
from .serializers import EconomicOverviewSerializer
|
||||
from config.swagger import status_response
|
||||
from external_api_adapter import request as external_api_request
|
||||
from farm_hub.models import FarmHub
|
||||
from .models import EconomicOverviewLog
|
||||
from .serializers import EconomicOverviewRequestSerializer, EconomicOverviewSerializer
|
||||
|
||||
|
||||
class EconomicOverviewView(APIView):
|
||||
def get(self, request):
|
||||
serializer = EconomicOverviewSerializer(ECONOMIC_OVERVIEW)
|
||||
return Response({"status": "success", "result": serializer.data})
|
||||
class EconomyOverviewView(APIView):
|
||||
@staticmethod
|
||||
def _get_farm(request, farm_uuid):
|
||||
if not farm_uuid:
|
||||
return None, Response(
|
||||
{"code": 400, "msg": "error", "data": {"farm_uuid": ["This field is required."]}},
|
||||
status=status.HTTP_400_BAD_REQUEST,
|
||||
)
|
||||
try:
|
||||
return FarmHub.objects.get(farm_uuid=farm_uuid, owner=request.user), None
|
||||
except FarmHub.DoesNotExist:
|
||||
return None, Response(
|
||||
{"code": 404, "msg": "error", "data": {"farm_uuid": ["Farm not found."]}},
|
||||
status=status.HTTP_404_NOT_FOUND,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _extract_result(adapter_data):
|
||||
if not isinstance(adapter_data, dict):
|
||||
return {}
|
||||
|
||||
data = adapter_data.get("data")
|
||||
if isinstance(data, dict) and isinstance(data.get("result"), dict):
|
||||
return data["result"]
|
||||
if isinstance(data, dict):
|
||||
return data
|
||||
|
||||
result = adapter_data.get("result")
|
||||
if isinstance(result, dict):
|
||||
return result
|
||||
|
||||
return adapter_data
|
||||
|
||||
@staticmethod
|
||||
def _persist_log(farm, overview_data):
|
||||
if not isinstance(overview_data, dict):
|
||||
return
|
||||
EconomicOverviewLog.objects.create(
|
||||
farm=farm,
|
||||
economic_data=overview_data.get("economicData", []),
|
||||
chart_series=overview_data.get("chartSeries", []),
|
||||
chart_categories=overview_data.get("chartCategories", []),
|
||||
)
|
||||
|
||||
@extend_schema(
|
||||
tags=["Economy"],
|
||||
request=EconomicOverviewRequestSerializer,
|
||||
responses={200: status_response("EconomyOverviewResponse", data=EconomicOverviewSerializer())},
|
||||
)
|
||||
def post(self, request):
|
||||
serializer = EconomicOverviewRequestSerializer(data=request.data)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
|
||||
farm, error_response = self._get_farm(request, serializer.validated_data["farm_uuid"])
|
||||
if error_response is not None:
|
||||
return error_response
|
||||
|
||||
payload = {"farm_uuid": str(farm.farm_uuid)}
|
||||
adapter_response = external_api_request(
|
||||
"ai",
|
||||
"/api/economy/overview/",
|
||||
method="POST",
|
||||
payload=payload,
|
||||
)
|
||||
|
||||
if adapter_response.status_code >= 400:
|
||||
response_data = (
|
||||
adapter_response.data
|
||||
if isinstance(adapter_response.data, dict)
|
||||
else {"message": str(adapter_response.data)}
|
||||
)
|
||||
return Response(
|
||||
{"code": adapter_response.status_code, "msg": "error", "data": response_data},
|
||||
status=adapter_response.status_code,
|
||||
)
|
||||
|
||||
overview_data = self._extract_result(adapter_response.data)
|
||||
if isinstance(overview_data, dict):
|
||||
overview_data.setdefault("farm_uuid", str(farm.farm_uuid))
|
||||
self._persist_log(farm, overview_data)
|
||||
return Response({"code": 200, "msg": "success", "data": overview_data}, status=status.HTTP_200_OK)
|
||||
|
||||
Reference in New Issue
Block a user