from django.contrib.auth.decorators import login_required
from django.shortcuts import render
from django.utils.translation import gettext as _

from connections.models import Entry, POS, Status
from syntax.models import Schema, SchemaOpinion
from semantics.models import Frame, FrameOpinion

from entries.polish_strings import POS as POS_names, STATUS, SCHEMA_OPINION, FRAME_OPINION


@login_required
def dictionary_statistics(request):
    ALL = _('wszystkie')

    # ENTRIES
    entries = Entry.objects.filter(import_error=False)
    pos_names = POS_names()
    status_names = STATUS()
    entry_statuses = list(Status.objects.all())
    all_pos = ['verb', 'noun', 'adj', 'adv']
    entry_stats = {ALL: [entries.count()] + [entries.filter(pos__tag=pos).count() for pos in all_pos]}
    used_statuses = set()
    for status in entry_statuses:
        c = entries.filter(status=status).count()
        if c:
            used_statuses.add(status)
            entry_stats[status_names[status.key]] = [entries.filter(status=status).count()] + [
                entries.filter(status=status, pos__tag=pos).count() for pos in all_pos]
    all_statuses = [ALL] + [status_names[status.key] for status in entry_statuses if status in used_statuses]
    all_pos = [ALL] + [pos_names[pos] for pos in all_pos]

    # SCHEMATA
    schemata = Schema.objects.filter(subentries__entry__import_error=False)
    schema_opinions = SCHEMA_OPINION()
    schema_stats = [(ALL, 'all', schemata.count())] + [
        (schema_opinions[opinion.key], opinion.key, schemata.filter(opinion=opinion).count()) for opinion in
        SchemaOpinion.objects.all()]

    # FRAMES
    frames = Frame.objects.all()  # filter(arguments__argument_connections__schema_connections__subentry__entry__import_error=False)
    frame_opinions = FRAME_OPINION()
    frame_stats = [(ALL, 'all', frames.count())] + [
        (frame_opinions[opinion.key], opinion.key, frames.filter(opinion=opinion).count()) for opinion in
        FrameOpinion.objects.all()]

    return render(request, 'dictionary_statistics.html',
                  {'all_statuses': all_statuses, 'all_pos': all_pos, 'entry_stats': entry_stats,
                   'schema_stats': schema_stats, 'frame_stats': frame_stats})