Hipo / drf-extra-fields

Extra Fields for Django Rest Framework
Apache License 2.0
670 stars 91 forks source link
django django-rest-framework

DRF-EXTRA-FIELDS

Extra Fields for Django Rest Framework

Build Status codecov PyPI Version Python Versions

Latest Changes

Usage

Install the package

pip install drf-extra-fields

Note:

Fields:

Base64ImageField

An image representation for Base64ImageField

Inherited from ImageField

Signature: Base64ImageField()

Example:

# serializer

from drf_extra_fields.fields import Base64ImageField

class UploadedBase64ImageSerializer(serializers.Serializer):
    file = Base64ImageField(required=False)
    created = serializers.DateTimeField()

# use the serializer
file = 'R0lGODlhAQABAIAAAP///////yH5BAEKAAEALAAAAAABAAEAAAICTAEAOw=='
serializer = UploadedBase64ImageSerializer(data={'created': now, 'file': file})

Base64FileField

A file representation for Base64FileField

Inherited from FileField

Signature: Base64FileField()

Example:

class PDFBase64File(Base64FileField):
    ALLOWED_TYPES = ['pdf']

    def get_file_extension(self, filename, decoded_file):
        try:
            PyPDF2.PdfFileReader(io.BytesIO(decoded_file))
        except PyPDF2.utils.PdfReadError as e:
            logger.warning(e)
        else:
            return 'pdf'

PointField

Point field for GeoDjango

Signature: PointField()

Example:

# serializer

from drf_extra_fields.geo_fields import PointField

class PointFieldSerializer(serializers.Serializer):
    point = PointField(required=False)
    created = serializers.DateTimeField()

# use the serializer
point = {
    "latitude": 49.8782482189424,
    "longitude": 24.452545489
    }
serializer = PointFieldSerializer(data={'created': now, 'point': point})

RangeField

The Range Fields map to Django's PostgreSQL specific Range Fields.

Each accepts an optional parameter child_attrs, which allows passing parameters to the child field.

For example, calling IntegerRangeField(child_attrs={"allow_null": True}) allows deserializing data with a null value for lower and/or upper:

from rest_framework import serializers
from drf_extra_fields.fields import IntegerRangeField

class RangeSerializer(serializers.Serializer):
    ranges = IntegerRangeField(child_attrs={"allow_null": True})

serializer = RangeSerializer(data={'ranges': {'lower': 0, 'upper': None}})

IntegerRangeField

from rest_framework import serializers
from drf_extra_fields.fields import IntegerRangeField

class RangeSerializer(serializers.Serializer):
    ranges = IntegerRangeField()

serializer = RangeSerializer(data={'ranges': {'lower': 0, 'upper': 1}})

FloatRangeField

from rest_framework import serializers
from drf_extra_fields.fields import FloatRangeField

class RangeSerializer(serializers.Serializer):
    ranges = FloatRangeField()

serializer = RangeSerializer(data={'ranges': {'lower': 0., 'upper': 1.}})

DecimalRangeField

from rest_framework import serializers
from drf_extra_fields.fields import DecimalRangeField

class RangeSerializer(serializers.Serializer):
    ranges = DecimalRangeField()

serializer = RangeSerializer(data={'ranges': {'lower': 0., 'upper': 1.}}, )

DateRangeField

import datetime

from rest_framework import serializers
from drf_extra_fields.fields import DateRangeField

class RangeSerializer(serializers.Serializer):
    ranges = DateRangeField()

serializer = RangeSerializer(data={'ranges': {'lower': datetime.date(2015, 1, 1), 'upper': datetime.date(2015, 2, 1)}})

DateTimeRangeField

import datetime

from rest_framework import serializers
from drf_extra_fields.fields import DateTimeRangeField

class RangeSerializer(serializers.Serializer):
    ranges = DateTimeRangeField()

serializer = RangeSerializer(data={'ranges': {'lower': datetime.datetime(2015, 1, 1, 0), 'upper': datetime.datetime(2015, 2, 1, 0)}})

PresentablePrimaryKeyRelatedField

Represents related object with a serializer.

presentation_serializer could also be a string that represents a dotted path of a serializer, this is useful when you want to represent a related field with the same serializer.

from drf_extra_fields.relations import PresentablePrimaryKeyRelatedField

class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = (
            'id',
            "username",
        )

class PostSerializer(serializers.ModelSerializer):
    user = PresentablePrimaryKeyRelatedField(
        queryset=User.objects.all(),
        presentation_serializer=UserSerializer,
        presentation_serializer_kwargs={
            'example': [
                'of',
                'passing',
                'kwargs',
                'to',
                'serializer',
            ]
        },
        read_source=None
    )
    class Meta:
        model = Post
        fields = (
            "id",
            "title",
            "user",
        )

Serializer data:

{
    "user": 1,
    "title": "test"
}

Serialized data with PrimaryKeyRelatedField:

{
    "id":1,
    "user": 1,
    "title": "test"
}

Serialized data with PresentablePrimaryKeyRelatedField:

{
    "id":1,
    "user": {
        "id": 1,
        "username": "test"
    },
    "title": "test"
}

PresentableSlugRelatedField

Represents related object retrieved using slug with a serializer.

from drf_extra_fields.relations import PresentableSlugRelatedField

class CategorySerializer(serializers.ModelSerializer):
    class Meta:
        model = Category
        fields = (
            "id",
            "slug",
            "name"
        )

class ProductSerializer(serializers.ModelSerializer):
    category = PresentableSlugRelatedField(
        slug_field="slug",
        queryset=Category.objects.all(),
        presentation_serializer=CategorySerializer,
        presentation_serializer_kwargs={
            'example': [
                'of',
                'passing',
                'kwargs',
                'to',
                'serializer',
            ]
        },
        read_source=None
    )
    class Meta:
        model = Product
        fields = (
            "id",
            "name",
            "category",
        )

Serializer data:

{
    "category": "vegetables",
    "name": "Tomato"
}

Serialized data with SlugRelatedField:

{
    "id": 1,
    "name": "Tomato",
    "category": "vegetables"
}

Serialized data with PresentableSlugRelatedField:

{
    "id": 1,
    "name": "Tomato",
    "category": {
        "id": 1,
        "slug": "vegetables",
        "name": "Vegetables"
    }
}

read_source parameter

This parameter allows you to use different source for read operations and doesn't change field name for write operations. This is only used while representing the data.

HybridImageField

A django-rest-framework field for handling image-uploads through raw post data, with a fallback to multipart form data.

It first tries Base64ImageField. if it fails then tries ImageField.

from rest_framework import serializers
from drf_extra_fields.fields import HybridImageField

class HybridImageSerializer(serializers.Serializer):
    image = HybridImageField()

drf-yasg fix for BASE64 Fields:

The drf-yasg project seems to generate wrong documentation on Base64ImageField or Base64FileField. It marks those fields as readonly. Here is the workaround code for correct the generated document. (More detail on issue #66)

class PDFBase64FileField(Base64FileField):
    ALLOWED_TYPES = ['pdf']

    class Meta:
        swagger_schema_fields = {
            'type': 'string',
            'title': 'File Content',
            'description': 'Content of the file base64 encoded',
            'read_only': False  # <-- FIX
        }

    def get_file_extension(self, filename, decoded_file):
        try:
            PyPDF2.PdfFileReader(io.BytesIO(decoded_file))
        except PyPDF2.utils.PdfReadError as e:
            logger.warning(e)
        else:
            return 'pdf'

LowercaseEmailField

An enhancement over django-rest-framework's EmailField to allow case-insensitive serialization and deserialization of e-mail addresses.

from rest_framework import serializers
from drf_extra_fields.fields import LowercaseEmailField

class EmailSerializer(serializers.Serializer):
    email = LowercaseEmailField()

CONTRIBUTION

TESTS

$ pip install tox  # if not already installed
$ tox

Or, if you prefer using Docker (recommended):

tools/run_development.sh
tox

README

LICENSE

Copyright DRF EXTRA FIELDS HIPO

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.