Source code for ted17.dategr

# -*- coding: utf-8 -*-
"""
dategr - A library to work with Greek Format Dates
--------------------------------------------------
dategr is a libray that provides a set of functions to work with Greek
Formatted Dates.
"""
import datetime


def date2gr(date, removezero=False):
    '''
    :param date: iso date 'yyyy-mm-dd'
    :type date: iso_date
    :param removezero: Month, Day without trailing zeros (If True '2017-01-09'
     => '9/1/2017'. If False '2017-01-09' => '09/01/2017')
    :return: 'dd/mm/yyyy'

    Example::

        >>> import dategr
        >>> dategr.date2gr('2017-01-15)
        '15/01/2017'
        >>> dategr.dat2gr('2017-01-15, True)
        '15/1/2017'
    '''
    def remove_zero(stra):
        """Remove trailing zeros"""
        if len(stra) > 1 and stra[0] == '0':
            return stra[1:]
        else:
            return stra
    if not date:
        return ''
    year, month, day = date.split('-')
    if removezero:
        return '%s/%s/%s' % (remove_zero(day), remove_zero(month), year)
    else:
        return '%s/%s/%s' % (day, month, year)


def grdate2iso(greek_date):
    """Return iso date (yyy-mm-dd) from greek date(dd/mm/yyyy)

    :param greek_date: Date of the form 'dd/mm/yyyy'
    :return: Iso Date ('yyyy-mm-dd')
    """
    assert len(greek_date) in (8, 9, 10)
    assert len(greek_date.replace('/', '')) + 2 == len(greek_date)
    day, month, year = greek_date.split('/')
    if len(day) == 1:
        day = '0%s' % day
    if len(month) == 1:
        month = '0%s' % month
    return '%s-%s-%s' % (year, month, day)


def getymd_from_iso(isodate):
    '''Year, month, day From isodate

    :param isodate: Date of the form 'yyyy-mm-dd'
    :return: int year, int month, int day
    '''
    assert isodate[4] == '-'
    assert isodate[7] == '-'
    year, month, day = isodate.split('-')
    return int(year), int(month), int(day)


def saizon(iso_date, startmonth=10):
    '''Find the yearfrom-yearto period of iso_date

    :param iso_date: Date of the form 'yyyy-mm-dd'
    :param startmonth: int month value to split year
    :return: 'year-year_next(startmonth)'
    '''
    startmonth = int(startmonth)  # Make sure startmonth is integer
    assert startmonth <= 12
    year, month, _ = getymd_from_iso(iso_date)
    if int(month) >= startmonth:
        return '%s-%s(%s)' % (year, year+1, startmonth)
    else:
        return '%s-%s(%s)' % (year-1, year, startmonth)


def week(iso_date):
    '''Evaluate year-weeknumber of iso_date

    :param iso_date: Date of the form 'yyyy-mm-dd'
    :return: year-weekNumber
    '''
    year, month, day = getymd_from_iso(iso_date)
    week_number = '%s' % datetime.date(year, month, day).isocalendar()[1]
    if len(week_number) == 1:
        week_number = '0' + week_number
    return '%sw%s' % (year, week_number)


def period(iso_date, diast=2):
    '''Find the timespace iso_dete belongs

    :param iso_date: Date of the form 'yyyy-mm-dd'
    :param diast: Time space
    :return: 'year-diast-diastv'
    '''
    assert diast in (2, 3, 4, 6)
    dval = {2: u'οΔίμ', 3: u'οΤρίμ', 4: u'οΤετρ', 6: u'οΕξάμ'}
    year, month, date = getymd_from_iso(iso_date)
    imonth = int(month)
    aperiod = imonth // diast
    ypoloipo = imonth % diast
    if ypoloipo > 0:
        aperiod += 1
    return '%s-%s%s' % (year, aperiod, dval[diast])


def group_selector(iso_date, per='m'):
    """Selector

    :param iso_date: Iso Date to transform
    :param per: Text value according to following table

    +-----+-----------------+
    | per |  Time Interval  |
    +=====+=================+
    | m   | month           |
    +-----+-----------------+
    | d   | full date       |
    +-----+-----------------+
    | y   | year            |
    +-----+-----------------+
    | m2  | Βimonthly       |
    +-----+-----------------+
    | m3  | Quarter         |
    +-----+-----------------+
    | m4  | Quadrimester    |
    +-----+-----------------+
    | m6  | Semester        |
    +-----+-----------------+
    | w   | Week            |
    +-----+-----------------+
    | s   | Season          |
    +-----+-----------------+
    """
    if per == 'd':  #: Όλη η ημερομηνία
        dat = iso_date[:10]
    elif per == 'm':  #: Έτος + μήνας
        dat = iso_date[:7]
    elif per == 'y':  #: Έτος μόνο
        dat = iso_date[:4]
    elif per == 'm2':
        dat = period(iso_date, 2)
    elif per == 'm3':
        dat = period(iso_date, 3)
    elif per == 'm4':
        dat = period(iso_date, 4)
    elif per == 'm6':
        dat = period(iso_date, 6)
    elif per == 'w':
        dat = week(iso_date)
    elif per == 's':
        dat = saizon(iso_date)
    else:  # Default is Year
        dat = iso_date[:4]
    return dat