Move search app to a module under netbox/

netbox/circuits/search_indexes.py

@@ -2,7 +2,7 @@ import circuits.filtersets
 import circuits.tables
 from circuits.models import Circuit, Provider, ProviderNetwork
 from django.db import models
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from utilities.utils import count_related
 
 

netbox/dcim/search_indexes.py

@@ -4,7 +4,6 @@ from dcim.models import (
     Cable,
     Device,
     DeviceType,
-    Interface,
     Location,
     Module,
     ModuleType,
@@ -14,8 +13,7 @@ from dcim.models import (
     Site,
     VirtualChassis,
 )
-from django.db import models
+from netbox.search.models import SearchMixin
-from search.models import SearchMixin
 from utilities.utils import count_related
 
 

netbox/extras/search_indexes.py

@@ -2,7 +2,7 @@ import extras.filtersets
 import extras.tables
 from django.db import models
 from extras.models import JournalEntry
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from utilities.utils import count_related
 
 

netbox/ipam/search_indexes.py

@@ -2,7 +2,7 @@ import ipam.filtersets
 import ipam.tables
 from django.db import models
 from ipam.models import ASN, VLAN, VRF, Aggregate, IPAddress, Prefix, Service
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from utilities.utils import count_related
 
 

netbox/netbox/forms/__init__.py

@@ -1,7 +1,7 @@
 from django import forms
-from utilities.forms import BootstrapMixin
 
-from search.backends import default_search_engine
+from netbox.search.backends import default_search_engine
+from utilities.forms import BootstrapMixin
 
 from .base import *
 
@@ -25,7 +25,11 @@ class SearchForm(BootstrapMixin, forms.Form):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.fields["obj_type"] = forms.ChoiceField(choices=default_search_engine.get_search_choices(), required=False, label='Type')
+        self.fields["obj_type"] = forms.ChoiceField(
+            choices=default_search_engine.get_search_choices(),
+            required=False,
+            label='Type'
+        )
 
     def get_options(self):
         if not self.options:

netbox/netbox/search/backends.py

@@ -1,9 +1,7 @@
-from abc import ABC
 from importlib import import_module
 
 from django.conf import settings
 from django.core.exceptions import ImproperlyConfigured
-from django.db import models
 from django.db.models.signals import post_save, pre_delete
 from django.urls import reverse
 from netbox.constants import SEARCH_MAX_RESULTS
@@ -146,7 +144,7 @@ def get_backend(backend_name=None):
     """Initializes and returns the search backend."""
     global _backends_cache
     if not backend_name:
-        backend_name = getattr(settings, "SEARCH_BACKEND", "search.backends.PostgresIcontainsSearchBackend")
+        backend_name = getattr(settings, "SEARCH_BACKEND", "netbox.search.backends.PostgresIcontainsSearchBackend")
 
     # Try to use the cached backend.
     if backend_name in _backends_cache:

netbox/netbox/search/models.py

@@ -1,7 +1,3 @@
-from django.core.exceptions import ImproperlyConfigured
-from django.db import models
-
-
 class SearchMixin(object):
     """
     Base class for building search indexes.

netbox/netbox/search/register.py

@@ -0,0 +1,39 @@
+import importlib
+import inspect
+import sys
+
+from django.apps import apps
+from django.conf import settings
+from django.utils.module_loading import module_has_submodule
+
+from .backends import default_search_engine
+from .hierarchy import SEARCH_TYPES
+
+
+def get_app_modules():
+    """
+    Returns all app modules (installed apps) - yields tuples of (app_name, module)
+    """
+    for app in apps.get_app_configs():
+        yield app.name, app.module
+
+
+def register():
+    for name, module in SEARCH_TYPES.items():
+        default_search_engine.register(name, module)
+
+    for name, module in get_app_modules():
+        submodule_name = "search_indexes"
+        if module_has_submodule(module, submodule_name):
+            module_name = f"{name}.{submodule_name}"
+            if name in settings.PLUGINS:
+                search_module = importlib.import_module(module_name)
+            else:
+                search_module = sys.modules[module_name]
+
+            cls_objects = inspect.getmembers(search_module, predicate=inspect.isclass)
+            for cls_name, cls_obj in inspect.getmembers(search_module, predicate=inspect.isclass):
+                if getattr(cls_obj, "search_index", False) and getattr(cls_obj, "model", None):
+                    cls_name = cls_obj.model.__name__.lower()
+                    if not default_search_engine.is_registered(cls_name, cls_obj):
+                        default_search_engine.register(cls_name, cls_obj)

netbox/netbox/settings.py

@@ -319,7 +319,6 @@ INSTALLED_APPS = [
     'social_django',
     'taggit',
     'timezone_field',
-    'search',
     'circuits',
     'dcim',
     'ipam',

netbox/netbox/views/__init__.py

@@ -23,7 +23,7 @@ from extras.tables import ObjectChangeTable
 from ipam.models import Aggregate, IPAddress, IPRange, Prefix, VLAN, VRF
 from netbox.constants import SEARCH_MAX_RESULTS
 from netbox.forms import SearchForm
-from search.backends import default_search_engine
+from netbox.search.backends import default_search_engine
 from tenancy.models import Tenant
 from virtualization.models import Cluster, VirtualMachine
 from wireless.models import WirelessLAN, WirelessLink

netbox/search/apps.py

@@ -1,45 +0,0 @@
-import importlib
-import inspect
-import sys
-from django.apps import AppConfig
-
-from django.apps import apps
-from django.conf import settings
-from django.utils.module_loading import module_has_submodule
-from netbox import denormalized
-
-
-def get_app_modules():
-    """
-    Returns all app modules (installed apps) - yields tuples of (app_name, module)
-    """
-    for app in apps.get_app_configs():
-        yield app.name, app.module
-
-
-class SearchConfig(AppConfig):
-    name = "search"
-    verbose_name = "search"
-
-    def ready(self):
-        from .backends import default_search_engine
-        from .hierarchy import SEARCH_TYPES
-
-        for name, module in SEARCH_TYPES.items():
-            default_search_engine.register(name, module)
-
-        for name, module in get_app_modules():
-            submodule_name = "search_indexes"
-            if module_has_submodule(module, submodule_name):
-                module_name = f"{name}.{submodule_name}"
-                if name in settings.PLUGINS:
-                    search_module = importlib.import_module(module_name)
-                else:
-                    search_module = sys.modules[module_name]
-
-                cls_objects = inspect.getmembers(search_module, predicate=inspect.isclass)
-                for cls_name, cls_obj in inspect.getmembers(search_module, predicate=inspect.isclass):
-                    if getattr(cls_obj, "search_index", False) and getattr(cls_obj, "model", None):
-                        cls_name = cls_obj.model.__name__.lower()
-                        if not default_search_engine.is_registered(cls_name, cls_obj):
-                            default_search_engine.register(cls_name, cls_obj)

netbox/tenancy/search_indexes.py

@@ -1,7 +1,7 @@
 import tenancy.filtersets
 import tenancy.tables
 from django.db import models
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from tenancy.models import Contact, ContactAssignment, Tenant
 from utilities.utils import count_related
 

netbox/utilities/templatetags/search.py

@@ -1,9 +1,10 @@
 from typing import Dict
-from netbox.forms import SearchForm
+
 from django import template
 
-register = template.Library()
+from netbox.forms import SearchForm
 
+register = template.Library()
 search_form = None
 
 
@@ -14,7 +15,7 @@ def search_options(request) -> Dict:
     if not search_form:
         search_form = SearchForm()
 
-    """Provide search options to template."""
+    # Provide search options to template.
     return {
         'options': search_form.get_options(),
         'request': request,

netbox/virtualization/search_indexes.py

@@ -1,7 +1,7 @@
 import virtualization.filtersets
 import virtualization.tables
 from django.db import models
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from utilities.utils import count_related
 from virtualization.models import Cluster, Device, VirtualMachine
 

netbox/wireless/search_indexes.py

@@ -2,7 +2,7 @@ import wireless.filtersets
 import wireless.tables
 from dcim.models import Interface
 from django.db import models
-from search.models import SearchMixin
+from netbox.search.models import SearchMixin
 from utilities.utils import count_related
 from wireless.models import WirelessLAN, WirelessLink