Skip to content

Commit

Permalink
Fix virtual museum import harvester (#4223)
Browse files Browse the repository at this point in the history
  • Loading branch information
dimasciput authored Sep 4, 2024
1 parent 50d0c2b commit 7da84bf
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions bims/tasks/virtual_museum_import.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
from celery.utils.log import get_task_logger
from django.core.management import call_command
from django_tenants.utils import get_tenant_model, schema_context
from django.db import transaction

logger = get_task_logger(__name__)

Expand All @@ -20,13 +19,12 @@ def import_data_task(self, module, limit=10):

TenantModel = get_tenant_model()
tenants = TenantModel.objects.all().exclude(schema_name='public')

for tenant in tenants:
with schema_context(tenant.schema_name):
api_token = preferences.SiteSetting.virtual_museum_token
if not api_token:
logger.info(f"No virtual museum token found for {tenant.schema_name}.")
return
continue
try:
# Check for in-progress tasks
running_tasks = ImportTask.objects.filter(
Expand Down Expand Up @@ -111,7 +109,7 @@ def import_data_task(self, module, limit=10):
call_command('import_odonata_data',
start_index=start_index,
limit=limit,
module_name='Odonates (adult)',
module_name='Invertebrates',
token=api_token)
elif module == 'anurans':
call_command('import_frog_vm_data',
Expand Down

0 comments on commit 7da84bf

Please sign in to comment.