app.infrastructure.serializers

 9@extend_schema_serializer(
10    examples=[
11        OpenApiExample(
12            'Hospital Example',
13            value={
14                "id": 1,
15                "name": "General Hospital",
16                "geolocation_point": "POINT(77.5946 12.9716)",
17                "radius": 500,
18                "address": "123 Main St"
19            }
20        )
21    ]
22)
23class HospitalSerializer(serializers.ModelSerializer):
24    """
25    Serializer for hospital details.
26
27    Fields:
28        - id: Hospital ID.
29        - name: Hospital name.
30        - geolocation_point: Geolocation as POINT string.
31        - radius: Coverage radius.
32        - address: Hospital address.
33    """
34    class Meta:
35        model = Hospital
36        fields = ['id','name','geolocation_point','radius', 'address']
37        read_only_fields = ['id']

34    class Meta:
35        model = Hospital
36        fields = ['id','name','geolocation_point','radius', 'address']
37        read_only_fields = ['id']

39@extend_schema_serializer(
40    examples=[
41        OpenApiExample(
42            'Role Example',
43            value={
44                "id": 1,
45                "name": "Doctor",
46                "description": "Medical Doctor",
47                "hospital": "General Hospital",
48                "is_approver": True,
49                "is_responder": False,
50                "is_creator": True,
51                "is_protected": False
52            }
53        )
54    ]
55)
56class RoleSerializer(serializers.ModelSerializer):
57    """
58    Serializer for user roles within a hospital.
59
60    Fields:
61        - id: Role ID.
62        - name: Role name.
63        - description: Role description.
64        - hospital: Hospital name.
65        - is_approver: Is role an approver.
66        - is_responder: Is role a responder.
67        - is_creator: Is role a creator.
68        - is_protected: Is role protected.
69    """
70    hospital = serializers.CharField(source='hospital.name', read_only=True)
71    class Meta:
72        model = Role
73        fields = ['id', 'name', 'description','hospital', 'is_approver', 'is_responder', 'is_creator', 'is_protected']
74        read_only_fields = ['id','hospital']

71    class Meta:
72        model = Role
73        fields = ['id', 'name', 'description','hospital', 'is_approver', 'is_responder', 'is_creator', 'is_protected']
74        read_only_fields = ['id','hospital']

 76@extend_schema_serializer(
 77    examples=[
 78        OpenApiExample(
 79            'Hospital User Example',
 80            value={
 81                "id": 1,
 82                "institution_id": "EMP123",
 83                "role": 2,
 84                "role_name": "Doctor",
 85                "phone_number": "9876543210",
 86                "current_block": 1,
 87                "current_block_name": "Block A",
 88                "current_ward": 3,
 89                "current_ward_name": "Ward 3",
 90                "date_joined": "2025-09-20T12:00:00Z"
 91            }
 92        )
 93    ]
 94)
 95class HospitalUserSerializer(serializers.ModelSerializer):
 96    """
 97    Serializer for hospital users including role, block, and ward info.
 98
 99    Fields:
100        - id: User ID.
101        - institution_id: Institution ID.
102        - role: Role ID.
103        - role_name: Role name.
104        - phone_number: User's phone number.
105        - password: Password (write-only).
106        - fcm_token: FCM token (write-only).
107        - current_block: Block ID.
108        - current_block_name: Block name.
109        - current_ward: Ward ID.
110        - current_ward_name: Ward name.
111        - date_joined: Date joined.
112    """
113    role = serializers.PrimaryKeyRelatedField(queryset=Role.objects.all())
114    password = serializers.CharField(write_only=True, required=False, allow_blank=True)
115    current_block = serializers.PrimaryKeyRelatedField(queryset=Blocks.objects.all(), required=False, allow_null=True)
116    fcm_token = serializers.CharField(write_only=True, required=False, allow_blank=True)
117    current_block_name = serializers.CharField(source='current_block.name', read_only=True)
118    current_ward = serializers.PrimaryKeyRelatedField(queryset=Ward.objects.all(), required=False, allow_null=True)
119    current_ward_name = serializers.CharField(source='current_ward.ward_name', read_only=True)
120    role_name = serializers.CharField(source='role.name', read_only=True)
121
122    class Meta:
123        model = User
124        fields = ['id', 'institution_id', 'role','role_name', 'phone_number', 'password','fcm_token','current_block','current_block_name','current_ward','current_ward_name','date_joined']
125        read_only_fields = ['id']
126        write_only_fields = ['fcm_token']
127        required_fields = ['institution_id', 'role', 'phone_number']
128
129    def create(self, validated_data):
130        """Create a hospital user and set password."""
131        hospital_id = self.context.get('hospital_id')
132        user = User.objects.create(
133            institution_id=validated_data['institution_id'],
134            role=validated_data['role'],
135            phone_number=validated_data['phone_number'],
136            hospital=Hospital.objects.get(id=hospital_id)
137        )
138        password = validated_data.get('password')
139        if password:
140            user.set_password(password)
141            user.password_set = True
142        else:
143            user.set_unusable_password()
144            user.password_set = False
145        user.save()
146        return user
147
148    def update(self, instance, validated_data):
149        """Update hospital user details and password."""
150        try:
151            hospital_id = self.context.get('hospital_id')
152            instance.institution_id = validated_data.get('institution_id', instance.institution_id)
153            instance.role = validated_data.get('role', instance.role)
154            instance.phone_number = validated_data.get('phone_number', instance.phone_number)
155            instance.current_block = validated_data.get('current_block', instance.current_block)
156            instance.fcm_token = validated_data.get('fcm_token', instance.fcm_token)
157            instance.fcm_token_updated = time.time()
158            password = validated_data.get('password')
159            if password:
160                instance.set_password(password)
161                instance.password_set = True
162            instance.save()
163            return instance
164        except Hospital.DoesNotExist:
165            raise serializers.ValidationError("Hospital does not exist.")
166        except Blocks.DoesNotExist:
167            raise serializers.ValidationError("Block does not exist.")

122    class Meta:
123        model = User
124        fields = ['id', 'institution_id', 'role','role_name', 'phone_number', 'password','fcm_token','current_block','current_block_name','current_ward','current_ward_name','date_joined']
125        read_only_fields = ['id']
126        write_only_fields = ['fcm_token']
127        required_fields = ['institution_id', 'role', 'phone_number']

169@extend_schema_serializer(
170    examples=[
171        OpenApiExample(
172            'Block Example',
173            value={
174                "id": 1,
175                "name": "Block A",
176                "no_of_floors": 5,
177                "created_by": "EMP123",
178                "hospital": "General Hospital",
179                "hospital_id": 1,
180                "created_at": "2025-09-20T12:00:00Z",
181                "updated_at": "2025-09-21T12:00:00Z"
182            }
183        )
184    ]
185)
186class BlockSerializer(serializers.ModelSerializer):
187    """
188    Serializer for hospital blocks.
189
190    Fields:
191        - id: Block ID.
192        - name: Block name.
193        - no_of_floors: Number of floors.
194        - created_by: Institution ID of creator.
195        - hospital: Hospital name.
196        - hospital_id: Hospital ID.
197        - created_at: Creation timestamp.
198        - updated_at: Last update timestamp.
199    """
200    created_by = serializers.CharField(source='created_by.institution_id', read_only=True)
201    hospital = serializers.CharField(source='hospital.name', read_only=True)
202    hospital_id = serializers.IntegerField(source='hospital.id', read_only=True)
203    class Meta:
204        model = Blocks
205        fields = ['id', 'name', 'no_of_floors', 'created_by', 'hospital', 'hospital_id', 'created_at', 'updated_at']
206        read_only_fields = ['id', 'hospital']
207
208    def create(self, validated_data):
209        """Create a block for a hospital."""
210        hospital_id = self.context.get('hospital_id')
211        validated_data['hospital'] = Hospital.objects.get(id=hospital_id)
212        return super().create(validated_data)
213
214    def update(self, instance, validated_data):
215        """Update block details for a hospital."""
216        hospital_id = self.context.get('hospital_id')
217        instance.hospital = Hospital.objects.get(id=hospital_id)
218        return super().update(instance, validated_data)

203    class Meta:
204        model = Blocks
205        fields = ['id', 'name', 'no_of_floors', 'created_by', 'hospital', 'hospital_id', 'created_at', 'updated_at']
206        read_only_fields = ['id', 'hospital']

220@extend_schema_serializer(
221    examples=[
222        OpenApiExample(
223            'Shift Example',
224            value={
225                "id": 1,
226                "name": "Morning Shift",
227                "start_time": "08:00:00",
228                "end_time": "16:00:00",
229                "hospital": "General Hospital",
230                "hospital_id": 1
231            }
232        )
233    ]
234)
235class ShiftSerializer(serializers.ModelSerializer):
236    """
237    Serializer for hospital shifts.
238
239    Fields:
240        - id: Shift ID.
241        - name: Shift name.
242        - start_time: Shift start time.
243        - end_time: Shift end time.
244        - hospital: Hospital name.
245        - hospital_id: Hospital ID.
246    """
247    hospital = serializers.CharField(source='hospital.name', read_only=True)
248    hospital_id = serializers.IntegerField(source='hospital.id', read_only=True)
249    class Meta:
250        model = Shift
251        fields = ['id', 'name', 'start_time', 'end_time', 'hospital','hospital_id']
252        read_only_fields = ['id', 'hospital']
253
254    def create(self, validated_data):
255        """Create a shift for a hospital."""
256        hospital_id = self.context.get('hospital_id')
257        validated_data['hospital'] = Hospital.objects.get(id=hospital_id)
258        return super().create(validated_data)
259
260    def update(self, instance, validated_data):
261        """Update shift details for a hospital."""
262        hospital_id = self.context.get('hospital_id')
263        instance.hospital = Hospital.objects.get(id=hospital_id)
264        return super().update(instance, validated_data)

249    class Meta:
250        model = Shift
251        fields = ['id', 'name', 'start_time', 'end_time', 'hospital','hospital_id']
252        read_only_fields = ['id', 'hospital']

266@extend_schema_serializer(
267    examples=[
268        OpenApiExample(
269            'Ward Example',
270            value={
271                "id": 1,
272                "ward_name": "Ward 3",
273                "block": 1,
274                "block_name": "Block A",
275                "floor": 2,
276                "created_at": "2025-09-20T12:00:00Z",
277                "created_by": 5,
278                "created_by_name": "EMP123"
279            }
280        )
281    ]
282)
283class WardSerializer(serializers.ModelSerializer):
284    """
285    Serializer for hospital wards.
286
287    Fields:
288        - id: Ward ID.
289        - ward_name: Ward name.
290        - block: Block ID.
291        - block_name: Block name.
292        - floor: Floor number.
293        - created_at: Creation timestamp.
294        - created_by: User ID of creator.
295        - created_by_name: Institution ID of creator.
296    """
297    block = serializers.PrimaryKeyRelatedField(queryset=Blocks.objects.all())
298    block_name = serializers.CharField(source='block.name', read_only=True)
299    created_by_name = serializers.CharField(source='created_by.institution_id',read_only=True)
300    class Meta:
301        model = Ward
302        fields = ['id','ward_name','block','floor','created_at','created_by','created_by_name','block_name']
303        read_only = ['id']
304    def create(self, validated_data):
305        """Create a ward for a hospital."""
306        hospital_id = self.context.get('hospital_id')
307        return super().create(validated_data)
308
309    def update(self, instance, validated_data):
310        """Update ward details for a hospital."""
311        hospital_id = self.context.get('hospital_id')
312        return super().update(instance, validated_data)
313    
314    def validate(self, data):
315        """Validate that the floor is within the block's floor range."""
316        block = data.get('block')
317        floor = data.get('floor')
318        if block and floor:
319            if floor < 1 or floor > block.no_of_floors:
320                raise serializers.ValidationError({
321                    'floor': f"Floor must be between 1 and {block.no_of_floors} for the selected block."
322                })
323        return data

300    class Meta:
301        model = Ward
302        fields = ['id','ward_name','block','floor','created_at','created_by','created_by_name','block_name']
303        read_only = ['id']

325@extend_schema_serializer(
326    examples=[
327        OpenApiExample(
328            'Approver Level Example',
329            value={
330                "id": 1,
331                "role": 2,
332                "role_name": "Doctor",
333                "priority": 1
334            }
335        )
336    ]
337)
338class ApproverLevelSerializer(serializers.ModelSerializer):
339    """
340    Serializer for approver levels within a hierarchy.
341
342    Fields:
343        - id: Level ID.
344        - role: Role ID.
345        - role_name: Role name.
346        - priority: Priority (lower means higher).
347    """
348    role_name = serializers.CharField(source='role.name', read_only=True)
349    class Meta:
350        model = ApproverLevel
351        fields = ['id', 'role', 'role_name', 'priority']
352        read_only_fields = ['id', 'role_name']

349    class Meta:
350        model = ApproverLevel
351        fields = ['id', 'role', 'role_name', 'priority']
352        read_only_fields = ['id', 'role_name']

354@extend_schema_serializer(
355    examples=[
356        OpenApiExample(
357            'Approver Hierarchy Example',
358            value={
359                "id": 1,
360                "is_active": True,
361                "levels": [
362                    {"id": 1, "role": 2, "role_name": "Doctor", "priority": 1}
363                ],
364                "created_at": "2025-09-20T12:00:00Z"
365            }
366        )
367    ]
368)
369class ApproverHierarchySerializer(serializers.ModelSerializer):
370    """
371    Serializer for approver hierarchy with levels.
372
373    Fields:
374        - id: Hierarchy ID.
375        - is_active: Is hierarchy active.
376        - levels: List of approver levels.
377        - created_at: Creation timestamp.
378    """
379    levels = ApproverLevelSerializer(many=True)
380    class Meta:
381        model = ApproverHierarchy
382        fields = ['id', 'is_active', 'levels', 'created_at']
383        read_only_fields = ['id', 'created_at']
384
385    def validate_levels(self, value):
386        """Ensure at least one level is provided."""
387        if not value:
388            raise serializers.ValidationError("At least one level is required")
389        return value
390
391    def create(self, validated_data):
392        """Create an approver hierarchy with levels."""
393        levels_data = validated_data.pop('levels')
394        hierarchy = ApproverHierarchy.objects.create(**validated_data)
395        for lvl in levels_data:
396            ApproverLevel.objects.create(hierarchy=hierarchy, **lvl)
397        return hierarchy
398
399    def update(self, instance, validated_data):
400        """Update approver hierarchy and its levels."""
401        levels_data = validated_data.pop('levels', None)
402        instance.is_active = validated_data.get('is_active', instance.is_active)
403        instance.save()
404        if levels_data is not None:
405            instance.levels.all().delete()
406            for lvl in levels_data:
407                ApproverLevel.objects.create(hierarchy=instance, **lvl)
408        return instance

380    class Meta:
381        model = ApproverHierarchy
382        fields = ['id', 'is_active', 'levels', 'created_at']
383        read_only_fields = ['id', 'created_at']