AdventureLog/backend/server/adventures/serializers.py

from django.utils import timezone
import os
from .models import Adventure, AdventureImage, ChecklistItem, Collection, Note, Transportation, Checklist, Visit, Category
from rest_framework import serializers
from main.utils import CustomModelSerializer


class AdventureImageSerializer(CustomModelSerializer):
    class Meta:
        model = AdventureImage
        fields = ['id', 'image', 'adventure']
        read_only_fields = ['id']

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        if instance.image:
            public_url = os.environ.get('PUBLIC_URL', 'http://127.0.0.1:8000').rstrip('/')
            #print(public_url)
            # remove any  ' from the url
            public_url = public_url.replace("'", "")
            representation['image'] = f"{public_url}/media/{instance.image.name}"
        return representation
    
class CategorySerializer(CustomModelSerializer):
    num_adventures = serializers.SerializerMethodField()
    class Meta:
        model = Category
        fields = ['id', 'name', 'display_name', 'icon', 'user_id', 'num_adventures']
        read_only_fields = ['id', 'user_id', 'num_adventures']

    def validate_name(self, value):
        if Category.objects.filter(name=value).exists():
            raise serializers.ValidationError('Category with this name already exists.')
    
        return value
    
    def get_num_adventures(self, obj):
        return Adventure.objects.filter(category=obj, user_id=obj.user_id).count()
    
class VisitSerializer(serializers.ModelSerializer):

    class Meta:
        model = Visit
        fields = ['id', 'start_date', 'end_date', 'notes']
        read_only_fields = ['id']
                                   
class AdventureSerializer(CustomModelSerializer):
    images = AdventureImageSerializer(many=True, read_only=True)
    visits = VisitSerializer(many=True, read_only=False)
    category = serializers.PrimaryKeyRelatedField(
        queryset=Category.objects.all(),
        write_only=True,
        required=False
    )
    category_object = CategorySerializer(source='category', read_only=True)
    is_visited = serializers.SerializerMethodField()

    class Meta:
        model = Adventure
        fields = [
            'id', 'user_id', 'name', 'description', 'rating', 'activity_types', 'location', 
            'is_public', 'collection', 'created_at', 'updated_at', 'images', 'link', 'longitude', 
            'latitude', 'visits', 'is_visited', 'category', 'category_object'
        ]
        read_only_fields = ['id', 'created_at', 'updated_at', 'user_id', 'is_visited']

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        representation['category'] = representation.pop('category_object')
        return representation

    def validate_category(self, category):
        # Check that the category belongs to the same user
        if category.user_id != self.context['request'].user:
            raise serializers.ValidationError('Category does not belong to the user.')
        return category

    def get_is_visited(self, obj):
        current_date = timezone.now().date()
        for visit in obj.visits.all():
            if visit.start_date and visit.end_date and (visit.start_date <= current_date):
                return True
            elif visit.start_date and not visit.end_date and (visit.start_date <= current_date):
                return True
        return False

    def create(self, validated_data):
        visits_data = validated_data.pop('visits', [])
        adventure = Adventure.objects.create(**validated_data)
        for visit_data in visits_data:
            Visit.objects.create(adventure=adventure, **visit_data)
        return adventure

    def update(self, instance, validated_data):
        visits_data = validated_data.pop('visits', [])

        for attr, value in validated_data.items():
            setattr(instance, attr, value)
        instance.save()

        current_visits = instance.visits.all()
        current_visit_ids = set(current_visits.values_list('id', flat=True))

        updated_visit_ids = set()
        for visit_data in visits_data:
            visit_id = visit_data.get('id')
            if visit_id and visit_id in current_visit_ids:
                visit = current_visits.get(id=visit_id)
                for attr, value in visit_data.items():
                    setattr(visit, attr, value)
                visit.save()
                updated_visit_ids.add(visit_id)
            else:
                new_visit = Visit.objects.create(adventure=instance, **visit_data)
                updated_visit_ids.add(new_visit.id)

        visits_to_delete = current_visit_ids - updated_visit_ids
        instance.visits.filter(id__in=visits_to_delete).delete()

        return instance

class TransportationSerializer(CustomModelSerializer):

    class Meta:
        model = Transportation
        fields = [
            'id', 'user_id', 'type', 'name', 'description', 'rating', 
            'link', 'date', 'flight_number', 'from_location', 'to_location', 
            'is_public', 'collection', 'created_at', 'updated_at', 'end_date'
        ]
        read_only_fields = ['id', 'created_at', 'updated_at', 'user_id']

class NoteSerializer(CustomModelSerializer):

    class Meta:
        model = Note
        fields = [
            'id', 'user_id', 'name', 'content', 'date', 'links', 
            'is_public', 'collection', 'created_at', 'updated_at'
        ]
        read_only_fields = ['id', 'created_at', 'updated_at', 'user_id']
    
class ChecklistItemSerializer(CustomModelSerializer):
        class Meta:
            model = ChecklistItem
            fields = [
                'id', 'user_id', 'name', 'is_checked', 'checklist', 'created_at', 'updated_at'
            ]
            read_only_fields = ['id', 'created_at', 'updated_at', 'user_id', 'checklist']
  
class ChecklistSerializer(CustomModelSerializer):
    items = ChecklistItemSerializer(many=True, source='checklistitem_set')
    class Meta:
        model = Checklist
        fields = [
            'id', 'user_id', 'name', 'date', 'is_public', 'collection', 'created_at', 'updated_at', 'items'
        ]
        read_only_fields = ['id', 'created_at', 'updated_at', 'user_id']
    
    def create(self, validated_data):
        items_data = validated_data.pop('checklistitem_set')
        checklist = Checklist.objects.create(**validated_data)
        for item_data in items_data:
            ChecklistItem.objects.create(checklist=checklist, **item_data)
        return checklist
    
    def update(self, instance, validated_data):
        items_data = validated_data.pop('checklistitem_set', [])
        
        # Update Checklist fields
        for attr, value in validated_data.items():
            setattr(instance, attr, value)
        instance.save()
        
        # Get current items
        current_items = instance.checklistitem_set.all()
        current_item_ids = set(current_items.values_list('id', flat=True))
        
        # Update or create items
        updated_item_ids = set()
        for item_data in items_data:
            item_id = item_data.get('id')
            if item_id:
                if item_id in current_item_ids:
                    item = current_items.get(id=item_id)
                    for attr, value in item_data.items():
                        setattr(item, attr, value)
                    item.save()
                    updated_item_ids.add(item_id)
                else:
                    # If ID is provided but doesn't exist, create new item
                    ChecklistItem.objects.create(checklist=instance, **item_data)
            else:
                # If no ID is provided, create new item
                ChecklistItem.objects.create(checklist=instance, **item_data)
        
        # Delete items that are not in the updated data
        items_to_delete = current_item_ids - updated_item_ids
        instance.checklistitem_set.filter(id__in=items_to_delete).delete()
        
        return instance

    def validate(self, data):
        # Check if the collection is public and the checklist is not
        collection = data.get('collection')
        is_public = data.get('is_public', False)
        if collection and collection.is_public and not is_public:
            raise serializers.ValidationError(
                'Checklists associated with a public collection must be public.'
            )

        return data

class CollectionSerializer(CustomModelSerializer):
    adventures = AdventureSerializer(many=True, read_only=True, source='adventure_set')
    transportations = TransportationSerializer(many=True, read_only=True, source='transportation_set')
    notes = NoteSerializer(many=True, read_only=True, source='note_set')
    checklists = ChecklistSerializer(many=True, read_only=True, source='checklist_set')

    class Meta:
        model = Collection
        fields = ['id', 'description', 'user_id', 'name', 'is_public', 'adventures', 'created_at', 'start_date', 'end_date', 'transportations', 'notes', 'updated_at', 'checklists', 'is_archived', 'shared_with', 'link']
        read_only_fields = ['id', 'created_at', 'updated_at', 'user_id']

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        # Make it display the user uuid for the shared users instead of the PK
        shared_uuids = []
        for user in instance.shared_with.all():
            shared_uuids.append(str(user.uuid))
        representation['shared_with'] = shared_uuids
        return representation