Cleanup table name resolution

Author: arthanson

netbox_custom_objects/field_types.py

@@ -14,18 +14,27 @@
 from django.utils.translation import gettext_lazy as _
 from extras.choices import CustomFieldTypeChoices, CustomFieldUIEditableChoices
 from utilities.api import get_serializer_for_model
-from utilities.forms.fields import (CSVChoiceField, CSVMultipleChoiceField,
-                                    DynamicChoiceField,
-                                    DynamicMultipleChoiceField, JSONField,
-                                    LaxURLField)
+from utilities.forms.fields import (
+    CSVChoiceField,
+    CSVMultipleChoiceField,
+    DynamicChoiceField,
+    DynamicMultipleChoiceField,
+    JSONField,
+    LaxURLField,
+)
 from utilities.forms.utils import add_blank_choice
-from utilities.forms.widgets import (APISelect, APISelectMultiple, DatePicker,
-                                     DateTimePicker)
+from utilities.forms.widgets import (
+    APISelect,
+    APISelectMultiple,
+    DatePicker,
+    DateTimePicker,
+)
 from utilities.templatetags.builtins.filters import linkify, render_markdown
 
 from netbox_custom_objects.constants import APP_LABEL
 from netbox_custom_objects.utilities import get_custom_object_type_from_content_type
 
+
 class FieldType:
 
     def get_display_value(self, instance, field_name):
@@ -335,7 +344,9 @@ def get_model_field(self, field, **kwargs):
         if content_type.app_label == APP_LABEL:
             from netbox_custom_objects.models import CustomObjectType
 
-            custom_object_type_id = get_custom_object_type_from_content_type(content_type)
+            custom_object_type_id = get_custom_object_type_from_content_type(
+                content_type
+            )
             custom_object_type = CustomObjectType.objects.get(pk=custom_object_type_id)
             model = custom_object_type.get_model()
         else:
@@ -356,11 +367,14 @@ def get_form_field(self, field, for_csv_import=False, **kwargs):
         content_type = ContentType.objects.get(pk=field.related_object_type_id)
 
         from utilities.forms.fields import DynamicModelChoiceField
+
         if content_type.app_label == APP_LABEL:
             # This is a custom object type
             from netbox_custom_objects.models import CustomObjectType
 
-            custom_object_type_id = get_custom_object_type_from_content_type(content_type)
+            custom_object_type_id = get_custom_object_type_from_content_type(
+                content_type
+            )
             custom_object_type = CustomObjectType.objects.get(pk=custom_object_type_id)
             model = custom_object_type.get_model()
             field_class = DynamicModelChoiceField
@@ -641,7 +655,9 @@ def get_form_field(self, field, for_csv_import=False, **kwargs):
             # This is a custom object type
             from netbox_custom_objects.models import CustomObjectType
 
-            custom_object_type_id = get_custom_object_type_from_content_type(content_type)
+            custom_object_type_id = get_custom_object_type_from_content_type(
+                content_type
+            )
             custom_object_type = CustomObjectType.objects.get(pk=custom_object_type_id)
             model = custom_object_type.get_model()
         else:
@@ -701,7 +717,9 @@ def after_model_generation(self, instance, model, field_name):
         if content_type.app_label == APP_LABEL:
             from netbox_custom_objects.models import CustomObjectType
 
-            custom_object_type_id = get_custom_object_type_from_content_type(content_type)
+            custom_object_type_id = get_custom_object_type_from_content_type(
+                content_type
+            )
             custom_object_type = CustomObjectType.objects.get(pk=custom_object_type_id)
             to_model = custom_object_type.get_model()
         else:
@@ -741,7 +759,9 @@ def create_m2m_table(self, instance, model, field_name):
             if content_type.app_label == APP_LABEL:
                 from netbox_custom_objects.models import CustomObjectType
 
-                custom_object_type_id = get_custom_object_type_from_content_type(content_type)
+                custom_object_type_id = get_custom_object_type_from_content_type(
+                    content_type
+                )
                 custom_object_type = CustomObjectType.objects.get(
                     pk=custom_object_type_id
                 )