CGM_Public/pretix_original
2
0
Fork 1
mirror of https://github.com/pretix/pretix.git synced 2026-05-05 15:14:04 +00:00
Code Issues Packages Projects Releases Wiki 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):
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):
verbosity = int(options['verbosity'])
@@ -20,8 +24,13 @@ class Command(BaseCommand):
return
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:
self.stdout.write(f'Running {receiver.__module__}.{receiver.__name__}…')
self.stdout.write(f'Running {name}…')
t0 = time.time()
try:
r = receiver(signal=periodic_task, sender=self)
@@ -38,6 +47,6 @@ class Command(BaseCommand):
else:
if options.get('verbosity') > 1:
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:
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'))