import datetime
from collections import defaultdict

#from shifts import RotaBuilder, days, sites


class Worker:
    def __init__(
        self, Rota, id, name, site, grade, fte=100, nwd=None, end_date=None, oop=None
    ):
        self.id = id
        self.name = name
        self.site = site
        self.grade = grade
        self.fte = fte
        self.nwd = nwd
        self.proportion_rota_to_work = 1

        self.shift_target_number = defaultdict(int)

        days_to_work = Rota.rota_days_length

        if end_date is None:
            self.end_date = None
        else:
            self.end_date = datetime.datetime.strptime(end_date, "%Y/%m/%d").date()

            if self.end_date > Rota.rota_end_date:
                self.end_date = Rota.rota_end_date
            else:
                days_to_work = (self.end_date - Rota.start_date).days

                # add unavalabilities
                for weeks_days in Rota.weeks_days_product[days_to_work:]:
                    week, day = weeks_days
                    Rota.unavailable_to_work.add((self.id, week, day))

        if oop is not None:
            start_oop, end_oop = oop
            start_oop_date = datetime.datetime.strptime(start_oop, "%Y/%m/%d").date()
            end_oop_date = datetime.datetime.strptime(end_oop, "%Y/%m/%d").date()

            if end_oop_date > Rota.rota_end_date:
                end_oop_date = Rota.rota_end_date

            oop_length = (end_oop_date - start_oop_date).days
            days_to_work = days_to_work - oop_length

            days_until_oop = (start_oop_date - Rota.start_date).days

            for weeks_days in Rota.weeks_days_product[
                days_until_oop : days_until_oop + oop_length
            ]:
                week, day = weeks_days
                Rota.unavailable_to_work.add((self.id, week, day))

        self.proportion_rota_to_work = days_to_work / Rota.rota_days_length

        # We had to adjust the full time equivalent for people who CCT / leave the rota early
        self.fte_adj = self.fte * self.proportion_rota_to_work

    def __lt__(self, other):
        return (self.site, self.grade, self.name) < (
            other.site,
            other.grade,
            other.name,
        )

    def get_details(self):
        return "{} {} {}".format(self.id, self.site[0], self.grade)