#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2017  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.

import csv
import sys

from django.core.management.base import BaseCommand
from django.contrib.gis.geos import GEOSGeometry, Point

from ishtar_common.models import Town


class Command(BaseCommand):
    help = 'Import GEOFLA csv'

    def add_arguments(self, parser):
        parser.add_argument('csv_file')
        parser.add_argument(
            '--year', type=int, default=2014, dest='year',
            help='Default year to affect to the town')
        parser.add_argument(
            '--quiet', dest='quiet', action='store_true',
            help='Quiet output')
        parser.add_argument(
            '--srid', type=int, default=2154, dest='srid',
            help='SRID uses. Default: 2154.')

    def handle(self, *args, **options):
        csv_file = options['csv_file']
        default_year = options['year']
        srid = options['srid']
        quiet = options['quiet']
        if not quiet:
            sys.stdout.write('* using year {} as a default\n'.format(
                default_year))
            sys.stdout.write('* Opening file {}\n'.format(csv_file))
        nb_created, nb_changed = 0, 0
        with open(csv_file, 'rb') as csvfile:
            reader = csv.DictReader(csvfile)
            for idx, row in enumerate(reader):
                if not quiet:
                    sys.stdout.write('Processing town %d.\r' % (idx + 1))
                    sys.stdout.flush()
                num_insee = row['INSEE_COM']
                if len(num_insee) < 5:
                    num_insee = '0' + num_insee
                q = Town.objects.filter(numero_insee=num_insee)
                created = False
                if q.count():
                    if q.filter(year=default_year).count():
                        town = q.filter(year=default_year).all()[0]
                    else:
                        town = q.order_by('-year').all()[0]
                else:
                    created = True
                    nb_created += 1
                    town = Town(name=row['NOM_COM_M'],
                                numero_insee=num_insee)
                geom = row['wkt_geom'].upper()
                if 'MULTI' not in geom:
                    geom = geom.replace('POLYGON', 'MULTIPOLYGON(') + ')'
                town.limit = GEOSGeometry(geom, srid=srid)
                if 'X_CENTROID' in row:
                    town.center = Point(float(row['X_CENTROID']),
                                        float(row['Y_CENTROID']), srid=srid)
                else:
                    town.center = None
                if not town.year and default_year:
                    town.year = default_year
                if 'SUPERFICIE' in row:
                    town.surface = row['SUPERFICIE']
                else:
                    town.surface = None
                if not created:
                    nb_changed += 1
                town.save()
        if quiet:
            return
        sys.stdout.write('\n* {} town created'.format(nb_created))
        sys.stdout.write('\n* {} town changed\n'.format(nb_changed))
        sys.stdout.flush()