fga-gpp-mds/2017.2-QueroCultura

from django.shortcuts import render
from datetime import datetime
from .api_connections import RequestEventsRawData
from .api_connections import RequestEventsInPeriod
from .api_connections import EmptyRequest
from .models import EventData
from .models import EventLanguage
from .models import LastUpdateEventDate
from project_indicators.views import clean_url
from quero_cultura.views import ParserYAML
from mixed_indicators.views import populate_mixed_data
from quero_cultura.views import get_metabase_url
from celery.decorators import task


DEFAULT_INITIAL_DATE = "2012-01-01 00:00:00.000000"
SP_URL = "http://spcultura.prefeitura.sp.gov.br/api/"
ESTADO_SP_URL = "http://estadodacultura.sp.gov.br/api/"
DEFAULT_YEAR = 2013
CURRENT_YEAR = datetime.today().year + 1

# Get graphics urls from metabase
# To add new graphis, just add in the metabase_graphics variable
view_type = "question"
metabase_graphics = [{'id': 1, 'url': get_metabase_url(view_type, 14, "true")},
                     {'id': 2, 'url': get_metabase_url(view_type, 15, "true")},
                     {'id': 3, 'url': get_metabase_url(view_type, 52, "true")},
                     {'id': 4, 'url': get_metabase_url(view_type, 17, "true")},
                     {'id': 5, 'url': get_metabase_url(view_type, 51, "true")}]

detailed_data = [{'id': 1, 'url': get_metabase_url("dashboard", 4, "false")}]


page_type = "Eventos"
graphic_type = 'events_graphic_detail'
page_descripition = "Evento cultural é a parte mínima de um projeto cultural,"\
                    + " continuado ou não, multidisciplinar ou setorial, "\
                    + "que tenha objetivo definido e impacto imediato, os "\
                    + "gráficos abaixo geram indicadores baseados nos dados "\
                    + "de Eventos Culturais disponiveis na plataforma"


def index(request):
    return render(request, 'quero_cultura/indicators_page.html',
                  {'metabase_graphics': metabase_graphics,
                   'detailed_data': detailed_data,
                   'page_type': page_type,
                   'graphic_type': graphic_type,
                   'page_descripition': page_descripition})


def graphic_detail(request, graphic_id):
    try:
        graphic = metabase_graphics[int(graphic_id) - 1]
    except IndexError:
        return render(request, 'quero_cultura/not_found.html')
    return render(request, 'quero_cultura/graphic_detail.html',
                  {'graphic': graphic})


@task(name="load_events")
def populate_event_data():
    if len(LastUpdateEventDate.objects) == 0:
        LastUpdateEventDate(DEFAULT_INITIAL_DATE).save()

    size = LastUpdateEventDate.objects.count()
    last_update = LastUpdateEventDate.objects[size - 1].create_date
    LastUpdateEventDate(str(datetime.now())).save()

    parser_yaml = ParserYAML()
    urls = parser_yaml.get_multi_instances_urls

    for url in urls:
        if (last_update == DEFAULT_INITIAL_DATE) and (url == SP_URL or url == ESTADO_SP_URL):
            request = EmptyRequest()
            for year in range(DEFAULT_YEAR, CURRENT_YEAR):
                single_request = RequestEventsInPeriod(year, url)
                request.data += single_request.data
            request = request.data
        else:
            request = RequestEventsRawData(last_update, url).data

        new_url = clean_url(url)
        for event in request:
            date = event["createTimestamp"]['date']
            if str(event['classificacaoEtaria']) != '':
                EventData(new_url, str(
                    event['classificacaoEtaria']).title(), event['occurrences'], date).save()
            for language in event["terms"]["linguagem"]:
                EventLanguage(new_url, language).save()

    populate_mixed_data(last_update)