Source code for tikon.estruc.tiempo

import math as mat
from datetime import date, datetime, timedelta

import numpy as np
import pandas as pd


class Tiempo(object):
    def __init__(símismo, n_días, f_inic=None, paso=1):
        símismo._f_inic = _gen_fecha(f_inic)
        símismo._día = 0
        símismo.paso = paso
        símismo._n_días = n_días
        símismo.eje = EjeTiempo(días=range(0, mat.ceil(n_días / paso) + 1, paso), f_inic=f_inic)

    def n_pasos(símismo):
        return len(símismo.eje)

    def avanzar(símismo):
        símismo._día += símismo.paso
        return símismo._día <= símismo._n_días

    def fecha(símismo):
        if símismo._f_inic is not None:
            return símismo._f_inic + timedelta(days=símismo._día)

    def día(símismo):
        return símismo._día

    def índices(símismo, t):
        return símismo.eje.índices(t)

    def reinic(símismo):
        símismo._día = 0

    def __len__(símismo):
        return len(símismo.eje)


class EjeTiempo(object):
    def __init__(símismo, días, f_inic=None):
        if not isinstance(días, np.ndarray):
            días = np.array(días)
        símismo.días = días
        símismo.f_inic = f_inic

    def índices(símismo, t):
        if símismo.f_inic is t.f_inic is None:
            dif = 0
        else:
            dif = (t.f_inic - símismo.f_inic).days
        return t.días - dif

    def vec(símismo):
        if símismo.f_inic:
            return pd.to_timedelta(símismo.días, unit='D') + pd.to_datetime(símismo.f_inic)
        else:
            return np.array(símismo.días)

    def cortar(símismo, eje):
        índs_eje = símismo.índices(eje)
        máx = índs_eje.max()
        mín = índs_eje.min()

        días = símismo.días
        return EjeTiempo(días=días[np.logical_and(días >= mín, días <= máx)], f_inic=símismo.f_inic)

    def a_dic(símismo):
        return {'días': símismo.días, 'f_inic': símismo.f_inic.strftime('%Y-%m-%d') if símismo.f_inic else None}

    @classmethod
    def de_dic(cls, dic):
        return cls(días=dic['días'],
                   f_inic=datetime.strptime(dic['f_inic'], '%Y-%m-%d').date() if dic['f_inic'] else None)

    def __len__(símismo):
        return len(símismo.días)


def _gen_fecha(f):
    if f is None:
        return
    elif isinstance(f, str):
        return datetime.strptime(str, '%Y-%m-%d')
    elif isinstance(f, date):
        return f
    elif isinstance(f, datetime):
        return f.date()
    else:
        raise TypeError(type(f))