from collections import defaultdict
import csv
import datetime
import errno
import hmac
import hashlib
import itertools
import json
import os
import re
import requests
import sys

from django.core.management.base import NoArgsCommand, CommandError
from django.template.defaultfilters import slugify

from django_date_extensions.fields import ApproximateDate

from settings import IEBC_API_ID, IEBC_API_SECRET
from optparse import make_option

from core.models import Place, PlaceKind, Person, ParliamentarySession, Position, PositionTitle, Organisation, OrganisationKind

from iebc_api import *

data_directory = os.path.join(sys.path[0], 'kenya', '2013-election-data')

headings = ['Place Name',
            'Place Type',
            'Race Type',
            'Old?',
            'Existing Aspirant Position ID',
            'Existing Aspirant Person ID',
            'Existing Aspirant External ID',
            'Existing Aspirant Legal Name',
            'Existing Aspirant Other Names',
            'API Normalized Name',
            'API Code',
            'Action']

class Command(NoArgsCommand):
    help = 'Update the database with aspirants from the IEBC website'

    option_list = NoArgsCommand.option_list + (
        make_option('--commit', action='store_true', dest='commit', help='Actually update the database'),
        )

    def handle_noargs(self, **options):

        csv_filename = os.path.join(data_directory, 'positions-to-end-delete-and-alternative-names.csv')

        with open(csv_filename) as fp:

            reader = csv.DictReader(fp)

            for row in reader:

                alternative_names_to_add = row['Alternative Names To Add']
                if not alternative_names_to_add:
                    continue

                try:
                    position = Position.objects.get(pk=row["Existing Aspirant Position ID"])
                except Position.DoesNotExist:
                    # Probably it's already been removed by the remove duplicates script
                    continue

                if alternative_names_to_add == '[endpos]':
                    position.end_date = yesterday_approximate_date
                    maybe_save(position, **options)
                elif alternative_names_to_add == '[delpos]':
                    if options['commit']:
                        position.delete()
                else:
                    print "------------------------------------------------------------------------"
                    print alternative_names_to_add
                    names_to_add = [an.title().strip() for an in alternative_names_to_add.split(', ')]
                    for n in names_to_add:
                        person = Person.objects.get(pk=row['Existing Aspirant Person ID'])
                        person.add_alternative_name(n)
                        maybe_save(person, **options)

#         for each county, representative, ward:
#             for each contest_type:
#                 get all the current aspirants
#                 for each aspirant:
#                     find each other aspirant that has this aspirant as an alternative name
#                     (make a mapping of (person with multiple names) => all people whose names match those)
#                     for each value in that mapping, check that they have the same API CODE
#                     set the key person's API CODE
#                     check that there's no extra data attached to the values peope, then remove the position + the person



# check - if we're deleting a position, because there's already an older one there, make sure any IEBC code of the former is applied to the latter

@@ -2871,796 +2871,4 @@
 ns)%0A
-%0A#         for each county, representative, ward:%0A#             for each contest_type:%0A#                 get all the current aspirants%0A#                 for each aspirant:%0A#                     find each other aspirant that has this aspirant as an alternative name%0A#                     (make a mapping of (person with multiple names) =%3E all people whose names match those)%0A#                     for each value in that mapping, check that they have the same API CODE%0A#                     set the key person's API CODE%0A#                     check that there's no extra data attached to the values peope, then remove the position + the person%0A%0A%0A%0A# check - if we're deleting a position, because there's already an older one there, make sure any IEBC code of the former is applied to the latter%0A

Remove commented-out notes