from .models import *
from django.db.models import Q


def get_chat_page_content_Dict(request, receiver_id=None):
    from AuthApp.models import User

    msgs = []
    try:
        cur_receiver = User.objects.get(id=receiver_id)

        msgs = get_msgs_for_chat_w_users(request.user, cur_receiver)
        msgs.filter(receiver=request.user).update(status='seen')

    except User.DoesNotExist:
        cur_receiver = None

    receivers, unread_msgs_count = get_chat_receivers_for_user(request.user)

    Dict = {
        'cur_receiver': cur_receiver,
        'messages': msgs,
        'receivers': receivers,
        'page': 'chat',
    }
    return Dict


def get_msgs_for_chat_w_users(user1, user2):

    msgs = Message.objects.filter(
        Q(sender=user1) | Q(receiver=user1),
        Q(sender=user2) | Q(receiver=user2),
        group=None
    ).order_by('-modifiedDT')
    return msgs

def get_chat_receivers_for_user(user, cur_receiver=None):
    msgs = Message.objects.filter(
        Q(sender=user) | Q(receiver=user),
        group=None
    ).order_by('-modifiedDT')#.select_related('sender', 'receiver').values('sender', 'receiver').distinct()

    receivers_list = []

    unread_msgs_count = 0

    for msg in msgs:
        if msg.receiver not in receivers_list and msg.receiver != user:
            msg.receiver.unread_msgs_count = 0
            msg.receiver.last_msg = msg.text
            receivers_list.append(msg.receiver)
        if msg.sender not in receivers_list and msg.sender != user:
            msg.sender.unread_msgs_count = 0
            receivers_list.append(msg.sender)
            msg.sender.last_msg = msg.text

        if msg.receiver == user and (not cur_receiver or msg.sender != cur_receiver):
            if msg.status == 'sended':
                i = receivers_list.index(msg.sender)
                receivers_list[i].unread_msgs_count += 1
                unread_msgs_count += 1

    return receivers_list, unread_msgs_count

    # receivers_list.extend((item.sender for item in receivers if item.sender != user))
    # receivers_list.extend((item.receiver for item in receivers if item.receiver != user))
    #
    # return list(set(receivers_list))


def get_tickets_Dict_by_manager(user):
    tickets = MsgGroup.objects.filter(enable=True, manager=user).order_by('-modifiedDT')

    Dict = {
        'tickets': tickets,
    }
    return Dict

def get_messages_for_ticket(ticket):
    return ticket.rel_messages_for_group.filter(enable=True).order_by('-modifiedDT')


def get_ticketsDict_for_staff(user):
    Dict = {
        'tickets_wo_manager': get_tickets_wo_manager(),
        'tickets_for_manager': get_tickets_for_manager(user),
    }
    return Dict

def get_tickets_wo_manager():

    tickets = MsgGroup.objects.filter(enable=True, manager=None).order_by('-modifiedDT')

    return tickets

def get_tickets_for_manager(user):

    tickets = MsgGroup.objects.filter(enable=True, manager=user).order_by('-modifiedDT')

    return tickets