feat: add levee work records

2026-04-04 11:32:26 +09:00
parent c773c7d3b8
commit b7b9818855
20 changed files with 929 additions and 6 deletions
--- a/backend/apps/levee_work/views.py
+++ b/backend/apps/levee_work/views.py
@@ -0,0 +1,70 @@
+from rest_framework import status, viewsets
+from rest_framework.permissions import IsAuthenticated
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from apps.plans.models import Plan
+from .models import LeveeWorkSession
+from .serializers import LeveeWorkSessionSerializer, LeveeWorkSessionWriteSerializer
+
+
+class LeveeWorkSessionViewSet(viewsets.ModelViewSet):
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        queryset = LeveeWorkSession.objects.prefetch_related(
+            'items',
+            'items__field',
+            'items__plan',
+        ).select_related('work_record')
+        year = self.request.query_params.get('year')
+        if year:
+            queryset = queryset.filter(year=year)
+        return queryset
+
+    def get_serializer_class(self):
+        if self.action in ['create', 'update', 'partial_update']:
+            return LeveeWorkSessionWriteSerializer
+        return LeveeWorkSessionSerializer
+
+
+class LeveeWorkCandidatesView(APIView):
+    permission_classes = [IsAuthenticated]
+
+    def get(self, request):
+        year = request.query_params.get('year')
+        if not year:
+            return Response(
+                {'detail': 'year が必要です。'},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+        try:
+            year = int(year)
+        except (TypeError, ValueError):
+            return Response(
+                {'detail': 'year は数値で指定してください。'},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+        plans = (
+            Plan.objects.select_related('field', 'crop', 'variety')
+            .filter(year=year, crop__name='水稲')
+            .order_by('field__display_order', 'field__id')
+        )
+
+        data = [
+            {
+                'field_id': plan.field_id,
+                'field_name': plan.field.name,
+                'field_area_tan': str(plan.field.area_tan),
+                'group_name': plan.field.group_name,
+                'plan_id': plan.id,
+                'crop_name': plan.crop.name,
+                'variety_name': plan.variety.name if plan.variety else '',
+                'selected': True,
+            }
+            for plan in plans
+        ]
+        return Response(data)
+