Moved all project files under a "helpdesk" directory. Should make it easier to use setup.py / pypi / whatever in the future.

helpdesk/management/commands/create_escalation_exclusions.py

@@ -0,0 +1,156 @@
+#!/usr/bin/python
+"""
+Jutda Helpdesk - A Django powered ticket tracker for small enterprise.
+
+(c) Copyright 2008 Jutda. All Rights Reserved. See LICENSE for details.
+
+scripts/create_escalation_exclusion.py - Easy way to routinely add particular
+                                         days to the list of days on which no
+                                         escalation should take place.
+"""
+
+from datetime import datetime, timedelta, date
+import getopt
+from optparse import make_option
+import sys
+
+from django.core.management.base import BaseCommand, CommandError
+from django.db.models import Q
+
+from helpdesk.models import EscalationExclusion, Queue
+
+
+class Command(BaseCommand):
+    def __init__(self):
+        BaseCommand.__init__(self)
+
+        self.option_list += (
+            make_option(
+                '--days', '-d',
+                help='Days of week (monday, tuesday, etc)'),
+            make_option(
+                '--occurrences', '-o',
+                type='int',
+                default=1,
+                help='Occurrences: How many weeks ahead to exclude this day'),
+            make_option(
+                '--queues', '-q',
+                help='Queues to include (default: all). Use queue slugs'),
+            make_option(
+                '--escalate-verbosely', '-x',
+                action='store_true',
+                default=False,
+                help='Display a list of dates excluded'),
+            )
+
+    def handle(self, *args, **options):
+        days = options['days']
+        occurrences = options['occurrences']
+        verbose = False
+        queue_slugs = options['queues']
+        queues = []
+
+        if options['escalate-verbosely']:
+            verbose = True
+
+        # this should already be handled by optparse
+        if not occurrences: occurrences = 1
+        if not (days and occurrences):
+            raise CommandError('One or more occurrences must be specified.')
+
+        if queue_slugs is not None:
+            queue_set = queue_slugs.split(',')
+            for queue in queue_set:
+                try:
+                    q = Queue.objects.get(slug__exact=queue)
+                except Queue.DoesNotExist:
+                    raise CommandError("Queue %s does not exist." % queue)
+                queues.append(q)
+
+        create_exclusions(days=days, occurrences=occurrences, verbose=verbose, queues=queues)
+
+
+day_names = {
+    'monday': 0,
+    'tuesday': 1,
+    'wednesday': 2,
+    'thursday': 3,
+    'friday': 4,
+    'saturday': 5,
+    'sunday': 6,
+}
+
+
+def create_exclusions(days, occurrences, verbose, queues):
+    days = days.split(',')
+    for day in days:
+        day_name = day
+        day = day_names[day]
+        workdate = date.today()
+        i = 0
+        while i < occurrences:
+            if day == workdate.weekday():
+                if EscalationExclusion.objects.filter(date=workdate).count() == 0:
+                    esc = EscalationExclusion(name='Auto Exclusion for %s' % day_name, date=workdate)
+                    esc.save()
+
+                    if verbose:
+                        print "Created exclusion for %s %s" % (day_name, workdate)
+
+                    for q in queues:
+                        esc.queues.add(q)
+                        if verbose:
+                            print "  - for queue %s" % q
+
+                i += 1
+            workdate += timedelta(days=1)
+
+
+def usage():
+    print "Options:"
+    print " --days, -d: Days of week (monday, tuesday, etc)"
+    print " --occurrences, -o: Occurrences: How many weeks ahead to exclude this day"
+    print " --queues, -q: Queues to include (default: all). Use queue slugs"
+    print " --verbose, -v: Display a list of dates excluded"
+
+
+if __name__ == '__main__':
+    # This script can be run from the command-line or via Django's manage.py.
+    try:
+        opts, args = getopt.getopt(sys.argv[1:], 'd:o:q:v', ['days=', 'occurrences=', 'verbose', 'queues='])
+    except getopt.GetoptError:
+        usage()
+        sys.exit(2)
+
+    days = None
+    occurrences = None
+    verbose = False
+    queue_slugs = None
+    queues = []
+
+    for o, a in opts:
+        if o in ('-x', '--escalate-verbosely'):
+            verbose = True
+        if o in ('-d', '--days'):
+            days = a
+        if o in ('-q', '--queues'):
+            queue_slugs = a
+        if o in ('-o', '--occurrences'):
+            occurrences = int(a)
+
+    if not occurrences: occurrences = 1
+    if not (days and occurrences):
+        usage()
+        sys.exit(2)
+
+    if queue_slugs is not None:
+        queue_set = queue_slugs.split(',')
+        for queue in queue_set:
+            try:
+                q = Queue.objects.get(slug__exact=queue)
+            except Queue.DoesNotExist:
+                print "Queue %s does not exist." % queue
+                sys.exit(2)
+            queues.append(q)
+
+    create_exclusions(days=days, occurrences=occurrences, verbose=verbose, queues=queues)