django-helpdesk/feeds.py

from django.contrib.auth.models import User
from django.contrib.syndication.feeds import Feed
from django.core.urlresolvers import reverse
from django.db.models import Q

from models import Ticket, FollowUp, Queue


class OpenTicketsByUser(Feed):
    title_template = 'helpdesk/rss/ticket_title.html'
    description_template = 'helpdesk/rss/ticket_description.html'

    def get_object(self, bits):
        if len(bits) < 1:
            raise ObjectDoesNotExist
        user = User.objects.get(username__exact=bits[0])
        if len(bits) == 2:
            queue = Queue.objects.get(slug__exact=bits[1])
        else: queue = False

        return {'user': user, 'queue': queue}

    def title(self, obj):
        if obj['queue']:
            return "Helpdesk: Open Tickets in queue %s for %s" % (obj['queue'].title, obj['user'].username)
        else:
            return "Helpdesk: Open Tickets for %s" % obj['user'].username

    def description(self, obj):
        if obj['queue']:
            return "Open and Reopened Tickets in queue %s for %s" % (obj['queue'].title, obj['user'].username)
        else:
            return "Open and Reopened Tickets for %s" % obj['user'].username

    def link(self, obj):
        if obj['queue']:
            return '%s?assigned_to=%s&queue=%s' % (reverse('helpdesk_list'), obj['user'].id, obj['queue'].id)
        else:
            return '%s?assigned_to=%s' % (reverse('helpdesk_list'), obj['user'].id)

    def items(self, obj):
        if obj['queue']:
            return Ticket.objects.filter(assigned_to=obj['user']).filter(queue=obj['queue']).filter(Q(status=Ticket.OPEN_STATUS) | Q(status=Ticket.REOPENED_STATUS))
        else:
            return Ticket.objects.filter(assigned_to=obj['user']).filter(Q(status=Ticket.OPEN_STATUS) | Q(status=Ticket.REOPENED_STATUS))
    
    def item_pubdate(self, item):
        return item.created

    def item_author_name(self, item):
        if item.assigned_to:
            return item.assigned_to.username
        else:
            return "Unassigned"


class UnassignedTickets(Feed):
    title_template = 'helpdesk/rss/ticket_title.html'
    description_template = 'helpdesk/rss/ticket_description.html'
    
    title = "Helpdesk: Unassigned Tickets"
    description = "Unassigned Open and Reopened tickets"
    link = ''#%s?assigned_to=' % reverse('helpdesk_list')

    def items(self, obj):
        return Ticket.objects.filter(assigned_to__isnull=True).filter(Q(status=Ticket.OPEN_STATUS) | Q(status=Ticket.REOPENED_STATUS))
    
    def item_pubdate(self, item):
        return item.created
    
    
    def item_author_name(self, item):
        if item.assigned_to:
            return item.assigned_to.username
        else:
            return "Unassigned"


class RecentFollowUps(Feed):
    title_template = 'helpdesk/rss/recent_activity_title.html'
    description_template = 'helpdesk/rss/recent_activity_description.html'

    title = "Helpdesk: Recent Followups"
    description = "Recent FollowUps, such as e-mail replies, comments, attachments and resolutions"
    link = '/tickets/' # reverse('helpdesk_list')

    def items(self):
        return FollowUp.objects.order_by('-date')[:20]


class OpenTicketsByQueue(Feed):
    title_template = 'helpdesk/rss/ticket_title.html'
    description_template = 'helpdesk/rss/ticket_description.html'

    def get_object(self, bits):
        if len(bits) != 1:
            raise ObjectDoesNotExist
        return Queue.objects.get(slug__exact=bits[0])

    def title(self, obj):
        return "Helpdesk: Open Tickets in queue %s" % obj.title

    def description(self, obj):
        return "Open and Reopened Tickets in queue %s" % obj.title

    def link(self, obj):
        return '%s?queue=%s' % (reverse('helpdesk_list'), obj.id)

    def items(self, obj):
        return Ticket.objects.filter(queue=obj).filter(Q(status=Ticket.OPEN_STATUS) | Q(status=Ticket.REOPENED_STATUS))

    def item_pubdate(self, item):
        return item.created
    
    def item_author_name(self, item):
        if item.assigned_to:
            return item.assigned_to.username
        else:
            return "Unassigned"


feed_setup = {
    'user': OpenTicketsByUser,
    'queue': OpenTicketsByQueue,
    'recent_activity': RecentFollowUps,
    'unassigned': UnassignedTickets,
}