# -*- coding: utf-8 -*-

import base64

from datetime import datetime

from decimal import Decimal

from io import BytesIO

from django.core.exceptions import ValidationError

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.models import User

import pyqrcode

from symposion import constants

from symposion.text_parser import parse

from registrasion.util import generate_access_code as generate_code

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"),

        )

    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)

    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)

    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)

    @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