Refactor register_model() to accept multiple models

netbox/circuits/apps.py

@@ -6,9 +6,8 @@ class CircuitsConfig(AppConfig):
     verbose_name = "Circuits"
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import signals, search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/core/apps.py

@@ -17,9 +17,8 @@ class CoreConfig(AppConfig):
 
     def ready(self):
         from core.api import schema  # noqa
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import data_backends, search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/dcim/apps.py

@@ -8,14 +8,13 @@ class DCIMConfig(AppConfig):
     verbose_name = "DCIM"
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from utilities.counters import connect_counters
         from . import signals, search
         from .models import CableTermination, Device, DeviceType, VirtualChassis
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())
 
         # Register denormalized fields
         denormalized.register(CableTermination, '_device', {

netbox/extras/apps.py

@@ -5,9 +5,8 @@ class ExtrasConfig(AppConfig):
     name = "extras"
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import dashboard, lookups, search, signals
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/ipam/apps.py

@@ -6,9 +6,8 @@ class IPAMConfig(AppConfig):
     verbose_name = "IPAM"
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import signals, search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/netbox/models/features.py

@@ -35,7 +35,7 @@ __all__ = (
     'JournalingMixin',
     'SyncedDataMixin',
     'TagsMixin',
-    'register_model',
+    'register_models',
 )
 
 
@@ -575,9 +575,9 @@ registry['model_features'].update({
 })
 
 
-def register_model(model, **kwargs):
+def register_models(*models):
     """
-    Register a model in NetBox. This entails:
+    Register one or more models in NetBox. This entails:
 
      - Determining whether the model is considered "public" (available for reference by other models)
      - Registering which features the model supports (e.g. bookmarks, custom fields, etc.)
@@ -585,6 +585,7 @@ def register_model(model, **kwargs):
 
     register_model() should be called for each relevant model under the ready() of an app's AppConfig class.
     """
+    for model in models:
         app_label, model_name = model._meta.label_lower.split('.')
 
         # Register public models

netbox/netbox/plugins/__init__.py

@@ -94,11 +94,10 @@ class PluginConfig(AppConfig):
             pass
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())
 
         plugin_name = self.name.rsplit('.', 1)[-1]
 

netbox/tenancy/apps.py

@@ -5,9 +5,8 @@ class TenancyConfig(AppConfig):
     name = 'tenancy'
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/users/apps.py

@@ -5,9 +5,8 @@ class UsersConfig(AppConfig):
     name = 'users'
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import signals
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/virtualization/apps.py

@@ -7,14 +7,13 @@ class VirtualizationConfig(AppConfig):
     name = 'virtualization'
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from utilities.counters import connect_counters
         from . import search, signals
         from .models import VirtualMachine
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())
 
         # Register denormalized fields
         denormalized.register(VirtualMachine, 'cluster', {

netbox/vpn/apps.py

@@ -6,9 +6,8 @@ class VPNConfig(AppConfig):
     verbose_name = 'VPN'
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())

netbox/wireless/apps.py

@@ -5,9 +5,8 @@ class WirelessConfig(AppConfig):
     name = 'wireless'
 
     def ready(self):
-        from netbox.models.features import register_model
+        from netbox.models.features import register_models
         from . import signals, search
 
         # Register models
-        for model in self.get_models():
-            register_model(model)
+        register_models(*self.get_models())