nautobot 2.1.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

nautobot/core/tests/test_navigations.py

@@ -1,6 +1,7 @@
-from django.test import TestCase
 from unittest.mock import patch
 
+from django.test import TestCase
+
 from nautobot.core.apps import NAV_CONTEXT_NAMES, NavContext, NavGrouping, NavItem, register_new_ui_menu_items
 
 

nautobot/core/tests/test_openapi.py

@@ -1,6 +1,5 @@
 from drf_spectacular.renderers import OpenApiYamlRenderer
 from drf_spectacular.settings import spectacular_settings
-
 import yaml
 
 from nautobot.core.testing import TestCase

nautobot/core/tests/test_paginator.py

@@ -3,13 +3,13 @@
 from constance.test import override_config
 from django.conf import settings
 from django.contrib.auth import get_user_model
-from django.test import RequestFactory, override_settings
+from django.test import override_settings, RequestFactory
 from django.urls import reverse
 
 from nautobot.circuits import models as circuits_models
-from nautobot.dcim import models as dcim_models
 from nautobot.core import testing
 from nautobot.core.views import paginator
+from nautobot.dcim import models as dcim_models
 from nautobot.extras import models as extras_models
 
 

nautobot/core/tests/test_releases.py

@@ -2,11 +2,11 @@ from io import BytesIO
 from logging import ERROR
 from unittest.mock import Mock, patch
 
-import requests
 from django.conf import settings
 from django.core.cache import cache
-from django.test import SimpleTestCase, override_settings
+from django.test import override_settings, SimpleTestCase
 from packaging import version
+import requests
 from requests import Response
 
 from nautobot.core.tasks import get_releases

nautobot/core/tests/test_tables.py

@@ -1,6 +1,6 @@
 from django.test import TestCase
 
-from nautobot.dcim.tables import ManufacturerTable, DeviceTypeTable, LocationTable, LocationTypeTable, RackGroupTable
+from nautobot.dcim.tables import DeviceTypeTable, LocationTable, LocationTypeTable, ManufacturerTable, RackGroupTable
 
 
 class TableTestCase(TestCase):

nautobot/core/tests/test_templatetags_helpers.py

@@ -3,13 +3,14 @@ from unittest import skipIf
 from constance.test import override_config
 from django.conf import settings
 from django.templatetags.static import static
-from django.test import TestCase, override_settings
-from example_plugin.models import AnotherExampleModel, ExampleModel
+from django.test import override_settings, TestCase
 
 from nautobot.core.templatetags import helpers
 from nautobot.dcim import models
 from nautobot.ipam.models import VLAN
 
+from example_plugin.models import AnotherExampleModel, ExampleModel
+
 
 @skipIf(
     "example_plugin" not in settings.PLUGINS,

nautobot/core/tests/test_templatetags_netutils.py

@@ -3,7 +3,6 @@ import inspect
 from django.template import Context, Engine
 from django.template.exceptions import TemplateSyntaxError
 from django.test import TestCase
-
 from netutils.utils import jinja2_convenience_function
 
 
@@ -47,6 +46,6 @@ class NautobotTemplateTagsNetutilsTest(TestCase):
                 except ConnectionError:
                     # Yes, some netutils functions such as tcp_ping actually make network calls when invoked. Eeek.
                     pass
-                except Exception:
+                except Exception:  # noqa: S110  # try-except-pass -- an antipattern in general, but OK here
                     # Catch-all - at least it wasn't a TemplateSyntaxError, so good enough for now.
                     pass

nautobot/core/tests/test_utils.py

@@ -8,24 +8,20 @@ from django.db.models import Q
 from django.http import QueryDict
 from django.test import TestCase
 
-from example_plugin.models import ExampleModel
-
 from nautobot.circuits import models as circuits_models
 from nautobot.core import exceptions, forms, settings_funcs
 from nautobot.core.api import utils as api_utils
 from nautobot.core.models import fields as core_fields, utils as models_utils
 from nautobot.core.utils import data as data_utils, filtering, lookup, requests
 from nautobot.core.utils.migrations import update_object_change_ct_for_replaced_models
-from nautobot.dcim import filters as dcim_filters
-from nautobot.dcim import forms as dcim_forms
-from nautobot.dcim import models as dcim_models
-from nautobot.dcim import tables
-from nautobot.extras import models as extras_models
-from nautobot.extras import utils as extras_utils
+from nautobot.dcim import filters as dcim_filters, forms as dcim_forms, models as dcim_models, tables
+from nautobot.extras import models as extras_models, utils as extras_utils
 from nautobot.extras.choices import ObjectChangeActionChoices, RelationshipTypeChoices
 from nautobot.extras.models import ObjectChange
 from nautobot.extras.registry import registry
 
+from example_plugin.models import ExampleModel
+
 
 class DictToFilterParamsTest(TestCase):
     """

nautobot/core/tests/test_views.py

@@ -4,7 +4,7 @@ import urllib.parse
 
 from django.contrib.contenttypes.models import ContentType
 from django.core.files.uploadedfile import SimpleUploadedFile
-from django.test import RequestFactory, override_settings
+from django.test import override_settings, RequestFactory
 from django.test.utils import override_script_prefix
 from django.urls import get_script_prefix, reverse
 from prometheus_client.parser import text_string_to_metric_families
@@ -41,7 +41,10 @@ class GetReturnURLMixinTestCase(TestCase):
         self.assertEqual(self.mixin.get_return_url(request=request, obj=None), reverse("home"))
 
     def test_get_return_url_explicit_punycode(self):
-        request = self.factory.get("/", {"return_url": "/dcım/devices/"})
+        """
+        Replace the 'i' in '/dcim/' with a unicode dotless 'ı' and make sure we're not fooled by it.
+        """  # noqa: RUF002  # ambiguous-unicode-character-docstring -- fully intentional here!
+        request = self.factory.get("/", {"return_url": "/dcım/devices/"})  # noqa: RUF001  # ambiguous-unicode-character-string -- fully intentional here!
         self.assertEqual(self.mixin.get_return_url(request=request, obj=None), "/dc%C4%B1m/devices/")
 
     def test_get_return_url_default_with_obj(self):

nautobot/core/tests/test_views_utils.py

@@ -3,7 +3,6 @@ from django.test import TestCase
 
 from nautobot.core.models.querysets import count_related
 from nautobot.core.views.utils import check_filter_for_display
-
 from nautobot.dcim.filters import DeviceFilterSet
 from nautobot.dcim.models import Device, DeviceRedundancyGroup, DeviceType, InventoryItem, Location, Manufacturer
 from nautobot.extras.models import Role, Status

nautobot/core/urls.py

@@ -6,20 +6,20 @@ from django.views.static import serve
 
 from nautobot.core.views import (
     CustomGraphQLView,
+    get_file_with_authorization,
     HomeView,
-    StaticMediaFailureView,
+    nautobot_metrics_view,
     SearchView,
+    StaticMediaFailureView,
     ThemePreviewView,
-    nautobot_metrics_view,
-    get_file_with_authorization,
 )
 from nautobot.extras.plugins.urls import (
     plugin_admin_patterns,
     plugin_patterns,
 )
 from nautobot.users.views import LoginView, LogoutView
-from .admin import admin_site
 
+from .admin import admin_site
 
 urlpatterns = [
     # Base views

nautobot/core/utils/color.py

@@ -5,7 +5,7 @@ def hex_to_rgb(hex_str):
     """
     Map a hex string like "00ff00" to individual r, g, b integer values.
     """
-    return [int(hex_str[c : c + 2], 16) for c in (0, 2, 4)]  # noqa: E203
+    return [int(hex_str[c : c + 2], 16) for c in (0, 2, 4)]
 
 
 def rgb_to_hex(r, g, b):

nautobot/core/utils/data.py

@@ -1,4 +1,4 @@
-from collections import OrderedDict, namedtuple
+from collections import namedtuple, OrderedDict
 from decimal import Decimal
 import uuid
 
@@ -7,7 +7,6 @@ from django.template import engines
 
 from nautobot.dcim import choices  # TODO move dcim.choices.CableLengthUnitChoices into core
 
-
 # Setup UtilizationData named tuple for use by multiple methods
 UtilizationData = namedtuple("UtilizationData", ["numerator", "denominator"])
 

nautobot/core/utils/filtering.py

@@ -13,7 +13,6 @@ from django_filters.utils import verbose_lookup_expr
 from nautobot.core import exceptions
 from nautobot.core.utils.lookup import get_filterset_for_model
 
-
 # Check if field name contains a lookup expr
 # e.g `name__ic` has lookup expr `ic (icontains)` while `name` has no lookup expr
 CONTAINS_LOOKUP_EXPR_RE = re.compile(r"(?<=__)\w+")
@@ -88,10 +87,6 @@ def get_filterset_parameter_form_field(model, parameter, filterset=None):
     Return the relevant form field instance for a filterset parameter e.g DynamicModelMultipleChoiceField, forms.IntegerField e.t.c
     """
     # Avoid circular import
-    from nautobot.dcim.models import Device
-    from nautobot.extras.filters import ContentTypeMultipleChoiceFilter, CustomFieldFilterMixin, StatusFilter
-    from nautobot.extras.models import ConfigContext, Role, Status, Tag
-    from nautobot.extras.utils import ChangeLoggedModelsQuery, RoleModelsQuery, TaggableClassesQuery
     from nautobot.core.filters import MultiValueDecimalFilter, MultiValueFloatFilter
     from nautobot.core.forms import (
         BOOLEAN_CHOICES,
@@ -100,6 +95,10 @@ def get_filterset_parameter_form_field(model, parameter, filterset=None):
         StaticSelect2,
         StaticSelect2Multiple,
     )
+    from nautobot.dcim.models import Device
+    from nautobot.extras.filters import ContentTypeMultipleChoiceFilter, CustomFieldFilterMixin, StatusFilter
+    from nautobot.extras.models import ConfigContext, Role, Status, Tag
+    from nautobot.extras.utils import ChangeLoggedModelsQuery, RoleModelsQuery, TaggableClassesQuery
     from nautobot.virtualization.models import VirtualMachine
 
     if filterset is None or filterset.Meta.model != model:

nautobot/core/utils/git.py

@@ -5,6 +5,7 @@ import logging
 import os
 
 from git import Repo
+
 from nautobot.core.utils.logging import sanitize
 
 logger = logging.getLogger(__name__)
@@ -121,7 +122,7 @@ class GitRepo:
                 branch_head.set_tracking_branch(self.repo.remotes.origin.refs[branch])
             except IndexError as git_error:
                 logger.error(
-                    "Branch %s does not exist at %s. %s", branch, list(self.repo.remotes.origin.urls)[0], git_error
+                    "Branch %s does not exist at %s. %s", branch, next(iter(self.repo.remotes.origin.urls)), git_error
                 )
                 raise BranchDoesNotExist(
                     f"Please create branch '{branch}' in upstream and try again."
@@ -146,7 +147,7 @@ class GitRepo:
             self.repo.remotes.origin.refs[branch]
         except IndexError as git_error:
             logger.error(
-                "Branch %s does not exist at %s. %s", branch, list(self.repo.remotes.origin.urls)[0], git_error
+                "Branch %s does not exist at %s. %s", branch, next(iter(self.repo.remotes.origin.urls)), git_error
             )
             raise BranchDoesNotExist(
                 f"Please create branch '{branch}' in upstream and try again."

nautobot/core/utils/logging.py

@@ -17,7 +17,8 @@ def sanitize(string, replacement="(redacted)"):
     Uses settings.SANITIZER_PATTERNS as the list of (regexp, repl) tuples to apply.
     """
     # Don't allow regex match groups to be referenced in the replacement string!
-    assert not re.search(r"\\\d|\\g<\d+>", replacement)
+    if re.search(r"\\\d|\\g<\d+>", replacement):
+        raise RuntimeError("Invalid replacement string! Must not contain regex match group references.")
 
     for sanitizer, repl in settings.SANITIZER_PATTERNS:
         try:

nautobot/core/utils/migrations.py

@@ -1,4 +1,5 @@
 import sys
+
 from django.db import models
 
 

nautobot/core/utils/permissions.py

@@ -58,12 +58,10 @@ def permission_is_exempt(name):
     if action == "view":
         if (
             # All models (excluding those in EXEMPT_EXCLUDE_MODELS) are exempt from view permission enforcement
-            "*" in settings.EXEMPT_VIEW_PERMISSIONS
-            and (app_label, model_name) not in settings.EXEMPT_EXCLUDE_MODELS
+            "*" in settings.EXEMPT_VIEW_PERMISSIONS and (app_label, model_name) not in settings.EXEMPT_EXCLUDE_MODELS
         ) or (
             # This specific model is exempt from view permission enforcement
-            f"{app_label}.{model_name}"
-            in settings.EXEMPT_VIEW_PERMISSIONS
+            f"{app_label}.{model_name}" in settings.EXEMPT_VIEW_PERMISSIONS
         ):
             return True
 

nautobot/core/views/__init__.py

@@ -5,22 +5,22 @@ import time
 
 from db_file_storage.views import get_file
 from django.apps import apps
-import prometheus_client
 from django.conf import settings
 from django.contrib.auth.decorators import permission_required
 from django.contrib.auth.mixins import AccessMixin, LoginRequiredMixin
 from django.contrib.contenttypes.models import ContentType
-from django.http import HttpResponseServerError, JsonResponse, HttpResponseForbidden, HttpResponse
+from django.http import HttpResponse, HttpResponseForbidden, HttpResponseServerError, JsonResponse
 from django.shortcuts import get_object_or_404, redirect, render
 from django.template import loader, RequestContext, Template
 from django.template.exceptions import TemplateDoesNotExist
 from django.urls import resolve, reverse
+from django.views.csrf import csrf_failure as _csrf_failure
 from django.views.decorators.csrf import requires_csrf_token
 from django.views.defaults import ERROR_500_TEMPLATE_NAME, page_not_found
-from django.views.csrf import csrf_failure as _csrf_failure
 from django.views.generic import TemplateView, View
-from packaging import version
 from graphene_django.views import GraphQLView
+from packaging import version
+import prometheus_client
 from prometheus_client import multiprocess
 from prometheus_client.metrics_core import GaugeMetricFamily
 from prometheus_client.registry import Collector
@@ -30,9 +30,9 @@ from nautobot.core.forms import SearchForm
 from nautobot.core.releases import get_latest_release
 from nautobot.core.utils.lookup import get_route_for_model
 from nautobot.core.utils.permissions import get_permission_for_model
+from nautobot.extras.forms import GraphQLQueryForm
 from nautobot.extras.models import FileProxy, GraphQLQuery, Status
 from nautobot.extras.registry import registry
-from nautobot.extras.forms import GraphQLQueryForm
 
 
 class HomeView(AccessMixin, TemplateView):

nautobot/core/views/generic.py

@@ -11,7 +11,7 @@ from django.core.exceptions import (
     ObjectDoesNotExist,
     ValidationError,
 )
-from django.db import transaction, IntegrityError
+from django.db import IntegrityError, transaction
 from django.db.models import ManyToManyField, ProtectedError
 from django.forms import Form, ModelMultipleChoiceField, MultipleHiddenInput
 from django.http import HttpResponse, JsonResponse
@@ -25,7 +25,6 @@ from rest_framework.exceptions import ParseError
 
 from nautobot.core.api.parsers import NautobotCSVParser
 from nautobot.core.api.utils import get_serializer_for_model
-from nautobot.core.forms import SearchForm
 from nautobot.core.exceptions import AbortTransaction
 from nautobot.core.forms import (
     BootstrapMixin,
@@ -34,8 +33,9 @@ from nautobot.core.forms import (
     CSVDataField,
     CSVFileField,
     ImportForm,
-    TableConfigForm,
     restrict_form_fields,
+    SearchForm,
+    TableConfigForm,
 )
 from nautobot.core.forms.forms import DynamicFilterFormSet
 from nautobot.core.templatetags.helpers import bettertitle, validated_viewname
@@ -47,8 +47,8 @@ from nautobot.core.utils.requests import (
     get_filterable_params_from_filter_params,
     normalize_querydict,
 )
-from nautobot.core.views.paginator import EnhancedPaginator, get_paginate_count
 from nautobot.core.views.mixins import GetReturnURLMixin, ObjectPermissionRequiredMixin
+from nautobot.core.views.paginator import EnhancedPaginator, get_paginate_count
 from nautobot.core.views.utils import (
     check_filter_for_display,
     get_csv_form_fields_from_serializer_class,

nautobot/core/views/mixins.py

@@ -2,8 +2,8 @@ from io import BytesIO
 import logging
 
 from django.contrib import messages
-from django.contrib.contenttypes.models import ContentType
 from django.contrib.auth.mixins import AccessMixin
+from django.contrib.contenttypes.models import ContentType
 from django.core.exceptions import (
     FieldDoesNotExist,
     ImproperlyConfigured,
@@ -19,18 +19,16 @@ from django.template.loader import select_template, TemplateDoesNotExist
 from django.urls import reverse
 from django.urls.exceptions import NoReverseMatch
 from django.utils.encoding import iri_to_uri
-from django.utils.http import url_has_allowed_host_and_scheme
 from django.utils.html import format_html
+from django.utils.http import url_has_allowed_host_and_scheme
 from django.views.generic.edit import FormView
-
-from rest_framework import mixins, exceptions
+from drf_spectacular.utils import extend_schema
+from rest_framework import exceptions, mixins
 from rest_framework.decorators import action as drf_action
 from rest_framework.parsers import FormParser, MultiPartParser
 from rest_framework.response import Response
 from rest_framework.viewsets import GenericViewSet
 
-from drf_spectacular.utils import extend_schema
-
 from nautobot.core.api.parsers import NautobotCSVParser
 from nautobot.core.api.views import BulkDestroyModelMixin, BulkUpdateModelMixin
 from nautobot.core.forms import (
@@ -41,16 +39,16 @@ from nautobot.core.forms import (
     restrict_form_fields,
 )
 from nautobot.core.utils import lookup, permissions
-from nautobot.core.views.renderers import NautobotHTMLRenderer
 from nautobot.core.utils.requests import get_filterable_params_from_filter_params
+from nautobot.core.views.renderers import NautobotHTMLRenderer
 from nautobot.core.views.utils import (
     get_csv_form_fields_from_serializer_class,
     handle_protectederror,
     prepare_cloned_fields,
 )
-from nautobot.extras.models import ExportTemplate
 from nautobot.extras.forms import NoteForm
-from nautobot.extras.tables import ObjectChangeTable, NoteTable
+from nautobot.extras.models import ExportTemplate
+from nautobot.extras.tables import NoteTable, ObjectChangeTable
 from nautobot.extras.utils import remove_prefix_from_cf_key
 
 PERMISSIONS_ACTION_MAP = {
@@ -305,9 +303,10 @@ class NautobotViewSetMixin(GenericViewSet, AccessMixin, GetReturnURLMixin, FormV
         elif self.action == "changelog":
             return ObjectChangeTable
 
-        assert (
-            self.table_class is not None
-        ), f"'{self.__class__.__name__}' should include a `table_class` attribute for bulk operations"
+        if self.table_class is None:
+            raise NotImplementedError(
+                f"'{self.__class__.__name__}' should include a `table_class` attribute for bulk operations"
+            )
 
         return self.table_class
 

nautobot/core/views/renderers.py

@@ -7,9 +7,9 @@ from django_tables2 import RequestConfig
 from rest_framework import renderers
 
 from nautobot.core.forms import (
+    restrict_form_fields,
     SearchForm,
     TableConfigForm,
-    restrict_form_fields,
 )
 from nautobot.core.forms.forms import DynamicFilterFormSet
 from nautobot.core.templatetags.helpers import bettertitle, validated_viewname

nautobot/core/views/utils.py

@@ -5,7 +5,6 @@ from django.core.exceptions import FieldError
 from django.db.models import ForeignKey
 from django.utils.html import format_html, format_html_join
 from django.utils.safestring import mark_safe
-
 from rest_framework import serializers
 
 from nautobot.core.api.fields import ChoiceField, ContentTypeField, TimeZoneSerializerField
@@ -97,6 +96,8 @@ def get_csv_form_fields_from_serializer_class(serializer_class):
     """From the given serializer class, build a list of field dicts suitable for rendering in the CSV import form."""
     serializer = serializer_class(context={"request": None, "depth": 0})
     fields = []
+    # Note lots of "noqa: S308" in this function. That's `suspicious-mark-safe-usage`, but in all of the below cases
+    # we control the input string and it's known to be safe, so mark_safe() is being used correctly here.
     for field_name, field in serializer.fields.items():
         if field.read_only:
             continue

nautobot/dcim/api/urls.py

@@ -1,6 +1,6 @@
 from nautobot.core.api.routers import OrderedDefaultRouter
-from . import views
 
+from . import views
 
 router = OrderedDefaultRouter()
 router.APIRootView = views.DCIMRootView

nautobot/dcim/api/views.py

@@ -1,10 +1,10 @@
-import socket
 from collections import OrderedDict
+import socket
 
 from django.conf import settings
 from django.core.exceptions import ObjectDoesNotExist
 from django.db.models import F
-from django.http import HttpResponseForbidden, HttpResponse
+from django.http import HttpResponse, HttpResponseForbidden
 from django.shortcuts import get_object_or_404
 from django.views.decorators.clickjacking import xframe_options_sameorigin
 from drf_spectacular.types import OpenApiTypes
@@ -39,10 +39,10 @@ from nautobot.dcim.models import (
     InterfaceRedundancyGroup,
     InterfaceRedundancyGroupAssociation,
     InterfaceTemplate,
+    InventoryItem,
     Location,
     LocationType,
     Manufacturer,
-    InventoryItem,
     Platform,
     PowerFeed,
     PowerOutlet,
@@ -65,6 +65,7 @@ from nautobot.extras.choices import SecretsGroupAccessTypeChoices, SecretsGroupS
 from nautobot.extras.secrets.exceptions import SecretError
 from nautobot.ipam.models import Prefix, VLAN
 from nautobot.virtualization.models import VirtualMachine
+
 from . import serializers
 from .exceptions import MissingFilterException
 
@@ -468,9 +469,9 @@ class DeviceViewSet(ConfigContextQuerySetMixin, NautobotModelViewSet):
         # Get NAPALM enable-secret from the device if present
         if device.secrets_group:
             # Work around inconsistent enable password arg in NAPALM drivers
-            enable_password_arg = "secret"
+            enable_password_arg = "secret"  # noqa: S105  # hardcoded-password-string -- false positive
             if device.platform.napalm_driver.lower() == "eos":
-                enable_password_arg = "enable_password"
+                enable_password_arg = "enable_password"  # noqa: S105  # hardcoded-password-string -- false positive
             try:
                 optional_args[enable_password_arg] = device.secrets_group.get_secret_value(
                     SecretsGroupAccessTypeChoices.TYPE_GENERIC,

nautobot/dcim/apps.py

@@ -17,4 +17,4 @@ class DCIMConfig(NautobotConfig):
 
     def ready(self):
         super().ready()
-        import nautobot.dcim.signals  # noqa: F401
+        import nautobot.dcim.signals  # noqa: F401  # unused-import -- but this import installs the signals

nautobot/dcim/choices.py

@@ -1,6 +1,5 @@
 from nautobot.core.choices import ChoiceSet
 
-
 #
 # Locations
 #

nautobot/dcim/constants.py

@@ -2,7 +2,6 @@ from django.db.models import Q
 
 from .choices import InterfaceTypeChoices
 
-
 #
 # Racks
 #

nautobot/dcim/elevations.py

@@ -1,8 +1,7 @@
-import svgwrite
-
 from django.conf import settings
 from django.urls import reverse
 from django.utils.http import urlencode
+import svgwrite
 
 from .choices import DeviceFaceChoices
 from .constants import RACK_ELEVATION_BORDER_WIDTH

nautobot/dcim/factory.py

@@ -1,22 +1,20 @@
-import factory
 import logging
-import pytz
-import random
-
-from faker import Faker
 
 from django.contrib.auth import get_user_model
 from django.contrib.contenttypes.models import ContentType
 from django.db.models import Q
+import factory
+from faker import Faker
+import pytz
 
+from nautobot.circuits.models import CircuitTermination
 from nautobot.core.factory import (
     NautobotBoolIterator,
     OrganizationalModelFactory,
     PrimaryModelFactory,
-    UniqueFaker,
     random_instance,
+    UniqueFaker,
 )
-from nautobot.circuits.models import CircuitTermination
 from nautobot.dcim.choices import (
     DeviceRedundancyGroupFailoverStrategyChoices,
     RackDimensionUnitChoices,
@@ -28,14 +26,14 @@ from nautobot.dcim.models import (
     Device,
     DeviceRedundancyGroup,
     DeviceType,
-    Manufacturer,
-    Platform,
     Location,
     LocationType,
+    Manufacturer,
+    Platform,
+    PowerPanel,
     Rack,
     RackGroup,
     RackReservation,
-    PowerPanel,
 )
 from nautobot.extras.models import Role, Status
 from nautobot.extras.utils import FeatureQuery
@@ -43,7 +41,6 @@ from nautobot.ipam.models import Prefix, VLAN, VLANGroup
 from nautobot.tenancy.models import Tenant
 from nautobot.virtualization.models import Cluster
 
-
 logger = logging.getLogger(__name__)
 
 User = get_user_model()
@@ -298,7 +295,7 @@ class PlatformFactory(OrganizationalModelFactory):
     # If it has a manufacturer, it *might* have a napalm_driver.
     napalm_driver = factory.Maybe(
         "has_manufacturer",
-        factory.LazyAttribute(lambda o: random.choice(NAPALM_DRIVERS.get(o.manufacturer.name, [""]))),
+        factory.LazyAttribute(lambda o: factory.random.randgen.choice(NAPALM_DRIVERS.get(o.manufacturer.name, [""]))),
         "",
     )
 
@@ -310,7 +307,7 @@ class PlatformFactory(OrganizationalModelFactory):
     has_description = NautobotBoolIterator()
     network_driver = factory.Maybe(
         "has_manufacturer",
-        factory.LazyAttribute(lambda o: random.choice(NETWORK_DRIVERS.get(o.manufacturer.name, [""]))),
+        factory.LazyAttribute(lambda o: factory.random.randgen.choice(NETWORK_DRIVERS.get(o.manufacturer.name, [""]))),
         "",
     )
 

nautobot/dcim/fields.py

@@ -3,6 +3,7 @@ from django.db import models
 
 from nautobot.core.models.fields import JSONArrayField
 from nautobot.ipam.constants import BGP_ASN_MAX, BGP_ASN_MIN
+
 from .lookups import PathContains