12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336 |
- # -*- coding: utf-8 -*-
- from __future__ import unicode_literals
- import collections
- import copy
- import datetime
- import decimal
- import math
- import uuid
- import warnings
- from base64 import b64decode, b64encode
- from itertools import tee
- from django.apps import apps
- from django.db import connection
- from django.db.models.lookups import default_lookups, RegisterLookupMixin
- from django.db.models.query_utils import QueryWrapper
- from django.conf import settings
- from django import forms
- from django.core import exceptions, validators, checks
- from django.utils.datastructures import DictWrapper
- from django.utils.dateparse import parse_date, parse_datetime, parse_time, parse_duration
- from django.utils.duration import duration_string
- from django.utils.functional import cached_property, curry, total_ordering, Promise
- from django.utils.text import capfirst
- from django.utils import timezone
- from django.utils.translation import ugettext_lazy as _
- from django.utils.encoding import (smart_text, force_text, force_bytes,
- python_2_unicode_compatible)
- from django.utils.ipv6 import clean_ipv6_address
- from django.utils import six
- from django.utils.itercompat import is_iterable
- # imported for backwards compatibility
- from django.core.exceptions import FieldDoesNotExist # NOQA
- # Avoid "TypeError: Item in ``from list'' not a string" -- unicode_literals
- # makes these strings unicode
- __all__ = [str(x) for x in (
- 'AutoField', 'BLANK_CHOICE_DASH', 'BigIntegerField', 'BinaryField',
- 'BooleanField', 'CharField', 'CommaSeparatedIntegerField', 'DateField',
- 'DateTimeField', 'DecimalField', 'DurationField', 'EmailField', 'Empty',
- 'Field', 'FieldDoesNotExist', 'FilePathField', 'FloatField',
- 'GenericIPAddressField', 'IPAddressField', 'IntegerField', 'NOT_PROVIDED',
- 'NullBooleanField', 'PositiveIntegerField', 'PositiveSmallIntegerField',
- 'SlugField', 'SmallIntegerField', 'TextField', 'TimeField', 'URLField',
- 'UUIDField',
- )]
- class Empty(object):
- pass
- class NOT_PROVIDED:
- pass
- # The values to use for "blank" in SelectFields. Will be appended to the start
- # of most "choices" lists.
- BLANK_CHOICE_DASH = [("", "---------")]
- def _load_field(app_label, model_name, field_name):
- return apps.get_model(app_label, model_name)._meta.get_field_by_name(field_name)[0]
- # A guide to Field parameters:
- #
- # * name: The name of the field specified in the model.
- # * attname: The attribute to use on the model object. This is the same as
- # "name", except in the case of ForeignKeys, where "_id" is
- # appended.
- # * db_column: The db_column specified in the model (or None).
- # * column: The database column for this field. This is the same as
- # "attname", except if db_column is specified.
- #
- # Code that introspects values, or does other dynamic things, should use
- # attname. For example, this gets the primary key value of object "obj":
- #
- # getattr(obj, opts.pk.attname)
- def _empty(of_cls):
- new = Empty()
- new.__class__ = of_cls
- return new
- @total_ordering
- @python_2_unicode_compatible
- class Field(RegisterLookupMixin):
- """Base class for all field types"""
- # Designates whether empty strings fundamentally are allowed at the
- # database level.
- empty_strings_allowed = True
- empty_values = list(validators.EMPTY_VALUES)
- # These track each time a Field instance is created. Used to retain order.
- # The auto_creation_counter is used for fields that Django implicitly
- # creates, creation_counter is used for all user-specified fields.
- creation_counter = 0
- auto_creation_counter = -1
- default_validators = [] # Default set of validators
- default_error_messages = {
- 'invalid_choice': _('Value %(value)r is not a valid choice.'),
- 'null': _('This field cannot be null.'),
- 'blank': _('This field cannot be blank.'),
- 'unique': _('%(model_name)s with this %(field_label)s '
- 'already exists.'),
- # Translators: The 'lookup_type' is one of 'date', 'year' or 'month'.
- # Eg: "Title must be unique for pub_date year"
- 'unique_for_date': _("%(field_label)s must be unique for "
- "%(date_field_label)s %(lookup_type)s."),
- }
- class_lookups = default_lookups.copy()
- # Generic field type description, usually overridden by subclasses
- def _description(self):
- return _('Field of type: %(field_type)s') % {
- 'field_type': self.__class__.__name__
- }
- description = property(_description)
- def __init__(self, verbose_name=None, name=None, primary_key=False,
- max_length=None, unique=False, blank=False, null=False,
- db_index=False, rel=None, default=NOT_PROVIDED, editable=True,
- serialize=True, unique_for_date=None, unique_for_month=None,
- unique_for_year=None, choices=None, help_text='', db_column=None,
- db_tablespace=None, auto_created=False, validators=[],
- error_messages=None):
- self.name = name
- self.verbose_name = verbose_name # May be set by set_attributes_from_name
- self._verbose_name = verbose_name # Store original for deconstruction
- self.primary_key = primary_key
- self.max_length, self._unique = max_length, unique
- self.blank, self.null = blank, null
- self.rel = rel
- self.default = default
- self.editable = editable
- self.serialize = serialize
- self.unique_for_date = unique_for_date
- self.unique_for_month = unique_for_month
- self.unique_for_year = unique_for_year
- self._choices = choices or []
- self.help_text = help_text
- self.db_column = db_column
- self.db_tablespace = db_tablespace or settings.DEFAULT_INDEX_TABLESPACE
- self.auto_created = auto_created
- # Set db_index to True if the field has a relationship and doesn't
- # explicitly set db_index.
- self.db_index = db_index
- # Adjust the appropriate creation counter, and save our local copy.
- if auto_created:
- self.creation_counter = Field.auto_creation_counter
- Field.auto_creation_counter -= 1
- else:
- self.creation_counter = Field.creation_counter
- Field.creation_counter += 1
- self._validators = validators # Store for deconstruction later
- messages = {}
- for c in reversed(self.__class__.__mro__):
- messages.update(getattr(c, 'default_error_messages', {}))
- messages.update(error_messages or {})
- self._error_messages = error_messages # Store for deconstruction later
- self.error_messages = messages
- def __str__(self):
- """ Return "app_label.model_label.field_name". """
- model = self.model
- app = model._meta.app_label
- return '%s.%s.%s' % (app, model._meta.object_name, self.name)
- def __repr__(self):
- """
- Displays the module, class and name of the field.
- """
- path = '%s.%s' % (self.__class__.__module__, self.__class__.__name__)
- name = getattr(self, 'name', None)
- if name is not None:
- return '<%s: %s>' % (path, name)
- return '<%s>' % path
- def check(self, **kwargs):
- errors = []
- errors.extend(self._check_field_name())
- errors.extend(self._check_choices())
- errors.extend(self._check_db_index())
- errors.extend(self._check_null_allowed_for_primary_keys())
- errors.extend(self._check_backend_specific_checks(**kwargs))
- return errors
- def _check_field_name(self):
- """ Check if field name is valid, i.e. 1) does not end with an
- underscore, 2) does not contain "__" and 3) is not "pk". """
- if self.name.endswith('_'):
- return [
- checks.Error(
- 'Field names must not end with an underscore.',
- hint=None,
- obj=self,
- id='fields.E001',
- )
- ]
- elif '__' in self.name:
- return [
- checks.Error(
- 'Field names must not contain "__".',
- hint=None,
- obj=self,
- id='fields.E002',
- )
- ]
- elif self.name == 'pk':
- return [
- checks.Error(
- "'pk' is a reserved word that cannot be used as a field name.",
- hint=None,
- obj=self,
- id='fields.E003',
- )
- ]
- else:
- return []
- def _check_choices(self):
- if self.choices:
- if (isinstance(self.choices, six.string_types) or
- not is_iterable(self.choices)):
- return [
- checks.Error(
- "'choices' must be an iterable (e.g., a list or tuple).",
- hint=None,
- obj=self,
- id='fields.E004',
- )
- ]
- elif any(isinstance(choice, six.string_types) or
- not is_iterable(choice) or len(choice) != 2
- for choice in self.choices):
- return [
- checks.Error(
- ("'choices' must be an iterable containing "
- "(actual value, human readable name) tuples."),
- hint=None,
- obj=self,
- id='fields.E005',
- )
- ]
- else:
- return []
- else:
- return []
- def _check_db_index(self):
- if self.db_index not in (None, True, False):
- return [
- checks.Error(
- "'db_index' must be None, True or False.",
- hint=None,
- obj=self,
- id='fields.E006',
- )
- ]
- else:
- return []
- def _check_null_allowed_for_primary_keys(self):
- if (self.primary_key and self.null and
- not connection.features.interprets_empty_strings_as_nulls):
- # We cannot reliably check this for backends like Oracle which
- # consider NULL and '' to be equal (and thus set up
- # character-based fields a little differently).
- return [
- checks.Error(
- 'Primary keys must not have null=True.',
- hint=('Set null=False on the field, or '
- 'remove primary_key=True argument.'),
- obj=self,
- id='fields.E007',
- )
- ]
- else:
- return []
- def _check_backend_specific_checks(self, **kwargs):
- return connection.validation.check_field(self, **kwargs)
- def deconstruct(self):
- """
- Returns enough information to recreate the field as a 4-tuple:
- * The name of the field on the model, if contribute_to_class has been run
- * The import path of the field, including the class: django.db.models.IntegerField
- This should be the most portable version, so less specific may be better.
- * A list of positional arguments
- * A dict of keyword arguments
- Note that the positional or keyword arguments must contain values of the
- following types (including inner values of collection types):
- * None, bool, str, unicode, int, long, float, complex, set, frozenset, list, tuple, dict
- * UUID
- * datetime.datetime (naive), datetime.date
- * top-level classes, top-level functions - will be referenced by their full import path
- * Storage instances - these have their own deconstruct() method
- This is because the values here must be serialized into a text format
- (possibly new Python code, possibly JSON) and these are the only types
- with encoding handlers defined.
- There's no need to return the exact way the field was instantiated this time,
- just ensure that the resulting field is the same - prefer keyword arguments
- over positional ones, and omit parameters with their default values.
- """
- # Short-form way of fetching all the default parameters
- keywords = {}
- possibles = {
- "verbose_name": None,
- "primary_key": False,
- "max_length": None,
- "unique": False,
- "blank": False,
- "null": False,
- "db_index": False,
- "default": NOT_PROVIDED,
- "editable": True,
- "serialize": True,
- "unique_for_date": None,
- "unique_for_month": None,
- "unique_for_year": None,
- "choices": [],
- "help_text": '',
- "db_column": None,
- "db_tablespace": settings.DEFAULT_INDEX_TABLESPACE,
- "auto_created": False,
- "validators": [],
- "error_messages": None,
- }
- attr_overrides = {
- "unique": "_unique",
- "choices": "_choices",
- "error_messages": "_error_messages",
- "validators": "_validators",
- "verbose_name": "_verbose_name",
- }
- equals_comparison = {"choices", "validators", "db_tablespace"}
- for name, default in possibles.items():
- value = getattr(self, attr_overrides.get(name, name))
- # Unroll anything iterable for choices into a concrete list
- if name == "choices" and isinstance(value, collections.Iterable):
- value = list(value)
- # Do correct kind of comparison
- if name in equals_comparison:
- if value != default:
- keywords[name] = value
- else:
- if value is not default:
- keywords[name] = value
- # Work out path - we shorten it for known Django core fields
- path = "%s.%s" % (self.__class__.__module__, self.__class__.__name__)
- if path.startswith("django.db.models.fields.related"):
- path = path.replace("django.db.models.fields.related", "django.db.models")
- if path.startswith("django.db.models.fields.files"):
- path = path.replace("django.db.models.fields.files", "django.db.models")
- if path.startswith("django.db.models.fields.proxy"):
- path = path.replace("django.db.models.fields.proxy", "django.db.models")
- if path.startswith("django.db.models.fields"):
- path = path.replace("django.db.models.fields", "django.db.models")
- # Return basic info - other fields should override this.
- return (
- force_text(self.name, strings_only=True),
- path,
- [],
- keywords,
- )
- def clone(self):
- """
- Uses deconstruct() to clone a new copy of this Field.
- Will not preserve any class attachments/attribute names.
- """
- name, path, args, kwargs = self.deconstruct()
- return self.__class__(*args, **kwargs)
- def __eq__(self, other):
- # Needed for @total_ordering
- if isinstance(other, Field):
- return self.creation_counter == other.creation_counter
- return NotImplemented
- def __lt__(self, other):
- # This is needed because bisect does not take a comparison function.
- if isinstance(other, Field):
- return self.creation_counter < other.creation_counter
- return NotImplemented
- def __hash__(self):
- return hash(self.creation_counter)
- def __deepcopy__(self, memodict):
- # We don't have to deepcopy very much here, since most things are not
- # intended to be altered after initial creation.
- obj = copy.copy(self)
- if self.rel:
- obj.rel = copy.copy(self.rel)
- if hasattr(self.rel, 'field') and self.rel.field is self:
- obj.rel.field = obj
- memodict[id(self)] = obj
- return obj
- def __copy__(self):
- # We need to avoid hitting __reduce__, so define this
- # slightly weird copy construct.
- obj = Empty()
- obj.__class__ = self.__class__
- obj.__dict__ = self.__dict__.copy()
- return obj
- def __reduce__(self):
- """
- Pickling should return the model._meta.fields instance of the field,
- not a new copy of that field. So, we use the app registry to load the
- model and then the field back.
- """
- if not hasattr(self, 'model'):
- # Fields are sometimes used without attaching them to models (for
- # example in aggregation). In this case give back a plain field
- # instance. The code below will create a new empty instance of
- # class self.__class__, then update its dict with self.__dict__
- # values - so, this is very close to normal pickle.
- return _empty, (self.__class__,), self.__dict__
- if self.model._deferred:
- # Deferred model will not be found from the app registry. This
- # could be fixed by reconstructing the deferred model on unpickle.
- raise RuntimeError("Fields of deferred models can't be reduced")
- return _load_field, (self.model._meta.app_label, self.model._meta.object_name,
- self.name)
- def to_python(self, value):
- """
- Converts the input value into the expected Python data type, raising
- django.core.exceptions.ValidationError if the data can't be converted.
- Returns the converted value. Subclasses should override this.
- """
- return value
- @cached_property
- def validators(self):
- # Some validators can't be created at field initialization time.
- # This method provides a way to delay their creation until required.
- return self.default_validators + self._validators
- def run_validators(self, value):
- if value in self.empty_values:
- return
- errors = []
- for v in self.validators:
- try:
- v(value)
- except exceptions.ValidationError as e:
- if hasattr(e, 'code') and e.code in self.error_messages:
- e.message = self.error_messages[e.code]
- errors.extend(e.error_list)
- if errors:
- raise exceptions.ValidationError(errors)
- def validate(self, value, model_instance):
- """
- Validates value and throws ValidationError. Subclasses should override
- this to provide validation logic.
- """
- if not self.editable:
- # Skip validation for non-editable fields.
- return
- if self._choices and value not in self.empty_values:
- for option_key, option_value in self.choices:
- if isinstance(option_value, (list, tuple)):
- # This is an optgroup, so look inside the group for
- # options.
- for optgroup_key, optgroup_value in option_value:
- if value == optgroup_key:
- return
- elif value == option_key:
- return
- raise exceptions.ValidationError(
- self.error_messages['invalid_choice'],
- code='invalid_choice',
- params={'value': value},
- )
- if value is None and not self.null:
- raise exceptions.ValidationError(self.error_messages['null'], code='null')
- if not self.blank and value in self.empty_values:
- raise exceptions.ValidationError(self.error_messages['blank'], code='blank')
- def clean(self, value, model_instance):
- """
- Convert the value's type and run validation. Validation errors
- from to_python and validate are propagated. The correct value is
- returned if no error is raised.
- """
- value = self.to_python(value)
- self.validate(value, model_instance)
- self.run_validators(value)
- return value
- def db_type(self, connection):
- """
- Returns the database column data type for this field, for the provided
- connection.
- """
- # The default implementation of this method looks at the
- # backend-specific data_types dictionary, looking up the field by its
- # "internal type".
- #
- # A Field class can implement the get_internal_type() method to specify
- # which *preexisting* Django Field class it's most similar to -- i.e.,
- # a custom field might be represented by a TEXT column type, which is
- # the same as the TextField Django field type, which means the custom
- # field's get_internal_type() returns 'TextField'.
- #
- # But the limitation of the get_internal_type() / data_types approach
- # is that it cannot handle database column types that aren't already
- # mapped to one of the built-in Django field types. In this case, you
- # can implement db_type() instead of get_internal_type() to specify
- # exactly which wacky database column type you want to use.
- data = DictWrapper(self.__dict__, connection.ops.quote_name, "qn_")
- try:
- return connection.data_types[self.get_internal_type()] % data
- except KeyError:
- return None
- def db_parameters(self, connection):
- """
- Extension of db_type(), providing a range of different return
- values (type, checks).
- This will look at db_type(), allowing custom model fields to override it.
- """
- data = DictWrapper(self.__dict__, connection.ops.quote_name, "qn_")
- type_string = self.db_type(connection)
- try:
- check_string = connection.data_type_check_constraints[self.get_internal_type()] % data
- except KeyError:
- check_string = None
- return {
- "type": type_string,
- "check": check_string,
- }
- def db_type_suffix(self, connection):
- return connection.data_types_suffix.get(self.get_internal_type())
- def get_db_converters(self, connection):
- if hasattr(self, 'from_db_value'):
- return [self.from_db_value]
- return []
- @property
- def unique(self):
- return self._unique or self.primary_key
- def set_attributes_from_name(self, name):
- if not self.name:
- self.name = name
- self.attname, self.column = self.get_attname_column()
- if self.verbose_name is None and self.name:
- self.verbose_name = self.name.replace('_', ' ')
- def contribute_to_class(self, cls, name, virtual_only=False):
- self.set_attributes_from_name(name)
- self.model = cls
- if virtual_only:
- cls._meta.add_virtual_field(self)
- else:
- cls._meta.add_field(self)
- if self.choices:
- setattr(cls, 'get_%s_display' % self.name,
- curry(cls._get_FIELD_display, field=self))
- def get_attname(self):
- return self.name
- def get_attname_column(self):
- attname = self.get_attname()
- column = self.db_column or attname
- return attname, column
- def get_cache_name(self):
- return '_%s_cache' % self.name
- def get_internal_type(self):
- return self.__class__.__name__
- def pre_save(self, model_instance, add):
- """
- Returns field's value just before saving.
- """
- return getattr(model_instance, self.attname)
- def get_prep_value(self, value):
- """
- Perform preliminary non-db specific value checks and conversions.
- """
- if isinstance(value, Promise):
- value = value._proxy____cast()
- return value
- def get_db_prep_value(self, value, connection, prepared=False):
- """Returns field's value prepared for interacting with the database
- backend.
- Used by the default implementations of ``get_db_prep_save``and
- `get_db_prep_lookup```
- """
- if not prepared:
- value = self.get_prep_value(value)
- return value
- def get_db_prep_save(self, value, connection):
- """
- Returns field's value prepared for saving into a database.
- """
- return self.get_db_prep_value(value, connection=connection,
- prepared=False)
- def get_prep_lookup(self, lookup_type, value):
- """
- Perform preliminary non-db specific lookup checks and conversions
- """
- if hasattr(value, '_prepare'):
- return value._prepare()
- if lookup_type in {
- 'iexact', 'contains', 'icontains',
- 'startswith', 'istartswith', 'endswith', 'iendswith',
- 'month', 'day', 'week_day', 'hour', 'minute', 'second',
- 'isnull', 'search', 'regex', 'iregex',
- }:
- return value
- elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
- return self.get_prep_value(value)
- elif lookup_type in ('range', 'in'):
- return [self.get_prep_value(v) for v in value]
- elif lookup_type == 'year':
- try:
- return int(value)
- except ValueError:
- raise ValueError("The __year lookup type requires an integer "
- "argument")
- return self.get_prep_value(value)
- def get_db_prep_lookup(self, lookup_type, value, connection,
- prepared=False):
- """
- Returns field's value prepared for database lookup.
- """
- if not prepared:
- value = self.get_prep_lookup(lookup_type, value)
- prepared = True
- if hasattr(value, 'get_compiler'):
- value = value.get_compiler(connection=connection)
- if hasattr(value, 'as_sql') or hasattr(value, '_as_sql'):
- # If the value has a relabeled_clone method it means the
- # value will be handled later on.
- if hasattr(value, 'relabeled_clone'):
- return value
- if hasattr(value, 'as_sql'):
- sql, params = value.as_sql()
- else:
- sql, params = value._as_sql(connection=connection)
- return QueryWrapper(('(%s)' % sql), params)
- if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute',
- 'second', 'search', 'regex', 'iregex'):
- return [value]
- elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
- return [self.get_db_prep_value(value, connection=connection,
- prepared=prepared)]
- elif lookup_type in ('range', 'in'):
- return [self.get_db_prep_value(v, connection=connection,
- prepared=prepared) for v in value]
- elif lookup_type in ('contains', 'icontains'):
- return ["%%%s%%" % connection.ops.prep_for_like_query(value)]
- elif lookup_type == 'iexact':
- return [connection.ops.prep_for_iexact_query(value)]
- elif lookup_type in ('startswith', 'istartswith'):
- return ["%s%%" % connection.ops.prep_for_like_query(value)]
- elif lookup_type in ('endswith', 'iendswith'):
- return ["%%%s" % connection.ops.prep_for_like_query(value)]
- elif lookup_type == 'isnull':
- return []
- elif lookup_type == 'year':
- if isinstance(self, DateTimeField):
- return connection.ops.year_lookup_bounds_for_datetime_field(value)
- elif isinstance(self, DateField):
- return connection.ops.year_lookup_bounds_for_date_field(value)
- else:
- return [value] # this isn't supposed to happen
- else:
- return [value]
- def has_default(self):
- """
- Returns a boolean of whether this field has a default value.
- """
- return self.default is not NOT_PROVIDED
- def get_default(self):
- """
- Returns the default value for this field.
- """
- if self.has_default():
- if callable(self.default):
- return self.default()
- return self.default
- if (not self.empty_strings_allowed or (self.null and
- not connection.features.interprets_empty_strings_as_nulls)):
- return None
- return ""
- def get_choices(self, include_blank=True, blank_choice=BLANK_CHOICE_DASH, limit_choices_to=None):
- """Returns choices with a default blank choices included, for use
- as SelectField choices for this field."""
- blank_defined = False
- choices = list(self.choices) if self.choices else []
- named_groups = choices and isinstance(choices[0][1], (list, tuple))
- if not named_groups:
- for choice, __ in choices:
- if choice in ('', None):
- blank_defined = True
- break
- first_choice = (blank_choice if include_blank and
- not blank_defined else [])
- if self.choices:
- return first_choice + choices
- rel_model = self.rel.to
- limit_choices_to = limit_choices_to or self.get_limit_choices_to()
- if hasattr(self.rel, 'get_related_field'):
- lst = [(getattr(x, self.rel.get_related_field().attname),
- smart_text(x))
- for x in rel_model._default_manager.complex_filter(
- limit_choices_to)]
- else:
- lst = [(x._get_pk_val(), smart_text(x))
- for x in rel_model._default_manager.complex_filter(
- limit_choices_to)]
- return first_choice + lst
- def get_choices_default(self):
- return self.get_choices()
- def get_flatchoices(self, include_blank=True,
- blank_choice=BLANK_CHOICE_DASH):
- """
- Returns flattened choices with a default blank choice included.
- """
- first_choice = blank_choice if include_blank else []
- return first_choice + list(self.flatchoices)
- def _get_val_from_obj(self, obj):
- if obj is not None:
- return getattr(obj, self.attname)
- else:
- return self.get_default()
- def value_to_string(self, obj):
- """
- Returns a string value of this field from the passed obj.
- This is used by the serialization framework.
- """
- return smart_text(self._get_val_from_obj(obj))
- def _get_choices(self):
- if isinstance(self._choices, collections.Iterator):
- choices, self._choices = tee(self._choices)
- return choices
- else:
- return self._choices
- choices = property(_get_choices)
- def _get_flatchoices(self):
- """Flattened version of choices tuple."""
- flat = []
- for choice, value in self.choices:
- if isinstance(value, (list, tuple)):
- flat.extend(value)
- else:
- flat.append((choice, value))
- return flat
- flatchoices = property(_get_flatchoices)
- def save_form_data(self, instance, data):
- setattr(instance, self.name, data)
- def formfield(self, form_class=None, choices_form_class=None, **kwargs):
- """
- Returns a django.forms.Field instance for this database Field.
- """
- defaults = {'required': not self.blank,
- 'label': capfirst(self.verbose_name),
- 'help_text': self.help_text}
- if self.has_default():
- if callable(self.default):
- defaults['initial'] = self.default
- defaults['show_hidden_initial'] = True
- else:
- defaults['initial'] = self.get_default()
- if self.choices:
- # Fields with choices get special treatment.
- include_blank = (self.blank or
- not (self.has_default() or 'initial' in kwargs))
- defaults['choices'] = self.get_choices(include_blank=include_blank)
- defaults['coerce'] = self.to_python
- if self.null:
- defaults['empty_value'] = None
- if choices_form_class is not None:
- form_class = choices_form_class
- else:
- form_class = forms.TypedChoiceField
- # Many of the subclass-specific formfield arguments (min_value,
- # max_value) don't apply for choice fields, so be sure to only pass
- # the values that TypedChoiceField will understand.
- for k in list(kwargs):
- if k not in ('coerce', 'empty_value', 'choices', 'required',
- 'widget', 'label', 'initial', 'help_text',
- 'error_messages', 'show_hidden_initial'):
- del kwargs[k]
- defaults.update(kwargs)
- if form_class is None:
- form_class = forms.CharField
- return form_class(**defaults)
- def value_from_object(self, obj):
- """
- Returns the value of this field in the given model instance.
- """
- return getattr(obj, self.attname)
- class AutoField(Field):
- description = _("Integer")
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be an integer."),
- }
- def __init__(self, *args, **kwargs):
- kwargs['blank'] = True
- super(AutoField, self).__init__(*args, **kwargs)
- def check(self, **kwargs):
- errors = super(AutoField, self).check(**kwargs)
- errors.extend(self._check_primary_key())
- return errors
- def _check_primary_key(self):
- if not self.primary_key:
- return [
- checks.Error(
- 'AutoFields must set primary_key=True.',
- hint=None,
- obj=self,
- id='fields.E100',
- ),
- ]
- else:
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(AutoField, self).deconstruct()
- del kwargs['blank']
- kwargs['primary_key'] = True
- return name, path, args, kwargs
- def get_internal_type(self):
- return "AutoField"
- def to_python(self, value):
- if value is None:
- return value
- try:
- return int(value)
- except (TypeError, ValueError):
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def validate(self, value, model_instance):
- pass
- def get_db_prep_value(self, value, connection, prepared=False):
- if not prepared:
- value = self.get_prep_value(value)
- value = connection.ops.validate_autopk_value(value)
- return value
- def get_prep_value(self, value):
- value = super(AutoField, self).get_prep_value(value)
- if value is None:
- return None
- return int(value)
- def contribute_to_class(self, cls, name, **kwargs):
- assert not cls._meta.has_auto_field, \
- "A model can't have more than one AutoField."
- super(AutoField, self).contribute_to_class(cls, name, **kwargs)
- cls._meta.has_auto_field = True
- cls._meta.auto_field = self
- def formfield(self, **kwargs):
- return None
- class BooleanField(Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be either True or False."),
- }
- description = _("Boolean (Either True or False)")
- def __init__(self, *args, **kwargs):
- kwargs['blank'] = True
- super(BooleanField, self).__init__(*args, **kwargs)
- def check(self, **kwargs):
- errors = super(BooleanField, self).check(**kwargs)
- errors.extend(self._check_null(**kwargs))
- return errors
- def _check_null(self, **kwargs):
- if getattr(self, 'null', False):
- return [
- checks.Error(
- 'BooleanFields do not accept null values.',
- hint='Use a NullBooleanField instead.',
- obj=self,
- id='fields.E110',
- )
- ]
- else:
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(BooleanField, self).deconstruct()
- del kwargs['blank']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "BooleanField"
- def to_python(self, value):
- if value in (True, False):
- # if value is 1 or 0 than it's equal to True or False, but we want
- # to return a true bool for semantic reasons.
- return bool(value)
- if value in ('t', 'True', '1'):
- return True
- if value in ('f', 'False', '0'):
- return False
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def get_prep_lookup(self, lookup_type, value):
- # Special-case handling for filters coming from a Web request (e.g. the
- # admin interface). Only works for scalar values (not lists). If you're
- # passing in a list, you might as well make things the right type when
- # constructing the list.
- if value in ('1', '0'):
- value = bool(int(value))
- return super(BooleanField, self).get_prep_lookup(lookup_type, value)
- def get_prep_value(self, value):
- value = super(BooleanField, self).get_prep_value(value)
- if value is None:
- return None
- return bool(value)
- def formfield(self, **kwargs):
- # Unlike most fields, BooleanField figures out include_blank from
- # self.null instead of self.blank.
- if self.choices:
- include_blank = not (self.has_default() or 'initial' in kwargs)
- defaults = {'choices': self.get_choices(include_blank=include_blank)}
- else:
- defaults = {'form_class': forms.BooleanField}
- defaults.update(kwargs)
- return super(BooleanField, self).formfield(**defaults)
- class CharField(Field):
- description = _("String (up to %(max_length)s)")
- def __init__(self, *args, **kwargs):
- super(CharField, self).__init__(*args, **kwargs)
- self.validators.append(validators.MaxLengthValidator(self.max_length))
- def check(self, **kwargs):
- errors = super(CharField, self).check(**kwargs)
- errors.extend(self._check_max_length_attribute(**kwargs))
- return errors
- def _check_max_length_attribute(self, **kwargs):
- try:
- max_length = int(self.max_length)
- if max_length <= 0:
- raise ValueError()
- except TypeError:
- return [
- checks.Error(
- "CharFields must define a 'max_length' attribute.",
- hint=None,
- obj=self,
- id='fields.E120',
- )
- ]
- except ValueError:
- return [
- checks.Error(
- "'max_length' must be a positive integer.",
- hint=None,
- obj=self,
- id='fields.E121',
- )
- ]
- else:
- return []
- def get_internal_type(self):
- return "CharField"
- def to_python(self, value):
- if isinstance(value, six.string_types) or value is None:
- return value
- return smart_text(value)
- def get_prep_value(self, value):
- value = super(CharField, self).get_prep_value(value)
- return self.to_python(value)
- def formfield(self, **kwargs):
- # Passing max_length to forms.CharField means that the value's length
- # will be validated twice. This is considered acceptable since we want
- # the value in the form field (to pass into widget for example).
- defaults = {'max_length': self.max_length}
- defaults.update(kwargs)
- return super(CharField, self).formfield(**defaults)
- # TODO: Maybe move this into contrib, because it's specialized.
- class CommaSeparatedIntegerField(CharField):
- default_validators = [validators.validate_comma_separated_integer_list]
- description = _("Comma-separated integers")
- def formfield(self, **kwargs):
- defaults = {
- 'error_messages': {
- 'invalid': _('Enter only digits separated by commas.'),
- }
- }
- defaults.update(kwargs)
- return super(CommaSeparatedIntegerField, self).formfield(**defaults)
- class DateTimeCheckMixin(object):
- def check(self, **kwargs):
- errors = super(DateTimeCheckMixin, self).check(**kwargs)
- errors.extend(self._check_mutually_exclusive_options())
- errors.extend(self._check_fix_default_value())
- return errors
- def _check_mutually_exclusive_options(self):
- # auto_now, auto_now_add, and default are mutually exclusive
- # options. The use of more than one of these options together
- # will trigger an Error
- mutually_exclusive_options = [self.auto_now_add, self.auto_now,
- self.has_default()]
- enabled_options = [option not in (None, False)
- for option in mutually_exclusive_options].count(True)
- if enabled_options > 1:
- return [
- checks.Error(
- "The options auto_now, auto_now_add, and default "
- "are mutually exclusive. Only one of these options "
- "may be present.",
- hint=None,
- obj=self,
- id='fields.E160',
- )
- ]
- else:
- return []
- def _check_fix_default_value(self):
- return []
- class DateField(DateTimeCheckMixin, Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value has an invalid date format. It must be "
- "in YYYY-MM-DD format."),
- 'invalid_date': _("'%(value)s' value has the correct format (YYYY-MM-DD) "
- "but it is an invalid date."),
- }
- description = _("Date (without time)")
- def __init__(self, verbose_name=None, name=None, auto_now=False,
- auto_now_add=False, **kwargs):
- self.auto_now, self.auto_now_add = auto_now, auto_now_add
- if auto_now or auto_now_add:
- kwargs['editable'] = False
- kwargs['blank'] = True
- super(DateField, self).__init__(verbose_name, name, **kwargs)
- def _check_fix_default_value(self):
- """
- Adds a warning to the checks framework stating, that using an actual
- date or datetime value is probably wrong; it's only being evaluated on
- server start-up.
- For details see ticket #21905
- """
- if not self.has_default():
- return []
- now = timezone.now()
- if not timezone.is_naive(now):
- now = timezone.make_naive(now, timezone.utc)
- value = self.default
- if isinstance(value, datetime.datetime):
- if not timezone.is_naive(value):
- value = timezone.make_naive(value, timezone.utc)
- value = value.date()
- elif isinstance(value, datetime.date):
- # Nothing to do, as dates don't have tz information
- pass
- else:
- # No explicit date / datetime value -- no checks necessary
- return []
- offset = datetime.timedelta(days=1)
- lower = (now - offset).date()
- upper = (now + offset).date()
- if lower <= value <= upper:
- return [
- checks.Warning(
- 'Fixed default value provided.',
- hint='It seems you set a fixed date / time / datetime '
- 'value as default for this field. This may not be '
- 'what you want. If you want to have the current date '
- 'as default, use `django.utils.timezone.now`',
- obj=self,
- id='fields.W161',
- )
- ]
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(DateField, self).deconstruct()
- if self.auto_now:
- kwargs['auto_now'] = True
- if self.auto_now_add:
- kwargs['auto_now_add'] = True
- if self.auto_now or self.auto_now_add:
- del kwargs['editable']
- del kwargs['blank']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "DateField"
- def to_python(self, value):
- if value is None:
- return value
- if isinstance(value, datetime.datetime):
- if settings.USE_TZ and timezone.is_aware(value):
- # Convert aware datetimes to the default time zone
- # before casting them to dates (#17742).
- default_timezone = timezone.get_default_timezone()
- value = timezone.make_naive(value, default_timezone)
- return value.date()
- if isinstance(value, datetime.date):
- return value
- try:
- parsed = parse_date(value)
- if parsed is not None:
- return parsed
- except ValueError:
- raise exceptions.ValidationError(
- self.error_messages['invalid_date'],
- code='invalid_date',
- params={'value': value},
- )
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def pre_save(self, model_instance, add):
- if self.auto_now or (self.auto_now_add and add):
- value = datetime.date.today()
- setattr(model_instance, self.attname, value)
- return value
- else:
- return super(DateField, self).pre_save(model_instance, add)
- def contribute_to_class(self, cls, name, **kwargs):
- super(DateField, self).contribute_to_class(cls, name, **kwargs)
- if not self.null:
- setattr(cls, 'get_next_by_%s' % self.name,
- curry(cls._get_next_or_previous_by_FIELD, field=self,
- is_next=True))
- setattr(cls, 'get_previous_by_%s' % self.name,
- curry(cls._get_next_or_previous_by_FIELD, field=self,
- is_next=False))
- def get_prep_lookup(self, lookup_type, value):
- # For dates lookups, convert the value to an int
- # so the database backend always sees a consistent type.
- if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute', 'second'):
- return int(value)
- return super(DateField, self).get_prep_lookup(lookup_type, value)
- def get_prep_value(self, value):
- value = super(DateField, self).get_prep_value(value)
- return self.to_python(value)
- def get_db_prep_value(self, value, connection, prepared=False):
- # Casts dates into the format expected by the backend
- if not prepared:
- value = self.get_prep_value(value)
- return connection.ops.value_to_db_date(value)
- def value_to_string(self, obj):
- val = self._get_val_from_obj(obj)
- return '' if val is None else val.isoformat()
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.DateField}
- defaults.update(kwargs)
- return super(DateField, self).formfield(**defaults)
- class DateTimeField(DateField):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value has an invalid format. It must be in "
- "YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ] format."),
- 'invalid_date': _("'%(value)s' value has the correct format "
- "(YYYY-MM-DD) but it is an invalid date."),
- 'invalid_datetime': _("'%(value)s' value has the correct format "
- "(YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ]) "
- "but it is an invalid date/time."),
- }
- description = _("Date (with time)")
- # __init__ is inherited from DateField
- def _check_fix_default_value(self):
- """
- Adds a warning to the checks framework stating, that using an actual
- date or datetime value is probably wrong; it's only being evaluated on
- server start-up.
- For details see ticket #21905
- """
- if not self.has_default():
- return []
- now = timezone.now()
- if not timezone.is_naive(now):
- now = timezone.make_naive(now, timezone.utc)
- value = self.default
- if isinstance(value, datetime.datetime):
- second_offset = datetime.timedelta(seconds=10)
- lower = now - second_offset
- upper = now + second_offset
- if timezone.is_aware(value):
- value = timezone.make_naive(value, timezone.utc)
- elif isinstance(value, datetime.date):
- second_offset = datetime.timedelta(seconds=10)
- lower = now - second_offset
- lower = datetime.datetime(lower.year, lower.month, lower.day)
- upper = now + second_offset
- upper = datetime.datetime(upper.year, upper.month, upper.day)
- value = datetime.datetime(value.year, value.month, value.day)
- else:
- # No explicit date / datetime value -- no checks necessary
- return []
- if lower <= value <= upper:
- return [
- checks.Warning(
- 'Fixed default value provided.',
- hint='It seems you set a fixed date / time / datetime '
- 'value as default for this field. This may not be '
- 'what you want. If you want to have the current date '
- 'as default, use `django.utils.timezone.now`',
- obj=self,
- id='fields.W161',
- )
- ]
- return []
- def get_internal_type(self):
- return "DateTimeField"
- def to_python(self, value):
- if value is None:
- return value
- if isinstance(value, datetime.datetime):
- return value
- if isinstance(value, datetime.date):
- value = datetime.datetime(value.year, value.month, value.day)
- if settings.USE_TZ:
- # For backwards compatibility, interpret naive datetimes in
- # local time. This won't work during DST change, but we can't
- # do much about it, so we let the exceptions percolate up the
- # call stack.
- warnings.warn("DateTimeField %s.%s received a naive datetime "
- "(%s) while time zone support is active." %
- (self.model.__name__, self.name, value),
- RuntimeWarning)
- default_timezone = timezone.get_default_timezone()
- value = timezone.make_aware(value, default_timezone)
- return value
- try:
- parsed = parse_datetime(value)
- if parsed is not None:
- return parsed
- except ValueError:
- raise exceptions.ValidationError(
- self.error_messages['invalid_datetime'],
- code='invalid_datetime',
- params={'value': value},
- )
- try:
- parsed = parse_date(value)
- if parsed is not None:
- return datetime.datetime(parsed.year, parsed.month, parsed.day)
- except ValueError:
- raise exceptions.ValidationError(
- self.error_messages['invalid_date'],
- code='invalid_date',
- params={'value': value},
- )
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def pre_save(self, model_instance, add):
- if self.auto_now or (self.auto_now_add and add):
- value = timezone.now()
- setattr(model_instance, self.attname, value)
- return value
- else:
- return super(DateTimeField, self).pre_save(model_instance, add)
- # contribute_to_class is inherited from DateField, it registers
- # get_next_by_FOO and get_prev_by_FOO
- # get_prep_lookup is inherited from DateField
- def get_prep_value(self, value):
- value = super(DateTimeField, self).get_prep_value(value)
- value = self.to_python(value)
- if value is not None and settings.USE_TZ and timezone.is_naive(value):
- # For backwards compatibility, interpret naive datetimes in local
- # time. This won't work during DST change, but we can't do much
- # about it, so we let the exceptions percolate up the call stack.
- try:
- name = '%s.%s' % (self.model.__name__, self.name)
- except AttributeError:
- name = '(unbound)'
- warnings.warn("DateTimeField %s received a naive datetime (%s)"
- " while time zone support is active." %
- (name, value),
- RuntimeWarning)
- default_timezone = timezone.get_default_timezone()
- value = timezone.make_aware(value, default_timezone)
- return value
- def get_db_prep_value(self, value, connection, prepared=False):
- # Casts datetimes into the format expected by the backend
- if not prepared:
- value = self.get_prep_value(value)
- return connection.ops.value_to_db_datetime(value)
- def value_to_string(self, obj):
- val = self._get_val_from_obj(obj)
- return '' if val is None else val.isoformat()
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.DateTimeField}
- defaults.update(kwargs)
- return super(DateTimeField, self).formfield(**defaults)
- class DecimalField(Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be a decimal number."),
- }
- description = _("Decimal number")
- def __init__(self, verbose_name=None, name=None, max_digits=None,
- decimal_places=None, **kwargs):
- self.max_digits, self.decimal_places = max_digits, decimal_places
- super(DecimalField, self).__init__(verbose_name, name, **kwargs)
- def check(self, **kwargs):
- errors = super(DecimalField, self).check(**kwargs)
- digits_errors = self._check_decimal_places()
- digits_errors.extend(self._check_max_digits())
- if not digits_errors:
- errors.extend(self._check_decimal_places_and_max_digits(**kwargs))
- else:
- errors.extend(digits_errors)
- return errors
- def _check_decimal_places(self):
- try:
- decimal_places = int(self.decimal_places)
- if decimal_places < 0:
- raise ValueError()
- except TypeError:
- return [
- checks.Error(
- "DecimalFields must define a 'decimal_places' attribute.",
- hint=None,
- obj=self,
- id='fields.E130',
- )
- ]
- except ValueError:
- return [
- checks.Error(
- "'decimal_places' must be a non-negative integer.",
- hint=None,
- obj=self,
- id='fields.E131',
- )
- ]
- else:
- return []
- def _check_max_digits(self):
- try:
- max_digits = int(self.max_digits)
- if max_digits <= 0:
- raise ValueError()
- except TypeError:
- return [
- checks.Error(
- "DecimalFields must define a 'max_digits' attribute.",
- hint=None,
- obj=self,
- id='fields.E132',
- )
- ]
- except ValueError:
- return [
- checks.Error(
- "'max_digits' must be a positive integer.",
- hint=None,
- obj=self,
- id='fields.E133',
- )
- ]
- else:
- return []
- def _check_decimal_places_and_max_digits(self, **kwargs):
- if int(self.decimal_places) > int(self.max_digits):
- return [
- checks.Error(
- "'max_digits' must be greater or equal to 'decimal_places'.",
- hint=None,
- obj=self,
- id='fields.E134',
- )
- ]
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(DecimalField, self).deconstruct()
- if self.max_digits is not None:
- kwargs['max_digits'] = self.max_digits
- if self.decimal_places is not None:
- kwargs['decimal_places'] = self.decimal_places
- return name, path, args, kwargs
- def get_internal_type(self):
- return "DecimalField"
- def to_python(self, value):
- if value is None:
- return value
- try:
- return decimal.Decimal(value)
- except decimal.InvalidOperation:
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def _format(self, value):
- if isinstance(value, six.string_types):
- return value
- else:
- return self.format_number(value)
- def format_number(self, value):
- """
- Formats a number into a string with the requisite number of digits and
- decimal places.
- """
- # Method moved to django.db.backends.utils.
- #
- # It is preserved because it is used by the oracle backend
- # (django.db.backends.oracle.query), and also for
- # backwards-compatibility with any external code which may have used
- # this method.
- from django.db.backends import utils
- return utils.format_number(value, self.max_digits, self.decimal_places)
- def get_db_prep_save(self, value, connection):
- return connection.ops.value_to_db_decimal(self.to_python(value),
- self.max_digits, self.decimal_places)
- def get_prep_value(self, value):
- value = super(DecimalField, self).get_prep_value(value)
- return self.to_python(value)
- def formfield(self, **kwargs):
- defaults = {
- 'max_digits': self.max_digits,
- 'decimal_places': self.decimal_places,
- 'form_class': forms.DecimalField,
- }
- defaults.update(kwargs)
- return super(DecimalField, self).formfield(**defaults)
- class DurationField(Field):
- """Stores timedelta objects.
- Uses interval on postgres, INVERAL DAY TO SECOND on Oracle, and bigint of
- microseconds on other databases.
- """
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value has an invalid format. It must be in "
- "[DD] [HH:[MM:]]ss[.uuuuuu] format.")
- }
- description = _("Duration")
- def get_internal_type(self):
- return "DurationField"
- def to_python(self, value):
- if value is None:
- return value
- if isinstance(value, datetime.timedelta):
- return value
- try:
- parsed = parse_duration(value)
- except ValueError:
- pass
- else:
- if parsed is not None:
- return parsed
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def get_db_prep_value(self, value, connection, prepared=False):
- if connection.features.has_native_duration_field:
- return value
- return value.total_seconds() * 1000000
- def get_db_converters(self, connection):
- converters = []
- if not connection.features.has_native_duration_field:
- converters.append(connection.ops.convert_durationfield_value)
- return converters + super(DurationField, self).get_db_converters(connection)
- def value_to_string(self, obj):
- val = self._get_val_from_obj(obj)
- return '' if val is None else duration_string(val)
- class EmailField(CharField):
- default_validators = [validators.validate_email]
- description = _("Email address")
- def __init__(self, *args, **kwargs):
- # max_length=254 to be compliant with RFCs 3696 and 5321
- kwargs['max_length'] = kwargs.get('max_length', 254)
- super(EmailField, self).__init__(*args, **kwargs)
- def deconstruct(self):
- name, path, args, kwargs = super(EmailField, self).deconstruct()
- # We do not exclude max_length if it matches default as we want to change
- # the default in future.
- return name, path, args, kwargs
- def formfield(self, **kwargs):
- # As with CharField, this will cause email validation to be performed
- # twice.
- defaults = {
- 'form_class': forms.EmailField,
- }
- defaults.update(kwargs)
- return super(EmailField, self).formfield(**defaults)
- class FilePathField(Field):
- description = _("File path")
- def __init__(self, verbose_name=None, name=None, path='', match=None,
- recursive=False, allow_files=True, allow_folders=False, **kwargs):
- self.path, self.match, self.recursive = path, match, recursive
- self.allow_files, self.allow_folders = allow_files, allow_folders
- kwargs['max_length'] = kwargs.get('max_length', 100)
- super(FilePathField, self).__init__(verbose_name, name, **kwargs)
- def check(self, **kwargs):
- errors = super(FilePathField, self).check(**kwargs)
- errors.extend(self._check_allowing_files_or_folders(**kwargs))
- return errors
- def _check_allowing_files_or_folders(self, **kwargs):
- if not self.allow_files and not self.allow_folders:
- return [
- checks.Error(
- "FilePathFields must have either 'allow_files' or 'allow_folders' set to True.",
- hint=None,
- obj=self,
- id='fields.E140',
- )
- ]
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(FilePathField, self).deconstruct()
- if self.path != '':
- kwargs['path'] = self.path
- if self.match is not None:
- kwargs['match'] = self.match
- if self.recursive is not False:
- kwargs['recursive'] = self.recursive
- if self.allow_files is not True:
- kwargs['allow_files'] = self.allow_files
- if self.allow_folders is not False:
- kwargs['allow_folders'] = self.allow_folders
- if kwargs.get("max_length", None) == 100:
- del kwargs["max_length"]
- return name, path, args, kwargs
- def get_prep_value(self, value):
- value = super(FilePathField, self).get_prep_value(value)
- if value is None:
- return None
- return six.text_type(value)
- def formfield(self, **kwargs):
- defaults = {
- 'path': self.path,
- 'match': self.match,
- 'recursive': self.recursive,
- 'form_class': forms.FilePathField,
- 'allow_files': self.allow_files,
- 'allow_folders': self.allow_folders,
- }
- defaults.update(kwargs)
- return super(FilePathField, self).formfield(**defaults)
- def get_internal_type(self):
- return "FilePathField"
- class FloatField(Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be a float."),
- }
- description = _("Floating point number")
- def get_prep_value(self, value):
- value = super(FloatField, self).get_prep_value(value)
- if value is None:
- return None
- return float(value)
- def get_internal_type(self):
- return "FloatField"
- def to_python(self, value):
- if value is None:
- return value
- try:
- return float(value)
- except (TypeError, ValueError):
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.FloatField}
- defaults.update(kwargs)
- return super(FloatField, self).formfield(**defaults)
- class IntegerField(Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be an integer."),
- }
- description = _("Integer")
- def check(self, **kwargs):
- errors = super(IntegerField, self).check(**kwargs)
- errors.extend(self._check_max_length_warning())
- return errors
- def _check_max_length_warning(self):
- if self.max_length is not None:
- return [
- checks.Warning(
- "'max_length' is ignored when used with IntegerField",
- hint="Remove 'max_length' from field",
- obj=self,
- id='fields.W122',
- )
- ]
- return []
- @cached_property
- def validators(self):
- # These validators can't be added at field initialization time since
- # they're based on values retrieved from `connection`.
- range_validators = []
- internal_type = self.get_internal_type()
- min_value, max_value = connection.ops.integer_field_range(internal_type)
- if min_value is not None:
- range_validators.append(validators.MinValueValidator(min_value))
- if max_value is not None:
- range_validators.append(validators.MaxValueValidator(max_value))
- return super(IntegerField, self).validators + range_validators
- def get_prep_value(self, value):
- value = super(IntegerField, self).get_prep_value(value)
- if value is None:
- return None
- return int(value)
- def get_prep_lookup(self, lookup_type, value):
- if ((lookup_type == 'gte' or lookup_type == 'lt')
- and isinstance(value, float)):
- value = math.ceil(value)
- return super(IntegerField, self).get_prep_lookup(lookup_type, value)
- def get_internal_type(self):
- return "IntegerField"
- def to_python(self, value):
- if value is None:
- return value
- try:
- return int(value)
- except (TypeError, ValueError):
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.IntegerField}
- defaults.update(kwargs)
- return super(IntegerField, self).formfield(**defaults)
- class BigIntegerField(IntegerField):
- empty_strings_allowed = False
- description = _("Big (8 byte) integer")
- MAX_BIGINT = 9223372036854775807
- def get_internal_type(self):
- return "BigIntegerField"
- def formfield(self, **kwargs):
- defaults = {'min_value': -BigIntegerField.MAX_BIGINT - 1,
- 'max_value': BigIntegerField.MAX_BIGINT}
- defaults.update(kwargs)
- return super(BigIntegerField, self).formfield(**defaults)
- class IPAddressField(Field):
- empty_strings_allowed = False
- description = _("IPv4 address")
- def __init__(self, *args, **kwargs):
- kwargs['max_length'] = 15
- super(IPAddressField, self).__init__(*args, **kwargs)
- def deconstruct(self):
- name, path, args, kwargs = super(IPAddressField, self).deconstruct()
- del kwargs['max_length']
- return name, path, args, kwargs
- def get_prep_value(self, value):
- value = super(IPAddressField, self).get_prep_value(value)
- if value is None:
- return None
- return six.text_type(value)
- def get_internal_type(self):
- return "IPAddressField"
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.IPAddressField}
- defaults.update(kwargs)
- return super(IPAddressField, self).formfield(**defaults)
- def check(self, **kwargs):
- errors = super(IPAddressField, self).check(**kwargs)
- errors.append(
- checks.Warning(
- 'IPAddressField has been deprecated. Support for it '
- '(except in historical migrations) will be removed in Django 1.9.',
- hint='Use GenericIPAddressField instead.',
- obj=self,
- id='fields.W900',
- )
- )
- return errors
- class GenericIPAddressField(Field):
- empty_strings_allowed = True
- description = _("IP address")
- default_error_messages = {}
- def __init__(self, verbose_name=None, name=None, protocol='both',
- unpack_ipv4=False, *args, **kwargs):
- self.unpack_ipv4 = unpack_ipv4
- self.protocol = protocol
- self.default_validators, invalid_error_message = \
- validators.ip_address_validators(protocol, unpack_ipv4)
- self.default_error_messages['invalid'] = invalid_error_message
- kwargs['max_length'] = 39
- super(GenericIPAddressField, self).__init__(verbose_name, name, *args,
- **kwargs)
- def check(self, **kwargs):
- errors = super(GenericIPAddressField, self).check(**kwargs)
- errors.extend(self._check_blank_and_null_values(**kwargs))
- return errors
- def _check_blank_and_null_values(self, **kwargs):
- if not getattr(self, 'null', False) and getattr(self, 'blank', False):
- return [
- checks.Error(
- ('GenericIPAddressFields cannot have blank=True if null=False, '
- 'as blank values are stored as nulls.'),
- hint=None,
- obj=self,
- id='fields.E150',
- )
- ]
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(GenericIPAddressField, self).deconstruct()
- if self.unpack_ipv4 is not False:
- kwargs['unpack_ipv4'] = self.unpack_ipv4
- if self.protocol != "both":
- kwargs['protocol'] = self.protocol
- if kwargs.get("max_length", None) == 39:
- del kwargs['max_length']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "GenericIPAddressField"
- def to_python(self, value):
- if value and ':' in value:
- return clean_ipv6_address(value,
- self.unpack_ipv4, self.error_messages['invalid'])
- return value
- def get_db_prep_value(self, value, connection, prepared=False):
- if not prepared:
- value = self.get_prep_value(value)
- return value or None
- def get_prep_value(self, value):
- value = super(GenericIPAddressField, self).get_prep_value(value)
- if value is None:
- return None
- if value and ':' in value:
- try:
- return clean_ipv6_address(value, self.unpack_ipv4)
- except exceptions.ValidationError:
- pass
- return six.text_type(value)
- def formfield(self, **kwargs):
- defaults = {
- 'protocol': self.protocol,
- 'form_class': forms.GenericIPAddressField,
- }
- defaults.update(kwargs)
- return super(GenericIPAddressField, self).formfield(**defaults)
- class NullBooleanField(Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value must be either None, True or False."),
- }
- description = _("Boolean (Either True, False or None)")
- def __init__(self, *args, **kwargs):
- kwargs['null'] = True
- kwargs['blank'] = True
- super(NullBooleanField, self).__init__(*args, **kwargs)
- def deconstruct(self):
- name, path, args, kwargs = super(NullBooleanField, self).deconstruct()
- del kwargs['null']
- del kwargs['blank']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "NullBooleanField"
- def to_python(self, value):
- if value is None:
- return None
- if value in (True, False):
- return bool(value)
- if value in ('None',):
- return None
- if value in ('t', 'True', '1'):
- return True
- if value in ('f', 'False', '0'):
- return False
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def get_prep_lookup(self, lookup_type, value):
- # Special-case handling for filters coming from a Web request (e.g. the
- # admin interface). Only works for scalar values (not lists). If you're
- # passing in a list, you might as well make things the right type when
- # constructing the list.
- if value in ('1', '0'):
- value = bool(int(value))
- return super(NullBooleanField, self).get_prep_lookup(lookup_type,
- value)
- def get_prep_value(self, value):
- value = super(NullBooleanField, self).get_prep_value(value)
- if value is None:
- return None
- return bool(value)
- def formfield(self, **kwargs):
- defaults = {
- 'form_class': forms.NullBooleanField,
- 'required': not self.blank,
- 'label': capfirst(self.verbose_name),
- 'help_text': self.help_text}
- defaults.update(kwargs)
- return super(NullBooleanField, self).formfield(**defaults)
- class PositiveIntegerField(IntegerField):
- description = _("Positive integer")
- def get_internal_type(self):
- return "PositiveIntegerField"
- def formfield(self, **kwargs):
- defaults = {'min_value': 0}
- defaults.update(kwargs)
- return super(PositiveIntegerField, self).formfield(**defaults)
- class PositiveSmallIntegerField(IntegerField):
- description = _("Positive small integer")
- def get_internal_type(self):
- return "PositiveSmallIntegerField"
- def formfield(self, **kwargs):
- defaults = {'min_value': 0}
- defaults.update(kwargs)
- return super(PositiveSmallIntegerField, self).formfield(**defaults)
- class SlugField(CharField):
- default_validators = [validators.validate_slug]
- description = _("Slug (up to %(max_length)s)")
- def __init__(self, *args, **kwargs):
- kwargs['max_length'] = kwargs.get('max_length', 50)
- # Set db_index=True unless it's been set manually.
- if 'db_index' not in kwargs:
- kwargs['db_index'] = True
- super(SlugField, self).__init__(*args, **kwargs)
- def deconstruct(self):
- name, path, args, kwargs = super(SlugField, self).deconstruct()
- if kwargs.get("max_length", None) == 50:
- del kwargs['max_length']
- if self.db_index is False:
- kwargs['db_index'] = False
- else:
- del kwargs['db_index']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "SlugField"
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.SlugField}
- defaults.update(kwargs)
- return super(SlugField, self).formfield(**defaults)
- class SmallIntegerField(IntegerField):
- description = _("Small integer")
- def get_internal_type(self):
- return "SmallIntegerField"
- class TextField(Field):
- description = _("Text")
- def get_internal_type(self):
- return "TextField"
- def get_prep_value(self, value):
- value = super(TextField, self).get_prep_value(value)
- if isinstance(value, six.string_types) or value is None:
- return value
- return smart_text(value)
- def formfield(self, **kwargs):
- # Passing max_length to forms.CharField means that the value's length
- # will be validated twice. This is considered acceptable since we want
- # the value in the form field (to pass into widget for example).
- defaults = {'max_length': self.max_length, 'widget': forms.Textarea}
- defaults.update(kwargs)
- return super(TextField, self).formfield(**defaults)
- class TimeField(DateTimeCheckMixin, Field):
- empty_strings_allowed = False
- default_error_messages = {
- 'invalid': _("'%(value)s' value has an invalid format. It must be in "
- "HH:MM[:ss[.uuuuuu]] format."),
- 'invalid_time': _("'%(value)s' value has the correct format "
- "(HH:MM[:ss[.uuuuuu]]) but it is an invalid time."),
- }
- description = _("Time")
- def __init__(self, verbose_name=None, name=None, auto_now=False,
- auto_now_add=False, **kwargs):
- self.auto_now, self.auto_now_add = auto_now, auto_now_add
- if auto_now or auto_now_add:
- kwargs['editable'] = False
- kwargs['blank'] = True
- super(TimeField, self).__init__(verbose_name, name, **kwargs)
- def _check_fix_default_value(self):
- """
- Adds a warning to the checks framework stating, that using an actual
- time or datetime value is probably wrong; it's only being evaluated on
- server start-up.
- For details see ticket #21905
- """
- if not self.has_default():
- return []
- now = timezone.now()
- if not timezone.is_naive(now):
- now = timezone.make_naive(now, timezone.utc)
- value = self.default
- if isinstance(value, datetime.datetime):
- second_offset = datetime.timedelta(seconds=10)
- lower = now - second_offset
- upper = now + second_offset
- if timezone.is_aware(value):
- value = timezone.make_naive(value, timezone.utc)
- elif isinstance(value, datetime.time):
- second_offset = datetime.timedelta(seconds=10)
- lower = now - second_offset
- upper = now + second_offset
- value = datetime.datetime.combine(now.date(), value)
- if timezone.is_aware(value):
- value = timezone.make_naive(value, timezone.utc).time()
- else:
- # No explicit time / datetime value -- no checks necessary
- return []
- if lower <= value <= upper:
- return [
- checks.Warning(
- 'Fixed default value provided.',
- hint='It seems you set a fixed date / time / datetime '
- 'value as default for this field. This may not be '
- 'what you want. If you want to have the current date '
- 'as default, use `django.utils.timezone.now`',
- obj=self,
- id='fields.W161',
- )
- ]
- return []
- def deconstruct(self):
- name, path, args, kwargs = super(TimeField, self).deconstruct()
- if self.auto_now is not False:
- kwargs["auto_now"] = self.auto_now
- if self.auto_now_add is not False:
- kwargs["auto_now_add"] = self.auto_now_add
- if self.auto_now or self.auto_now_add:
- del kwargs['blank']
- del kwargs['editable']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "TimeField"
- def to_python(self, value):
- if value is None:
- return None
- if isinstance(value, datetime.time):
- return value
- if isinstance(value, datetime.datetime):
- # Not usually a good idea to pass in a datetime here (it loses
- # information), but this can be a side-effect of interacting with a
- # database backend (e.g. Oracle), so we'll be accommodating.
- return value.time()
- try:
- parsed = parse_time(value)
- if parsed is not None:
- return parsed
- except ValueError:
- raise exceptions.ValidationError(
- self.error_messages['invalid_time'],
- code='invalid_time',
- params={'value': value},
- )
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- def pre_save(self, model_instance, add):
- if self.auto_now or (self.auto_now_add and add):
- value = datetime.datetime.now().time()
- setattr(model_instance, self.attname, value)
- return value
- else:
- return super(TimeField, self).pre_save(model_instance, add)
- def get_prep_value(self, value):
- value = super(TimeField, self).get_prep_value(value)
- return self.to_python(value)
- def get_db_prep_value(self, value, connection, prepared=False):
- # Casts times into the format expected by the backend
- if not prepared:
- value = self.get_prep_value(value)
- return connection.ops.value_to_db_time(value)
- def value_to_string(self, obj):
- val = self._get_val_from_obj(obj)
- return '' if val is None else val.isoformat()
- def formfield(self, **kwargs):
- defaults = {'form_class': forms.TimeField}
- defaults.update(kwargs)
- return super(TimeField, self).formfield(**defaults)
- class URLField(CharField):
- default_validators = [validators.URLValidator()]
- description = _("URL")
- def __init__(self, verbose_name=None, name=None, **kwargs):
- kwargs['max_length'] = kwargs.get('max_length', 200)
- super(URLField, self).__init__(verbose_name, name, **kwargs)
- def deconstruct(self):
- name, path, args, kwargs = super(URLField, self).deconstruct()
- if kwargs.get("max_length", None) == 200:
- del kwargs['max_length']
- return name, path, args, kwargs
- def formfield(self, **kwargs):
- # As with CharField, this will cause URL validation to be performed
- # twice.
- defaults = {
- 'form_class': forms.URLField,
- }
- defaults.update(kwargs)
- return super(URLField, self).formfield(**defaults)
- class BinaryField(Field):
- description = _("Raw binary data")
- empty_values = [None, b'']
- def __init__(self, *args, **kwargs):
- kwargs['editable'] = False
- super(BinaryField, self).__init__(*args, **kwargs)
- if self.max_length is not None:
- self.validators.append(validators.MaxLengthValidator(self.max_length))
- def deconstruct(self):
- name, path, args, kwargs = super(BinaryField, self).deconstruct()
- del kwargs['editable']
- return name, path, args, kwargs
- def get_internal_type(self):
- return "BinaryField"
- def get_default(self):
- if self.has_default() and not callable(self.default):
- return self.default
- default = super(BinaryField, self).get_default()
- if default == '':
- return b''
- return default
- def get_db_prep_value(self, value, connection, prepared=False):
- value = super(BinaryField, self).get_db_prep_value(value, connection, prepared)
- if value is not None:
- return connection.Database.Binary(value)
- return value
- def value_to_string(self, obj):
- """Binary data is serialized as base64"""
- return b64encode(force_bytes(self._get_val_from_obj(obj))).decode('ascii')
- def to_python(self, value):
- # If it's a string, it should be base64-encoded data
- if isinstance(value, six.text_type):
- return six.memoryview(b64decode(force_bytes(value)))
- return value
- class UUIDField(Field):
- default_error_messages = {
- 'invalid': _("'%(value)s' is not a valid UUID."),
- }
- description = 'Universally unique identifier'
- empty_strings_allowed = False
- def __init__(self, **kwargs):
- kwargs['max_length'] = 32
- super(UUIDField, self).__init__(**kwargs)
- def get_internal_type(self):
- return "UUIDField"
- def get_prep_value(self, value):
- if isinstance(value, uuid.UUID):
- return value.hex
- if isinstance(value, six.string_types):
- return value.replace('-', '')
- return value
- def to_python(self, value):
- if value and not isinstance(value, uuid.UUID):
- try:
- return uuid.UUID(value)
- except ValueError:
- raise exceptions.ValidationError(
- self.error_messages['invalid'],
- code='invalid',
- params={'value': value},
- )
- return value
- def formfield(self, **kwargs):
- defaults = {
- 'form_class': forms.UUIDField,
- }
- defaults.update(kwargs)
- return super(UUIDField, self).formfield(**defaults)
|