carom-server/billard/tasks.py

from __future__ import absolute_import, unicode_literals

import logging
import billard.utils as utils
from celery import shared_task

from billard.models import LocationData, Client, Accounting

log = logging.getLogger(__name__)

@shared_task
def process_location_data():
    data = LocationData.objects.filter(processed=False)
    for ld in data:
        cli = Client.objects.filter(uuid=ld.client_id, desks__desk_no=ld.desk_no)
        if cli.count() < 1:
            ld.processed = True
            ld.error_msg = 'No client object found. Stopp processing! {}, {}'.format(ld.client_id, ld.desk_no)
            ld.save()
            log.error(ld.error_msg)
        else:
            cli = cli[0]
            desk = cli.desks.filter(desk_no=ld.desk_no, enabled=True)
            if desk.count() != 1:
                ld.processed = True
                ld.error_msg = 'No desk object found. Stopp processing! {}, {}'.format(ld.client_id, ld.desk_no)
                ld.save()
                log.error(ld.error_msg)
            desk = desk[0]
            ac = desk.accounting_set.order_by('time_from').reverse()
            if ld.on_off:
                acc = None
                if ac.count() > 0 and ac[0].time_to is None:
                    log.error('Vorheriges Accounting nicht abgeschlossen: Desk_id {}, Accounting_id {}'
                              .format(desk.id, ac[0].id))
                    acc = ac[0]
                if acc is None:
                    acc = Accounting(
                        desk=desk,
                        time_from=ld.tst,
                    )
                acc.save()
                ld.delete()
            else:
                if len(ac) > 0:
                    acc = ac[0]
                    acc.time_to = ld.tst
                    acc.prize, acc.prize_normal, acc.prize_hh = utils.get_prize_for(
                        start=acc.time_from,
                        end=ld.tst,
                        pph=desk.prize,
                        hh_start=cli.location.happy_hour_start,
                        hh_end=cli.location.happy_hour_end,
                        pphh=desk.prize_hh,
                    )
                    acc.save()
                    ld.delete()
                else:
                    ld.processed = True
                    ld.error_msg = 'No existing accountings found. Stopp processing! {}, {}'.format(ld.client_id, ld.desk_no)
                    ld.save()
                    log.error(ld.error_msg)