encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-25 19:14:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

unique_for_month, unique_for_year

Browse Source
This commit is contained in:
Tom Christie 2014-10-22 16:29:09 +01:00
parent ae53fdff9c
commit f9c0e6ee1b
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
rest_framework/serializers.py
View File

@ -24,7 +24,8 @@ from rest_framework.utils.field_mapping import (
ClassLookupDict
)
from rest_framework.validators import (
UniqueForDateValidator, UniqueTogetherValidator
UniqueForDateValidator, UniqueForMonthValidator, UniqueForYearValidator,
UniqueTogetherValidator
)
import copy
import inspect
@ -601,6 +602,22 @@ class ModelSerializer(Serializer):
)
validators.append(validator)
if field.unique_for_month and field_name in field_names:
validator = UniqueForMonthValidator(
queryset=model_class._default_manager,
field=field_name,
date_field=field.unique_for_month
)
validators.append(validator)
if field.unique_for_year and field_name in field_names:
validator = UniqueForYearValidator(
queryset=model_class._default_manager,
field=field_name,
date_field=field.unique_for_year
)
validators.append(validator)
return validators
def _get_base_fields(self):