admin uchun ham plan apilar ozgartirildi

core/apps/dashboard/serializers/plan.py

@@ -5,24 +5,31 @@ from django.db import transaction
 from rest_framework import serializers
 
 # shared
-from core.apps.shared.models import Plan
+from core.apps.shared.models import Plan, Doctor, Pharmacy
 # accounts
 from core.apps.accounts.models import User
 
 
 class PlanListSerializer(serializers.ModelSerializer):
     user = serializers.SerializerMethodField(method_name='get_user')
-    
+    doctor = serializers.SerializerMethodField(method_name='get_doctor')
+    pharmacy = serializers.SerializerMethodField(method_name='get_pharmacy')
+
     class Meta:
         model = Plan
         fields = [
-            'id',
-            'title',
-            'description',
+            'id', 
+            'title', 
+            'description', 
             'date',
+            'comment', 
+            'doctor', 
+            'pharmacy',
             'user',
-            'is_done',
-            'created_at'
+            'longitude', 
+            'latitude', 
+            'extra_location',
+            'created_at',
         ]
     
     def get_user(self, obj):
@@ -31,6 +38,19 @@ class PlanListSerializer(serializers.ModelSerializer):
             'first_name': obj.user.first_name,
             'last_name': obj.user.last_name,
         }
+    
+    def get_doctor(self, obj):
+        return {
+            "id": obj.doctor.id,
+            "first_name": obj.doctor.first_name,
+            "last_name": obj.doctor.last_name
+        } if obj.doctor is not None else None
+
+    def get_pharmacy(self, obj):
+        return {
+            "id": obj.pharmacy.id,
+            "name": obj.pharmacy.name,
+        } if obj.pharmacy is not None else None
 
 
 class AdminPlanCreateSerializer(serializers.Serializer):
@@ -38,12 +58,29 @@ class AdminPlanCreateSerializer(serializers.Serializer):
     description = serializers.CharField()
     date = serializers.DateField()
     user_id = serializers.IntegerField()
+    doctor_id = serializers.IntegerField(required=False, allow_null=True)
+    pharmacy_id = serializers.IntegerField(required=False, allow_null=True)
+
+    longitude = serializers.FloatField(required=False, default=0.0)
+    latitude = serializers.FloatField(required=False, default=0.0)
+    extra_location = serializers.JSONField(required=False, allow_null=True)
+
 
     def validate(self, data):
         user = User.objects.filter(id=data['user_id']).first()
         if not user:
             raise serializers.ValidationError({"user_id": "Foydalanuvchi topilmadi"})
         data['user'] = user
+        if data.get('doctor_id'):
+            doctor = Doctor.objects.filter(id=data.get('doctor_id')).first()
+            if not doctor:
+                raise serializers.ValidationError({"doctor": "Doctor not found"})
+            data['doctor'] = doctor
+        if data.get('pharmacy_id'):
+            pharmacy = Pharmacy.objects.filter(id=data.get('pharmacy_id')).first()
+            if not pharmacy:
+                raise serializers.ValidationError({"pharmacy_id": "Pharmacy not found"})
+            data['pharmacy'] = pharmacy
         return data
 
     def create(self, validated_data):
@@ -53,6 +90,11 @@ class AdminPlanCreateSerializer(serializers.Serializer):
                 description=validated_data.get('description'),
                 user=validated_data.get('user'),
                 date=validated_data.get('date'),
+                doctor=validated_data.get('doctor'),
+                pharmacy=validated_data.get('pharmacy'),
+                longitude=validated_data.get('longitude'),
+                latitude=validated_data.get('latitude'),
+                extra_location=validated_data.get('extra_location'),
             )
 
 
@@ -60,10 +102,9 @@ class PlanUpdateSerializer(serializers.ModelSerializer):
     class Meta:
         model = Plan
         fields = [
-            'title',
-            'description',
-            'date',
-            'user',
+            'title', 'description', 'date',
+            'comment', 'doctor', 'pharmacy',
+            'longitude', 'latitude', 'extra_location',
         ]
     extra_kwargs = {
         "user": {"required": False}
@@ -71,10 +112,8 @@ class PlanUpdateSerializer(serializers.ModelSerializer):
     
     def update(self, instance, validated_data):
         with transaction.atomic():
-            instance.title = validated_data.get('title', instance.title)
-            instance.description = validated_data.get('description', instance.description)
-            instance.user = validated_data.get('user', instance.user)
-            instance.date = validated_data.get('date', instance.date)
+            for field, value in validated_data.items():
+                setattr(instance, field, value)
             instance.save()
             return instance
-    
+