from rest_framework import serializers
from .models import Ticket, Message, Attachment
from django.utils.timezone import localtime
from account.serializers import ProfileSerializer

class AttachmentSerializer(serializers.ModelSerializer):
    file = serializers.FileField(write_only=True)
    link = serializers.SerializerMethodField()

    class Meta:
        model = Attachment
        fields = ['id', 'name', 'file', 'link' , 'created_at', 'size']
        read_only_fields = ('size', 'name', )
        
    def get_link(self, obj):
        request = self.context.get('request')
        if request is not None:
            return request.build_absolute_uri(obj.file.url)
        return obj.file.url

class MessageSerializer(serializers.ModelSerializer):
    class Meta:
        model = Message
        exclude = ('sender', )
        extra_kwargs = {'ticket': {'write_only': True}}

class MessageForTicketSerializer(serializers.ModelSerializer):
    class Meta:
        model = Message
        exclude = ('sender', 'ticket')
class TicketSerializer(serializers.ModelSerializer):
    status = serializers.SerializerMethodField()
    ticket_category = serializers.SerializerMethodField()
    messages = MessageSerializer(many=True, read_only=True)
    message = MessageForTicketSerializer(write_only=True)
    class Meta:
        model = Ticket
        exclude = ('customer', 'admin')
        read_only_fields = ('status',)

    def create(self, validated_data):
        message = validated_data.pop('message', None)
        ticket = super().create(validated_data)  
        print(f'fck this shit id {ticket.pk}')
        message['ticket'] = ticket.pk
        message_obj = MessageSerializer(data=message)
        if message_obj.is_valid():
            message_obj.save(sender=ticket.customer)
            return ticket
        else:
            raise Exception('bullshit data for message')
    def get_status(self, obj):
        return obj.get_status_display()

    def get_ticket_category(self, obj):
        return obj.get_ticket_category_display()

class TicketListSerializer(serializers.ModelSerializer):
    status = serializers.SerializerMethodField()
    ticket_category = serializers.SerializerMethodField()
    class Meta:
        model = Ticket
        exclude = ('customer', 'admin', )
        read_only_fields = ('status',)
    def get_status(self, obj):
        return obj.get_status_display()

    def get_ticket_category(self, obj):
        return obj.get_ticket_category_display()