carom-server/billard/models.py

import uuid
from django.db import models
from django.contrib.auth.models import User
from datetime import datetime, timezone
from . import utils
from . import tasks
from django.contrib.auth.models import User
from django.db.models.signals import post_save


class LocationData(models.Model):
    location_id = models.UUIDField(blank=False, null=False, verbose_name="Standort-ID")
    table_no = models.IntegerField(blank=False, null=False, verbose_name="Tischnummer")
    tst = models.DateTimeField(blank=False, null=False, verbose_name="Zeitstempel")
    on_off = models.BooleanField(blank=False, null=False, verbose_name="Ein/Ausgebucht")
    processed = models.BooleanField(default=False, verbose_name="Verarbeitet")
    error_msg = models.TextField(blank=True, null=True, verbose_name="Fehlermeldung")

    def __str__(self):
        return str(self.location_id)

    class Meta:
        verbose_name = "Standortlog"
        verbose_name_plural = "Standortlogs"

class Location(models.Model):
    users = models.ManyToManyField(User, related_name='locations', verbose_name="Benutzer")
    code = models.CharField(max_length=16, unique=True, verbose_name="Code")
    name = models.CharField(max_length=64, unique=True, verbose_name="Name")
    street = models.CharField(max_length=64, blank=True, null=True, verbose_name="Straße")
    plz = models.CharField(max_length=8, blank=True, null=True, verbose_name="Postleitzahl")
    city = models.CharField(max_length=64, blank=True, null=True, verbose_name="Stadt")
    phone = models.CharField(max_length=64, blank=True, null=True, verbose_name="Telefon")
    email = models.EmailField(blank=True, null=True, verbose_name="Email")
    url = models.URLField(blank=True,null=True, verbose_name="URL")

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = "Standort"
        verbose_name_plural = "Standorte"


class Client(models.Model):
    uuid = models.UUIDField(unique=True, default=uuid.uuid4, verbose_name="Identifier")
    location = models.ForeignKey(Location, verbose_name="Standort")
    desk1_enable = models.BooleanField()
    desk1_name = models.CharField(max_length=32, blank=True, null=True)
    desk1_prize_nt = models.DecimalField(max_digits=4, decimal_places=2, blank=True, null=True)
    desk1_prize_ht = models.DecimalField(max_digits=4, decimal_places=2, blank=True, null=True)
    desk2_enable = models.BooleanField()
    desk2_name = models.CharField(max_length=32, blank=True, null=True)
    desk2_prize_nt = models.DecimalField(max_digits=4, decimal_places=2, blank=True, null=True)
    desk2_prize_ht = models.DecimalField(max_digits=4, decimal_places=2, blank=True, null=True)

    def accounting_for(self, desk_no, pht, pnt):
        t = Accounting.objects.filter(client_id=self.id, desk_no=desk_no)[:3][::-1]
        if t.__len__() > 0:
            a = t[t.__len__() - 1]
            if a.time_to is None:
                prize = utils.get_prize_for(start=a.time_from, end=datetime.now(timezone.utc), pph=pht)
                if prize != a.prize:
                    a.prize = prize
        return t

    def accounting_1(self):
        return self.accounting_for(1, self.desk1_prize_ht, self.desk1_prize_nt)

    def accounting_2(self):
        return self.accounting_for(2, self.desk2_prize_ht, self.desk2_prize_nt)

    def __str__(self):
        return '{}, {}'.format(self.location.name, self.uuid)

    class Meta:
        verbose_name = "Client"
        verbose_name_plural = "Clienten"

class Accounting(models.Model):
    client = models.ForeignKey(Client, verbose_name="Client")
    desk_no = models.IntegerField(verbose_name="Tischnummer")
    time_from = models.DateTimeField(verbose_name="Beginn")
    time_to = models.DateTimeField(blank=True, null=True, verbose_name="Ende")
    prize = models.DecimalField(max_digits=5, decimal_places=2, blank=True, null=True, verbose_name="Preis")

    def __str__(self):
        return '{}: {} -> {}, {}'.format(self.client.uuid, self.time_from, self.time_to, self.prize)

    class Meta:
        ordering = ['-time_from']
        verbose_name = "Buchhaltungseintrag"
        verbose_name_plural = "Buchhaltungseinträge"


def update_location_data(sender, instance, **kwargs):
    tasks.process_location_data.delay()

post_save.connect(update_location_data, sender=LocationData)