Add search indexers for DataSource, DataFile

netbox/core/apps.py

@@ -0,0 +1,8 @@
+from django.apps import AppConfig
+
+
+class CoreConfig(AppConfig):
+    name = "core"
+
+    def ready(self):
+        from . import data_backends, search

netbox/core/jobs.py

@@ -1,6 +1,7 @@
 import logging
 
 from extras.choices import JobResultStatusChoices
+from netbox.search.backends import search_backend
 from .choices import *
 from .exceptions import SyncError
 from .models import DataSource
@@ -17,6 +18,10 @@ def sync_datasource(job_result, *args, **kwargs):
     try:
         job_result.start()
         datasource.sync()
+
+        # Update the search cache for DataFiles belonging to this source
+        search_backend.cache(datasource.datafiles.iterator())
+
     except SyncError as e:
         job_result.set_status(JobResultStatusChoices.STATUS_ERRORED)
         job_result.save()

netbox/core/search.py

@@ -0,0 +1,20 @@
+from netbox.search import SearchIndex, register_search
+from . import models
+
+
+@register_search
+class DataSourceIndex(SearchIndex):
+    model = models.DataSource
+    fields = (
+        ('name', 100),
+        ('url', 300),
+        ('description', 500),
+    )
+
+
+@register_search
+class DataFileIndex(SearchIndex):
+    model = models.DataFile
+    fields = (
+        ('path', 200),
+    )