licencia.py

#!/usr/bin/env python
# Author Dario Clavijo 2016
# Licence: GPLv3

from holidays import *
from workdays import *
from datetime import date
import sys

# calculamos licencia sin contar sabados ni domingos ni feriados
def lic(dstart, days, franqueo, holydays):

    r = [0, 0, 0, 0]

    r[0] = workday(dstart, days)
    r[1] = workday(dstart, days + franqueo)
    r[2] = workday(dstart, days, holydays)
    r[3] = workday(dstart, days + franqueo, holydays)

    d = [0, 0, 0, 0]

    d[0] = (r[0] - dstart).days
    d[1] = (r[1] - dstart).days
    d[2] = (r[2] - dstart).days
    d[3] = (r[3] - dstart).days

    return (r, d)


def init_holidays(year):
    pascua = easter(year)  # pascuas

    semana_santa = [0, 0, 0, 0, 0]

    semana_santa[0] = pascua - timedelta(days=6)  # lunes santo
    semana_santa[1] = pascua - timedelta(days=5)  # martes santo
    semana_santa[2] = pascua - timedelta(days=4)  # miercoles santo
    semana_santa[3] = pascua - timedelta(days=3)  # jueves santo
    semana_santa[4] = pascua - timedelta(days=2)  # viernes santo

    carnaval = [0, 0]
    carnaval[0] = pascua - timedelta(days=7) - timedelta(days=40)
    carnaval[1] = carnaval[0] + timedelta(days=1)

    return [  # feriados no laborables
        date(year=year, month=1, day=1),  # dia del transportista
        date(year=year, month=5, day=1),  # dia de los trabajadores
        date(year=year, month=7, day=18),  # dia de la independencia
        date(year=year, month=8, day=25),  # dia de la declaratoria de la independencia
        date(year=year, month=12, day=25),  # vino papa noel y se fue...
        # semana santa
        semana_santa[0],  # lunes
        semana_santa[1],  # martes
        semana_santa[2],  # miercoles
        semana_santa[3],  # jueves
        semana_santa[
            4
        ],  # viernes de la muerte, no se puede comer carne pero yo la como igual.
        carnaval[0],  # dia de las morenas carnosas bailando
        carnaval[1],  # dia de las morenas carnosas bailando
        # feriados laborables
        date(
            year=year, month=6, day=19
        ),  # el dia en que artigas se emborracho (porque era su cumplenios)
        date(year=year, month=5, day=18),  # batalla de las piedras
        date(year=year, month=4, day=19),  # desembarco de los 33 (eran 34)
        date(year=year, month=11, day=2),  # dia de los muertos
        date(
            year=year, month=10, day=12
        ),  # los espanioles descubrieron que no estaban solos
        date(year=year, month=1, day=6),  # dia mas regalos (de reyes)
    ]


def print_holidays():
    for h in holidays:
        print(h)


def display_res(dstart, days, franqueo, holydays, r, d):
    print(
        "+---------------------------------------------------------------------------------------------------+"
    )
    print("|Sin contar feriados										   |")
    print(
        "|Empieza:",
        dstart,
        ", Dias laborales:",
        days,
        ", Termina:",
        r[0],
        " Dias Totales: ",
        (d[0]),
    )
    print(
        "|Empieza:",
        dstart,
        ", Dias laborales:",
        days + franqueo,
        ", Termina:",
        r[1],
        ", Dias totales:",
        (d[1]),
    )
    print("|												   |")

    print("|Contando feriados										   |")
    print(
        "|Empieza:",
        dstart,
        ", Dias laborales:",
        days,
        ", Termina:",
        r[2],
        " Dias Totales: ",
        (d[2]),
    )
    print(
        "|Empieza:",
        dstart,
        ", Dias laborales:",
        days + franqueo,
        ", Termina:",
        r[3],
        " Dias Totales: ",
        (d[3]),
    )


def calculate_best_candidate(start_date, days, franqueo):

    start_date = datetime.strptime(start_date, "%d-%m-%Y").date()
    print("parametros:", start_date, days, franqueo)
    dstart = start_date

    last_best = 0
    hs = init_holidays(dstart.year)
    best = []
    for _ in range(0, 365):
        # dstart += timedelta(days=1)

        r, d = lic(dstart, days, franqueo, hs)
        best_d = max(d)
        if best_d > last_best:
            # if dstart >= start_date:
            last_best = best_d
            best.append((dstart, r, d))

            # best.append((dstart,r,d))
        # sorted(best)
        dstart += timedelta(days=1)

    for dstart, r, d in best:
        # print dstart,r[3],d[3]
        display_res(dstart, days, franqueo, hs, r, d)
    print(
        "+---------------------------------------------------------------------------------------------------+"
    )


if len(sys.argv) > 1:
    calculate_best_candidate(sys.argv[1], int(sys.argv[2]), int(sys.argv[3]))
else:
    print(
        "uso ./licencia.py dia_del_inicio_de_licencia dias_de_licencia horas_de_franqueo"
    )
    print("ejemplo: ./licencia.py '01-01-2016' 20 0")
    print()