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

from django.conf import settings
from django.core.management import BaseCommand
from p4l.models import Country
from p4l.utils import show_progress
import logging
import requests

logger = logging.getLogger(__name__)

class Command(BaseCommand):
    
    query = """
PREFIX skos:<http://www.w3.org/2004/02/skos/core#>
PREFIX rdf:<http://www.w3.org/1999/02/22-rdf-syntax-ns#>
PREFIX owl:<http://www.w3.org/2002/07/owl#>
PREFIX rdfs:<http://www.w3.org/2000/01/rdf-schema#>
SELECT ?uri ?label 
WHERE {
    ?uri skos:inScheme <http://skos.um.es/unescothes/CS000/Countries> .
    ?uri skos:prefLabel|skos:label ?label .
    FILTER (?uri = $root)
}
"""
    
    def fill_label(self):
        # Loads Countries label from sparkl query
        countries = Country.objects.filter(label=None)
        total_countries = len(countries)
        writer = None
        i = 0
        found = 0
        for c in countries:
            i += 1
            res = requests.get(
                settings.SPARQL_QUERY_ENDPOINT,
                params={'query':self.query, 'timeout':10, '$root' : "<"+c.country+">"},
                headers={'accept':'application/sparql-results+json'},
            )
            if not res.ok:
                continue
            elif res.text:
                json_res = res.json()
                if 'results' in json_res and 'bindings' in json_res['results'] and len(json_res['results']['bindings'])>0:
                    # json_res['results']['bindings'] has several languages. If we find french, we save the french label.
                    # If not, we save the first one.
                    tmp_dict = {}
                    first_label = None
                    # We create a temporary dict with the lang code and the label
                    for b in json_res['results']['bindings']:
                        if 'label' in b and 'value' in b['label'] and 'xml:lang' in b['label']:
                            tmp_dict[b['label']['xml:lang']] = b['label']['value']
                            if not first_label:
                                first_label = b['label']['value']
                    if 'fr' in tmp_dict or first_label:
                        if 'fr' in tmp_dict:
                            c.label = tmp_dict['fr']
                        else:
                            c.label = first_label
                        l = c.label
                        c.save()
                        found += 1
                        writer = show_progress(i, total_countries, l, 50, writer=writer)
        print("Processing Subjects Sparql Done. %d found on %d" % (found, total_countries))

    def handle(self, *args, **options):
        self.fill_label()