Refactor imports

This commit is contained in:
Jeremy Stretch 2020-07-09 09:13:58 -04:00
parent 0a44ed1355
commit b535608519

View File

@ -8,14 +8,13 @@ from django.core.exceptions import FieldError, MultipleObjectsReturned, ObjectDo
from django.db import transaction from django.db import transaction
from django.db.models import ManyToManyField, ProtectedError from django.db.models import ManyToManyField, ProtectedError
from django.urls import reverse from django.urls import reverse
from rest_framework.exceptions import APIException from rest_framework import serializers
from rest_framework.exceptions import APIException, ValidationError
from rest_framework.permissions import BasePermission from rest_framework.permissions import BasePermission
from rest_framework.relations import PrimaryKeyRelatedField, RelatedField from rest_framework.relations import PrimaryKeyRelatedField, RelatedField
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.serializers import Field, ModelSerializer, ValidationError
from rest_framework.viewsets import ModelViewSet as _ModelViewSet from rest_framework.viewsets import ModelViewSet as _ModelViewSet
from netbox.api import TokenPermissions
from .utils import dict_to_filter_params, dynamic_import from .utils import dict_to_filter_params, dynamic_import
@ -70,7 +69,7 @@ class IsAuthenticatedOrLoginNotRequired(BasePermission):
# Fields # Fields
# #
class ChoiceField(Field): class ChoiceField(serializers.Field):
""" """
Represent a ChoiceField as {'value': <DB value>, 'label': <string>}. Accepts a single value on write. Represent a ChoiceField as {'value': <DB value>, 'label': <string>}. Accepts a single value on write.
@ -177,7 +176,7 @@ class ContentTypeField(RelatedField):
return "{}.{}".format(obj.app_label, obj.model) return "{}.{}".format(obj.app_label, obj.model)
class TimeZoneField(Field): class TimeZoneField(serializers.Field):
""" """
Represent a pytz time zone. Represent a pytz time zone.
""" """
@ -212,7 +211,7 @@ class SerializedPKRelatedField(PrimaryKeyRelatedField):
# TODO: We should probably take a fresh look at exactly what we're doing with this. There might be a more elegant # TODO: We should probably take a fresh look at exactly what we're doing with this. There might be a more elegant
# way to enforce model validation on the serializer. # way to enforce model validation on the serializer.
class ValidatedModelSerializer(ModelSerializer): class ValidatedModelSerializer(serializers.ModelSerializer):
""" """
Extends the built-in ModelSerializer to enforce calling clean() on the associated model during validation. Extends the built-in ModelSerializer to enforce calling clean() on the associated model during validation.
""" """
@ -241,7 +240,7 @@ class ValidatedModelSerializer(ModelSerializer):
return data return data
class WritableNestedSerializer(ModelSerializer): class WritableNestedSerializer(serializers.ModelSerializer):
""" """
Returns a nested representation of an object on read, but accepts only a primary key on write. Returns a nested representation of an object on read, but accepts only a primary key on write.
""" """