from django.conf import settings #@UnresolvedImport

from ldt.ldt_utils.models import Segment, ContentStat, POL_INDICES

import datetime

from django.utils import timezone

def update_stat_content(content):

    nb_division = settings.DIVISIONS_FOR_STAT_ANNOTATION

    segments = Segment.objects.filter(content=content).select_related('project_obj').only('start_ts','duration','polemics','project_obj__modification_date') #@UndefinedVariable

    buckets = [0] * nb_division 

    size_division = content.duration / nb_division

    limits = [x * size_division for x in range(nb_division+1)]

    nb_annotation = segments.count()

    contentstat, created = ContentStat.objects.get_or_create(content=content) #@UnusedVariable @UndefinedVariable

    pol_stats = [0]*len(POL_INDICES)

    min_ts = datetime.datetime.min

    if settings.USE_TZ:

        min_ts = min_ts.replace(tzinfo=timezone.utc)

    min_update_ts = min_ts #@UndefinedVariable

    for segment in segments:

        begin = segment.start_ts

        end = segment.start_ts + segment.duration

        if segment.is_polemic('OK'):

            pol_stats[POL_INDICES['pol_positive']] += 1

        if segment.is_polemic('KO'):

            pol_stats[POL_INDICES['pol_negative']] += 1

        if segment.is_polemic('Q'):

            pol_stats[POL_INDICES['pol_question']] += 1

        if segment.is_polemic('REF'):

            pol_stats[POL_INDICES['pol_reference']] += 1

        buckets = find_buckets(buckets, limits, begin, end)

        if segment.project_obj is not None and segment.project_obj.modification_date > min_update_ts:

            min_update_ts = segment.project_obj.modification_date

    contentstat.polemics_volume = pol_stats

    contentstat.annotation_volume = buckets

    contentstat.nb_annotations = nb_annotation

    contentstat.last_annotated = min_update_ts if min_update_ts > min_ts else content.update_date #@UndefinedVariable

    contentstat.save()

def update_stat_project(project, contents_only=[]):

    if contents_only:

        contents = contents_only

    else:

        contents = project.contents.all()

    for c in contents:

        update_stat_content(c)

    return True

def add_annotation_to_stat(content, begin, end):

    nb_division = settings.DIVISIONS_FOR_STAT_ANNOTATION

    size_division = content.duration / nb_division

    limits = [x * size_division for x in range(nb_division+1)]    

    buckets = find_buckets(content.annotation_volume, limits, begin, end)

    contentstat, created = ContentStat.objects.get_or_create(content=content) #@UnusedVariable @UndefinedVariable

    contentstat.annotation_volume = buckets

    contentstat.last_annotated = timezone.now()

    contentstat.save()        

def get_string_from_buckets(buckets):

    return ','.join([str(b) for b in buckets])

def find_buckets(buckets, limits, begin, end):

    if len(buckets)+1 != len(limits):

        raise ValueError("There should be as many buckets as those defined by limits.")

    has_started = False  

    for i in range(len(limits)-1):

        if not has_started:

            if limits[i] <= begin and begin <= limits[i+1]:

                buckets[i] += 1

                has_started = True

                #print "Starts after timecode %s" % limits[i]

        elif limits[i] > end:

            #print "Ends before timecode %s" % limits[i]

            break

        else:

            buckets[i] += 1

    return buckets