CGM_Public/pretix_cgo
1
0
Fork 0
forked from CGM_Public/pretix_original
Code Pull Requests Actions Activity

runperiodic: Allow to execute specific tasks only

Browse Source
This commit is contained in:
Raphael Michel
2020-10-22 21:05:10 +02:00
parent a62c7939ae
commit 8241ddf5be
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/pretix/base/management/commands/runperiodic.py
View File

@@ -13,6 +13,10 @@ from ...signals import periodic_task
class Command(BaseCommand): class Command(BaseCommand):
help = "Run periodic tasks" help = "Run periodic tasks"
def add_arguments(self, parser):
parser.add_argument('--tasks', action='store', type=str, help='Only execute the tasks with this name '
'(dotted path, comma separation)')
def handle(self, *args, **options): def handle(self, *args, **options):
verbosity = int(options['verbosity']) verbosity = int(options['verbosity'])
@@ -20,8 +24,13 @@ class Command(BaseCommand):
return return
for receiver in periodic_task._live_receivers(self): for receiver in periodic_task._live_receivers(self):
name = f'{receiver.__module__}.{receiver.__name__}'
if options.get('tasks'):
if name not in options.get('tasks').split(','):
continue
if verbosity > 1: if verbosity > 1:
self.stdout.write(f'Running {receiver.__module__}.{receiver.__name__}…') self.stdout.write(f'Running {name}…')
t0 = time.time() t0 = time.time()
try: try:
r = receiver(signal=periodic_task, sender=self) r = receiver(signal=periodic_task, sender=self)
@@ -38,6 +47,6 @@ class Command(BaseCommand):
else: else:
if options.get('verbosity') > 1: if options.get('verbosity') > 1:
if r is SKIPPED: if r is SKIPPED:
self.stdout.write(self.style.SUCCESS(f'Skipped {receiver.__module__}.{receiver.__name__}')) self.stdout.write(self.style.SUCCESS(f'Skipped {name}'))
else: else:
self.stdout.write(self.style.SUCCESS(f'Completed {receiver.__module__}.{receiver.__name__} in {round(time.time() - t0, 3)}s')) self.stdout.write(self.style.SUCCESS(f'Completed {name} in {round(time.time() - t0, 3)}s'))