# Note: Most of the code below is hard coded with values. Please change it
# required. Reprensented as (HC) for Hard Coded values.

from sage_days.sdi.events import send_acco_confirm_mail
from sage_days.sdi.events import send_acco_reject_mail
from sage_days.sdi.models import ParticipantInfo, RegistrantInfo


def allot():
    """Performs the allotment of accomodation based on availability for the
       participants who have requested accommodation.
    """

    # (HC)
    # Availability for number of people at various locations. Here in the
    # Order of IITB Hostel, IITB Guest house and then NITIE
    availability = [30, 20, 20]

    # (HC)
    acco_info = ParticipantInfo.objects.filter(
        participant__acco_required=True)
    acco_given = acco_info[:65]
    acco_rejected = acco_info[65:]

    for participant in acco_given:
        if availability[0] <= 0:
            break
        if participant.will_sprint == '3' or participant.will_sprint == '2':
            if participant.participant.profession.lower() == 'student':
                reg_info = participant.participant.registrantinfo_set.get()
                reg_info.status_of_accomodation = '2'
                reg_info.acco_location = '0'
                availability[0] -= 1
                reg_info.save()

    for participant in acco_given:
        if availability[1] <= 0:
            break
        if participant.participant.gender == 'F':
            reg_info = participant.participant.registrantinfo_set.get()
            reg_info.status_of_accomodation = '2'
            reg_info.acco_location = '1'
            availability[1] -= 1
            reg_info.save()
        if participant.participant.gender == 'M':
            if (participant.will_sprint == '3' or
                participant.will_sprint == '2'):
                if participant.participant.profession.lower() != 'student':
                    reg_info = participant.participant.registrantinfo_set.get()
                    reg_info.status_of_accomodation = '2'
                    reg_info.acco_location = '1'
                    availability[1] -= 1
                    reg_info.save()

    for participant in acco_given:
        if availability[2] <= 0:
            break
        if (participant.will_sprint == '1'
            and participant.participant.gender == 'M'):
            reg_info = participant.participant.registrantinfo_set.get()
            reg_info.status_of_accomodation = '2'
            reg_info.acco_location = '2'
            availability[2] -= 1
            reg_info.save()

    for participant in acco_rejected:
        reg_info = participant.participant.registrantinfo_set.get()
        reg_info.status_of_accomodation = '3'
        reg_info.save()

def send_mails():
    """Sends mails to all those who have been alloted accommodation.
    """

    hostel_participants = RegistrantInfo.objects.filter(
        status_of_accomodation=2, acco_location=0)
    ghouse_partiticpants = RegistrantInfo.objects.filter(
        status_of_accomodation=2, acco_location=1)
    nitie_participants = RegistrantInfo.objects.filter(
        status_of_accomodation=2, acco_location=2)
    acco_rejected = RegistrantInfo.objects.filter(
        status_of_accomodation=3)

    for p in hostel_participants:
        send_acco_confirm_mail(p.registrant, 'IIT Bombay Hostel',
                               '150', True)

    for p in ghouse_partiticpants:
        send_acco_confirm_mail(p.registrant, 'IIT Bombay Guest House', '300')

    for p in nitie_participants:
        send_acco_confirm_mail(p.registrant, 'NITIE House', '500')
  
    for p in acco_rejected:
        send_acco_reject_mail(p.registrant)
  
    
#65
#Participants attending sprints: 21
#Female participants attending sprints: 2
#Male participants attending sprints: 19
#Female participants may be attending sprints: 4
#Male participants may beattending sprints: 21