From 5338c5d5fbceb6e4a2e9fb5ea92f20a91a4a766c Mon Sep 17 00:00:00 2001 From: Vermaas <vermaas@astron.nl> Date: Fri, 16 Feb 2024 07:15:51 +0100 Subject: [PATCH] add batchsize to UpdateIngestQTasks --- atdb/taskdatabase/templates/taskdatabase/index.html | 2 +- atdb/taskdatabase/urls.py | 2 +- atdb/taskdatabase/views.py | 12 ++++++------ 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/atdb/taskdatabase/templates/taskdatabase/index.html b/atdb/taskdatabase/templates/taskdatabase/index.html index 2324fdb0..f4fb3278 100644 --- a/atdb/taskdatabase/templates/taskdatabase/index.html +++ b/atdb/taskdatabase/templates/taskdatabase/index.html @@ -31,7 +31,7 @@ {% include 'taskdatabase/pagination.html' %} </div> </div> - <p class="footer"> Version 15 Feb 2024 (19:50) + <p class="footer"> Version 16 Feb 2024 (7:00) </div> {% include 'taskdatabase/refresh.html' %} diff --git a/atdb/taskdatabase/urls.py b/atdb/taskdatabase/urls.py index 9f0f7207..bf8ed0a4 100644 --- a/atdb/taskdatabase/urls.py +++ b/atdb/taskdatabase/urls.py @@ -134,6 +134,6 @@ urlpatterns = [ path('tasks/repair/associate-activities/', views.AssociateActivities, name='associate-activities'), path('tasks/repair/update-all-activities/', views.UpdateAllActivities, name='update-all-activities'), path('tasks/repair/update-failed-tasks/', views.UpdateFailedTasks, name='update-failed-tasks'), - path('tasks/repair/update-ingestq-tasks/', views.UpdateIngestQTasks, name='update-ingestq-tasks'), + path('tasks/repair/update-ingestq-tasks/<batch_size>', views.UpdateIngestQTasks, name='update-ingestq-tasks'), path('tasks/repair/update-finished-tasks/', views.UpdateFinishedTasks, name='update-finished-tasks'), ] diff --git a/atdb/taskdatabase/views.py b/atdb/taskdatabase/views.py index 2d95d2c9..fe602715 100644 --- a/atdb/taskdatabase/views.py +++ b/atdb/taskdatabase/views.py @@ -1684,8 +1684,8 @@ def AssociateActivities(request): # disconnect the signals to avoid save recursion disconnect_signals() - #tasks = Task.objects.all().only('sas_id') - tasks = Task.objects.filter(activity__isnull=True)[:10000] + #tasks = Task.objects.filter(activity__isnull=True)[:10000] + tasks = Task.objects.filter(activity__isnull=True) total = tasks.count() i = 0 for task in tasks: @@ -1712,8 +1712,7 @@ def UpdateAllActivities(request): try: i += 1 task = Task.objects.filter(sas_id=activity.sas_id)[0] - if i>2680: - activities_handler.update_activity(task) + activities_handler.update_activity(task) logger.info(f'{i} of {total}') except Exception as error: logger.error(error) @@ -1742,17 +1741,18 @@ def UpdateFailedTasks(request): return redirect('index') @staff_member_required -def UpdateIngestQTasks(request): +def UpdateIngestQTasks(request, batch_size): tasks = Task.objects.only('sas_id').filter( Q(status__icontains=State.SCRUBBED.value) | Q(status__icontains=State.ARCHIVING.value) | Q(status__icontains=State.ARCHIVED.value) | - Q(status__icontains=State.FINISHING.value)) + Q(status__icontains=State.FINISHING.value)).filter(activity__ingested_fraction__isnull=True)[:int(batch_size)] total = tasks.count() i = 0 for task in tasks: i+=1 + activities_handler.update_activity(task) logger.info(f'{i} of {total}') -- GitLab