costs.py

import numpy as np
import pandas as pd

import config

config.salary_per_department


def gship_tuition(date_range, instate_date):
    costs = []
    for one_date in date_range:
        month_cost = config.gship_tuition["Gship"]

        if one_date.month not in [6, 7, 8]:  # Summer
            month_cost += config.gship_tuition["Instate"]
            if one_date < instate_date:
                month_cost += config.gship_tuition["Non-resident"]
        costs.append(month_cost)

    return costs


def salary(date_range, department, overhead, candidacy=None):
    if candidacy is None:
        candidacy = config.salary_per_department[department]["candidacy year"]
    candidacy = date_range[0] + pd.to_timedelta(f"{candidacy}Y")
    costs = []
    for one_date in date_range:
        if one_date < candidacy:
            month_cost = config.salary_per_department[department]["pre-candidacy"]
        else:
            month_cost = config.salary_per_department[department]["post-candidacy"]

        if overhead:
            month_cost *= 1.555

        costs.append(month_cost)
    return costs


def student(student_dict):
    date_range = pd.date_range(
        start=student_dict["start"], end=student_dict["end"], freq="M"
    )
    instate_date = student_dict["instate"]
    overhead = student_dict["overhead"]
    candidacy = student_dict["candidacy"]
    department = student_dict["department"]
    return [
        a + b
        for a, b in zip(
            gship_tuition(date_range, instate_date),
            salary(date_range, department, overhead, candidacy),
        )
    ]


def sra(date_range):
    return [config.salary["SRA"] for _ in date_range]


def over_date_range(start, end, start_member, end_member, costs_member):
    costs_zeros_start = [0] * len(
        pd.date_range(start=start, end=start_member, freq="M")
    )
    costs_zeros_end = [0] * len(pd.date_range(start=end_member, end=end, freq="M"))
    costs = costs_zeros_start + costs_member + costs_zeros_end
    return costs