from django.db.models.signals import post_save

from conversations.constants import UPDATED_CONVERSATION_UID
from conversations.enums import MessageType
from conversations.models import Conversation
from conversations.receivers import handle_updated_conversation

post_save.disconnect(
    sender=Conversation,
    dispatch_uid=UPDATED_CONVERSATION_UID,
)

for conversation in Conversation.objects.all():
    candidate_messages = conversation.messages.filter(message_type=MessageType.candidate.value)
    if candidate_messages.exists():
        conversation.timestamp_first_response = candidate_messages.order_by('created').first().created
        conversation.timestamp_last_response = candidate_messages.order_by('created').last().created
    else:
        conversation.timestamp_first_response = candidate_timestamp_last_response = None
    conversation.save()

post_save.connect(
    handle_updated_conversation,
    sender=Conversation,
    dispatch_uid=UPDATED_CONVERSATION_UID,
)