# -*- coding: utf-8 -*-
from local_conf import LLL1
import importlib
oahpa_module = importlib.import_module(LLL1+'_oahpa')

from django.core.management.base import BaseCommand, CommandError

# from_yaml(cls, loader, node)

from optparse import make_option

import sys

Form = oahpa_module.drill.models.Form
Word = oahpa_module.drill.models.Word
Tag = oahpa_module.drill.models.Tag


def install_file(filename, pos):
    from collections import defaultdict
    words_to_install = defaultdict(list)
    with open(filename, 'r') as F:
        for l in F.readlines():
            _tag, _, form = l.strip().partition('\t')
            lemma, _, tag = _tag.partition('+')

            if lemma not in words_to_install:
                words_to_install[lemma] = {}

            if tag not in words_to_install[lemma]:
                words_to_install[lemma][tag] = []

            words_to_install[lemma][tag].append(form)

    from itertools import izip_longest

    for lemma, forms in words_to_install.iteritems():
        print 'lemma: ', lemma
        ws = Word.objects.filter(lemma=lemma, pos=pos)
        try:
            w = ws[0]
            for tag, wfs in forms.iteritems():
                fs = Form.objects.filter(word__lemma=lemma, tag__string=tag)
                fs.delete()
                print '  ', tag
                last_db = False
                t, _c = Tag.objects.get_or_create(string=tag)
                if _c:
                    t.save()
                for new_form in wfs:
                    print '    ', new_form
                    new = Form.objects.create(word=w,
                                        tag=t,
                                        fullform=new_form,)
                    new.save()
        except IndexError:
            print 'error:', IndexError





# # #
#
#  Command class
#
# # #

def mergetags(tfilter=False):
    if tfilter:
        qset = Tag.objects.filter(string=tfilter)
    else:
        qset = Tag.objects.all()

    strings = qset.values_list('string', flat=True)
    strings = list(set(strings))

    print 'Merging:'
    for string in strings:
        tag = Tag.objects.filter(string=string)

        if tag.count() > 1:
            print 'Merging conflict in %s' % tag[0].string
            merge(tag)


class Command(BaseCommand):
    args = '--tagelement'
    help = """
    Strips tags of an element and then merges them all.
    """
    def add_arguments(self, parser):
	parser.add_argument(
		"-f", "--filename",
            	dest="filename",
            	default=False,
            	help="Static file to read from")
	parser.add_argument(
            	"-p", "--pos",
                dest="pos",
                default=False,
         	help="Part of speech")


    def handle(self, *args, **options):
        import sys, os

        filename = options['filename']
        pos = options['pos']

        install_file(filename=filename, pos=pos)