# -*- coding: utf-8 -*-
'''
Importe des codes INSEE et les coordonnées géographiques d'un fichier CSV.
Crée des objects :class:`hdalab.models.dataviz.InseeCoords`.

Attention cette commande de remplace pas ni n'efface de données existante dans la base (la clef étant le numéro INSEE).

**Usage**: ``django-admin import_insee_csv <chemin_vers_fichier_csv>``

'''
from django.core.management.base import BaseCommand, CommandError
from SPARQLWrapper import SPARQLWrapper, JSON
from hdalab.models import InseeCoords
import json
import csv
import re
import sys

class Command(BaseCommand):
    '''
    Command to export tags
    '''
    args = '<path_to_csv_file>'
    options = ''
    help = """Imports Insee codes and geographic coordinates from a csv file"""

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

        if len(args) == 0 or not args[0]:
            raise CommandError("Give a CSV File to import")

        filename = args[0]

        csvfile = open(filename, "rb")
        dialect = csv.Sniffer().sniff(csvfile.read(1024))
        csvfile.seek(0)
        reader = csv.reader(csvfile, dialect)
        fieldstoget = [ 'ville', 'insee', 'latitude', 'longitude' ]

        for i,line in enumerate(reader):
            if i == 0:
                fields = {}
                minlength = 0
                for j,field in enumerate(line):
                    for fieldname in fieldstoget:
                        if re.search('(?i)%s' % fieldname, field):
                            fields[fieldname] = j
                            minlength = max(j,minlength)
            else:
                if len(line) > minlength:
                    rawdata = dict([(k,line[v].strip()) for k,v in fields.iteritems()])
                    #print "Processing line %d" % i
                    #print rawdata
                    try:
                        latitude = float(rawdata['latitude'].replace(',','.'))
                        longitude = float(rawdata['longitude'].replace(',','.')) if rawdata['longitude'] != '-' else 0
                        insee = int(rawdata['insee'])
                        ville = unicode(rawdata['ville'], 'iso-8859-1')

                        InseeCoords.objects.get_or_create(insee=insee, city_name=ville, latitude=latitude, longitude=longitude)

                    except:
                        print line, "Error :", sys.exc_info()[1]

        csvfile.close()