add_statuses_to_real_lemmas.py 3.1 KB
#-*- coding:utf-8 -*-

#Copyright (c) 2013, Bartłomiej Nitoń
#All rights reserved.

#Redistribution and use in source and binary forms, with or without modification, are permitted provided 
#that the following conditions are met:

#    Redistributions of source code must retain the above copyright notice, this list of conditions and 
#    the following disclaimer.
#    Redistributions in binary form must reproduce the above copyright notice, this list of conditions 
#    and the following disclaimer in the documentation and/or other materials provided with the distribution.

# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED 
# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 
# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR 
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 
# TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
# POSSIBILITY OF SUCH DAMAGE.
'''
Sript for adding proper statuses to reaalized lemmas.
@author: bniton
'''

import codecs
import operator

from django.core.management.base import BaseCommand
from django.db.models import Q

from accounts.models import RealizedLemma
from dictionary.models import Lemma_Status, Lemma

class Command(BaseCommand):
    args = 'none'
    help = """
    Add proper statuses to realized lemmas.
    """

    def handle(self, **options):
        add_statuses()

def add_statuses():
    """
    Add proper statuses to realized lemmas.
    """
    for real_lemma in RealizedLemma.objects.all():
        real_lemma.counted = True
        if real_lemma.corr_frames > 0 or real_lemma.ncorr_frames > 0:
            real_lemma.status = Lemma_Status.objects.get(status=u'sprawdzone')
            real_lemma.bonus = False         
        elif real_lemma.made_frames > 0:
            real_lemma.status = Lemma_Status.objects.get(status=u'gotowe')
            real_lemma.bonus = False
        elif real_lemma.prop_frames > 0 or real_lemma.wrong_frames > 0:
            real_lemma.status = Lemma_Status.objects.get(status=u'sprawdzone')
            real_lemma.bonus = True
        else:
            real_lemma.status = Lemma_Status.objects.get(status=u'gotowe')
            real_lemma.bonus = False
        real_lemma.save()
    for lemma in Lemma.objects.filter(old=False, status__status=u'sprawdzone'):
        checked_real_lemmas = RealizedLemma.objects.filter(status__status='sprawdzone', 
                                                           lemma__entry=lemma.entry)
        if not checked_real_lemmas.exists():
            print lemma
            RealizedLemma(lemma=lemma, paid=True, counted=True,
                          status=Lemma_Status.objects.get(status=u'sprawdzone')).save()