mirror of
https://github.com/django-helpdesk/django-helpdesk.git
synced 2024-12-14 02:41:26 +01:00
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
#!/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 date, timedelta
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from helpdesk.models import EscalationExclusion, Queue
|
|
|
|
day_names = {
|
|
'monday': 0,
|
|
'tuesday': 1,
|
|
'wednesday': 2,
|
|
'thursday': 3,
|
|
'friday': 4,
|
|
'saturday': 5,
|
|
'sunday': 6,
|
|
}
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'-d',
|
|
'--days',
|
|
nargs='*',
|
|
choices=list(day_names.keys()),
|
|
required=True,
|
|
help='Days of week (monday, tuesday, etc). Enter the days as space separated list.'
|
|
)
|
|
parser.add_argument(
|
|
'-o',
|
|
'--occurrences',
|
|
default=1,
|
|
type=int,
|
|
help='Occurrences: How many weeks ahead to exclude this day'
|
|
)
|
|
parser.add_argument(
|
|
'-q',
|
|
'--queues',
|
|
nargs='*',
|
|
choices=list(Queue.objects.values_list('slug', flat=True)),
|
|
help='Queues to include (default: all). Enter the queues slug as space separated list.'
|
|
)
|
|
parser.add_argument(
|
|
'-x',
|
|
'--exclude-verbosely',
|
|
action='store_true',
|
|
default=False,
|
|
help='Display a list of dates excluded'
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
days = options['days']
|
|
occurrences = options['occurrences']
|
|
verbose = options['exclude_verbosely']
|
|
queue_slugs = options['queues']
|
|
|
|
if not (days and occurrences):
|
|
raise CommandError('One or more occurrences must be specified.')
|
|
|
|
queues = []
|
|
if queue_slugs is not None:
|
|
queues = Queue.objects.filter(slug__in=queue_slugs)
|
|
|
|
for day_name in days:
|
|
day = day_names[day_name]
|
|
workdate = date.today()
|
|
i = 0
|
|
while i < occurrences:
|
|
if day == workdate.weekday():
|
|
if EscalationExclusion.objects.filter(date=workdate).count() == 0:
|
|
esc = EscalationExclusion.objects.create(
|
|
name=f'Auto Exclusion for {day_name}',
|
|
date=workdate
|
|
)
|
|
|
|
if verbose:
|
|
self.stdout.write(f"Created exclusion for {day_name} {workdate}")
|
|
|
|
for q in queues:
|
|
esc.queues.add(q)
|
|
if verbose:
|
|
self.stdout.write(f" - for queue {q}")
|
|
|
|
i += 1
|
|
workdate += timedelta(days=1)
|