Files
@ d9e8a72d140c
Branch filter:
Location: symposion_app/vendor/regidesk/regidesk/models.py
d9e8a72d140c
7.5 KiB
text/x-python
Update Venueless tokens
Determine traits based on ticket type.
Set different expiry of token for miniconf only tickets.
Determine traits based on ticket type.
Set different expiry of token for miniconf only tickets.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 | # -*- coding: utf-8 -*-
import base64
import datetime
from decimal import Decimal
from io import BytesIO
import jwt
from django.core.exceptions import ValidationError
from django.utils import timezone
from django.conf import settings
from django.db import models
from django.db.models import Q, F
from django.db.models import Case, When, Value
from django.db.models import Count
from django.db.models.signals import post_save
from django.contrib.auth import get_user_model
import pyqrcode
from symposion import constants
from symposion.text_parser import parse
from registrasion.models import commerce
from registrasion.util import generate_access_code as generate_code
User = get_user_model()
class BoardingPassTemplate(models.Model):
label = models.CharField(max_length=100, verbose_name="Label")
from_address = models.EmailField(verbose_name="From address")
subject = models.CharField(max_length=100, verbose_name="Subject")
body = models.TextField(verbose_name="Body")
html_body = models.TextField(verbose_name="HTML Body",null=True)
class Meta:
verbose_name = ("Boarding Pass template")
verbose_name_plural = ("Boarding Pass templates")
class BoardingPass(models.Model):
template = models.ForeignKey(BoardingPassTemplate, null=True, blank=True,
on_delete=models.SET_NULL, verbose_name="Template")
created = models.DateTimeField(auto_now_add=True, verbose_name="Created")
sent = models.DateTimeField(null=True, verbose_name="Sent")
to_address = models.EmailField(verbose_name="To address")
from_address = models.EmailField(verbose_name="From address")
subject = models.CharField(max_length=255, verbose_name="Subject")
body = models.TextField(verbose_name="Body")
html_body = models.TextField(verbose_name="HTML Body", null=True)
class Meta:
permissions = (
("view_boarding_pass", "Can view sent boarding passes"),
("send_boarding_pass", "Can send boarding passes"),
)
verbose_name = ("Boarding Pass")
verbose_name_plural = ("Boarding Passes")
def __unicode__(self):
return self.checkin.attendee.attendeeprofilebase.attendeeprofile.name + ' ' + self.timestamp.strftime('%Y-%m-%d %H:%M:%S')
@property
def email_args(self):
return (self.subject, self.body, self.from_address, self.user.email)
class CheckIn(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
boardingpass = models.OneToOneField(BoardingPass, null=True,
blank=True, on_delete=models.SET_NULL)
seen = models.DateTimeField(null=True,blank=True)
checked_in = models.DateTimeField(null=True,blank=True)
checkin_code = models.CharField(
max_length=6,
unique=True,
db_index=True,
)
_checkin_code_png=models.TextField(max_length=512,null=True,blank=True)
badge_printed = models.BooleanField(default=False)
schwag_given = models.BooleanField(default=False)
checked_in_bool = models.BooleanField(default=False)
needs_review = models.BooleanField(default=False)
review_text = models.TextField(blank=True)
venueless_user_id = models.TextField(max_length=40, null=True, blank=True)
venueless_traits = models.TextField(max_length=256, null=True, blank=True)
_venueless_token = models.TextField(null=True, blank=True)
class Meta:
permissions = (
("view_checkin_details", "Can view the details of other user's checkins"),
)
def save(self, *a, **k):
while not self.checkin_code:
checkin_code = generate_code()
if CheckIn.objects.filter(checkin_code=checkin_code).count() == 0:
self.checkin_code = checkin_code
return super(CheckIn, self).save(*a, **k)
def mark_checked_in(self):
self.checked_in_bool = True
self.checked_in = timezone.now()
self.save()
def mark_badge_printed(self):
self.badge_printed = True
self.save()
def unset_badge(self):
self.badge_printed = False
self.save()
def mark_schwag_given(self):
self.schwag_given = True
self.save()
def bulk_mark_given(self):
self.checked_in_bool = True
self.schwag_given = True
self.save()
def set_exception(self, text):
self.needs_review = True
self.review_text = text
self.save()
@property
def code(self):
return self.checkin_code
@property
def qrcode(self):
"""Returns the QR Code for this checkin's code.
If this is the first time the QR code has been generated, cache it on the object.
If a code has already been cached, serve that.
Returns the raw PNG blob, unless b64=True, in which case the return value
is the base64encoded PNG blob."""
if not self.code:
return None
if not self._checkin_code_png:
qrcode = pyqrcode.create(self.code)
qr_io = BytesIO()
qrcode.png(qr_io, scale=6)
qr_io.seek(0)
self._checkin_code_png = base64.b64encode(qr_io.read()).decode('UTF-8')
self.save()
return self._checkin_code_png
@property
def venueless_token(self):
"""Returns the Venueless JWT token for this checkin's code."""
updated = False
if not self.venueless_user_id:
self.venueless_user_id = self.checkin_code
updated = True
if not self.venueless_traits:
traits = self._get_venueless_traits()
self.venueless_traits = ",".join(traits)
updated = True
if not self._venueless_token:
self._venueless_token = self._generate_venueless_token()
updated = True
if updated:
self.save()
return self._venueless_token
def _get_venueless_traits(self):
traits = ["attendee"]
ticket_type = self.user.attendee.ticket_type
if ticket_type == "Conference Volunteer":
traits.append("volunteer")
elif ticket_type == "Conference Organiser":
traits.append("organiser")
if hasattr(self.user, "speaker_profile"):
best = 0
for presentation in self.user.speaker_profile.presentations.all():
if presentation.section.id == 1:
# Main Conference Speaker
best = 1
if best == 0 and presentation.section.id == 2:
# Miniconf Organiser
best = 2
if best == 1:
traits.append("speaker")
elif best == 2:
traits.append("miniconf_org")
else:
traits.append("miniconf_speaker")
return traits
def _generate_venueless_token(self):
""" Generate token for Venueless login """
issued_at = datetime.datetime.utcnow()
expiry = settings.LCA_END + datetime.timedelta(days=1)
if self.user.attendee.ticket_type == "Miniconf Only":
# Miniconf only ticket holders have limited access
expiry = settings.LCA_MINICONF_END
payload = {
"iss": settings.VENUELESS_TOKEN_ISSUER,
"aud": settings.VENUELESS_AUDIENCE,
"iat": issued_at,
"exp": expiry,
"uid": self.venueless_user_id,
"traits": self.venueless_traits.split(','),
}
token = jwt.encode(payload, settings.VENUELESS_SECRET, algorithm="HS256")
return token
|