forked from rc/aircox
110 lines
4.0 KiB
Python
110 lines
4.0 KiB
Python
"""
|
|
Manage diffusions using schedules, to update, clean up or check diffusions.
|
|
A diffusion generated using this utility is considered has type "unconfirmed",
|
|
and is not considered as ready for diffusion; To do so, users must confirm the
|
|
diffusion case by changing it's type to "default".
|
|
|
|
Different actions are available:
|
|
- "update" is the process that is used to generated them using programs
|
|
schedules for the (given) month.
|
|
|
|
- "clean" will remove all diffusions that are still unconfirmed and have been
|
|
planified before the (given) month.
|
|
|
|
- "check" will remove all diffusions that are unconfirmed and have been planified
|
|
from the (given) month and later.
|
|
"""
|
|
from argparse import RawTextHelpFormatter
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from django.utils import timezone as tz
|
|
from aircox_programs.models import *
|
|
|
|
|
|
class Actions:
|
|
@staticmethod
|
|
def update (date):
|
|
items = []
|
|
for schedule in Schedule.objects.filter(program__active = True):
|
|
items += schedule.diffusions_of_month(date, exclude_saved = True)
|
|
print('> {} new diffusions for schedule #{} ({})'.format(
|
|
len(items), schedule.id, str(schedule)
|
|
))
|
|
|
|
print('total of {} diffusions will be created. To be used, they need '
|
|
'manual approval.'.format(len(items)))
|
|
Diffusion.objects.bulk_create(items)
|
|
|
|
@staticmethod
|
|
def clean (date):
|
|
qs = Diffusion.objects.filter(type = Diffusion.Type['unconfirmed'],
|
|
date__lt = date)
|
|
print('{} diffusions will be removed'.format(qs.count()))
|
|
qs.delete()
|
|
|
|
@staticmethod
|
|
def check (date):
|
|
qs = Diffusion.objects.filter(type = Diffusion.Type['unconfirmed'],
|
|
date__gt = date)
|
|
items = []
|
|
for diffusion in qs:
|
|
schedules = Schedule.objects.filter(program = diffusion.program)
|
|
for schedule in schedules:
|
|
if schedule.match(diffusion.date):
|
|
break
|
|
else:
|
|
print('> #{}: {}'.format(diffusion.pk, str(diffusion)))
|
|
items.append(diffusion.id)
|
|
|
|
print('{} diffusions will be removed'.format(len(items)))
|
|
if len(items):
|
|
Diffusion.objects.filter(id__in = items).delete()
|
|
|
|
|
|
class Command (BaseCommand):
|
|
help= __doc__
|
|
|
|
def add_arguments (self, parser):
|
|
parser.formatter_class=RawTextHelpFormatter
|
|
now = tz.datetime.today()
|
|
|
|
group = parser.add_argument_group('action')
|
|
group.add_argument(
|
|
'--update', action='store_true',
|
|
help='generate (unconfirmed) diffusions for the given month. '
|
|
'These diffusions must be confirmed manually by changing '
|
|
'their type to "normal"')
|
|
group.add_argument(
|
|
'--clean', action='store_true',
|
|
help='remove unconfirmed diffusions older than the given month')
|
|
|
|
group.add_argument(
|
|
'--check', action='store_true',
|
|
help='check future unconfirmed diffusions from the given date '
|
|
'agains\'t schedules and remove it if that do not match any '
|
|
'schedule')
|
|
|
|
group = parser.add_argument_group(
|
|
'date')
|
|
group.add_argument(
|
|
'--year', type=int, default=now.year,
|
|
help='used by update, default is today\'s year')
|
|
group.add_argument(
|
|
'--month', type=int, default=now.month,
|
|
help='used by update, default is today\'s month')
|
|
|
|
def handle (self, *args, **options):
|
|
date = tz.datetime(year = options.get('year'),
|
|
month = options.get('month'),
|
|
day = 1)
|
|
date = tz.make_aware(date)
|
|
|
|
if options.get('update'):
|
|
Actions.update(date)
|
|
elif options.get('clean'):
|
|
Actions.clean(date)
|
|
elif options.get('check'):
|
|
Actions.check(date)
|
|
else:
|
|
raise CommandError('no action has been given')
|
|
|