__init__.py 84 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336
  1. # -*- coding: utf-8 -*-
  2. from __future__ import unicode_literals
  3. import collections
  4. import copy
  5. import datetime
  6. import decimal
  7. import math
  8. import uuid
  9. import warnings
  10. from base64 import b64decode, b64encode
  11. from itertools import tee
  12. from django.apps import apps
  13. from django.db import connection
  14. from django.db.models.lookups import default_lookups, RegisterLookupMixin
  15. from django.db.models.query_utils import QueryWrapper
  16. from django.conf import settings
  17. from django import forms
  18. from django.core import exceptions, validators, checks
  19. from django.utils.datastructures import DictWrapper
  20. from django.utils.dateparse import parse_date, parse_datetime, parse_time, parse_duration
  21. from django.utils.duration import duration_string
  22. from django.utils.functional import cached_property, curry, total_ordering, Promise
  23. from django.utils.text import capfirst
  24. from django.utils import timezone
  25. from django.utils.translation import ugettext_lazy as _
  26. from django.utils.encoding import (smart_text, force_text, force_bytes,
  27. python_2_unicode_compatible)
  28. from django.utils.ipv6 import clean_ipv6_address
  29. from django.utils import six
  30. from django.utils.itercompat import is_iterable
  31. # imported for backwards compatibility
  32. from django.core.exceptions import FieldDoesNotExist # NOQA
  33. # Avoid "TypeError: Item in ``from list'' not a string" -- unicode_literals
  34. # makes these strings unicode
  35. __all__ = [str(x) for x in (
  36. 'AutoField', 'BLANK_CHOICE_DASH', 'BigIntegerField', 'BinaryField',
  37. 'BooleanField', 'CharField', 'CommaSeparatedIntegerField', 'DateField',
  38. 'DateTimeField', 'DecimalField', 'DurationField', 'EmailField', 'Empty',
  39. 'Field', 'FieldDoesNotExist', 'FilePathField', 'FloatField',
  40. 'GenericIPAddressField', 'IPAddressField', 'IntegerField', 'NOT_PROVIDED',
  41. 'NullBooleanField', 'PositiveIntegerField', 'PositiveSmallIntegerField',
  42. 'SlugField', 'SmallIntegerField', 'TextField', 'TimeField', 'URLField',
  43. 'UUIDField',
  44. )]
  45. class Empty(object):
  46. pass
  47. class NOT_PROVIDED:
  48. pass
  49. # The values to use for "blank" in SelectFields. Will be appended to the start
  50. # of most "choices" lists.
  51. BLANK_CHOICE_DASH = [("", "---------")]
  52. def _load_field(app_label, model_name, field_name):
  53. return apps.get_model(app_label, model_name)._meta.get_field_by_name(field_name)[0]
  54. # A guide to Field parameters:
  55. #
  56. # * name: The name of the field specified in the model.
  57. # * attname: The attribute to use on the model object. This is the same as
  58. # "name", except in the case of ForeignKeys, where "_id" is
  59. # appended.
  60. # * db_column: The db_column specified in the model (or None).
  61. # * column: The database column for this field. This is the same as
  62. # "attname", except if db_column is specified.
  63. #
  64. # Code that introspects values, or does other dynamic things, should use
  65. # attname. For example, this gets the primary key value of object "obj":
  66. #
  67. # getattr(obj, opts.pk.attname)
  68. def _empty(of_cls):
  69. new = Empty()
  70. new.__class__ = of_cls
  71. return new
  72. @total_ordering
  73. @python_2_unicode_compatible
  74. class Field(RegisterLookupMixin):
  75. """Base class for all field types"""
  76. # Designates whether empty strings fundamentally are allowed at the
  77. # database level.
  78. empty_strings_allowed = True
  79. empty_values = list(validators.EMPTY_VALUES)
  80. # These track each time a Field instance is created. Used to retain order.
  81. # The auto_creation_counter is used for fields that Django implicitly
  82. # creates, creation_counter is used for all user-specified fields.
  83. creation_counter = 0
  84. auto_creation_counter = -1
  85. default_validators = [] # Default set of validators
  86. default_error_messages = {
  87. 'invalid_choice': _('Value %(value)r is not a valid choice.'),
  88. 'null': _('This field cannot be null.'),
  89. 'blank': _('This field cannot be blank.'),
  90. 'unique': _('%(model_name)s with this %(field_label)s '
  91. 'already exists.'),
  92. # Translators: The 'lookup_type' is one of 'date', 'year' or 'month'.
  93. # Eg: "Title must be unique for pub_date year"
  94. 'unique_for_date': _("%(field_label)s must be unique for "
  95. "%(date_field_label)s %(lookup_type)s."),
  96. }
  97. class_lookups = default_lookups.copy()
  98. # Generic field type description, usually overridden by subclasses
  99. def _description(self):
  100. return _('Field of type: %(field_type)s') % {
  101. 'field_type': self.__class__.__name__
  102. }
  103. description = property(_description)
  104. def __init__(self, verbose_name=None, name=None, primary_key=False,
  105. max_length=None, unique=False, blank=False, null=False,
  106. db_index=False, rel=None, default=NOT_PROVIDED, editable=True,
  107. serialize=True, unique_for_date=None, unique_for_month=None,
  108. unique_for_year=None, choices=None, help_text='', db_column=None,
  109. db_tablespace=None, auto_created=False, validators=[],
  110. error_messages=None):
  111. self.name = name
  112. self.verbose_name = verbose_name # May be set by set_attributes_from_name
  113. self._verbose_name = verbose_name # Store original for deconstruction
  114. self.primary_key = primary_key
  115. self.max_length, self._unique = max_length, unique
  116. self.blank, self.null = blank, null
  117. self.rel = rel
  118. self.default = default
  119. self.editable = editable
  120. self.serialize = serialize
  121. self.unique_for_date = unique_for_date
  122. self.unique_for_month = unique_for_month
  123. self.unique_for_year = unique_for_year
  124. self._choices = choices or []
  125. self.help_text = help_text
  126. self.db_column = db_column
  127. self.db_tablespace = db_tablespace or settings.DEFAULT_INDEX_TABLESPACE
  128. self.auto_created = auto_created
  129. # Set db_index to True if the field has a relationship and doesn't
  130. # explicitly set db_index.
  131. self.db_index = db_index
  132. # Adjust the appropriate creation counter, and save our local copy.
  133. if auto_created:
  134. self.creation_counter = Field.auto_creation_counter
  135. Field.auto_creation_counter -= 1
  136. else:
  137. self.creation_counter = Field.creation_counter
  138. Field.creation_counter += 1
  139. self._validators = validators # Store for deconstruction later
  140. messages = {}
  141. for c in reversed(self.__class__.__mro__):
  142. messages.update(getattr(c, 'default_error_messages', {}))
  143. messages.update(error_messages or {})
  144. self._error_messages = error_messages # Store for deconstruction later
  145. self.error_messages = messages
  146. def __str__(self):
  147. """ Return "app_label.model_label.field_name". """
  148. model = self.model
  149. app = model._meta.app_label
  150. return '%s.%s.%s' % (app, model._meta.object_name, self.name)
  151. def __repr__(self):
  152. """
  153. Displays the module, class and name of the field.
  154. """
  155. path = '%s.%s' % (self.__class__.__module__, self.__class__.__name__)
  156. name = getattr(self, 'name', None)
  157. if name is not None:
  158. return '<%s: %s>' % (path, name)
  159. return '<%s>' % path
  160. def check(self, **kwargs):
  161. errors = []
  162. errors.extend(self._check_field_name())
  163. errors.extend(self._check_choices())
  164. errors.extend(self._check_db_index())
  165. errors.extend(self._check_null_allowed_for_primary_keys())
  166. errors.extend(self._check_backend_specific_checks(**kwargs))
  167. return errors
  168. def _check_field_name(self):
  169. """ Check if field name is valid, i.e. 1) does not end with an
  170. underscore, 2) does not contain "__" and 3) is not "pk". """
  171. if self.name.endswith('_'):
  172. return [
  173. checks.Error(
  174. 'Field names must not end with an underscore.',
  175. hint=None,
  176. obj=self,
  177. id='fields.E001',
  178. )
  179. ]
  180. elif '__' in self.name:
  181. return [
  182. checks.Error(
  183. 'Field names must not contain "__".',
  184. hint=None,
  185. obj=self,
  186. id='fields.E002',
  187. )
  188. ]
  189. elif self.name == 'pk':
  190. return [
  191. checks.Error(
  192. "'pk' is a reserved word that cannot be used as a field name.",
  193. hint=None,
  194. obj=self,
  195. id='fields.E003',
  196. )
  197. ]
  198. else:
  199. return []
  200. def _check_choices(self):
  201. if self.choices:
  202. if (isinstance(self.choices, six.string_types) or
  203. not is_iterable(self.choices)):
  204. return [
  205. checks.Error(
  206. "'choices' must be an iterable (e.g., a list or tuple).",
  207. hint=None,
  208. obj=self,
  209. id='fields.E004',
  210. )
  211. ]
  212. elif any(isinstance(choice, six.string_types) or
  213. not is_iterable(choice) or len(choice) != 2
  214. for choice in self.choices):
  215. return [
  216. checks.Error(
  217. ("'choices' must be an iterable containing "
  218. "(actual value, human readable name) tuples."),
  219. hint=None,
  220. obj=self,
  221. id='fields.E005',
  222. )
  223. ]
  224. else:
  225. return []
  226. else:
  227. return []
  228. def _check_db_index(self):
  229. if self.db_index not in (None, True, False):
  230. return [
  231. checks.Error(
  232. "'db_index' must be None, True or False.",
  233. hint=None,
  234. obj=self,
  235. id='fields.E006',
  236. )
  237. ]
  238. else:
  239. return []
  240. def _check_null_allowed_for_primary_keys(self):
  241. if (self.primary_key and self.null and
  242. not connection.features.interprets_empty_strings_as_nulls):
  243. # We cannot reliably check this for backends like Oracle which
  244. # consider NULL and '' to be equal (and thus set up
  245. # character-based fields a little differently).
  246. return [
  247. checks.Error(
  248. 'Primary keys must not have null=True.',
  249. hint=('Set null=False on the field, or '
  250. 'remove primary_key=True argument.'),
  251. obj=self,
  252. id='fields.E007',
  253. )
  254. ]
  255. else:
  256. return []
  257. def _check_backend_specific_checks(self, **kwargs):
  258. return connection.validation.check_field(self, **kwargs)
  259. def deconstruct(self):
  260. """
  261. Returns enough information to recreate the field as a 4-tuple:
  262. * The name of the field on the model, if contribute_to_class has been run
  263. * The import path of the field, including the class: django.db.models.IntegerField
  264. This should be the most portable version, so less specific may be better.
  265. * A list of positional arguments
  266. * A dict of keyword arguments
  267. Note that the positional or keyword arguments must contain values of the
  268. following types (including inner values of collection types):
  269. * None, bool, str, unicode, int, long, float, complex, set, frozenset, list, tuple, dict
  270. * UUID
  271. * datetime.datetime (naive), datetime.date
  272. * top-level classes, top-level functions - will be referenced by their full import path
  273. * Storage instances - these have their own deconstruct() method
  274. This is because the values here must be serialized into a text format
  275. (possibly new Python code, possibly JSON) and these are the only types
  276. with encoding handlers defined.
  277. There's no need to return the exact way the field was instantiated this time,
  278. just ensure that the resulting field is the same - prefer keyword arguments
  279. over positional ones, and omit parameters with their default values.
  280. """
  281. # Short-form way of fetching all the default parameters
  282. keywords = {}
  283. possibles = {
  284. "verbose_name": None,
  285. "primary_key": False,
  286. "max_length": None,
  287. "unique": False,
  288. "blank": False,
  289. "null": False,
  290. "db_index": False,
  291. "default": NOT_PROVIDED,
  292. "editable": True,
  293. "serialize": True,
  294. "unique_for_date": None,
  295. "unique_for_month": None,
  296. "unique_for_year": None,
  297. "choices": [],
  298. "help_text": '',
  299. "db_column": None,
  300. "db_tablespace": settings.DEFAULT_INDEX_TABLESPACE,
  301. "auto_created": False,
  302. "validators": [],
  303. "error_messages": None,
  304. }
  305. attr_overrides = {
  306. "unique": "_unique",
  307. "choices": "_choices",
  308. "error_messages": "_error_messages",
  309. "validators": "_validators",
  310. "verbose_name": "_verbose_name",
  311. }
  312. equals_comparison = {"choices", "validators", "db_tablespace"}
  313. for name, default in possibles.items():
  314. value = getattr(self, attr_overrides.get(name, name))
  315. # Unroll anything iterable for choices into a concrete list
  316. if name == "choices" and isinstance(value, collections.Iterable):
  317. value = list(value)
  318. # Do correct kind of comparison
  319. if name in equals_comparison:
  320. if value != default:
  321. keywords[name] = value
  322. else:
  323. if value is not default:
  324. keywords[name] = value
  325. # Work out path - we shorten it for known Django core fields
  326. path = "%s.%s" % (self.__class__.__module__, self.__class__.__name__)
  327. if path.startswith("django.db.models.fields.related"):
  328. path = path.replace("django.db.models.fields.related", "django.db.models")
  329. if path.startswith("django.db.models.fields.files"):
  330. path = path.replace("django.db.models.fields.files", "django.db.models")
  331. if path.startswith("django.db.models.fields.proxy"):
  332. path = path.replace("django.db.models.fields.proxy", "django.db.models")
  333. if path.startswith("django.db.models.fields"):
  334. path = path.replace("django.db.models.fields", "django.db.models")
  335. # Return basic info - other fields should override this.
  336. return (
  337. force_text(self.name, strings_only=True),
  338. path,
  339. [],
  340. keywords,
  341. )
  342. def clone(self):
  343. """
  344. Uses deconstruct() to clone a new copy of this Field.
  345. Will not preserve any class attachments/attribute names.
  346. """
  347. name, path, args, kwargs = self.deconstruct()
  348. return self.__class__(*args, **kwargs)
  349. def __eq__(self, other):
  350. # Needed for @total_ordering
  351. if isinstance(other, Field):
  352. return self.creation_counter == other.creation_counter
  353. return NotImplemented
  354. def __lt__(self, other):
  355. # This is needed because bisect does not take a comparison function.
  356. if isinstance(other, Field):
  357. return self.creation_counter < other.creation_counter
  358. return NotImplemented
  359. def __hash__(self):
  360. return hash(self.creation_counter)
  361. def __deepcopy__(self, memodict):
  362. # We don't have to deepcopy very much here, since most things are not
  363. # intended to be altered after initial creation.
  364. obj = copy.copy(self)
  365. if self.rel:
  366. obj.rel = copy.copy(self.rel)
  367. if hasattr(self.rel, 'field') and self.rel.field is self:
  368. obj.rel.field = obj
  369. memodict[id(self)] = obj
  370. return obj
  371. def __copy__(self):
  372. # We need to avoid hitting __reduce__, so define this
  373. # slightly weird copy construct.
  374. obj = Empty()
  375. obj.__class__ = self.__class__
  376. obj.__dict__ = self.__dict__.copy()
  377. return obj
  378. def __reduce__(self):
  379. """
  380. Pickling should return the model._meta.fields instance of the field,
  381. not a new copy of that field. So, we use the app registry to load the
  382. model and then the field back.
  383. """
  384. if not hasattr(self, 'model'):
  385. # Fields are sometimes used without attaching them to models (for
  386. # example in aggregation). In this case give back a plain field
  387. # instance. The code below will create a new empty instance of
  388. # class self.__class__, then update its dict with self.__dict__
  389. # values - so, this is very close to normal pickle.
  390. return _empty, (self.__class__,), self.__dict__
  391. if self.model._deferred:
  392. # Deferred model will not be found from the app registry. This
  393. # could be fixed by reconstructing the deferred model on unpickle.
  394. raise RuntimeError("Fields of deferred models can't be reduced")
  395. return _load_field, (self.model._meta.app_label, self.model._meta.object_name,
  396. self.name)
  397. def to_python(self, value):
  398. """
  399. Converts the input value into the expected Python data type, raising
  400. django.core.exceptions.ValidationError if the data can't be converted.
  401. Returns the converted value. Subclasses should override this.
  402. """
  403. return value
  404. @cached_property
  405. def validators(self):
  406. # Some validators can't be created at field initialization time.
  407. # This method provides a way to delay their creation until required.
  408. return self.default_validators + self._validators
  409. def run_validators(self, value):
  410. if value in self.empty_values:
  411. return
  412. errors = []
  413. for v in self.validators:
  414. try:
  415. v(value)
  416. except exceptions.ValidationError as e:
  417. if hasattr(e, 'code') and e.code in self.error_messages:
  418. e.message = self.error_messages[e.code]
  419. errors.extend(e.error_list)
  420. if errors:
  421. raise exceptions.ValidationError(errors)
  422. def validate(self, value, model_instance):
  423. """
  424. Validates value and throws ValidationError. Subclasses should override
  425. this to provide validation logic.
  426. """
  427. if not self.editable:
  428. # Skip validation for non-editable fields.
  429. return
  430. if self._choices and value not in self.empty_values:
  431. for option_key, option_value in self.choices:
  432. if isinstance(option_value, (list, tuple)):
  433. # This is an optgroup, so look inside the group for
  434. # options.
  435. for optgroup_key, optgroup_value in option_value:
  436. if value == optgroup_key:
  437. return
  438. elif value == option_key:
  439. return
  440. raise exceptions.ValidationError(
  441. self.error_messages['invalid_choice'],
  442. code='invalid_choice',
  443. params={'value': value},
  444. )
  445. if value is None and not self.null:
  446. raise exceptions.ValidationError(self.error_messages['null'], code='null')
  447. if not self.blank and value in self.empty_values:
  448. raise exceptions.ValidationError(self.error_messages['blank'], code='blank')
  449. def clean(self, value, model_instance):
  450. """
  451. Convert the value's type and run validation. Validation errors
  452. from to_python and validate are propagated. The correct value is
  453. returned if no error is raised.
  454. """
  455. value = self.to_python(value)
  456. self.validate(value, model_instance)
  457. self.run_validators(value)
  458. return value
  459. def db_type(self, connection):
  460. """
  461. Returns the database column data type for this field, for the provided
  462. connection.
  463. """
  464. # The default implementation of this method looks at the
  465. # backend-specific data_types dictionary, looking up the field by its
  466. # "internal type".
  467. #
  468. # A Field class can implement the get_internal_type() method to specify
  469. # which *preexisting* Django Field class it's most similar to -- i.e.,
  470. # a custom field might be represented by a TEXT column type, which is
  471. # the same as the TextField Django field type, which means the custom
  472. # field's get_internal_type() returns 'TextField'.
  473. #
  474. # But the limitation of the get_internal_type() / data_types approach
  475. # is that it cannot handle database column types that aren't already
  476. # mapped to one of the built-in Django field types. In this case, you
  477. # can implement db_type() instead of get_internal_type() to specify
  478. # exactly which wacky database column type you want to use.
  479. data = DictWrapper(self.__dict__, connection.ops.quote_name, "qn_")
  480. try:
  481. return connection.data_types[self.get_internal_type()] % data
  482. except KeyError:
  483. return None
  484. def db_parameters(self, connection):
  485. """
  486. Extension of db_type(), providing a range of different return
  487. values (type, checks).
  488. This will look at db_type(), allowing custom model fields to override it.
  489. """
  490. data = DictWrapper(self.__dict__, connection.ops.quote_name, "qn_")
  491. type_string = self.db_type(connection)
  492. try:
  493. check_string = connection.data_type_check_constraints[self.get_internal_type()] % data
  494. except KeyError:
  495. check_string = None
  496. return {
  497. "type": type_string,
  498. "check": check_string,
  499. }
  500. def db_type_suffix(self, connection):
  501. return connection.data_types_suffix.get(self.get_internal_type())
  502. def get_db_converters(self, connection):
  503. if hasattr(self, 'from_db_value'):
  504. return [self.from_db_value]
  505. return []
  506. @property
  507. def unique(self):
  508. return self._unique or self.primary_key
  509. def set_attributes_from_name(self, name):
  510. if not self.name:
  511. self.name = name
  512. self.attname, self.column = self.get_attname_column()
  513. if self.verbose_name is None and self.name:
  514. self.verbose_name = self.name.replace('_', ' ')
  515. def contribute_to_class(self, cls, name, virtual_only=False):
  516. self.set_attributes_from_name(name)
  517. self.model = cls
  518. if virtual_only:
  519. cls._meta.add_virtual_field(self)
  520. else:
  521. cls._meta.add_field(self)
  522. if self.choices:
  523. setattr(cls, 'get_%s_display' % self.name,
  524. curry(cls._get_FIELD_display, field=self))
  525. def get_attname(self):
  526. return self.name
  527. def get_attname_column(self):
  528. attname = self.get_attname()
  529. column = self.db_column or attname
  530. return attname, column
  531. def get_cache_name(self):
  532. return '_%s_cache' % self.name
  533. def get_internal_type(self):
  534. return self.__class__.__name__
  535. def pre_save(self, model_instance, add):
  536. """
  537. Returns field's value just before saving.
  538. """
  539. return getattr(model_instance, self.attname)
  540. def get_prep_value(self, value):
  541. """
  542. Perform preliminary non-db specific value checks and conversions.
  543. """
  544. if isinstance(value, Promise):
  545. value = value._proxy____cast()
  546. return value
  547. def get_db_prep_value(self, value, connection, prepared=False):
  548. """Returns field's value prepared for interacting with the database
  549. backend.
  550. Used by the default implementations of ``get_db_prep_save``and
  551. `get_db_prep_lookup```
  552. """
  553. if not prepared:
  554. value = self.get_prep_value(value)
  555. return value
  556. def get_db_prep_save(self, value, connection):
  557. """
  558. Returns field's value prepared for saving into a database.
  559. """
  560. return self.get_db_prep_value(value, connection=connection,
  561. prepared=False)
  562. def get_prep_lookup(self, lookup_type, value):
  563. """
  564. Perform preliminary non-db specific lookup checks and conversions
  565. """
  566. if hasattr(value, '_prepare'):
  567. return value._prepare()
  568. if lookup_type in {
  569. 'iexact', 'contains', 'icontains',
  570. 'startswith', 'istartswith', 'endswith', 'iendswith',
  571. 'month', 'day', 'week_day', 'hour', 'minute', 'second',
  572. 'isnull', 'search', 'regex', 'iregex',
  573. }:
  574. return value
  575. elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
  576. return self.get_prep_value(value)
  577. elif lookup_type in ('range', 'in'):
  578. return [self.get_prep_value(v) for v in value]
  579. elif lookup_type == 'year':
  580. try:
  581. return int(value)
  582. except ValueError:
  583. raise ValueError("The __year lookup type requires an integer "
  584. "argument")
  585. return self.get_prep_value(value)
  586. def get_db_prep_lookup(self, lookup_type, value, connection,
  587. prepared=False):
  588. """
  589. Returns field's value prepared for database lookup.
  590. """
  591. if not prepared:
  592. value = self.get_prep_lookup(lookup_type, value)
  593. prepared = True
  594. if hasattr(value, 'get_compiler'):
  595. value = value.get_compiler(connection=connection)
  596. if hasattr(value, 'as_sql') or hasattr(value, '_as_sql'):
  597. # If the value has a relabeled_clone method it means the
  598. # value will be handled later on.
  599. if hasattr(value, 'relabeled_clone'):
  600. return value
  601. if hasattr(value, 'as_sql'):
  602. sql, params = value.as_sql()
  603. else:
  604. sql, params = value._as_sql(connection=connection)
  605. return QueryWrapper(('(%s)' % sql), params)
  606. if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute',
  607. 'second', 'search', 'regex', 'iregex'):
  608. return [value]
  609. elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
  610. return [self.get_db_prep_value(value, connection=connection,
  611. prepared=prepared)]
  612. elif lookup_type in ('range', 'in'):
  613. return [self.get_db_prep_value(v, connection=connection,
  614. prepared=prepared) for v in value]
  615. elif lookup_type in ('contains', 'icontains'):
  616. return ["%%%s%%" % connection.ops.prep_for_like_query(value)]
  617. elif lookup_type == 'iexact':
  618. return [connection.ops.prep_for_iexact_query(value)]
  619. elif lookup_type in ('startswith', 'istartswith'):
  620. return ["%s%%" % connection.ops.prep_for_like_query(value)]
  621. elif lookup_type in ('endswith', 'iendswith'):
  622. return ["%%%s" % connection.ops.prep_for_like_query(value)]
  623. elif lookup_type == 'isnull':
  624. return []
  625. elif lookup_type == 'year':
  626. if isinstance(self, DateTimeField):
  627. return connection.ops.year_lookup_bounds_for_datetime_field(value)
  628. elif isinstance(self, DateField):
  629. return connection.ops.year_lookup_bounds_for_date_field(value)
  630. else:
  631. return [value] # this isn't supposed to happen
  632. else:
  633. return [value]
  634. def has_default(self):
  635. """
  636. Returns a boolean of whether this field has a default value.
  637. """
  638. return self.default is not NOT_PROVIDED
  639. def get_default(self):
  640. """
  641. Returns the default value for this field.
  642. """
  643. if self.has_default():
  644. if callable(self.default):
  645. return self.default()
  646. return self.default
  647. if (not self.empty_strings_allowed or (self.null and
  648. not connection.features.interprets_empty_strings_as_nulls)):
  649. return None
  650. return ""
  651. def get_choices(self, include_blank=True, blank_choice=BLANK_CHOICE_DASH, limit_choices_to=None):
  652. """Returns choices with a default blank choices included, for use
  653. as SelectField choices for this field."""
  654. blank_defined = False
  655. choices = list(self.choices) if self.choices else []
  656. named_groups = choices and isinstance(choices[0][1], (list, tuple))
  657. if not named_groups:
  658. for choice, __ in choices:
  659. if choice in ('', None):
  660. blank_defined = True
  661. break
  662. first_choice = (blank_choice if include_blank and
  663. not blank_defined else [])
  664. if self.choices:
  665. return first_choice + choices
  666. rel_model = self.rel.to
  667. limit_choices_to = limit_choices_to or self.get_limit_choices_to()
  668. if hasattr(self.rel, 'get_related_field'):
  669. lst = [(getattr(x, self.rel.get_related_field().attname),
  670. smart_text(x))
  671. for x in rel_model._default_manager.complex_filter(
  672. limit_choices_to)]
  673. else:
  674. lst = [(x._get_pk_val(), smart_text(x))
  675. for x in rel_model._default_manager.complex_filter(
  676. limit_choices_to)]
  677. return first_choice + lst
  678. def get_choices_default(self):
  679. return self.get_choices()
  680. def get_flatchoices(self, include_blank=True,
  681. blank_choice=BLANK_CHOICE_DASH):
  682. """
  683. Returns flattened choices with a default blank choice included.
  684. """
  685. first_choice = blank_choice if include_blank else []
  686. return first_choice + list(self.flatchoices)
  687. def _get_val_from_obj(self, obj):
  688. if obj is not None:
  689. return getattr(obj, self.attname)
  690. else:
  691. return self.get_default()
  692. def value_to_string(self, obj):
  693. """
  694. Returns a string value of this field from the passed obj.
  695. This is used by the serialization framework.
  696. """
  697. return smart_text(self._get_val_from_obj(obj))
  698. def _get_choices(self):
  699. if isinstance(self._choices, collections.Iterator):
  700. choices, self._choices = tee(self._choices)
  701. return choices
  702. else:
  703. return self._choices
  704. choices = property(_get_choices)
  705. def _get_flatchoices(self):
  706. """Flattened version of choices tuple."""
  707. flat = []
  708. for choice, value in self.choices:
  709. if isinstance(value, (list, tuple)):
  710. flat.extend(value)
  711. else:
  712. flat.append((choice, value))
  713. return flat
  714. flatchoices = property(_get_flatchoices)
  715. def save_form_data(self, instance, data):
  716. setattr(instance, self.name, data)
  717. def formfield(self, form_class=None, choices_form_class=None, **kwargs):
  718. """
  719. Returns a django.forms.Field instance for this database Field.
  720. """
  721. defaults = {'required': not self.blank,
  722. 'label': capfirst(self.verbose_name),
  723. 'help_text': self.help_text}
  724. if self.has_default():
  725. if callable(self.default):
  726. defaults['initial'] = self.default
  727. defaults['show_hidden_initial'] = True
  728. else:
  729. defaults['initial'] = self.get_default()
  730. if self.choices:
  731. # Fields with choices get special treatment.
  732. include_blank = (self.blank or
  733. not (self.has_default() or 'initial' in kwargs))
  734. defaults['choices'] = self.get_choices(include_blank=include_blank)
  735. defaults['coerce'] = self.to_python
  736. if self.null:
  737. defaults['empty_value'] = None
  738. if choices_form_class is not None:
  739. form_class = choices_form_class
  740. else:
  741. form_class = forms.TypedChoiceField
  742. # Many of the subclass-specific formfield arguments (min_value,
  743. # max_value) don't apply for choice fields, so be sure to only pass
  744. # the values that TypedChoiceField will understand.
  745. for k in list(kwargs):
  746. if k not in ('coerce', 'empty_value', 'choices', 'required',
  747. 'widget', 'label', 'initial', 'help_text',
  748. 'error_messages', 'show_hidden_initial'):
  749. del kwargs[k]
  750. defaults.update(kwargs)
  751. if form_class is None:
  752. form_class = forms.CharField
  753. return form_class(**defaults)
  754. def value_from_object(self, obj):
  755. """
  756. Returns the value of this field in the given model instance.
  757. """
  758. return getattr(obj, self.attname)
  759. class AutoField(Field):
  760. description = _("Integer")
  761. empty_strings_allowed = False
  762. default_error_messages = {
  763. 'invalid': _("'%(value)s' value must be an integer."),
  764. }
  765. def __init__(self, *args, **kwargs):
  766. kwargs['blank'] = True
  767. super(AutoField, self).__init__(*args, **kwargs)
  768. def check(self, **kwargs):
  769. errors = super(AutoField, self).check(**kwargs)
  770. errors.extend(self._check_primary_key())
  771. return errors
  772. def _check_primary_key(self):
  773. if not self.primary_key:
  774. return [
  775. checks.Error(
  776. 'AutoFields must set primary_key=True.',
  777. hint=None,
  778. obj=self,
  779. id='fields.E100',
  780. ),
  781. ]
  782. else:
  783. return []
  784. def deconstruct(self):
  785. name, path, args, kwargs = super(AutoField, self).deconstruct()
  786. del kwargs['blank']
  787. kwargs['primary_key'] = True
  788. return name, path, args, kwargs
  789. def get_internal_type(self):
  790. return "AutoField"
  791. def to_python(self, value):
  792. if value is None:
  793. return value
  794. try:
  795. return int(value)
  796. except (TypeError, ValueError):
  797. raise exceptions.ValidationError(
  798. self.error_messages['invalid'],
  799. code='invalid',
  800. params={'value': value},
  801. )
  802. def validate(self, value, model_instance):
  803. pass
  804. def get_db_prep_value(self, value, connection, prepared=False):
  805. if not prepared:
  806. value = self.get_prep_value(value)
  807. value = connection.ops.validate_autopk_value(value)
  808. return value
  809. def get_prep_value(self, value):
  810. value = super(AutoField, self).get_prep_value(value)
  811. if value is None:
  812. return None
  813. return int(value)
  814. def contribute_to_class(self, cls, name, **kwargs):
  815. assert not cls._meta.has_auto_field, \
  816. "A model can't have more than one AutoField."
  817. super(AutoField, self).contribute_to_class(cls, name, **kwargs)
  818. cls._meta.has_auto_field = True
  819. cls._meta.auto_field = self
  820. def formfield(self, **kwargs):
  821. return None
  822. class BooleanField(Field):
  823. empty_strings_allowed = False
  824. default_error_messages = {
  825. 'invalid': _("'%(value)s' value must be either True or False."),
  826. }
  827. description = _("Boolean (Either True or False)")
  828. def __init__(self, *args, **kwargs):
  829. kwargs['blank'] = True
  830. super(BooleanField, self).__init__(*args, **kwargs)
  831. def check(self, **kwargs):
  832. errors = super(BooleanField, self).check(**kwargs)
  833. errors.extend(self._check_null(**kwargs))
  834. return errors
  835. def _check_null(self, **kwargs):
  836. if getattr(self, 'null', False):
  837. return [
  838. checks.Error(
  839. 'BooleanFields do not accept null values.',
  840. hint='Use a NullBooleanField instead.',
  841. obj=self,
  842. id='fields.E110',
  843. )
  844. ]
  845. else:
  846. return []
  847. def deconstruct(self):
  848. name, path, args, kwargs = super(BooleanField, self).deconstruct()
  849. del kwargs['blank']
  850. return name, path, args, kwargs
  851. def get_internal_type(self):
  852. return "BooleanField"
  853. def to_python(self, value):
  854. if value in (True, False):
  855. # if value is 1 or 0 than it's equal to True or False, but we want
  856. # to return a true bool for semantic reasons.
  857. return bool(value)
  858. if value in ('t', 'True', '1'):
  859. return True
  860. if value in ('f', 'False', '0'):
  861. return False
  862. raise exceptions.ValidationError(
  863. self.error_messages['invalid'],
  864. code='invalid',
  865. params={'value': value},
  866. )
  867. def get_prep_lookup(self, lookup_type, value):
  868. # Special-case handling for filters coming from a Web request (e.g. the
  869. # admin interface). Only works for scalar values (not lists). If you're
  870. # passing in a list, you might as well make things the right type when
  871. # constructing the list.
  872. if value in ('1', '0'):
  873. value = bool(int(value))
  874. return super(BooleanField, self).get_prep_lookup(lookup_type, value)
  875. def get_prep_value(self, value):
  876. value = super(BooleanField, self).get_prep_value(value)
  877. if value is None:
  878. return None
  879. return bool(value)
  880. def formfield(self, **kwargs):
  881. # Unlike most fields, BooleanField figures out include_blank from
  882. # self.null instead of self.blank.
  883. if self.choices:
  884. include_blank = not (self.has_default() or 'initial' in kwargs)
  885. defaults = {'choices': self.get_choices(include_blank=include_blank)}
  886. else:
  887. defaults = {'form_class': forms.BooleanField}
  888. defaults.update(kwargs)
  889. return super(BooleanField, self).formfield(**defaults)
  890. class CharField(Field):
  891. description = _("String (up to %(max_length)s)")
  892. def __init__(self, *args, **kwargs):
  893. super(CharField, self).__init__(*args, **kwargs)
  894. self.validators.append(validators.MaxLengthValidator(self.max_length))
  895. def check(self, **kwargs):
  896. errors = super(CharField, self).check(**kwargs)
  897. errors.extend(self._check_max_length_attribute(**kwargs))
  898. return errors
  899. def _check_max_length_attribute(self, **kwargs):
  900. try:
  901. max_length = int(self.max_length)
  902. if max_length <= 0:
  903. raise ValueError()
  904. except TypeError:
  905. return [
  906. checks.Error(
  907. "CharFields must define a 'max_length' attribute.",
  908. hint=None,
  909. obj=self,
  910. id='fields.E120',
  911. )
  912. ]
  913. except ValueError:
  914. return [
  915. checks.Error(
  916. "'max_length' must be a positive integer.",
  917. hint=None,
  918. obj=self,
  919. id='fields.E121',
  920. )
  921. ]
  922. else:
  923. return []
  924. def get_internal_type(self):
  925. return "CharField"
  926. def to_python(self, value):
  927. if isinstance(value, six.string_types) or value is None:
  928. return value
  929. return smart_text(value)
  930. def get_prep_value(self, value):
  931. value = super(CharField, self).get_prep_value(value)
  932. return self.to_python(value)
  933. def formfield(self, **kwargs):
  934. # Passing max_length to forms.CharField means that the value's length
  935. # will be validated twice. This is considered acceptable since we want
  936. # the value in the form field (to pass into widget for example).
  937. defaults = {'max_length': self.max_length}
  938. defaults.update(kwargs)
  939. return super(CharField, self).formfield(**defaults)
  940. # TODO: Maybe move this into contrib, because it's specialized.
  941. class CommaSeparatedIntegerField(CharField):
  942. default_validators = [validators.validate_comma_separated_integer_list]
  943. description = _("Comma-separated integers")
  944. def formfield(self, **kwargs):
  945. defaults = {
  946. 'error_messages': {
  947. 'invalid': _('Enter only digits separated by commas.'),
  948. }
  949. }
  950. defaults.update(kwargs)
  951. return super(CommaSeparatedIntegerField, self).formfield(**defaults)
  952. class DateTimeCheckMixin(object):
  953. def check(self, **kwargs):
  954. errors = super(DateTimeCheckMixin, self).check(**kwargs)
  955. errors.extend(self._check_mutually_exclusive_options())
  956. errors.extend(self._check_fix_default_value())
  957. return errors
  958. def _check_mutually_exclusive_options(self):
  959. # auto_now, auto_now_add, and default are mutually exclusive
  960. # options. The use of more than one of these options together
  961. # will trigger an Error
  962. mutually_exclusive_options = [self.auto_now_add, self.auto_now,
  963. self.has_default()]
  964. enabled_options = [option not in (None, False)
  965. for option in mutually_exclusive_options].count(True)
  966. if enabled_options > 1:
  967. return [
  968. checks.Error(
  969. "The options auto_now, auto_now_add, and default "
  970. "are mutually exclusive. Only one of these options "
  971. "may be present.",
  972. hint=None,
  973. obj=self,
  974. id='fields.E160',
  975. )
  976. ]
  977. else:
  978. return []
  979. def _check_fix_default_value(self):
  980. return []
  981. class DateField(DateTimeCheckMixin, Field):
  982. empty_strings_allowed = False
  983. default_error_messages = {
  984. 'invalid': _("'%(value)s' value has an invalid date format. It must be "
  985. "in YYYY-MM-DD format."),
  986. 'invalid_date': _("'%(value)s' value has the correct format (YYYY-MM-DD) "
  987. "but it is an invalid date."),
  988. }
  989. description = _("Date (without time)")
  990. def __init__(self, verbose_name=None, name=None, auto_now=False,
  991. auto_now_add=False, **kwargs):
  992. self.auto_now, self.auto_now_add = auto_now, auto_now_add
  993. if auto_now or auto_now_add:
  994. kwargs['editable'] = False
  995. kwargs['blank'] = True
  996. super(DateField, self).__init__(verbose_name, name, **kwargs)
  997. def _check_fix_default_value(self):
  998. """
  999. Adds a warning to the checks framework stating, that using an actual
  1000. date or datetime value is probably wrong; it's only being evaluated on
  1001. server start-up.
  1002. For details see ticket #21905
  1003. """
  1004. if not self.has_default():
  1005. return []
  1006. now = timezone.now()
  1007. if not timezone.is_naive(now):
  1008. now = timezone.make_naive(now, timezone.utc)
  1009. value = self.default
  1010. if isinstance(value, datetime.datetime):
  1011. if not timezone.is_naive(value):
  1012. value = timezone.make_naive(value, timezone.utc)
  1013. value = value.date()
  1014. elif isinstance(value, datetime.date):
  1015. # Nothing to do, as dates don't have tz information
  1016. pass
  1017. else:
  1018. # No explicit date / datetime value -- no checks necessary
  1019. return []
  1020. offset = datetime.timedelta(days=1)
  1021. lower = (now - offset).date()
  1022. upper = (now + offset).date()
  1023. if lower <= value <= upper:
  1024. return [
  1025. checks.Warning(
  1026. 'Fixed default value provided.',
  1027. hint='It seems you set a fixed date / time / datetime '
  1028. 'value as default for this field. This may not be '
  1029. 'what you want. If you want to have the current date '
  1030. 'as default, use `django.utils.timezone.now`',
  1031. obj=self,
  1032. id='fields.W161',
  1033. )
  1034. ]
  1035. return []
  1036. def deconstruct(self):
  1037. name, path, args, kwargs = super(DateField, self).deconstruct()
  1038. if self.auto_now:
  1039. kwargs['auto_now'] = True
  1040. if self.auto_now_add:
  1041. kwargs['auto_now_add'] = True
  1042. if self.auto_now or self.auto_now_add:
  1043. del kwargs['editable']
  1044. del kwargs['blank']
  1045. return name, path, args, kwargs
  1046. def get_internal_type(self):
  1047. return "DateField"
  1048. def to_python(self, value):
  1049. if value is None:
  1050. return value
  1051. if isinstance(value, datetime.datetime):
  1052. if settings.USE_TZ and timezone.is_aware(value):
  1053. # Convert aware datetimes to the default time zone
  1054. # before casting them to dates (#17742).
  1055. default_timezone = timezone.get_default_timezone()
  1056. value = timezone.make_naive(value, default_timezone)
  1057. return value.date()
  1058. if isinstance(value, datetime.date):
  1059. return value
  1060. try:
  1061. parsed = parse_date(value)
  1062. if parsed is not None:
  1063. return parsed
  1064. except ValueError:
  1065. raise exceptions.ValidationError(
  1066. self.error_messages['invalid_date'],
  1067. code='invalid_date',
  1068. params={'value': value},
  1069. )
  1070. raise exceptions.ValidationError(
  1071. self.error_messages['invalid'],
  1072. code='invalid',
  1073. params={'value': value},
  1074. )
  1075. def pre_save(self, model_instance, add):
  1076. if self.auto_now or (self.auto_now_add and add):
  1077. value = datetime.date.today()
  1078. setattr(model_instance, self.attname, value)
  1079. return value
  1080. else:
  1081. return super(DateField, self).pre_save(model_instance, add)
  1082. def contribute_to_class(self, cls, name, **kwargs):
  1083. super(DateField, self).contribute_to_class(cls, name, **kwargs)
  1084. if not self.null:
  1085. setattr(cls, 'get_next_by_%s' % self.name,
  1086. curry(cls._get_next_or_previous_by_FIELD, field=self,
  1087. is_next=True))
  1088. setattr(cls, 'get_previous_by_%s' % self.name,
  1089. curry(cls._get_next_or_previous_by_FIELD, field=self,
  1090. is_next=False))
  1091. def get_prep_lookup(self, lookup_type, value):
  1092. # For dates lookups, convert the value to an int
  1093. # so the database backend always sees a consistent type.
  1094. if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute', 'second'):
  1095. return int(value)
  1096. return super(DateField, self).get_prep_lookup(lookup_type, value)
  1097. def get_prep_value(self, value):
  1098. value = super(DateField, self).get_prep_value(value)
  1099. return self.to_python(value)
  1100. def get_db_prep_value(self, value, connection, prepared=False):
  1101. # Casts dates into the format expected by the backend
  1102. if not prepared:
  1103. value = self.get_prep_value(value)
  1104. return connection.ops.value_to_db_date(value)
  1105. def value_to_string(self, obj):
  1106. val = self._get_val_from_obj(obj)
  1107. return '' if val is None else val.isoformat()
  1108. def formfield(self, **kwargs):
  1109. defaults = {'form_class': forms.DateField}
  1110. defaults.update(kwargs)
  1111. return super(DateField, self).formfield(**defaults)
  1112. class DateTimeField(DateField):
  1113. empty_strings_allowed = False
  1114. default_error_messages = {
  1115. 'invalid': _("'%(value)s' value has an invalid format. It must be in "
  1116. "YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ] format."),
  1117. 'invalid_date': _("'%(value)s' value has the correct format "
  1118. "(YYYY-MM-DD) but it is an invalid date."),
  1119. 'invalid_datetime': _("'%(value)s' value has the correct format "
  1120. "(YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ]) "
  1121. "but it is an invalid date/time."),
  1122. }
  1123. description = _("Date (with time)")
  1124. # __init__ is inherited from DateField
  1125. def _check_fix_default_value(self):
  1126. """
  1127. Adds a warning to the checks framework stating, that using an actual
  1128. date or datetime value is probably wrong; it's only being evaluated on
  1129. server start-up.
  1130. For details see ticket #21905
  1131. """
  1132. if not self.has_default():
  1133. return []
  1134. now = timezone.now()
  1135. if not timezone.is_naive(now):
  1136. now = timezone.make_naive(now, timezone.utc)
  1137. value = self.default
  1138. if isinstance(value, datetime.datetime):
  1139. second_offset = datetime.timedelta(seconds=10)
  1140. lower = now - second_offset
  1141. upper = now + second_offset
  1142. if timezone.is_aware(value):
  1143. value = timezone.make_naive(value, timezone.utc)
  1144. elif isinstance(value, datetime.date):
  1145. second_offset = datetime.timedelta(seconds=10)
  1146. lower = now - second_offset
  1147. lower = datetime.datetime(lower.year, lower.month, lower.day)
  1148. upper = now + second_offset
  1149. upper = datetime.datetime(upper.year, upper.month, upper.day)
  1150. value = datetime.datetime(value.year, value.month, value.day)
  1151. else:
  1152. # No explicit date / datetime value -- no checks necessary
  1153. return []
  1154. if lower <= value <= upper:
  1155. return [
  1156. checks.Warning(
  1157. 'Fixed default value provided.',
  1158. hint='It seems you set a fixed date / time / datetime '
  1159. 'value as default for this field. This may not be '
  1160. 'what you want. If you want to have the current date '
  1161. 'as default, use `django.utils.timezone.now`',
  1162. obj=self,
  1163. id='fields.W161',
  1164. )
  1165. ]
  1166. return []
  1167. def get_internal_type(self):
  1168. return "DateTimeField"
  1169. def to_python(self, value):
  1170. if value is None:
  1171. return value
  1172. if isinstance(value, datetime.datetime):
  1173. return value
  1174. if isinstance(value, datetime.date):
  1175. value = datetime.datetime(value.year, value.month, value.day)
  1176. if settings.USE_TZ:
  1177. # For backwards compatibility, interpret naive datetimes in
  1178. # local time. This won't work during DST change, but we can't
  1179. # do much about it, so we let the exceptions percolate up the
  1180. # call stack.
  1181. warnings.warn("DateTimeField %s.%s received a naive datetime "
  1182. "(%s) while time zone support is active." %
  1183. (self.model.__name__, self.name, value),
  1184. RuntimeWarning)
  1185. default_timezone = timezone.get_default_timezone()
  1186. value = timezone.make_aware(value, default_timezone)
  1187. return value
  1188. try:
  1189. parsed = parse_datetime(value)
  1190. if parsed is not None:
  1191. return parsed
  1192. except ValueError:
  1193. raise exceptions.ValidationError(
  1194. self.error_messages['invalid_datetime'],
  1195. code='invalid_datetime',
  1196. params={'value': value},
  1197. )
  1198. try:
  1199. parsed = parse_date(value)
  1200. if parsed is not None:
  1201. return datetime.datetime(parsed.year, parsed.month, parsed.day)
  1202. except ValueError:
  1203. raise exceptions.ValidationError(
  1204. self.error_messages['invalid_date'],
  1205. code='invalid_date',
  1206. params={'value': value},
  1207. )
  1208. raise exceptions.ValidationError(
  1209. self.error_messages['invalid'],
  1210. code='invalid',
  1211. params={'value': value},
  1212. )
  1213. def pre_save(self, model_instance, add):
  1214. if self.auto_now or (self.auto_now_add and add):
  1215. value = timezone.now()
  1216. setattr(model_instance, self.attname, value)
  1217. return value
  1218. else:
  1219. return super(DateTimeField, self).pre_save(model_instance, add)
  1220. # contribute_to_class is inherited from DateField, it registers
  1221. # get_next_by_FOO and get_prev_by_FOO
  1222. # get_prep_lookup is inherited from DateField
  1223. def get_prep_value(self, value):
  1224. value = super(DateTimeField, self).get_prep_value(value)
  1225. value = self.to_python(value)
  1226. if value is not None and settings.USE_TZ and timezone.is_naive(value):
  1227. # For backwards compatibility, interpret naive datetimes in local
  1228. # time. This won't work during DST change, but we can't do much
  1229. # about it, so we let the exceptions percolate up the call stack.
  1230. try:
  1231. name = '%s.%s' % (self.model.__name__, self.name)
  1232. except AttributeError:
  1233. name = '(unbound)'
  1234. warnings.warn("DateTimeField %s received a naive datetime (%s)"
  1235. " while time zone support is active." %
  1236. (name, value),
  1237. RuntimeWarning)
  1238. default_timezone = timezone.get_default_timezone()
  1239. value = timezone.make_aware(value, default_timezone)
  1240. return value
  1241. def get_db_prep_value(self, value, connection, prepared=False):
  1242. # Casts datetimes into the format expected by the backend
  1243. if not prepared:
  1244. value = self.get_prep_value(value)
  1245. return connection.ops.value_to_db_datetime(value)
  1246. def value_to_string(self, obj):
  1247. val = self._get_val_from_obj(obj)
  1248. return '' if val is None else val.isoformat()
  1249. def formfield(self, **kwargs):
  1250. defaults = {'form_class': forms.DateTimeField}
  1251. defaults.update(kwargs)
  1252. return super(DateTimeField, self).formfield(**defaults)
  1253. class DecimalField(Field):
  1254. empty_strings_allowed = False
  1255. default_error_messages = {
  1256. 'invalid': _("'%(value)s' value must be a decimal number."),
  1257. }
  1258. description = _("Decimal number")
  1259. def __init__(self, verbose_name=None, name=None, max_digits=None,
  1260. decimal_places=None, **kwargs):
  1261. self.max_digits, self.decimal_places = max_digits, decimal_places
  1262. super(DecimalField, self).__init__(verbose_name, name, **kwargs)
  1263. def check(self, **kwargs):
  1264. errors = super(DecimalField, self).check(**kwargs)
  1265. digits_errors = self._check_decimal_places()
  1266. digits_errors.extend(self._check_max_digits())
  1267. if not digits_errors:
  1268. errors.extend(self._check_decimal_places_and_max_digits(**kwargs))
  1269. else:
  1270. errors.extend(digits_errors)
  1271. return errors
  1272. def _check_decimal_places(self):
  1273. try:
  1274. decimal_places = int(self.decimal_places)
  1275. if decimal_places < 0:
  1276. raise ValueError()
  1277. except TypeError:
  1278. return [
  1279. checks.Error(
  1280. "DecimalFields must define a 'decimal_places' attribute.",
  1281. hint=None,
  1282. obj=self,
  1283. id='fields.E130',
  1284. )
  1285. ]
  1286. except ValueError:
  1287. return [
  1288. checks.Error(
  1289. "'decimal_places' must be a non-negative integer.",
  1290. hint=None,
  1291. obj=self,
  1292. id='fields.E131',
  1293. )
  1294. ]
  1295. else:
  1296. return []
  1297. def _check_max_digits(self):
  1298. try:
  1299. max_digits = int(self.max_digits)
  1300. if max_digits <= 0:
  1301. raise ValueError()
  1302. except TypeError:
  1303. return [
  1304. checks.Error(
  1305. "DecimalFields must define a 'max_digits' attribute.",
  1306. hint=None,
  1307. obj=self,
  1308. id='fields.E132',
  1309. )
  1310. ]
  1311. except ValueError:
  1312. return [
  1313. checks.Error(
  1314. "'max_digits' must be a positive integer.",
  1315. hint=None,
  1316. obj=self,
  1317. id='fields.E133',
  1318. )
  1319. ]
  1320. else:
  1321. return []
  1322. def _check_decimal_places_and_max_digits(self, **kwargs):
  1323. if int(self.decimal_places) > int(self.max_digits):
  1324. return [
  1325. checks.Error(
  1326. "'max_digits' must be greater or equal to 'decimal_places'.",
  1327. hint=None,
  1328. obj=self,
  1329. id='fields.E134',
  1330. )
  1331. ]
  1332. return []
  1333. def deconstruct(self):
  1334. name, path, args, kwargs = super(DecimalField, self).deconstruct()
  1335. if self.max_digits is not None:
  1336. kwargs['max_digits'] = self.max_digits
  1337. if self.decimal_places is not None:
  1338. kwargs['decimal_places'] = self.decimal_places
  1339. return name, path, args, kwargs
  1340. def get_internal_type(self):
  1341. return "DecimalField"
  1342. def to_python(self, value):
  1343. if value is None:
  1344. return value
  1345. try:
  1346. return decimal.Decimal(value)
  1347. except decimal.InvalidOperation:
  1348. raise exceptions.ValidationError(
  1349. self.error_messages['invalid'],
  1350. code='invalid',
  1351. params={'value': value},
  1352. )
  1353. def _format(self, value):
  1354. if isinstance(value, six.string_types):
  1355. return value
  1356. else:
  1357. return self.format_number(value)
  1358. def format_number(self, value):
  1359. """
  1360. Formats a number into a string with the requisite number of digits and
  1361. decimal places.
  1362. """
  1363. # Method moved to django.db.backends.utils.
  1364. #
  1365. # It is preserved because it is used by the oracle backend
  1366. # (django.db.backends.oracle.query), and also for
  1367. # backwards-compatibility with any external code which may have used
  1368. # this method.
  1369. from django.db.backends import utils
  1370. return utils.format_number(value, self.max_digits, self.decimal_places)
  1371. def get_db_prep_save(self, value, connection):
  1372. return connection.ops.value_to_db_decimal(self.to_python(value),
  1373. self.max_digits, self.decimal_places)
  1374. def get_prep_value(self, value):
  1375. value = super(DecimalField, self).get_prep_value(value)
  1376. return self.to_python(value)
  1377. def formfield(self, **kwargs):
  1378. defaults = {
  1379. 'max_digits': self.max_digits,
  1380. 'decimal_places': self.decimal_places,
  1381. 'form_class': forms.DecimalField,
  1382. }
  1383. defaults.update(kwargs)
  1384. return super(DecimalField, self).formfield(**defaults)
  1385. class DurationField(Field):
  1386. """Stores timedelta objects.
  1387. Uses interval on postgres, INVERAL DAY TO SECOND on Oracle, and bigint of
  1388. microseconds on other databases.
  1389. """
  1390. empty_strings_allowed = False
  1391. default_error_messages = {
  1392. 'invalid': _("'%(value)s' value has an invalid format. It must be in "
  1393. "[DD] [HH:[MM:]]ss[.uuuuuu] format.")
  1394. }
  1395. description = _("Duration")
  1396. def get_internal_type(self):
  1397. return "DurationField"
  1398. def to_python(self, value):
  1399. if value is None:
  1400. return value
  1401. if isinstance(value, datetime.timedelta):
  1402. return value
  1403. try:
  1404. parsed = parse_duration(value)
  1405. except ValueError:
  1406. pass
  1407. else:
  1408. if parsed is not None:
  1409. return parsed
  1410. raise exceptions.ValidationError(
  1411. self.error_messages['invalid'],
  1412. code='invalid',
  1413. params={'value': value},
  1414. )
  1415. def get_db_prep_value(self, value, connection, prepared=False):
  1416. if connection.features.has_native_duration_field:
  1417. return value
  1418. return value.total_seconds() * 1000000
  1419. def get_db_converters(self, connection):
  1420. converters = []
  1421. if not connection.features.has_native_duration_field:
  1422. converters.append(connection.ops.convert_durationfield_value)
  1423. return converters + super(DurationField, self).get_db_converters(connection)
  1424. def value_to_string(self, obj):
  1425. val = self._get_val_from_obj(obj)
  1426. return '' if val is None else duration_string(val)
  1427. class EmailField(CharField):
  1428. default_validators = [validators.validate_email]
  1429. description = _("Email address")
  1430. def __init__(self, *args, **kwargs):
  1431. # max_length=254 to be compliant with RFCs 3696 and 5321
  1432. kwargs['max_length'] = kwargs.get('max_length', 254)
  1433. super(EmailField, self).__init__(*args, **kwargs)
  1434. def deconstruct(self):
  1435. name, path, args, kwargs = super(EmailField, self).deconstruct()
  1436. # We do not exclude max_length if it matches default as we want to change
  1437. # the default in future.
  1438. return name, path, args, kwargs
  1439. def formfield(self, **kwargs):
  1440. # As with CharField, this will cause email validation to be performed
  1441. # twice.
  1442. defaults = {
  1443. 'form_class': forms.EmailField,
  1444. }
  1445. defaults.update(kwargs)
  1446. return super(EmailField, self).formfield(**defaults)
  1447. class FilePathField(Field):
  1448. description = _("File path")
  1449. def __init__(self, verbose_name=None, name=None, path='', match=None,
  1450. recursive=False, allow_files=True, allow_folders=False, **kwargs):
  1451. self.path, self.match, self.recursive = path, match, recursive
  1452. self.allow_files, self.allow_folders = allow_files, allow_folders
  1453. kwargs['max_length'] = kwargs.get('max_length', 100)
  1454. super(FilePathField, self).__init__(verbose_name, name, **kwargs)
  1455. def check(self, **kwargs):
  1456. errors = super(FilePathField, self).check(**kwargs)
  1457. errors.extend(self._check_allowing_files_or_folders(**kwargs))
  1458. return errors
  1459. def _check_allowing_files_or_folders(self, **kwargs):
  1460. if not self.allow_files and not self.allow_folders:
  1461. return [
  1462. checks.Error(
  1463. "FilePathFields must have either 'allow_files' or 'allow_folders' set to True.",
  1464. hint=None,
  1465. obj=self,
  1466. id='fields.E140',
  1467. )
  1468. ]
  1469. return []
  1470. def deconstruct(self):
  1471. name, path, args, kwargs = super(FilePathField, self).deconstruct()
  1472. if self.path != '':
  1473. kwargs['path'] = self.path
  1474. if self.match is not None:
  1475. kwargs['match'] = self.match
  1476. if self.recursive is not False:
  1477. kwargs['recursive'] = self.recursive
  1478. if self.allow_files is not True:
  1479. kwargs['allow_files'] = self.allow_files
  1480. if self.allow_folders is not False:
  1481. kwargs['allow_folders'] = self.allow_folders
  1482. if kwargs.get("max_length", None) == 100:
  1483. del kwargs["max_length"]
  1484. return name, path, args, kwargs
  1485. def get_prep_value(self, value):
  1486. value = super(FilePathField, self).get_prep_value(value)
  1487. if value is None:
  1488. return None
  1489. return six.text_type(value)
  1490. def formfield(self, **kwargs):
  1491. defaults = {
  1492. 'path': self.path,
  1493. 'match': self.match,
  1494. 'recursive': self.recursive,
  1495. 'form_class': forms.FilePathField,
  1496. 'allow_files': self.allow_files,
  1497. 'allow_folders': self.allow_folders,
  1498. }
  1499. defaults.update(kwargs)
  1500. return super(FilePathField, self).formfield(**defaults)
  1501. def get_internal_type(self):
  1502. return "FilePathField"
  1503. class FloatField(Field):
  1504. empty_strings_allowed = False
  1505. default_error_messages = {
  1506. 'invalid': _("'%(value)s' value must be a float."),
  1507. }
  1508. description = _("Floating point number")
  1509. def get_prep_value(self, value):
  1510. value = super(FloatField, self).get_prep_value(value)
  1511. if value is None:
  1512. return None
  1513. return float(value)
  1514. def get_internal_type(self):
  1515. return "FloatField"
  1516. def to_python(self, value):
  1517. if value is None:
  1518. return value
  1519. try:
  1520. return float(value)
  1521. except (TypeError, ValueError):
  1522. raise exceptions.ValidationError(
  1523. self.error_messages['invalid'],
  1524. code='invalid',
  1525. params={'value': value},
  1526. )
  1527. def formfield(self, **kwargs):
  1528. defaults = {'form_class': forms.FloatField}
  1529. defaults.update(kwargs)
  1530. return super(FloatField, self).formfield(**defaults)
  1531. class IntegerField(Field):
  1532. empty_strings_allowed = False
  1533. default_error_messages = {
  1534. 'invalid': _("'%(value)s' value must be an integer."),
  1535. }
  1536. description = _("Integer")
  1537. def check(self, **kwargs):
  1538. errors = super(IntegerField, self).check(**kwargs)
  1539. errors.extend(self._check_max_length_warning())
  1540. return errors
  1541. def _check_max_length_warning(self):
  1542. if self.max_length is not None:
  1543. return [
  1544. checks.Warning(
  1545. "'max_length' is ignored when used with IntegerField",
  1546. hint="Remove 'max_length' from field",
  1547. obj=self,
  1548. id='fields.W122',
  1549. )
  1550. ]
  1551. return []
  1552. @cached_property
  1553. def validators(self):
  1554. # These validators can't be added at field initialization time since
  1555. # they're based on values retrieved from `connection`.
  1556. range_validators = []
  1557. internal_type = self.get_internal_type()
  1558. min_value, max_value = connection.ops.integer_field_range(internal_type)
  1559. if min_value is not None:
  1560. range_validators.append(validators.MinValueValidator(min_value))
  1561. if max_value is not None:
  1562. range_validators.append(validators.MaxValueValidator(max_value))
  1563. return super(IntegerField, self).validators + range_validators
  1564. def get_prep_value(self, value):
  1565. value = super(IntegerField, self).get_prep_value(value)
  1566. if value is None:
  1567. return None
  1568. return int(value)
  1569. def get_prep_lookup(self, lookup_type, value):
  1570. if ((lookup_type == 'gte' or lookup_type == 'lt')
  1571. and isinstance(value, float)):
  1572. value = math.ceil(value)
  1573. return super(IntegerField, self).get_prep_lookup(lookup_type, value)
  1574. def get_internal_type(self):
  1575. return "IntegerField"
  1576. def to_python(self, value):
  1577. if value is None:
  1578. return value
  1579. try:
  1580. return int(value)
  1581. except (TypeError, ValueError):
  1582. raise exceptions.ValidationError(
  1583. self.error_messages['invalid'],
  1584. code='invalid',
  1585. params={'value': value},
  1586. )
  1587. def formfield(self, **kwargs):
  1588. defaults = {'form_class': forms.IntegerField}
  1589. defaults.update(kwargs)
  1590. return super(IntegerField, self).formfield(**defaults)
  1591. class BigIntegerField(IntegerField):
  1592. empty_strings_allowed = False
  1593. description = _("Big (8 byte) integer")
  1594. MAX_BIGINT = 9223372036854775807
  1595. def get_internal_type(self):
  1596. return "BigIntegerField"
  1597. def formfield(self, **kwargs):
  1598. defaults = {'min_value': -BigIntegerField.MAX_BIGINT - 1,
  1599. 'max_value': BigIntegerField.MAX_BIGINT}
  1600. defaults.update(kwargs)
  1601. return super(BigIntegerField, self).formfield(**defaults)
  1602. class IPAddressField(Field):
  1603. empty_strings_allowed = False
  1604. description = _("IPv4 address")
  1605. def __init__(self, *args, **kwargs):
  1606. kwargs['max_length'] = 15
  1607. super(IPAddressField, self).__init__(*args, **kwargs)
  1608. def deconstruct(self):
  1609. name, path, args, kwargs = super(IPAddressField, self).deconstruct()
  1610. del kwargs['max_length']
  1611. return name, path, args, kwargs
  1612. def get_prep_value(self, value):
  1613. value = super(IPAddressField, self).get_prep_value(value)
  1614. if value is None:
  1615. return None
  1616. return six.text_type(value)
  1617. def get_internal_type(self):
  1618. return "IPAddressField"
  1619. def formfield(self, **kwargs):
  1620. defaults = {'form_class': forms.IPAddressField}
  1621. defaults.update(kwargs)
  1622. return super(IPAddressField, self).formfield(**defaults)
  1623. def check(self, **kwargs):
  1624. errors = super(IPAddressField, self).check(**kwargs)
  1625. errors.append(
  1626. checks.Warning(
  1627. 'IPAddressField has been deprecated. Support for it '
  1628. '(except in historical migrations) will be removed in Django 1.9.',
  1629. hint='Use GenericIPAddressField instead.',
  1630. obj=self,
  1631. id='fields.W900',
  1632. )
  1633. )
  1634. return errors
  1635. class GenericIPAddressField(Field):
  1636. empty_strings_allowed = True
  1637. description = _("IP address")
  1638. default_error_messages = {}
  1639. def __init__(self, verbose_name=None, name=None, protocol='both',
  1640. unpack_ipv4=False, *args, **kwargs):
  1641. self.unpack_ipv4 = unpack_ipv4
  1642. self.protocol = protocol
  1643. self.default_validators, invalid_error_message = \
  1644. validators.ip_address_validators(protocol, unpack_ipv4)
  1645. self.default_error_messages['invalid'] = invalid_error_message
  1646. kwargs['max_length'] = 39
  1647. super(GenericIPAddressField, self).__init__(verbose_name, name, *args,
  1648. **kwargs)
  1649. def check(self, **kwargs):
  1650. errors = super(GenericIPAddressField, self).check(**kwargs)
  1651. errors.extend(self._check_blank_and_null_values(**kwargs))
  1652. return errors
  1653. def _check_blank_and_null_values(self, **kwargs):
  1654. if not getattr(self, 'null', False) and getattr(self, 'blank', False):
  1655. return [
  1656. checks.Error(
  1657. ('GenericIPAddressFields cannot have blank=True if null=False, '
  1658. 'as blank values are stored as nulls.'),
  1659. hint=None,
  1660. obj=self,
  1661. id='fields.E150',
  1662. )
  1663. ]
  1664. return []
  1665. def deconstruct(self):
  1666. name, path, args, kwargs = super(GenericIPAddressField, self).deconstruct()
  1667. if self.unpack_ipv4 is not False:
  1668. kwargs['unpack_ipv4'] = self.unpack_ipv4
  1669. if self.protocol != "both":
  1670. kwargs['protocol'] = self.protocol
  1671. if kwargs.get("max_length", None) == 39:
  1672. del kwargs['max_length']
  1673. return name, path, args, kwargs
  1674. def get_internal_type(self):
  1675. return "GenericIPAddressField"
  1676. def to_python(self, value):
  1677. if value and ':' in value:
  1678. return clean_ipv6_address(value,
  1679. self.unpack_ipv4, self.error_messages['invalid'])
  1680. return value
  1681. def get_db_prep_value(self, value, connection, prepared=False):
  1682. if not prepared:
  1683. value = self.get_prep_value(value)
  1684. return value or None
  1685. def get_prep_value(self, value):
  1686. value = super(GenericIPAddressField, self).get_prep_value(value)
  1687. if value is None:
  1688. return None
  1689. if value and ':' in value:
  1690. try:
  1691. return clean_ipv6_address(value, self.unpack_ipv4)
  1692. except exceptions.ValidationError:
  1693. pass
  1694. return six.text_type(value)
  1695. def formfield(self, **kwargs):
  1696. defaults = {
  1697. 'protocol': self.protocol,
  1698. 'form_class': forms.GenericIPAddressField,
  1699. }
  1700. defaults.update(kwargs)
  1701. return super(GenericIPAddressField, self).formfield(**defaults)
  1702. class NullBooleanField(Field):
  1703. empty_strings_allowed = False
  1704. default_error_messages = {
  1705. 'invalid': _("'%(value)s' value must be either None, True or False."),
  1706. }
  1707. description = _("Boolean (Either True, False or None)")
  1708. def __init__(self, *args, **kwargs):
  1709. kwargs['null'] = True
  1710. kwargs['blank'] = True
  1711. super(NullBooleanField, self).__init__(*args, **kwargs)
  1712. def deconstruct(self):
  1713. name, path, args, kwargs = super(NullBooleanField, self).deconstruct()
  1714. del kwargs['null']
  1715. del kwargs['blank']
  1716. return name, path, args, kwargs
  1717. def get_internal_type(self):
  1718. return "NullBooleanField"
  1719. def to_python(self, value):
  1720. if value is None:
  1721. return None
  1722. if value in (True, False):
  1723. return bool(value)
  1724. if value in ('None',):
  1725. return None
  1726. if value in ('t', 'True', '1'):
  1727. return True
  1728. if value in ('f', 'False', '0'):
  1729. return False
  1730. raise exceptions.ValidationError(
  1731. self.error_messages['invalid'],
  1732. code='invalid',
  1733. params={'value': value},
  1734. )
  1735. def get_prep_lookup(self, lookup_type, value):
  1736. # Special-case handling for filters coming from a Web request (e.g. the
  1737. # admin interface). Only works for scalar values (not lists). If you're
  1738. # passing in a list, you might as well make things the right type when
  1739. # constructing the list.
  1740. if value in ('1', '0'):
  1741. value = bool(int(value))
  1742. return super(NullBooleanField, self).get_prep_lookup(lookup_type,
  1743. value)
  1744. def get_prep_value(self, value):
  1745. value = super(NullBooleanField, self).get_prep_value(value)
  1746. if value is None:
  1747. return None
  1748. return bool(value)
  1749. def formfield(self, **kwargs):
  1750. defaults = {
  1751. 'form_class': forms.NullBooleanField,
  1752. 'required': not self.blank,
  1753. 'label': capfirst(self.verbose_name),
  1754. 'help_text': self.help_text}
  1755. defaults.update(kwargs)
  1756. return super(NullBooleanField, self).formfield(**defaults)
  1757. class PositiveIntegerField(IntegerField):
  1758. description = _("Positive integer")
  1759. def get_internal_type(self):
  1760. return "PositiveIntegerField"
  1761. def formfield(self, **kwargs):
  1762. defaults = {'min_value': 0}
  1763. defaults.update(kwargs)
  1764. return super(PositiveIntegerField, self).formfield(**defaults)
  1765. class PositiveSmallIntegerField(IntegerField):
  1766. description = _("Positive small integer")
  1767. def get_internal_type(self):
  1768. return "PositiveSmallIntegerField"
  1769. def formfield(self, **kwargs):
  1770. defaults = {'min_value': 0}
  1771. defaults.update(kwargs)
  1772. return super(PositiveSmallIntegerField, self).formfield(**defaults)
  1773. class SlugField(CharField):
  1774. default_validators = [validators.validate_slug]
  1775. description = _("Slug (up to %(max_length)s)")
  1776. def __init__(self, *args, **kwargs):
  1777. kwargs['max_length'] = kwargs.get('max_length', 50)
  1778. # Set db_index=True unless it's been set manually.
  1779. if 'db_index' not in kwargs:
  1780. kwargs['db_index'] = True
  1781. super(SlugField, self).__init__(*args, **kwargs)
  1782. def deconstruct(self):
  1783. name, path, args, kwargs = super(SlugField, self).deconstruct()
  1784. if kwargs.get("max_length", None) == 50:
  1785. del kwargs['max_length']
  1786. if self.db_index is False:
  1787. kwargs['db_index'] = False
  1788. else:
  1789. del kwargs['db_index']
  1790. return name, path, args, kwargs
  1791. def get_internal_type(self):
  1792. return "SlugField"
  1793. def formfield(self, **kwargs):
  1794. defaults = {'form_class': forms.SlugField}
  1795. defaults.update(kwargs)
  1796. return super(SlugField, self).formfield(**defaults)
  1797. class SmallIntegerField(IntegerField):
  1798. description = _("Small integer")
  1799. def get_internal_type(self):
  1800. return "SmallIntegerField"
  1801. class TextField(Field):
  1802. description = _("Text")
  1803. def get_internal_type(self):
  1804. return "TextField"
  1805. def get_prep_value(self, value):
  1806. value = super(TextField, self).get_prep_value(value)
  1807. if isinstance(value, six.string_types) or value is None:
  1808. return value
  1809. return smart_text(value)
  1810. def formfield(self, **kwargs):
  1811. # Passing max_length to forms.CharField means that the value's length
  1812. # will be validated twice. This is considered acceptable since we want
  1813. # the value in the form field (to pass into widget for example).
  1814. defaults = {'max_length': self.max_length, 'widget': forms.Textarea}
  1815. defaults.update(kwargs)
  1816. return super(TextField, self).formfield(**defaults)
  1817. class TimeField(DateTimeCheckMixin, Field):
  1818. empty_strings_allowed = False
  1819. default_error_messages = {
  1820. 'invalid': _("'%(value)s' value has an invalid format. It must be in "
  1821. "HH:MM[:ss[.uuuuuu]] format."),
  1822. 'invalid_time': _("'%(value)s' value has the correct format "
  1823. "(HH:MM[:ss[.uuuuuu]]) but it is an invalid time."),
  1824. }
  1825. description = _("Time")
  1826. def __init__(self, verbose_name=None, name=None, auto_now=False,
  1827. auto_now_add=False, **kwargs):
  1828. self.auto_now, self.auto_now_add = auto_now, auto_now_add
  1829. if auto_now or auto_now_add:
  1830. kwargs['editable'] = False
  1831. kwargs['blank'] = True
  1832. super(TimeField, self).__init__(verbose_name, name, **kwargs)
  1833. def _check_fix_default_value(self):
  1834. """
  1835. Adds a warning to the checks framework stating, that using an actual
  1836. time or datetime value is probably wrong; it's only being evaluated on
  1837. server start-up.
  1838. For details see ticket #21905
  1839. """
  1840. if not self.has_default():
  1841. return []
  1842. now = timezone.now()
  1843. if not timezone.is_naive(now):
  1844. now = timezone.make_naive(now, timezone.utc)
  1845. value = self.default
  1846. if isinstance(value, datetime.datetime):
  1847. second_offset = datetime.timedelta(seconds=10)
  1848. lower = now - second_offset
  1849. upper = now + second_offset
  1850. if timezone.is_aware(value):
  1851. value = timezone.make_naive(value, timezone.utc)
  1852. elif isinstance(value, datetime.time):
  1853. second_offset = datetime.timedelta(seconds=10)
  1854. lower = now - second_offset
  1855. upper = now + second_offset
  1856. value = datetime.datetime.combine(now.date(), value)
  1857. if timezone.is_aware(value):
  1858. value = timezone.make_naive(value, timezone.utc).time()
  1859. else:
  1860. # No explicit time / datetime value -- no checks necessary
  1861. return []
  1862. if lower <= value <= upper:
  1863. return [
  1864. checks.Warning(
  1865. 'Fixed default value provided.',
  1866. hint='It seems you set a fixed date / time / datetime '
  1867. 'value as default for this field. This may not be '
  1868. 'what you want. If you want to have the current date '
  1869. 'as default, use `django.utils.timezone.now`',
  1870. obj=self,
  1871. id='fields.W161',
  1872. )
  1873. ]
  1874. return []
  1875. def deconstruct(self):
  1876. name, path, args, kwargs = super(TimeField, self).deconstruct()
  1877. if self.auto_now is not False:
  1878. kwargs["auto_now"] = self.auto_now
  1879. if self.auto_now_add is not False:
  1880. kwargs["auto_now_add"] = self.auto_now_add
  1881. if self.auto_now or self.auto_now_add:
  1882. del kwargs['blank']
  1883. del kwargs['editable']
  1884. return name, path, args, kwargs
  1885. def get_internal_type(self):
  1886. return "TimeField"
  1887. def to_python(self, value):
  1888. if value is None:
  1889. return None
  1890. if isinstance(value, datetime.time):
  1891. return value
  1892. if isinstance(value, datetime.datetime):
  1893. # Not usually a good idea to pass in a datetime here (it loses
  1894. # information), but this can be a side-effect of interacting with a
  1895. # database backend (e.g. Oracle), so we'll be accommodating.
  1896. return value.time()
  1897. try:
  1898. parsed = parse_time(value)
  1899. if parsed is not None:
  1900. return parsed
  1901. except ValueError:
  1902. raise exceptions.ValidationError(
  1903. self.error_messages['invalid_time'],
  1904. code='invalid_time',
  1905. params={'value': value},
  1906. )
  1907. raise exceptions.ValidationError(
  1908. self.error_messages['invalid'],
  1909. code='invalid',
  1910. params={'value': value},
  1911. )
  1912. def pre_save(self, model_instance, add):
  1913. if self.auto_now or (self.auto_now_add and add):
  1914. value = datetime.datetime.now().time()
  1915. setattr(model_instance, self.attname, value)
  1916. return value
  1917. else:
  1918. return super(TimeField, self).pre_save(model_instance, add)
  1919. def get_prep_value(self, value):
  1920. value = super(TimeField, self).get_prep_value(value)
  1921. return self.to_python(value)
  1922. def get_db_prep_value(self, value, connection, prepared=False):
  1923. # Casts times into the format expected by the backend
  1924. if not prepared:
  1925. value = self.get_prep_value(value)
  1926. return connection.ops.value_to_db_time(value)
  1927. def value_to_string(self, obj):
  1928. val = self._get_val_from_obj(obj)
  1929. return '' if val is None else val.isoformat()
  1930. def formfield(self, **kwargs):
  1931. defaults = {'form_class': forms.TimeField}
  1932. defaults.update(kwargs)
  1933. return super(TimeField, self).formfield(**defaults)
  1934. class URLField(CharField):
  1935. default_validators = [validators.URLValidator()]
  1936. description = _("URL")
  1937. def __init__(self, verbose_name=None, name=None, **kwargs):
  1938. kwargs['max_length'] = kwargs.get('max_length', 200)
  1939. super(URLField, self).__init__(verbose_name, name, **kwargs)
  1940. def deconstruct(self):
  1941. name, path, args, kwargs = super(URLField, self).deconstruct()
  1942. if kwargs.get("max_length", None) == 200:
  1943. del kwargs['max_length']
  1944. return name, path, args, kwargs
  1945. def formfield(self, **kwargs):
  1946. # As with CharField, this will cause URL validation to be performed
  1947. # twice.
  1948. defaults = {
  1949. 'form_class': forms.URLField,
  1950. }
  1951. defaults.update(kwargs)
  1952. return super(URLField, self).formfield(**defaults)
  1953. class BinaryField(Field):
  1954. description = _("Raw binary data")
  1955. empty_values = [None, b'']
  1956. def __init__(self, *args, **kwargs):
  1957. kwargs['editable'] = False
  1958. super(BinaryField, self).__init__(*args, **kwargs)
  1959. if self.max_length is not None:
  1960. self.validators.append(validators.MaxLengthValidator(self.max_length))
  1961. def deconstruct(self):
  1962. name, path, args, kwargs = super(BinaryField, self).deconstruct()
  1963. del kwargs['editable']
  1964. return name, path, args, kwargs
  1965. def get_internal_type(self):
  1966. return "BinaryField"
  1967. def get_default(self):
  1968. if self.has_default() and not callable(self.default):
  1969. return self.default
  1970. default = super(BinaryField, self).get_default()
  1971. if default == '':
  1972. return b''
  1973. return default
  1974. def get_db_prep_value(self, value, connection, prepared=False):
  1975. value = super(BinaryField, self).get_db_prep_value(value, connection, prepared)
  1976. if value is not None:
  1977. return connection.Database.Binary(value)
  1978. return value
  1979. def value_to_string(self, obj):
  1980. """Binary data is serialized as base64"""
  1981. return b64encode(force_bytes(self._get_val_from_obj(obj))).decode('ascii')
  1982. def to_python(self, value):
  1983. # If it's a string, it should be base64-encoded data
  1984. if isinstance(value, six.text_type):
  1985. return six.memoryview(b64decode(force_bytes(value)))
  1986. return value
  1987. class UUIDField(Field):
  1988. default_error_messages = {
  1989. 'invalid': _("'%(value)s' is not a valid UUID."),
  1990. }
  1991. description = 'Universally unique identifier'
  1992. empty_strings_allowed = False
  1993. def __init__(self, **kwargs):
  1994. kwargs['max_length'] = 32
  1995. super(UUIDField, self).__init__(**kwargs)
  1996. def get_internal_type(self):
  1997. return "UUIDField"
  1998. def get_prep_value(self, value):
  1999. if isinstance(value, uuid.UUID):
  2000. return value.hex
  2001. if isinstance(value, six.string_types):
  2002. return value.replace('-', '')
  2003. return value
  2004. def to_python(self, value):
  2005. if value and not isinstance(value, uuid.UUID):
  2006. try:
  2007. return uuid.UUID(value)
  2008. except ValueError:
  2009. raise exceptions.ValidationError(
  2010. self.error_messages['invalid'],
  2011. code='invalid',
  2012. params={'value': value},
  2013. )
  2014. return value
  2015. def formfield(self, **kwargs):
  2016. defaults = {
  2017. 'form_class': forms.UUIDField,
  2018. }
  2019. defaults.update(kwargs)
  2020. return super(UUIDField, self).formfield(**defaults)