carom-server/billard/models.py

146 lines
5.8 KiB
Python
Raw Normal View History

2017-02-01 17:52:16 +01:00
import uuid
2017-01-03 19:26:51 +01:00
from django.db import models
2017-02-04 06:51:09 +01:00
from django.contrib.auth.models import User
2017-02-06 20:11:20 +01:00
from datetime import datetime, timezone
2017-02-11 11:31:27 +01:00
from billard import utils
2017-02-06 19:11:32 +01:00
from django.contrib.auth.models import User
from django.db.models.signals import post_save
2017-02-10 20:49:38 +01:00
from django.dispatch import receiver
2017-02-11 11:31:27 +01:00
from django.core.mail import mail_admins
2017-02-10 21:02:07 +01:00
from celery import shared_task
2017-01-03 19:26:51 +01:00
2017-02-01 13:51:15 +01:00
class LocationData(models.Model):
2017-02-10 20:49:38 +01:00
client_id = models.UUIDField(blank=False, null=False, verbose_name="Client-ID")
2017-02-10 19:55:31 +01:00
desk_no = models.IntegerField(blank=False, null=False, verbose_name="Tischnummer")
2017-02-06 20:27:11 +01:00
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")
2017-02-06 20:29:35 +01:00
error_msg = models.TextField(blank=True, null=True, verbose_name="Fehlermeldung")
2017-01-03 19:26:51 +01:00
2017-01-06 18:53:26 +01:00
def __str__(self):
2017-02-10 20:49:38 +01:00
return str(self.client_id)
2017-01-06 18:53:26 +01:00
2017-02-06 19:19:28 +01:00
class Meta:
verbose_name = "Standortlog"
verbose_name_plural = "Standortlogs"
2017-01-03 20:13:47 +01:00
2017-02-01 13:51:15 +01:00
class Location(models.Model):
2017-02-06 20:27:11 +01:00
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")
2017-01-03 19:26:51 +01:00
2017-01-06 18:53:26 +01:00
def __str__(self):
return self.name
2017-02-06 19:19:28 +01:00
class Meta:
verbose_name = "Standort"
verbose_name_plural = "Standorte"
2017-01-06 18:53:26 +01:00
2017-01-03 20:13:47 +01:00
2017-02-01 13:51:15 +01:00
class Client(models.Model):
2017-02-06 20:27:11 +01:00
uuid = models.UUIDField(unique=True, default=uuid.uuid4, verbose_name="Identifier")
location = models.ForeignKey(Location, verbose_name="Standort")
2017-02-01 20:25:33 +01:00
def __str__(self):
2017-02-04 12:39:28 +01:00
return '{}, {}'.format(self.location.name, self.uuid)
2017-02-01 20:25:33 +01:00
2017-02-06 19:19:28 +01:00
class Meta:
verbose_name = "Client"
verbose_name_plural = "Clienten"
2017-02-01 20:25:33 +01:00
class Desk(models.Model):
2017-02-10 19:39:21 +01:00
client = models.ForeignKey(Client, verbose_name='Client', related_name='desks')
desk_no = models.IntegerField(verbose_name='Tischnummer')
name = models.CharField(max_length=32, blank=True, null=True, verbose_name='Tischbezeichnung')
enabled = models.BooleanField(verbose_name='Tisch aktiv')
prize = models.DecimalField(max_digits=5, decimal_places=2, blank=True, null=True, verbose_name="Normelpreis")
prize_hh = models.DecimalField(max_digits=5, decimal_places=2, blank=True, null=True,
verbose_name="Preis Happy Hour")
def accounting_for(self):
t = Accounting.objects.filter(client=self.client, desk_no=self.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=self.prize)
if prize != a.prize:
a.prize = prize
return t
def __str__(self):
return '{}, {}'.format(self.client.uuid, self.name)
class Meta:
verbose_name = "Tisch"
verbose_name_plural = "Tische"
2017-02-01 20:25:33 +01:00
class Accounting(models.Model):
2017-02-11 10:23:03 +01:00
desk = models.ForeignKey(Desk, verbose_name="Tisch")
2017-02-06 20:27:11 +01:00
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")
2017-02-11 10:23:03 +01:00
billed = models.BooleanField(default=False, verbose_name="Abgerechnet")
2017-02-01 20:25:33 +01:00
2017-01-06 18:53:26 +01:00
def __str__(self):
2017-02-11 10:23:03 +01:00
return '{}: {} -> {}, {}, {}'.format(self.desk, self.time_from, self.time_to, self.prize, self.billed)
2017-01-06 18:53:26 +01:00
2017-02-01 20:25:33 +01:00
class Meta:
ordering = ['-time_from']
2017-02-06 19:19:28 +01:00
verbose_name = "Buchhaltungseintrag"
2017-02-06 19:11:32 +01:00
verbose_name_plural = "Buchhaltungseinträge"
2017-02-10 20:49:38 +01:00
@receiver(post_save, sender=LocationData)
def test(sender, **kwargs):
2017-02-10 21:02:07 +01:00
process_location_data()
@shared_task
def process_location_data():
data = LocationData.objects.filter(processed=False)
for ld in data:
2017-02-11 11:00:14 +01:00
cli = Client.objects.filter(uuid=ld.client_id, desks__desk_no=ld.desk_no)
2017-02-10 21:02:07 +01:00
if cli.count() < 1:
ld.processed = True
2017-02-11 11:31:27 +01:00
ld.error_msg = 'No location object found. Stopp processing! {}, {}'.format(ld.client_id, ld.desk_no)
2017-02-10 21:02:07 +01:00
ld.save()
2017-02-11 11:31:27 +01:00
mail_admins(subject=ld.error_msg, message=ld.error_msg)
2017-02-10 21:02:07 +01:00
# TODO Send error eMail to Admin
else:
cli = cli[0]
desk = cli.desks.filter(desk_no=ld.desk_no)[0]
2017-02-11 10:23:03 +01:00
ac = desk.accounting_set.order_by('time_from').reverse()
2017-02-10 21:02:07 +01:00
if ld.on_off:
if ac.count() > 0 and ac[0].time_to is None:
ac[0].time_to = datetime.now()
ac[0].save()
2017-02-11 11:31:27 +01:00
mail_admins(subject='Accounting passt nicht',
message='Vorheriges Accounting nicht abgeschlossen: Accounting_id {}'.format(ac[0].id))
2017-02-10 21:02:07 +01:00
# TODO Send error eMail to Admin
acc = Accounting(
2017-02-11 10:23:03 +01:00
desk=desk,
2017-02-10 21:02:07 +01:00
time_from=ld.tst,
)
acc.save()
ld.delete()
else:
acc = ac[0]
acc.time_to = ld.tst
acc.prize = utils.get_prize_for(
start=acc.time_from,
end=ld.tst,
pph=desk.prize
)
acc.save()
ld.delete()