__init__.py 74 KB

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