bornhack-website/src/tickets/models.py

134 lines
3.7 KiB
Python
Raw Normal View History

2017-08-17 15:51:24 +00:00
import io
import hashlib
import base64
import qrcode
from django.conf import settings
2018-04-03 16:44:10 +00:00
from django.urls import reverse_lazy
2017-08-17 15:51:24 +00:00
from django.utils.translation import ugettext_lazy as _
2019-06-16 12:32:24 +00:00
from utils.models import UUIDModel, CampRelatedModel
from utils.pdf import generate_pdf_letter
2017-08-19 23:05:30 +00:00
from django.db import models
import logging
2019-06-16 12:32:24 +00:00
2017-08-17 15:51:24 +00:00
logger = logging.getLogger("bornhack.%s" % __name__)
# TicketType can be full week, one day. etc.
2017-08-19 23:05:30 +00:00
class TicketType(CampRelatedModel, UUIDModel):
2017-08-17 15:51:24 +00:00
name = models.TextField()
2019-06-16 12:32:24 +00:00
camp = models.ForeignKey("camps.Camp", on_delete=models.PROTECT)
2017-08-17 15:51:24 +00:00
def __str__(self):
2019-06-16 12:32:24 +00:00
return "{} ({})".format(self.name, self.camp.title)
2017-08-17 15:51:24 +00:00
class BaseTicket(CampRelatedModel, UUIDModel):
2019-06-16 12:32:24 +00:00
ticket_type = models.ForeignKey("TicketType", on_delete=models.PROTECT)
used = models.BooleanField(default=False)
badge_handed_out = models.BooleanField(default=False)
2019-07-17 20:02:47 +00:00
token = models.CharField(max_length=64)
class Meta:
abstract = True
2017-08-17 15:51:24 +00:00
@property
def camp(self):
return self.ticket_type.camp
2019-07-17 20:02:47 +00:00
def save(self, **kwargs):
self.token = self._get_token()
super().save(**kwargs)
2017-08-17 15:51:24 +00:00
def _get_token(self):
return hashlib.sha256(
2019-06-16 12:32:24 +00:00
"{_id}{secret_key}".format(
2019-07-17 20:02:47 +00:00
_id=self.uuid, secret_key=settings.SECRET_KEY
2019-06-16 12:32:24 +00:00
).encode("utf-8")
2017-08-17 15:51:24 +00:00
).hexdigest()
2017-08-19 21:10:13 +00:00
def get_qr_code_base64(self):
2017-08-17 15:51:24 +00:00
qr = qrcode.make(
self._get_token(),
version=1,
2019-06-16 12:32:24 +00:00
error_correction=qrcode.constants.ERROR_CORRECT_H,
2017-08-17 15:51:24 +00:00
).resize((250, 250))
file_like = io.BytesIO()
2019-06-16 12:32:24 +00:00
qr.save(file_like, format="png")
2017-08-17 15:51:24 +00:00
qrcode_base64 = base64.b64encode(file_like.getvalue())
return qrcode_base64
def get_qr_code_url(self):
2019-06-16 12:32:24 +00:00
return "data:image/png;base64,{}".format(
self.get_qr_code_base64().decode("utf-8")
2018-01-07 18:59:53 +00:00
)
2017-08-17 15:51:24 +00:00
def generate_pdf(self):
return generate_pdf_letter(
2019-06-16 12:32:24 +00:00
filename="{}_ticket_{}.pdf".format(self.shortname, self.pk),
formatdict={"ticket": self},
template="pdf/ticket.html",
2017-08-17 15:51:24 +00:00
)
class SponsorTicket(BaseTicket):
2019-06-16 12:32:24 +00:00
sponsor = models.ForeignKey("sponsors.Sponsor", on_delete=models.PROTECT)
2017-08-17 15:51:24 +00:00
def __str__(self):
2019-06-16 12:32:24 +00:00
return "SponsorTicket: {}".format(self.pk)
2017-08-17 15:51:24 +00:00
2017-08-19 22:25:48 +00:00
@property
def shortname(self):
return "sponsor"
2017-08-17 15:51:24 +00:00
class DiscountTicket(BaseTicket):
2017-08-17 15:51:24 +00:00
price = models.IntegerField(
2019-06-16 12:32:24 +00:00
help_text=_("Price of the discounted ticket (in DKK, including VAT).")
2017-08-17 15:51:24 +00:00
)
def __str__(self):
2019-06-16 12:32:24 +00:00
return "DiscountTicket: {}".format(self.pk)
2017-08-17 15:51:24 +00:00
2017-08-19 22:25:48 +00:00
@property
def shortname(self):
return "discount"
2017-08-17 15:51:24 +00:00
class ShopTicket(BaseTicket):
2018-03-04 15:26:35 +00:00
order = models.ForeignKey(
2019-06-16 12:32:24 +00:00
"shop.Order", related_name="shoptickets", on_delete=models.PROTECT
2018-03-04 15:26:35 +00:00
)
2019-06-16 12:32:24 +00:00
product = models.ForeignKey("shop.Product", on_delete=models.PROTECT)
2017-08-17 15:51:24 +00:00
name = models.CharField(
max_length=100,
help_text=(
2019-06-16 12:32:24 +00:00
"Name of the person this ticket belongs to. "
"This can be different from the buying user."
2017-08-17 15:51:24 +00:00
),
null=True,
blank=True,
)
2019-06-16 12:32:24 +00:00
email = models.EmailField(null=True, blank=True)
2017-08-17 15:51:24 +00:00
# overwrite the _get_token method because old tickets use the user_id
def _get_token(self):
return hashlib.sha256(
2019-06-16 12:32:24 +00:00
"{_id}{user_id}{secret_key}".format(
_id=self.pk, user_id=self.order.user.pk, secret_key=settings.SECRET_KEY
).encode("utf-8")
2017-08-17 15:51:24 +00:00
).hexdigest()
def __str__(self):
2019-06-16 12:32:24 +00:00
return "Ticket {user} {product}".format(
user=self.order.user, product=self.product
2017-08-17 15:51:24 +00:00
)
def get_absolute_url(self):
2019-06-16 12:32:24 +00:00
return str(reverse_lazy("tickets:shopticket_edit", kwargs={"pk": self.pk}))
2017-08-19 22:25:48 +00:00
@property
def shortname(self):
return "shop"