DEMOS2/allauthdemo/polls/models.py

from __future__ import unicode_literals

import json
import uuid

from django.core.mail import send_mail
from django.db import models
from django.utils import timezone

from allauthdemo.auth.models import DemoUser


class EmailUser(models.Model):
    email = models.CharField(max_length=80, unique=True)

    def send_email(self, subject, message, from_email=None):
        """
        Sends an email to this User.
        """
        send_mail(subject, message, from_email, [self.email])

    def __unicode__(self):
        return self.email


class Event(models.Model):
    users_organisers = models.ManyToManyField(DemoUser, blank=True, related_name="organisers")
    users_trustees = models.ManyToManyField(EmailUser, blank=True, related_name="trustees")
    voters = models.ManyToManyField(EmailUser, blank=True, related_name="voters")
    start_time = models.DateTimeField()
    end_time = models.DateTimeField()
    prepared = models.BooleanField(default=False)
    ended = models.BooleanField(default=False)
    public_key = models.CharField(null=True, blank=False, max_length=1024)
    title = models.CharField(max_length=1024)
    EID = models.CharField(max_length=2048, blank=True)
    creator = models.CharField(max_length=256, blank=True)
    c_email = models.CharField(max_length=512, blank=True)
    trustees = models.CharField(max_length=4096)
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    # Custom helper methods
    def EID_hr(self):
        try:
            EID_json = json.loads(self.EID)
            return EID_json['hr']
        except ValueError:
            return self.EID

    def EID_crypto(self):
        try:
            EID_json = json.loads(self.EID)
            EID_crypto_str = EID_json['crypto']
            return json.dumps(json.loads(EID_crypto_str))
        except ValueError:
            return "None - Event not Initialised"

    def duration(self):
        duration_str = self.start_time_formatted()
        duration_str = duration_str + " - " + self.end_time_formatted_utc()
        return duration_str

    def start_time_formatted(self):
        return self.start_time.strftime("%d-%m-%y %H:%M")

    def start_time_formatted_utc(self):
        return self.start_time.strftime("%d-%m-%y %H:%M %Z")

    def end_time_formatted(self):
        return self.end_time.strftime("%d-%m-%y %H:%M")

    def end_time_formatted_utc(self):
        return self.end_time.strftime("%d-%m-%y %H:%M %Z")

    # Total number of options in all polls
    def total_num_opts(self):
        polls = self.polls.all()
        count = 0

        for poll in polls:
            count += poll.options.all().count()

        return count

    def total_num_partial_decs(self):
        polls = self.polls.all()
        count = 0

        for poll in polls:
            count += PartialBallotDecryption.objects.filter(poll=poll).count()

        return count

    def all_part_decs_received(self):
        received = False

        if self.total_num_partial_decs() == self.total_num_opts() * self.users_trustees.all().count():
            received = True

        return received

    def status(self):
        status_str = ""

        # Get the current date and time to compare against to establish if this is a past, current or
        # future event. Prepared means the public key has been initialised
        present = timezone.now()

        if self.ended is False:
            if present < self.start_time and self.prepared is False:
                status_str = "Future"
            elif present < self.start_time and self.prepared is True:
                status_str = "Prepared"
            elif present >= self.start_time and present <= self.end_time and self.prepared is True:
                status_str = "Active"
            elif present >= self.start_time and present <= self.end_time and self.prepared is False:
                status_str = "Future"
            elif present > self.end_time:
                status_str = "Expired"
        else:
            if self.all_part_decs_received():
                status_str = "Decrypted"
            else:
                status_str = "Ended"

        return status_str

    '''
        The result applies to all polls for an event so True will only be returned when votes have
        been received for every poll. 
    '''
    def has_received_votes(self):
        received_votes = True

        for poll in self.polls.all():
            if Ballot.objects.filter(poll=poll, cast=True).count() == 0:
                received_votes = False

        return received_votes

    def __str__(self):
        return self.title


class TrusteeKey(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="trustee_keys")
    user = models.ForeignKey(EmailUser, on_delete=models.CASCADE, related_name="trustee_keys")
    key = models.CharField(max_length=255, unique=True)


class AccessKey(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="keys")
    user = models.ForeignKey(EmailUser, on_delete=models.CASCADE, related_name="keys")
    key = models.CharField(max_length=255, unique=True)

    def has_started(self):
        return timezone.now() >= self.start

    def has_ended(self):
        return timezone.now() >= self.end

    def __unicode__(self):
        return self.title


class Poll(models.Model):
    question_text = models.CharField(max_length=200)
    total_votes = models.IntegerField(default=0)
    min_num_selections = models.IntegerField(default=0)
    max_num_selections = models.IntegerField(default=1)
    event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="polls")
    combined_ballots = models.CharField(max_length=4096, null=True)
    result_json = models.CharField(max_length=4096, null=True)
    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    def __str__(self):
        return self.question_text


class PollOption(models.Model):
    choice_text = models.CharField(max_length=200)
    votes = models.IntegerField(default=0)
    question = models.ForeignKey(Poll, on_delete=models.CASCADE, related_name="options")

    def __str__(self):
        return self.choice_text


class CombinedBallot(models.Model):
    poll = models.ForeignKey(Poll, on_delete=models.CASCADE, related_name="combined_ballot")
    option = models.ForeignKey(PollOption, on_delete=models.CASCADE, related_name="combined_ballot")
    cipher_text_c1 = models.CharField(max_length=4096)
    cipher_text_c2 = models.CharField(max_length=4096)


# A partial decryption supplied by a trustee for a combined ballot that relates to a poll option
class PartialBallotDecryption(models.Model):
    event = models.ForeignKey(Event, on_delete=models.CASCADE, related_name="decryption")
    poll = models.ForeignKey(Poll, on_delete=models.CASCADE, related_name="decryption")
    option = models.ForeignKey(PollOption, on_delete=models.CASCADE, related_name="decryption")
    user = models.ForeignKey(EmailUser, on_delete=models.CASCADE, related_name="decryption")
    text = models.CharField(max_length=4096)


class Ballot(models.Model):
    voter = models.ForeignKey(EmailUser, on_delete=models.CASCADE, related_name="ballots")
    poll = models.ForeignKey(Poll, on_delete=models.CASCADE, related_name="ballots")
    cast = models.BooleanField(default=False)


class EncBallot(models.Model):
    handle = models.CharField(primary_key=True, default=uuid.uuid4, editable=False, max_length=255)
    ballot = models.CharField(max_length=4096)


# Implements the new binary encoding scheme
class EncryptedVote(models.Model):
    ballot = models.ForeignKey(Ballot, on_delete=models.CASCADE, related_name="encrypted_vote")


class CombinedEncryptedVote(models.Model):
    ballot = models.ForeignKey(Ballot, on_delete=models.CASCADE, related_name="comb_encrypted_vote")


class VoteFragment(models.Model):
    encrypted_vote = models.ForeignKey(EncryptedVote, on_delete=models.CASCADE, related_name="fragment", null=True)
    comb_encrypted_vote = models.ForeignKey(CombinedEncryptedVote, on_delete=models.CASCADE, related_name="fragment", null=True)
    cipher_text_c1 = models.CharField(max_length=4096)
    cipher_text_c2 = models.CharField(max_length=4096)