utils.py

import collections
import datetime


ORDEM_REFEICAO = [
    'Desjejum', 'Colação', 'Almoço', 'Lanche', 'Refeição da Tarde',
    'Lanche - Permanência de 5 ou 6 horas',
    'Refeição',
    'Lanche - Permanência de 4 ou 8 horas',
    'Lanche Período Integral',
    'Merenda Seca', 'Merenda Especial Seca',
    'Merenda Inicial', 'Refeição - Professor', 'Pro Jovem (filhos)', 'Almoço - Professor',
    'Jantar - Professor',
    'Refeição 2', 'Refeição CEMEI', 'Lanche CEMEI', 'Sem Refeição',
    'Horário estendido', 'Lanche passeio',
    'Lanche 5h - Escola Aberta', 'Refeição - Escola Aberta',
    'Lanche 4h - Escola Aberta',
]


def sort_cardapio_por_refeicao(refeicoes_desord):
    ordenado = collections.OrderedDict()
    for ordem_refeicao in ORDEM_REFEICAO:
        for _ in refeicoes_desord:
            if (not (ordem_refeicao in ordenado)
                    and ordem_refeicao in refeicoes_desord
                    and refeicoes_desord[ordem_refeicao]):
                ordenado[ordem_refeicao] = refeicoes_desord[ordem_refeicao]
    return ordenado


def remove_refeicao_duplicada_sme_conv(refeicoes):
    retval = []
    refeicoes = sorted(refeicoes, key=lambda r: len(r['cardapio']), reverse=True)
    len_todas_as_idades = (list(refeicoes[0]['cardapio'])).__len__()
    todas_as_idades_contem_refeicao = 'Refeição' in list(refeicoes[0]['cardapio'])
    for refeicao in refeicoes:
        if len_todas_as_idades == 1 or not todas_as_idades_contem_refeicao or (len_todas_as_idades == 2 and refeicao['idade'] != 'Toda Idade'):
            retval.append(refeicao)
    return retval


def extract_digits(word):
    digits_extracted = ''.join([p for p in word if p in '0123456789']) or None
    if digits_extracted:
        return int(digits_extracted)
    return digits_extracted


def extract_chars(word):
    return ''.join([p for p in word if p not in '0123456789'])


def translate_date_month(month):
    month_dict = {
        1: 'Janeiro',
        2: 'Fevereiro',
        3: 'Março',
        4: 'Abril',
        5: 'Maio',
        6: 'Junho',
        7: 'Julho',
        8: 'Agosto',
        9: 'Setembro',
        10: 'Outubro',
        11: 'Novembro',
        12: 'Dezembro',
    }
    return month_dict[month]


def translate_date_week(day):
    week_day = {
        0: 'Segunda-feira',
        1: 'Terça-feira',
        2: 'Quarta-feira',
        3: 'Quinta-feira',
        4: 'Sexta-feira',
        5: 'Sábado',
        6: 'Domingo',
    }

    return week_day[day]


def datetime_range(start=None, end=None):
    dates = []
    if isinstance(start, str):
        start = datetime.datetime.strptime(start, '%Y%m%d').date()
    if isinstance(end, str):
        end = datetime.datetime.strptime(end, '%Y%m%d').date()
    span = end - start
    for i in range(span.days + 1):
        dates.append((start + datetime.timedelta(days=i)).strftime('%Y%m%d'))
    return dates


def limpar_cardapios(dados):
    try:
        keys_para_remover = set(dados[0]["cardapio"].keys())
        for obj in dados:
            for key, value in obj["cardapio"].items():
                if value != ['-']:
                    keys_para_remover.discard(key)
        for obj in dados:
            for key in keys_para_remover:
                obj["cardapio"].pop(key, None)
    except IndexError:
        return dados
    return dados