NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-08 08:38:16 -06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Undo changes to syncdatasource management command

Browse Source
This commit is contained in:
Jeremy Stretch 2024-07-24 14:47:08 -04:00
parent 1264302877
commit a091ad66a7
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
netbox/core/management/commands/syncdatasource.py
View File

@ -34,12 +34,13 @@ class Command(BaseCommand):
for i, datasource in enumerate(datasources, start=1): for i, datasource in enumerate(datasources, start=1):
self.stdout.write(f"[{i}] Syncing {datasource}... ", ending='') self.stdout.write(f"[{i}] Syncing {datasource}... ", ending='')
self.stdout.flush() self.stdout.flush()
try:
datasource.enqueue_sync_job() datasource.sync()
datasource.refresh_from_db()
self.stdout.write(datasource.get_status_display()) self.stdout.write(datasource.get_status_display())
self.stdout.flush() self.stdout.flush()
except Exception as e:
DataSource.objects.filter(pk=datasource.pk).update(status=DataSourceStatusChoices.FAILED)
raise e
if len(options['name']) > 1: if len(options['name']) > 1:
self.stdout.write(f"Finished.") self.stdout.write(f"Finished.")

5
netbox/core/models/data.py
View File

@ -153,7 +153,7 @@ class DataSource(JobsMixin, PrimaryModel):
return objectchange return objectchange
def enqueue_sync_job(self, request=None): def enqueue_sync_job(self, request):
""" """
Enqueue a background job to synchronize the DataSource by calling sync(). Enqueue a background job to synchronize the DataSource by calling sync().
""" """
@ -165,8 +165,7 @@ class DataSource(JobsMixin, PrimaryModel):
SyncDataSourceJob = import_string('core.jobs.SyncDataSourceJob') SyncDataSourceJob = import_string('core.jobs.SyncDataSourceJob')
return SyncDataSourceJob.enqueue( return SyncDataSourceJob.enqueue(
instance=self, instance=self,
user=(request.user if request else None), user=request.user
run_now=(request is None),
) )
def get_backend(self): def get_backend(self):