fga-gpp-mds/2017.2-QueroCultura

from datetime import datetime
from django.shortcuts import render
from .api_connections import RequestSpacesRawData
from .models import SpaceData
from .models import LastUpdateDate
from project_indicators.views import clean_url
from quero_cultura.views import ParserYAML
from quero_cultura.views import get_metabase_url
from celery.decorators import task
from .models import OccupationArea


DEFAULT_INITIAL_DATE = "2012-01-01 00:00:00.000000"

# 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, 2, "true")},
                     {'id': 2, 'url': get_metabase_url(view_type, 4, "true")},
                     {'id': 3, 'url': get_metabase_url(view_type, 3, "true")},
                     {'id': 4, 'url': get_metabase_url(view_type, 7, "true")},
                     {'id': 5, 'url': get_metabase_url(view_type, 6, "true")}]


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

page_type = "Espaços"
graphic_type = 'space_graphic_detail'
page_descripition = "Espaço cultural é qualquer lugar que possa ser "\
                    + "identificado como um ponto referenciado para a criação"\
                    + " formação e a fruição cultural, os gráficos abaixo "\
                    + "mostram indicadores gerados a partir dos dados"\
                    + " de Espaços Culturais contidos 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_spaces")
def populate_space_data():
    if len(LastUpdateDate.objects) == 0:
        LastUpdateDate(DEFAULT_INITIAL_DATE).save()

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

    parser_yaml = ParserYAML()
    urls = parser_yaml.get_multi_instances_urls

    for url in urls:
        request = RequestSpacesRawData(last_update, url).data
        new_url = clean_url(url)
        for space in request:
            date = space["createTimestamp"]['date']
            SpaceData(new_url, str(space['name']), date,
                      str(space['type']['name'])).save()
            for area in space["terms"]["area"]:
                OccupationArea(new_url, area).save()