Source code for dk.ttcal

# -*- coding: utf-8 -*-

"""Date classes (originally from TikTok).
"""

from .day import Day, Days, Today
from .duration import Duration
from .calfns import chop, isoweek
from .month import Month
from .week import Week  #, Weeks
from .year import Year


def from_idtag(idtag):
    """Return a class from idtag.
    """
    assert len(idtag) > 1
    assert idtag[0] in 'wdmy'

    return {
        'w': Week,
        'd': Day,
        'm': Month,
        'y': Year,
        }[idtag[0]].from_idtag(idtag)