SciPost Code Repository

Skip to content
Snippets Groups Projects
models.py 8.34 KiB
Newer Older
__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


from django.contrib.contenttypes.fields import GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.db import models
from django.template import Template
from django.urls import reverse
from django.utils import timezone

Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
from .constants import (
    TICKET_PRIORITIES, TICKET_PRIORITY_URGENT, TICKET_PRIORITY_HIGH, TICKET_PRIORITY_MEDIUM,
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    TICKET_STATUSES, TICKET_STATUS_UNASSIGNED, TICKET_STATUS_ASSIGNED,
    TICKET_STATUS_PICKEDUP, TICKET_STATUS_PASSED_ON,
    TICKET_STATUS_AWAITING_RESPONSE_ASSIGNEE, TICKET_STATUS_AWAITING_RESPONSE_USER,
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    TICKET_STATUS_RESOLVED, TICKET_STATUS_CLOSED,
    TICKET_FOLLOWUP_ACTION_TYPES)
from .managers import QueueQuerySet, TicketQuerySet


class Queue(models.Model):
    """
    A Queue is a container for a category of Tickets.
    Each Ticket has a ForeignKey relationship to a specific Queue.
    A Queue is managed by a specific Group specific by `managing_group`,
    and can be responsed to by users in a set of groups defined by the
    `response_groups` field.

    A Queue is visible only to users in managing or response groups,
    and is thus admin/internal only.

    Permissions are handled at object level using `django_guardian`.
    """
    name = models.CharField(max_length=64)
    slug = models.SlugField(allow_unicode=True, unique=True)
    description = models.TextField(
        help_text=('You can use plain text, Markdown or reStructuredText; see our '
                   '<a href="/markup/help/" target="_blank">markup help</a> pages.'))
    managing_group = models.ForeignKey('auth.Group', on_delete=models.CASCADE,
                                       related_name='managed_queues')
    response_groups = models.ManyToManyField('auth.Group')
    parent_queue = models.ForeignKey('helpdesk.Queue', on_delete=models.CASCADE,
                                     blank=True, null=True, related_name='sub_queues')
    objects = QueueQuerySet.as_manager()

    class Meta:
        ordering = ['name',]
        permissions = [
            ('can_manage_queue', 'Can manage Queue'),
            ('can_handle_queue', 'Can handle Queue'),
            ('can_view_queue', 'Can view Queue'),
        ]

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('helpdesk:queue_detail', kwargs={'slug': self.slug})

    @property
    def nr_unassigned(self):
        return self.tickets.unassigned().count()

    @property
    def nr_assigned(self):
        return self.tickets.assigned().count()

    @property
    def nr_resolved(self):
        return self.tickets.resolved().count()

    @property
    def nr_closed(self):
        return self.tickets.closed().count()

    @property
    def nr_awaiting_handling(self):
        return self.tickets.awaiting_handling().count()

    @property
    def nr_in_handling(self):
        return self.tickets.in_handling().count()

    @property
    def nr_handled(self):
        return self.tickets.handled().count()



class Ticket(models.Model):
    """
    User-created ticket, representing a query or request for assistance.

    Each ticket lives in a Queue.
    """
    queue = models.ForeignKey('helpdesk.Queue', on_delete=models.CASCADE,
                              related_name='tickets',
                              help_text=('Don\'t worry and just choose the one that '
                                         'seems most appropriate for your issue'))
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    title = models.CharField(max_length=64, default='')
    description = models.TextField(
        help_text=('You can use plain text, Markdown or reStructuredText; see our '
                   '<a href="/markup/help/" target="_blank">markup help</a> pages.'))
    publicly_visible = models.BooleanField(
        default=False,
        help_text=('Do you agree with this Ticket being made publicly visible '
                   '(and appearing, anonymized, in our public Knowledge Base)?'))
    defined_on = models.DateTimeField(default=timezone.now)
    defined_by = models.ForeignKey('auth.User', on_delete=models.CASCADE)
    priority = models.CharField(max_length=32, choices=TICKET_PRIORITIES)
    deadline = models.DateField(blank=True, null=True)
    status = models.CharField(max_length=32, choices=TICKET_STATUSES)
    assigned_to = models.ForeignKey('auth.User', on_delete=models.SET_NULL,
                                    blank=True, null=True, related_name='assigned_tickets')
    concerning_object_type = models.ForeignKey(ContentType, on_delete=models.CASCADE,
                                               blank=True, null=True)
    concerning_object_id = models.PositiveIntegerField(blank=True, null=True)
    concerning_object = GenericForeignKey('concerning_object_type', 'concerning_object_id')

    objects = TicketQuerySet.as_manager()

    class Meta:
        ordering = ['queue', 'priority']
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        permissions = [
            ('can_view_ticket', 'Can view Ticket'),
        ]
        return '%s-%s: %s' % (self.queue.slug, self.pk, self.title)
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed

    def get_absolute_url(self):
        return reverse('helpdesk:ticket_detail', kwargs={'pk': self.id})
    def latest_followup(self):
        return self.followups.order_by('timestamp').last()

    @property
    def latest_activity(self):
        try:
            return self.latest_followup().timestamp
        except AttributeError:
            return self.defined_on

    @property
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    def is_unassigned(self):
        return self.status == TICKET_STATUS_UNASSIGNED

    @property
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    def is_awaiting_handling(self):
        return self.status in [TICKET_STATUS_ASSIGNED, TICKET_STATUS_PASSED_ON]

    @property
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    def is_in_handling(self):
        return self.status in [TICKET_STATUS_PICKEDUP,
                               TICKET_STATUS_AWAITING_RESPONSE_ASSIGNEE,
                               TICKET_STATUS_AWAITING_RESPONSE_USER]

    @property
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    def is_handled(self):
        return self.status in [TICKET_STATUS_RESOLVED, TICKET_STATUS_CLOSED]

Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    @property
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    def is_open(self):
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        """Return True if the Ticket hasn't been resolved or closed."""
        return self.status not in [TICKET_STATUS_RESOLVED, TICKET_STATUS_CLOSED]

    @property
    def priority_level(self):
        if self.priority == TICKET_PRIORITY_URGENT:
            return 3
        elif self.priority == TICKET_PRIORITY_HIGH:
            return 2
        elif self.priority == TICKET_PRIORITY_MEDIUM:
            return 1
            return 0
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
    @property
    def status_classes(self):
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        if self.is_unassigned:
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
            return {'class': 'danger', 'text': 'white'}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        if self.is_awaiting_handling:
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
            return {'class': 'warning', 'text': 'dark'}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        if self.is_in_handling:
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
            return {'class': 'success', 'text': 'white'}
        if self.status == TICKET_STATUS_RESOLVED:
            return {'class': 'primary', 'text': 'white'}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        elif self.status == TICKET_STATUS_CLOSED:
            return {'class': 'secondary', 'text': 'dark'}
    @property
    def progress_level(self):
        """For fa box checking: 0 if unassigned, 1 if assigned, 2 if in handling, 3 if handled."""
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        if self.is_unassigned:
            return {'1': '', '2': '', '3': ''}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        elif self.is_awaiting_handling:
            return {'1': '-check', '2': '', '3': ''}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        elif self.is_in_handling:
            return {'1': '-check', '2': '-check', '3': ''}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed
        elif self.is_handled:
            return {'1': '-check', '2': '-check', '3': '-check'}
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed

class Followup(models.Model):
    """
    Response concerning a Ticket, from either the user or handler.
    """
    ticket = models.ForeignKey('helpdesk.Ticket', on_delete=models.CASCADE,
                               related_name='followups')
    text = models.TextField(blank=True, null=True)
    by = models.ForeignKey('auth.User', on_delete=models.CASCADE,
                           related_name='ticket_followups')
    timestamp = models.DateTimeField()
    action = models.CharField(max_length=32, choices=TICKET_FOLLOWUP_ACTION_TYPES)

    class Meta:
        ordering = ['timestamp']

    def __str__(self):
        return '%s, by %s on %s: %s' % (self.ticket, self.by, self.timestamp,
                                        self.get_action_display())
Jean-Sébastien Caux's avatar
Jean-Sébastien Caux committed

    def get_absolute_url(self):
        return '%s#%s' % (reverse('helpdesk:ticket_detail', kwargs={'pk': self.ticket.id}), self.id)