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

'''

Created on Feb 22, 2012

@author: raphv

'''

from django.core.management.base import NoArgsCommand

from django.core.management.color import no_style

from hdabo.utils import show_progress

from hdabo.models import Tag

from hdalab.models import Country, GeoInclusion

from SPARQLWrapper import SPARQLWrapper, JSON

import re

class Command(NoArgsCommand):

    def handle_noargs(self, **options):

        self.style = no_style()

        qs = Tag.objects.filter(category__label="Localisation").exclude(dbpedia_uri = None)

        total = qs.count()

        endpoint = SPARQLWrapper("http://dbpedia.org/sparql")

        endpoint.setReturnFormat(JSON)

        sparqltext = """

            SELECT ?resource WHERE {

             { <%s> ?resource <http://dbpedia.org/ontology/Country> . }

            UNION

             { <%s> <http://dbpedia.org/ontology/country> ?resource . }

            }

        """

        resourceprefix = "http://dbpedia.org/resource/"

        identityuri = "http://www.w3.org/1999/02/22-rdf-syntax-ns#type"

        writer = None

        for i,tag in enumerate(qs):

            endpoint.setQuery(sparqltext % (tag.dbpedia_uri, tag.dbpedia_uri))

            results = endpoint.query().convert()['results']['bindings']

            if len(results) == 1: # We don't want places located in multiple countries

                resourceuri = results[0]['resource']['value']

                if re.match(resourceprefix, resourceuri):

                    countrytxt = re.findall('([^/]+$)', resourceuri)[0]

                    country, created = Country.objects.get_or_create(dbpedia_uri=resourceuri)

                    GeoInclusion.objects.get_or_create(tag=tag, country=country)

                if resourceuri == identityuri:

                    countrytxt = '<is a country>'

                    country, created = Country.objects.get_or_create(dbpedia_uri=tag.dbpedia_uri)

                    GeoInclusion.objects.get_or_create(tag=tag, country=country)

            else:

                countrytxt = '<unknown>'

            writer = show_progress(i+1, total, '%s => %s'%(tag.label, countrytxt), 50, writer)