__init__.py 59 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627
  1. from __future__ import unicode_literals
  2. import copy
  3. import datetime
  4. import decimal
  5. import math
  6. import warnings
  7. from base64 import b64decode, b64encode
  8. from itertools import tee
  9. from django.db import connection
  10. from django.db.models.loading import get_model
  11. from django.db.models.query_utils import QueryWrapper
  12. from django.conf import settings
  13. from django import forms
  14. from django.core import exceptions, validators
  15. from django.utils.datastructures import DictWrapper
  16. from django.utils.dateparse import parse_date, parse_datetime, parse_time
  17. from django.utils.functional import curry, total_ordering
  18. from django.utils.itercompat import is_iterator
  19. from django.utils.text import capfirst
  20. from django.utils import timezone
  21. from django.utils.translation import ugettext_lazy as _
  22. from django.utils.encoding import smart_text, force_text, force_bytes
  23. from django.utils.ipv6 import clean_ipv6_address
  24. from django.utils import six
  25. class Empty(object):
  26. pass
  27. class NOT_PROVIDED:
  28. pass
  29. # The values to use for "blank" in SelectFields. Will be appended to the start
  30. # of most "choices" lists.
  31. BLANK_CHOICE_DASH = [("", "---------")]
  32. def _load_field(app_label, model_name, field_name):
  33. return get_model(app_label, model_name)._meta.get_field_by_name(field_name)[0]
  34. class FieldDoesNotExist(Exception):
  35. pass
  36. # A guide to Field parameters:
  37. #
  38. # * name: The name of the field specifed in the model.
  39. # * attname: The attribute to use on the model object. This is the same as
  40. # "name", except in the case of ForeignKeys, where "_id" is
  41. # appended.
  42. # * db_column: The db_column specified in the model (or None).
  43. # * column: The database column for this field. This is the same as
  44. # "attname", except if db_column is specified.
  45. #
  46. # Code that introspects values, or does other dynamic things, should use
  47. # attname. For example, this gets the primary key value of object "obj":
  48. #
  49. # getattr(obj, opts.pk.attname)
  50. def _empty(of_cls):
  51. new = Empty()
  52. new.__class__ = of_cls
  53. return new
  54. @total_ordering
  55. class Field(object):
  56. """Base class for all field types"""
  57. # Designates whether empty strings fundamentally are allowed at the
  58. # database level.
  59. empty_strings_allowed = True
  60. empty_values = list(validators.EMPTY_VALUES)
  61. # These track each time a Field instance is created. Used to retain order.
  62. # The auto_creation_counter is used for fields that Django implicitly
  63. # creates, creation_counter is used for all user-specified fields.
  64. creation_counter = 0
  65. auto_creation_counter = -1
  66. default_validators = [] # Default set of validators
  67. default_error_messages = {
  68. 'invalid_choice': _('Value %(value)r is not a valid choice.'),
  69. 'null': _('This field cannot be null.'),
  70. 'blank': _('This field cannot be blank.'),
  71. 'unique': _('%(model_name)s with this %(field_label)s '
  72. 'already exists.'),
  73. }
  74. # Generic field type description, usually overriden by subclasses
  75. def _description(self):
  76. return _('Field of type: %(field_type)s') % {
  77. 'field_type': self.__class__.__name__
  78. }
  79. description = property(_description)
  80. def __init__(self, verbose_name=None, name=None, primary_key=False,
  81. max_length=None, unique=False, blank=False, null=False,
  82. db_index=False, rel=None, default=NOT_PROVIDED, editable=True,
  83. serialize=True, unique_for_date=None, unique_for_month=None,
  84. unique_for_year=None, choices=None, help_text='', db_column=None,
  85. db_tablespace=None, auto_created=False, validators=[],
  86. error_messages=None):
  87. self.name = name
  88. self.verbose_name = verbose_name # May be set by set_attributes_from_name
  89. self._verbose_name = verbose_name # Store original for deconstruction
  90. self.primary_key = primary_key
  91. self.max_length, self._unique = max_length, unique
  92. self.blank, self.null = blank, null
  93. self.rel = rel
  94. self.default = default
  95. self.editable = editable
  96. self.serialize = serialize
  97. self.unique_for_date, self.unique_for_month = (unique_for_date,
  98. unique_for_month)
  99. self.unique_for_year = unique_for_year
  100. self._choices = choices or []
  101. self.help_text = help_text
  102. self.db_column = db_column
  103. self.db_tablespace = db_tablespace or settings.DEFAULT_INDEX_TABLESPACE
  104. self.auto_created = auto_created
  105. # Set db_index to True if the field has a relationship and doesn't
  106. # explicitly set db_index.
  107. self.db_index = db_index
  108. # Adjust the appropriate creation counter, and save our local copy.
  109. if auto_created:
  110. self.creation_counter = Field.auto_creation_counter
  111. Field.auto_creation_counter -= 1
  112. else:
  113. self.creation_counter = Field.creation_counter
  114. Field.creation_counter += 1
  115. self._validators = validators # Store for deconstruction later
  116. self.validators = self.default_validators + validators
  117. messages = {}
  118. for c in reversed(self.__class__.__mro__):
  119. messages.update(getattr(c, 'default_error_messages', {}))
  120. messages.update(error_messages or {})
  121. self._error_messages = error_messages # Store for deconstruction later
  122. self.error_messages = messages
  123. def deconstruct(self):
  124. """
  125. Returns enough information to recreate the field as a 4-tuple:
  126. * The name of the field on the model, if contribute_to_class has been run
  127. * The import path of the field, including the class: django.db.models.IntegerField
  128. This should be the most portable version, so less specific may be better.
  129. * A list of positional arguments
  130. * A dict of keyword arguments
  131. Note that the positional or keyword arguments must contain values of the
  132. following types (including inner values of collection types):
  133. * None, bool, str, unicode, int, long, float, complex, set, frozenset, list, tuple, dict
  134. * UUID
  135. * datetime.datetime (naive), datetime.date
  136. * top-level classes, top-level functions - will be referenced by their full import path
  137. * Storage instances - these have their own deconstruct() method
  138. This is because the values here must be serialised into a text format
  139. (possibly new Python code, possibly JSON) and these are the only types
  140. with encoding handlers defined.
  141. There's no need to return the exact way the field was instantiated this time,
  142. just ensure that the resulting field is the same - prefer keyword arguments
  143. over positional ones, and omit parameters with their default values.
  144. """
  145. # Short-form way of fetching all the default parameters
  146. keywords = {}
  147. possibles = {
  148. "verbose_name": None,
  149. "primary_key": False,
  150. "max_length": None,
  151. "unique": False,
  152. "blank": False,
  153. "null": False,
  154. "db_index": False,
  155. "default": NOT_PROVIDED,
  156. "editable": True,
  157. "serialize": True,
  158. "unique_for_date": None,
  159. "unique_for_month": None,
  160. "unique_for_year": None,
  161. "choices": [],
  162. "help_text": '',
  163. "db_column": None,
  164. "db_tablespace": settings.DEFAULT_INDEX_TABLESPACE,
  165. "auto_created": False,
  166. "validators": [],
  167. "error_messages": None,
  168. }
  169. attr_overrides = {
  170. "unique": "_unique",
  171. "choices": "_choices",
  172. "error_messages": "_error_messages",
  173. "validators": "_validators",
  174. "verbose_name": "_verbose_name",
  175. }
  176. equals_comparison = set(["choices", "validators", "db_tablespace"])
  177. for name, default in possibles.items():
  178. value = getattr(self, attr_overrides.get(name, name))
  179. if name in equals_comparison:
  180. if value != default:
  181. keywords[name] = value
  182. else:
  183. if value is not default:
  184. keywords[name] = value
  185. # Work out path - we shorten it for known Django core fields
  186. path = "%s.%s" % (self.__class__.__module__, self.__class__.__name__)
  187. if path.startswith("django.db.models.fields.related"):
  188. path = path.replace("django.db.models.fields.related", "django.db.models")
  189. if path.startswith("django.db.models.fields.files"):
  190. path = path.replace("django.db.models.fields.files", "django.db.models")
  191. if path.startswith("django.db.models.fields"):
  192. path = path.replace("django.db.models.fields", "django.db.models")
  193. # Return basic info - other fields should override this.
  194. return (
  195. self.name,
  196. path,
  197. [],
  198. keywords,
  199. )
  200. def __eq__(self, other):
  201. # Needed for @total_ordering
  202. if isinstance(other, Field):
  203. return self.creation_counter == other.creation_counter
  204. return NotImplemented
  205. def __lt__(self, other):
  206. # This is needed because bisect does not take a comparison function.
  207. if isinstance(other, Field):
  208. return self.creation_counter < other.creation_counter
  209. return NotImplemented
  210. def __hash__(self):
  211. return hash(self.creation_counter)
  212. def __deepcopy__(self, memodict):
  213. # We don't have to deepcopy very much here, since most things are not
  214. # intended to be altered after initial creation.
  215. obj = copy.copy(self)
  216. if self.rel:
  217. obj.rel = copy.copy(self.rel)
  218. if hasattr(self.rel, 'field') and self.rel.field is self:
  219. obj.rel.field = obj
  220. memodict[id(self)] = obj
  221. return obj
  222. def __copy__(self):
  223. # We need to avoid hitting __reduce__, so define this
  224. # slightly weird copy construct.
  225. obj = Empty()
  226. obj.__class__ = self.__class__
  227. obj.__dict__ = self.__dict__.copy()
  228. return obj
  229. def __reduce__(self):
  230. """
  231. Pickling should return the model._meta.fields instance of the field,
  232. not a new copy of that field. So, we use the app cache to load the
  233. model and then the field back.
  234. """
  235. if not hasattr(self, 'model'):
  236. # Fields are sometimes used without attaching them to models (for
  237. # example in aggregation). In this case give back a plain field
  238. # instance. The code below will create a new empty instance of
  239. # class self.__class__, then update its dict with self.__dict__
  240. # values - so, this is very close to normal pickle.
  241. return _empty, (self.__class__,), self.__dict__
  242. if self.model._deferred:
  243. # Deferred model will not be found from the app cache. This could
  244. # be fixed by reconstructing the deferred model on unpickle.
  245. raise RuntimeError("Fields of deferred models can't be reduced")
  246. return _load_field, (self.model._meta.app_label, self.model._meta.object_name,
  247. self.name)
  248. def to_python(self, value):
  249. """
  250. Converts the input value into the expected Python data type, raising
  251. django.core.exceptions.ValidationError if the data can't be converted.
  252. Returns the converted value. Subclasses should override this.
  253. """
  254. return value
  255. def run_validators(self, value):
  256. if value in self.empty_values:
  257. return
  258. errors = []
  259. for v in self.validators:
  260. try:
  261. v(value)
  262. except exceptions.ValidationError as e:
  263. if hasattr(e, 'code') and e.code in self.error_messages:
  264. e.message = self.error_messages[e.code]
  265. errors.extend(e.error_list)
  266. if errors:
  267. raise exceptions.ValidationError(errors)
  268. def validate(self, value, model_instance):
  269. """
  270. Validates value and throws ValidationError. Subclasses should override
  271. this to provide validation logic.
  272. """
  273. if not self.editable:
  274. # Skip validation for non-editable fields.
  275. return
  276. if self._choices and value not in self.empty_values:
  277. for option_key, option_value in self.choices:
  278. if isinstance(option_value, (list, tuple)):
  279. # This is an optgroup, so look inside the group for
  280. # options.
  281. for optgroup_key, optgroup_value in option_value:
  282. if value == optgroup_key:
  283. return
  284. elif value == option_key:
  285. return
  286. raise exceptions.ValidationError(
  287. self.error_messages['invalid_choice'],
  288. code='invalid_choice',
  289. params={'value': value},
  290. )
  291. if value is None and not self.null:
  292. raise exceptions.ValidationError(self.error_messages['null'], code='null')
  293. if not self.blank and value in self.empty_values:
  294. raise exceptions.ValidationError(self.error_messages['blank'], code='blank')
  295. def clean(self, value, model_instance):
  296. """
  297. Convert the value's type and run validation. Validation errors
  298. from to_python and validate are propagated. The correct value is
  299. returned if no error is raised.
  300. """
  301. value = self.to_python(value)
  302. self.validate(value, model_instance)
  303. self.run_validators(value)
  304. return value
  305. def db_type(self, connection):
  306. """
  307. Returns the database column data type for this field, for the provided
  308. connection.
  309. """
  310. # The default implementation of this method looks at the
  311. # backend-specific DATA_TYPES dictionary, looking up the field by its
  312. # "internal type".
  313. #
  314. # A Field class can implement the get_internal_type() method to specify
  315. # which *preexisting* Django Field class it's most similar to -- i.e.,
  316. # a custom field might be represented by a TEXT column type, which is
  317. # the same as the TextField Django field type, which means the custom
  318. # field's get_internal_type() returns 'TextField'.
  319. #
  320. # But the limitation of the get_internal_type() / data_types approach
  321. # is that it cannot handle database column types that aren't already
  322. # mapped to one of the built-in Django field types. In this case, you
  323. # can implement db_type() instead of get_internal_type() to specify
  324. # exactly which wacky database column type you want to use.
  325. data = DictWrapper(self.__dict__, connection.ops.quote_name, "qn_")
  326. try:
  327. return (connection.creation.data_types[self.get_internal_type()]
  328. % data)
  329. except KeyError:
  330. return None
  331. @property
  332. def unique(self):
  333. return self._unique or self.primary_key
  334. def set_attributes_from_name(self, name):
  335. if not self.name:
  336. self.name = name
  337. self.attname, self.column = self.get_attname_column()
  338. if self.verbose_name is None and self.name:
  339. self.verbose_name = self.name.replace('_', ' ')
  340. def contribute_to_class(self, cls, name, virtual_only=False):
  341. self.set_attributes_from_name(name)
  342. self.model = cls
  343. if virtual_only:
  344. cls._meta.add_virtual_field(self)
  345. else:
  346. cls._meta.add_field(self)
  347. if self.choices:
  348. setattr(cls, 'get_%s_display' % self.name,
  349. curry(cls._get_FIELD_display, field=self))
  350. def get_attname(self):
  351. return self.name
  352. def get_attname_column(self):
  353. attname = self.get_attname()
  354. column = self.db_column or attname
  355. return attname, column
  356. def get_cache_name(self):
  357. return '_%s_cache' % self.name
  358. def get_internal_type(self):
  359. return self.__class__.__name__
  360. def pre_save(self, model_instance, add):
  361. """
  362. Returns field's value just before saving.
  363. """
  364. return getattr(model_instance, self.attname)
  365. def get_prep_value(self, value):
  366. """
  367. Perform preliminary non-db specific value checks and conversions.
  368. """
  369. return value
  370. def get_db_prep_value(self, value, connection, prepared=False):
  371. """Returns field's value prepared for interacting with the database
  372. backend.
  373. Used by the default implementations of ``get_db_prep_save``and
  374. `get_db_prep_lookup```
  375. """
  376. if not prepared:
  377. value = self.get_prep_value(value)
  378. return value
  379. def get_db_prep_save(self, value, connection):
  380. """
  381. Returns field's value prepared for saving into a database.
  382. """
  383. return self.get_db_prep_value(value, connection=connection,
  384. prepared=False)
  385. def get_prep_lookup(self, lookup_type, value):
  386. """
  387. Perform preliminary non-db specific lookup checks and conversions
  388. """
  389. if hasattr(value, 'prepare'):
  390. return value.prepare()
  391. if hasattr(value, '_prepare'):
  392. return value._prepare()
  393. if lookup_type in {
  394. 'iexact', 'contains', 'icontains',
  395. 'startswith', 'istartswith', 'endswith', 'iendswith',
  396. 'month', 'day', 'week_day', 'hour', 'minute', 'second',
  397. 'isnull', 'search', 'regex', 'iregex',
  398. }:
  399. return value
  400. elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
  401. return self.get_prep_value(value)
  402. elif lookup_type in ('range', 'in'):
  403. return [self.get_prep_value(v) for v in value]
  404. elif lookup_type == 'year':
  405. try:
  406. return int(value)
  407. except ValueError:
  408. raise ValueError("The __year lookup type requires an integer "
  409. "argument")
  410. raise TypeError("Field has invalid lookup: %s" % lookup_type)
  411. def get_db_prep_lookup(self, lookup_type, value, connection,
  412. prepared=False):
  413. """
  414. Returns field's value prepared for database lookup.
  415. """
  416. if not prepared:
  417. value = self.get_prep_lookup(lookup_type, value)
  418. if hasattr(value, 'get_compiler'):
  419. value = value.get_compiler(connection=connection)
  420. if hasattr(value, 'as_sql') or hasattr(value, '_as_sql'):
  421. # If the value has a relabeled_clone method it means the
  422. # value will be handled later on.
  423. if hasattr(value, 'relabeled_clone'):
  424. return value
  425. if hasattr(value, 'as_sql'):
  426. sql, params = value.as_sql()
  427. else:
  428. sql, params = value._as_sql(connection=connection)
  429. return QueryWrapper(('(%s)' % sql), params)
  430. if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute',
  431. 'second', 'search', 'regex', 'iregex'):
  432. return [value]
  433. elif lookup_type in ('exact', 'gt', 'gte', 'lt', 'lte'):
  434. return [self.get_db_prep_value(value, connection=connection,
  435. prepared=prepared)]
  436. elif lookup_type in ('range', 'in'):
  437. return [self.get_db_prep_value(v, connection=connection,
  438. prepared=prepared) for v in value]
  439. elif lookup_type in ('contains', 'icontains'):
  440. return ["%%%s%%" % connection.ops.prep_for_like_query(value)]
  441. elif lookup_type == 'iexact':
  442. return [connection.ops.prep_for_iexact_query(value)]
  443. elif lookup_type in ('startswith', 'istartswith'):
  444. return ["%s%%" % connection.ops.prep_for_like_query(value)]
  445. elif lookup_type in ('endswith', 'iendswith'):
  446. return ["%%%s" % connection.ops.prep_for_like_query(value)]
  447. elif lookup_type == 'isnull':
  448. return []
  449. elif lookup_type == 'year':
  450. if isinstance(self, DateTimeField):
  451. return connection.ops.year_lookup_bounds_for_datetime_field(value)
  452. elif isinstance(self, DateField):
  453. return connection.ops.year_lookup_bounds_for_date_field(value)
  454. else:
  455. return [value] # this isn't supposed to happen
  456. def has_default(self):
  457. """
  458. Returns a boolean of whether this field has a default value.
  459. """
  460. return self.default is not NOT_PROVIDED
  461. def get_default(self):
  462. """
  463. Returns the default value for this field.
  464. """
  465. if self.has_default():
  466. if callable(self.default):
  467. return self.default()
  468. return force_text(self.default, strings_only=True)
  469. if (not self.empty_strings_allowed or (self.null and
  470. not connection.features.interprets_empty_strings_as_nulls)):
  471. return None
  472. return ""
  473. def get_validator_unique_lookup_type(self):
  474. return '%s__exact' % self.name
  475. def get_choices(self, include_blank=True, blank_choice=BLANK_CHOICE_DASH):
  476. """Returns choices with a default blank choices included, for use
  477. as SelectField choices for this field."""
  478. first_choice = blank_choice if include_blank else []
  479. if self.choices:
  480. return first_choice + list(self.choices)
  481. rel_model = self.rel.to
  482. if hasattr(self.rel, 'get_related_field'):
  483. lst = [(getattr(x, self.rel.get_related_field().attname),
  484. smart_text(x))
  485. for x in rel_model._default_manager.complex_filter(
  486. self.rel.limit_choices_to)]
  487. else:
  488. lst = [(x._get_pk_val(), smart_text(x))
  489. for x in rel_model._default_manager.complex_filter(
  490. self.rel.limit_choices_to)]
  491. return first_choice + lst
  492. def get_choices_default(self):
  493. return self.get_choices()
  494. def get_flatchoices(self, include_blank=True,
  495. blank_choice=BLANK_CHOICE_DASH):
  496. """
  497. Returns flattened choices with a default blank choice included.
  498. """
  499. first_choice = blank_choice if include_blank else []
  500. return first_choice + list(self.flatchoices)
  501. def _get_val_from_obj(self, obj):
  502. if obj is not None:
  503. return getattr(obj, self.attname)
  504. else:
  505. return self.get_default()
  506. def value_to_string(self, obj):
  507. """
  508. Returns a string value of this field from the passed obj.
  509. This is used by the serialization framework.
  510. """
  511. return smart_text(self._get_val_from_obj(obj))
  512. def bind(self, fieldmapping, original, bound_field_class):
  513. return bound_field_class(self, fieldmapping, original)
  514. def _get_choices(self):
  515. if is_iterator(self._choices):
  516. choices, self._choices = tee(self._choices)
  517. return choices
  518. else:
  519. return self._choices
  520. choices = property(_get_choices)
  521. def _get_flatchoices(self):
  522. """Flattened version of choices tuple."""
  523. flat = []
  524. for choice, value in self.choices:
  525. if isinstance(value, (list, tuple)):
  526. flat.extend(value)
  527. else:
  528. flat.append((choice, value))
  529. return flat
  530. flatchoices = property(_get_flatchoices)
  531. def save_form_data(self, instance, data):
  532. setattr(instance, self.name, data)
  533. def formfield(self, form_class=None, **kwargs):
  534. """
  535. Returns a django.forms.Field instance for this database Field.
  536. """
  537. defaults = {'required': not self.blank,
  538. 'label': capfirst(self.verbose_name),
  539. 'help_text': self.help_text}
  540. if self.has_default():
  541. if callable(self.default):
  542. defaults['initial'] = self.default
  543. defaults['show_hidden_initial'] = True
  544. else:
  545. defaults['initial'] = self.get_default()
  546. if self.choices:
  547. # Fields with choices get special treatment.
  548. include_blank = (self.blank or
  549. not (self.has_default() or 'initial' in kwargs))
  550. defaults['choices'] = self.get_choices(include_blank=include_blank)
  551. defaults['coerce'] = self.to_python
  552. if self.null:
  553. defaults['empty_value'] = None
  554. if form_class is None or not issubclass(form_class, forms.TypedChoiceField):
  555. form_class = forms.TypedChoiceField
  556. # Many of the subclass-specific formfield arguments (min_value,
  557. # max_value) don't apply for choice fields, so be sure to only pass
  558. # the values that TypedChoiceField will understand.
  559. for k in list(kwargs):
  560. if k not in ('coerce', 'empty_value', 'choices', 'required',
  561. 'widget', 'label', 'initial', 'help_text',
  562. 'error_messages', 'show_hidden_initial'):
  563. del kwargs[k]
  564. defaults.update(kwargs)
  565. if form_class is None:
  566. form_class = forms.CharField
  567. return form_class(**defaults)
  568. def value_from_object(self, obj):
  569. """
  570. Returns the value of this field in the given model instance.
  571. """
  572. return getattr(obj, self.attname)
  573. def __repr__(self):
  574. """
  575. Displays the module, class and name of the field.
  576. """
  577. path = '%s.%s' % (self.__class__.__module__, self.__class__.__name__)
  578. name = getattr(self, 'name', None)
  579. if name is not None:
  580. return '<%s: %s>' % (path, name)
  581. return '<%s>' % path
  582. class AutoField(Field):
  583. description = _("Integer")
  584. empty_strings_allowed = False
  585. default_error_messages = {
  586. 'invalid': _("'%(value)s' value must be an integer."),
  587. }
  588. def __init__(self, *args, **kwargs):
  589. assert kwargs.get('primary_key', False) is True, \
  590. "%ss must have primary_key=True." % self.__class__.__name__
  591. kwargs['blank'] = True
  592. Field.__init__(self, *args, **kwargs)
  593. def deconstruct(self):
  594. name, path, args, kwargs = super(AutoField, self).deconstruct()
  595. del kwargs['blank']
  596. kwargs['primary_key'] = True
  597. return name, path, args, kwargs
  598. def get_internal_type(self):
  599. return "AutoField"
  600. def to_python(self, value):
  601. if value is None:
  602. return value
  603. try:
  604. return int(value)
  605. except (TypeError, ValueError):
  606. raise exceptions.ValidationError(
  607. self.error_messages['invalid'],
  608. code='invalid',
  609. params={'value': value},
  610. )
  611. def validate(self, value, model_instance):
  612. pass
  613. def get_db_prep_value(self, value, connection, prepared=False):
  614. if not prepared:
  615. value = self.get_prep_value(value)
  616. value = connection.ops.validate_autopk_value(value)
  617. return value
  618. def get_prep_value(self, value):
  619. if value is None:
  620. return None
  621. return int(value)
  622. def contribute_to_class(self, cls, name):
  623. assert not cls._meta.has_auto_field, \
  624. "A model can't have more than one AutoField."
  625. super(AutoField, self).contribute_to_class(cls, name)
  626. cls._meta.has_auto_field = True
  627. cls._meta.auto_field = self
  628. def formfield(self, **kwargs):
  629. return None
  630. class BooleanField(Field):
  631. empty_strings_allowed = False
  632. default_error_messages = {
  633. 'invalid': _("'%(value)s' value must be either True or False."),
  634. }
  635. description = _("Boolean (Either True or False)")
  636. def __init__(self, *args, **kwargs):
  637. kwargs['blank'] = True
  638. Field.__init__(self, *args, **kwargs)
  639. def deconstruct(self):
  640. name, path, args, kwargs = super(BooleanField, self).deconstruct()
  641. del kwargs['blank']
  642. return name, path, args, kwargs
  643. def get_internal_type(self):
  644. return "BooleanField"
  645. def to_python(self, value):
  646. if value in (True, False):
  647. # if value is 1 or 0 than it's equal to True or False, but we want
  648. # to return a true bool for semantic reasons.
  649. return bool(value)
  650. if value in ('t', 'True', '1'):
  651. return True
  652. if value in ('f', 'False', '0'):
  653. return False
  654. raise exceptions.ValidationError(
  655. self.error_messages['invalid'],
  656. code='invalid',
  657. params={'value': value},
  658. )
  659. def get_prep_lookup(self, lookup_type, value):
  660. # Special-case handling for filters coming from a Web request (e.g. the
  661. # admin interface). Only works for scalar values (not lists). If you're
  662. # passing in a list, you might as well make things the right type when
  663. # constructing the list.
  664. if value in ('1', '0'):
  665. value = bool(int(value))
  666. return super(BooleanField, self).get_prep_lookup(lookup_type, value)
  667. def get_prep_value(self, value):
  668. if value is None:
  669. return None
  670. return bool(value)
  671. def formfield(self, **kwargs):
  672. # Unlike most fields, BooleanField figures out include_blank from
  673. # self.null instead of self.blank.
  674. if self.choices:
  675. include_blank = (self.null or
  676. not (self.has_default() or 'initial' in kwargs))
  677. defaults = {'choices': self.get_choices(include_blank=include_blank)}
  678. else:
  679. defaults = {'form_class': forms.BooleanField}
  680. defaults.update(kwargs)
  681. return super(BooleanField, self).formfield(**defaults)
  682. class CharField(Field):
  683. description = _("String (up to %(max_length)s)")
  684. def __init__(self, *args, **kwargs):
  685. super(CharField, self).__init__(*args, **kwargs)
  686. self.validators.append(validators.MaxLengthValidator(self.max_length))
  687. def get_internal_type(self):
  688. return "CharField"
  689. def to_python(self, value):
  690. if isinstance(value, six.string_types) or value is None:
  691. return value
  692. return smart_text(value)
  693. def get_prep_value(self, value):
  694. return self.to_python(value)
  695. def formfield(self, **kwargs):
  696. # Passing max_length to forms.CharField means that the value's length
  697. # will be validated twice. This is considered acceptable since we want
  698. # the value in the form field (to pass into widget for example).
  699. defaults = {'max_length': self.max_length}
  700. defaults.update(kwargs)
  701. return super(CharField, self).formfield(**defaults)
  702. # TODO: Maybe move this into contrib, because it's specialized.
  703. class CommaSeparatedIntegerField(CharField):
  704. default_validators = [validators.validate_comma_separated_integer_list]
  705. description = _("Comma-separated integers")
  706. def formfield(self, **kwargs):
  707. defaults = {
  708. 'error_messages': {
  709. 'invalid': _('Enter only digits separated by commas.'),
  710. }
  711. }
  712. defaults.update(kwargs)
  713. return super(CommaSeparatedIntegerField, self).formfield(**defaults)
  714. class DateField(Field):
  715. empty_strings_allowed = False
  716. default_error_messages = {
  717. 'invalid': _("'%(value)s' value has an invalid date format. It must be "
  718. "in YYYY-MM-DD format."),
  719. 'invalid_date': _("'%(value)s' value has the correct format (YYYY-MM-DD) "
  720. "but it is an invalid date."),
  721. }
  722. description = _("Date (without time)")
  723. def __init__(self, verbose_name=None, name=None, auto_now=False,
  724. auto_now_add=False, **kwargs):
  725. self.auto_now, self.auto_now_add = auto_now, auto_now_add
  726. if auto_now or auto_now_add:
  727. kwargs['editable'] = False
  728. kwargs['blank'] = True
  729. Field.__init__(self, verbose_name, name, **kwargs)
  730. def deconstruct(self):
  731. name, path, args, kwargs = super(DateField, self).deconstruct()
  732. if self.auto_now:
  733. kwargs['auto_now'] = True
  734. del kwargs['editable']
  735. del kwargs['blank']
  736. if self.auto_now_add:
  737. kwargs['auto_now_add'] = True
  738. del kwargs['editable']
  739. del kwargs['blank']
  740. return name, path, args, kwargs
  741. def get_internal_type(self):
  742. return "DateField"
  743. def to_python(self, value):
  744. if value is None:
  745. return value
  746. if isinstance(value, datetime.datetime):
  747. if settings.USE_TZ and timezone.is_aware(value):
  748. # Convert aware datetimes to the default time zone
  749. # before casting them to dates (#17742).
  750. default_timezone = timezone.get_default_timezone()
  751. value = timezone.make_naive(value, default_timezone)
  752. return value.date()
  753. if isinstance(value, datetime.date):
  754. return value
  755. try:
  756. parsed = parse_date(value)
  757. if parsed is not None:
  758. return parsed
  759. except ValueError:
  760. raise exceptions.ValidationError(
  761. self.error_messages['invalid_date'],
  762. code='invalid_date',
  763. params={'value': value},
  764. )
  765. raise exceptions.ValidationError(
  766. self.error_messages['invalid'],
  767. code='invalid',
  768. params={'value': value},
  769. )
  770. def pre_save(self, model_instance, add):
  771. if self.auto_now or (self.auto_now_add and add):
  772. value = datetime.date.today()
  773. setattr(model_instance, self.attname, value)
  774. return value
  775. else:
  776. return super(DateField, self).pre_save(model_instance, add)
  777. def contribute_to_class(self, cls, name):
  778. super(DateField, self).contribute_to_class(cls, name)
  779. if not self.null:
  780. setattr(cls, 'get_next_by_%s' % self.name,
  781. curry(cls._get_next_or_previous_by_FIELD, field=self,
  782. is_next=True))
  783. setattr(cls, 'get_previous_by_%s' % self.name,
  784. curry(cls._get_next_or_previous_by_FIELD, field=self,
  785. is_next=False))
  786. def get_prep_lookup(self, lookup_type, value):
  787. # For dates lookups, convert the value to an int
  788. # so the database backend always sees a consistent type.
  789. if lookup_type in ('month', 'day', 'week_day', 'hour', 'minute', 'second'):
  790. return int(value)
  791. return super(DateField, self).get_prep_lookup(lookup_type, value)
  792. def get_prep_value(self, value):
  793. return self.to_python(value)
  794. def get_db_prep_value(self, value, connection, prepared=False):
  795. # Casts dates into the format expected by the backend
  796. if not prepared:
  797. value = self.get_prep_value(value)
  798. return connection.ops.value_to_db_date(value)
  799. def value_to_string(self, obj):
  800. val = self._get_val_from_obj(obj)
  801. return '' if val is None else val.isoformat()
  802. def formfield(self, **kwargs):
  803. defaults = {'form_class': forms.DateField}
  804. defaults.update(kwargs)
  805. return super(DateField, self).formfield(**defaults)
  806. class DateTimeField(DateField):
  807. empty_strings_allowed = False
  808. default_error_messages = {
  809. 'invalid': _("'%(value)s' value has an invalid format. It must be in "
  810. "YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ] format."),
  811. 'invalid_date': _("'%(value)s' value has the correct format "
  812. "(YYYY-MM-DD) but it is an invalid date."),
  813. 'invalid_datetime': _("'%(value)s' value has the correct format "
  814. "(YYYY-MM-DD HH:MM[:ss[.uuuuuu]][TZ]) "
  815. "but it is an invalid date/time."),
  816. }
  817. description = _("Date (with time)")
  818. # __init__ is inherited from DateField
  819. def get_internal_type(self):
  820. return "DateTimeField"
  821. def to_python(self, value):
  822. if value is None:
  823. return value
  824. if isinstance(value, datetime.datetime):
  825. return value
  826. if isinstance(value, datetime.date):
  827. value = datetime.datetime(value.year, value.month, value.day)
  828. if settings.USE_TZ:
  829. # For backwards compatibility, interpret naive datetimes in
  830. # local time. This won't work during DST change, but we can't
  831. # do much about it, so we let the exceptions percolate up the
  832. # call stack.
  833. warnings.warn("DateTimeField received a naive datetime (%s)"
  834. " while time zone support is active." % value,
  835. RuntimeWarning)
  836. default_timezone = timezone.get_default_timezone()
  837. value = timezone.make_aware(value, default_timezone)
  838. return value
  839. try:
  840. parsed = parse_datetime(value)
  841. if parsed is not None:
  842. return parsed
  843. except ValueError:
  844. raise exceptions.ValidationError(
  845. self.error_messages['invalid_datetime'],
  846. code='invalid_datetime',
  847. params={'value': value},
  848. )
  849. try:
  850. parsed = parse_date(value)
  851. if parsed is not None:
  852. return datetime.datetime(parsed.year, parsed.month, parsed.day)
  853. except ValueError:
  854. raise exceptions.ValidationError(
  855. self.error_messages['invalid_date'],
  856. code='invalid_date',
  857. params={'value': value},
  858. )
  859. raise exceptions.ValidationError(
  860. self.error_messages['invalid'],
  861. code='invalid',
  862. params={'value': value},
  863. )
  864. def pre_save(self, model_instance, add):
  865. if self.auto_now or (self.auto_now_add and add):
  866. value = timezone.now()
  867. setattr(model_instance, self.attname, value)
  868. return value
  869. else:
  870. return super(DateTimeField, self).pre_save(model_instance, add)
  871. # contribute_to_class is inherited from DateField, it registers
  872. # get_next_by_FOO and get_prev_by_FOO
  873. # get_prep_lookup is inherited from DateField
  874. def get_prep_value(self, value):
  875. value = self.to_python(value)
  876. if value is not None and settings.USE_TZ and timezone.is_naive(value):
  877. # For backwards compatibility, interpret naive datetimes in local
  878. # time. This won't work during DST change, but we can't do much
  879. # about it, so we let the exceptions percolate up the call stack.
  880. warnings.warn("DateTimeField received a naive datetime (%s)"
  881. " while time zone support is active." % value,
  882. RuntimeWarning)
  883. default_timezone = timezone.get_default_timezone()
  884. value = timezone.make_aware(value, default_timezone)
  885. return value
  886. def get_db_prep_value(self, value, connection, prepared=False):
  887. # Casts datetimes into the format expected by the backend
  888. if not prepared:
  889. value = self.get_prep_value(value)
  890. return connection.ops.value_to_db_datetime(value)
  891. def value_to_string(self, obj):
  892. val = self._get_val_from_obj(obj)
  893. return '' if val is None else val.isoformat()
  894. def formfield(self, **kwargs):
  895. defaults = {'form_class': forms.DateTimeField}
  896. defaults.update(kwargs)
  897. return super(DateTimeField, self).formfield(**defaults)
  898. class DecimalField(Field):
  899. empty_strings_allowed = False
  900. default_error_messages = {
  901. 'invalid': _("'%(value)s' value must be a decimal number."),
  902. }
  903. description = _("Decimal number")
  904. def __init__(self, verbose_name=None, name=None, max_digits=None,
  905. decimal_places=None, **kwargs):
  906. self.max_digits, self.decimal_places = max_digits, decimal_places
  907. Field.__init__(self, verbose_name, name, **kwargs)
  908. def deconstruct(self):
  909. name, path, args, kwargs = super(DecimalField, self).deconstruct()
  910. if self.max_digits:
  911. kwargs['max_digits'] = self.max_digits
  912. if self.decimal_places:
  913. kwargs['decimal_places'] = self.decimal_places
  914. return name, path, args, kwargs
  915. def get_internal_type(self):
  916. return "DecimalField"
  917. def to_python(self, value):
  918. if value is None:
  919. return value
  920. try:
  921. return decimal.Decimal(value)
  922. except decimal.InvalidOperation:
  923. raise exceptions.ValidationError(
  924. self.error_messages['invalid'],
  925. code='invalid',
  926. params={'value': value},
  927. )
  928. def _format(self, value):
  929. if isinstance(value, six.string_types) or value is None:
  930. return value
  931. else:
  932. return self.format_number(value)
  933. def format_number(self, value):
  934. """
  935. Formats a number into a string with the requisite number of digits and
  936. decimal places.
  937. """
  938. # Method moved to django.db.backends.util.
  939. #
  940. # It is preserved because it is used by the oracle backend
  941. # (django.db.backends.oracle.query), and also for
  942. # backwards-compatibility with any external code which may have used
  943. # this method.
  944. from django.db.backends import util
  945. return util.format_number(value, self.max_digits, self.decimal_places)
  946. def get_db_prep_save(self, value, connection):
  947. return connection.ops.value_to_db_decimal(self.to_python(value),
  948. self.max_digits, self.decimal_places)
  949. def get_prep_value(self, value):
  950. return self.to_python(value)
  951. def formfield(self, **kwargs):
  952. defaults = {
  953. 'max_digits': self.max_digits,
  954. 'decimal_places': self.decimal_places,
  955. 'form_class': forms.DecimalField,
  956. }
  957. defaults.update(kwargs)
  958. return super(DecimalField, self).formfield(**defaults)
  959. class EmailField(CharField):
  960. default_validators = [validators.validate_email]
  961. description = _("Email address")
  962. def __init__(self, *args, **kwargs):
  963. # max_length should be overridden to 254 characters to be fully
  964. # compliant with RFCs 3696 and 5321
  965. kwargs['max_length'] = kwargs.get('max_length', 75)
  966. CharField.__init__(self, *args, **kwargs)
  967. def deconstruct(self):
  968. name, path, args, kwargs = super(EmailField, self).deconstruct()
  969. # We do not exclude max_length if it matches default as we want to change
  970. # the default in future.
  971. return name, path, args, kwargs
  972. def formfield(self, **kwargs):
  973. # As with CharField, this will cause email validation to be performed
  974. # twice.
  975. defaults = {
  976. 'form_class': forms.EmailField,
  977. }
  978. defaults.update(kwargs)
  979. return super(EmailField, self).formfield(**defaults)
  980. class FilePathField(Field):
  981. description = _("File path")
  982. def __init__(self, verbose_name=None, name=None, path='', match=None,
  983. recursive=False, allow_files=True, allow_folders=False, **kwargs):
  984. self.path, self.match, self.recursive = path, match, recursive
  985. self.allow_files, self.allow_folders = allow_files, allow_folders
  986. kwargs['max_length'] = kwargs.get('max_length', 100)
  987. Field.__init__(self, verbose_name, name, **kwargs)
  988. def deconstruct(self):
  989. name, path, args, kwargs = super(FilePathField, self).deconstruct()
  990. if self.path != '':
  991. kwargs['path'] = self.path
  992. if self.match is not None:
  993. kwargs['match'] = self.match
  994. if self.recursive is not False:
  995. kwargs['recursive'] = self.recursive
  996. if self.allow_files is not True:
  997. kwargs['allow_files'] = self.allow_files
  998. if self.allow_folders is not False:
  999. kwargs['allow_folders'] = self.allow_folders
  1000. if kwargs.get("max_length", None) == 100:
  1001. del kwargs["max_length"]
  1002. return name, path, args, kwargs
  1003. def formfield(self, **kwargs):
  1004. defaults = {
  1005. 'path': self.path,
  1006. 'match': self.match,
  1007. 'recursive': self.recursive,
  1008. 'form_class': forms.FilePathField,
  1009. 'allow_files': self.allow_files,
  1010. 'allow_folders': self.allow_folders,
  1011. }
  1012. defaults.update(kwargs)
  1013. return super(FilePathField, self).formfield(**defaults)
  1014. def get_internal_type(self):
  1015. return "FilePathField"
  1016. class FloatField(Field):
  1017. empty_strings_allowed = False
  1018. default_error_messages = {
  1019. 'invalid': _("'%(value)s' value must be a float."),
  1020. }
  1021. description = _("Floating point number")
  1022. def get_prep_value(self, value):
  1023. if value is None:
  1024. return None
  1025. return float(value)
  1026. def get_internal_type(self):
  1027. return "FloatField"
  1028. def to_python(self, value):
  1029. if value is None:
  1030. return value
  1031. try:
  1032. return float(value)
  1033. except (TypeError, ValueError):
  1034. raise exceptions.ValidationError(
  1035. self.error_messages['invalid'],
  1036. code='invalid',
  1037. params={'value': value},
  1038. )
  1039. def formfield(self, **kwargs):
  1040. defaults = {'form_class': forms.FloatField}
  1041. defaults.update(kwargs)
  1042. return super(FloatField, self).formfield(**defaults)
  1043. class IntegerField(Field):
  1044. empty_strings_allowed = False
  1045. default_error_messages = {
  1046. 'invalid': _("'%(value)s' value must be an integer."),
  1047. }
  1048. description = _("Integer")
  1049. def get_prep_value(self, value):
  1050. if value is None:
  1051. return None
  1052. return int(value)
  1053. def get_prep_lookup(self, lookup_type, value):
  1054. if ((lookup_type == 'gte' or lookup_type == 'lt')
  1055. and isinstance(value, float)):
  1056. value = math.ceil(value)
  1057. return super(IntegerField, self).get_prep_lookup(lookup_type, value)
  1058. def get_internal_type(self):
  1059. return "IntegerField"
  1060. def to_python(self, value):
  1061. if value is None:
  1062. return value
  1063. try:
  1064. return int(value)
  1065. except (TypeError, ValueError):
  1066. raise exceptions.ValidationError(
  1067. self.error_messages['invalid'],
  1068. code='invalid',
  1069. params={'value': value},
  1070. )
  1071. def formfield(self, **kwargs):
  1072. defaults = {'form_class': forms.IntegerField}
  1073. defaults.update(kwargs)
  1074. return super(IntegerField, self).formfield(**defaults)
  1075. class BigIntegerField(IntegerField):
  1076. empty_strings_allowed = False
  1077. description = _("Big (8 byte) integer")
  1078. MAX_BIGINT = 9223372036854775807
  1079. def get_internal_type(self):
  1080. return "BigIntegerField"
  1081. def formfield(self, **kwargs):
  1082. defaults = {'min_value': -BigIntegerField.MAX_BIGINT - 1,
  1083. 'max_value': BigIntegerField.MAX_BIGINT}
  1084. defaults.update(kwargs)
  1085. return super(BigIntegerField, self).formfield(**defaults)
  1086. class IPAddressField(Field):
  1087. empty_strings_allowed = False
  1088. description = _("IPv4 address")
  1089. def __init__(self, *args, **kwargs):
  1090. kwargs['max_length'] = 15
  1091. Field.__init__(self, *args, **kwargs)
  1092. def deconstruct(self):
  1093. name, path, args, kwargs = super(IPAddressField, self).deconstruct()
  1094. del kwargs['max_length']
  1095. return name, path, args, kwargs
  1096. def get_internal_type(self):
  1097. return "IPAddressField"
  1098. def formfield(self, **kwargs):
  1099. defaults = {'form_class': forms.IPAddressField}
  1100. defaults.update(kwargs)
  1101. return super(IPAddressField, self).formfield(**defaults)
  1102. class GenericIPAddressField(Field):
  1103. empty_strings_allowed = True
  1104. description = _("IP address")
  1105. default_error_messages = {}
  1106. def __init__(self, verbose_name=None, name=None, protocol='both',
  1107. unpack_ipv4=False, *args, **kwargs):
  1108. self.unpack_ipv4 = unpack_ipv4
  1109. self.protocol = protocol
  1110. self.default_validators, invalid_error_message = \
  1111. validators.ip_address_validators(protocol, unpack_ipv4)
  1112. self.default_error_messages['invalid'] = invalid_error_message
  1113. kwargs['max_length'] = 39
  1114. Field.__init__(self, verbose_name, name, *args, **kwargs)
  1115. def deconstruct(self):
  1116. name, path, args, kwargs = super(GenericIPAddressField, self).deconstruct()
  1117. if self.unpack_ipv4 is not False:
  1118. kwargs['unpack_ipv4'] = self.unpack_ipv4
  1119. if self.protocol != "both":
  1120. kwargs['protocol'] = self.protocol
  1121. if kwargs.get("max_length", None) == 39:
  1122. del kwargs['max_length']
  1123. return name, path, args, kwargs
  1124. def get_internal_type(self):
  1125. return "GenericIPAddressField"
  1126. def to_python(self, value):
  1127. if value and ':' in value:
  1128. return clean_ipv6_address(value,
  1129. self.unpack_ipv4, self.error_messages['invalid'])
  1130. return value
  1131. def get_db_prep_value(self, value, connection, prepared=False):
  1132. if not prepared:
  1133. value = self.get_prep_value(value)
  1134. return value or None
  1135. def get_prep_value(self, value):
  1136. if value and ':' in value:
  1137. try:
  1138. return clean_ipv6_address(value, self.unpack_ipv4)
  1139. except exceptions.ValidationError:
  1140. pass
  1141. return value
  1142. def formfield(self, **kwargs):
  1143. defaults = {'form_class': forms.GenericIPAddressField}
  1144. defaults.update(kwargs)
  1145. return super(GenericIPAddressField, self).formfield(**defaults)
  1146. class NullBooleanField(Field):
  1147. empty_strings_allowed = False
  1148. default_error_messages = {
  1149. 'invalid': _("'%(value)s' value must be either None, True or False."),
  1150. }
  1151. description = _("Boolean (Either True, False or None)")
  1152. def __init__(self, *args, **kwargs):
  1153. kwargs['null'] = True
  1154. kwargs['blank'] = True
  1155. Field.__init__(self, *args, **kwargs)
  1156. def deconstruct(self):
  1157. name, path, args, kwargs = super(NullBooleanField, self).deconstruct()
  1158. del kwargs['null']
  1159. del kwargs['blank']
  1160. return name, path, args, kwargs
  1161. def get_internal_type(self):
  1162. return "NullBooleanField"
  1163. def to_python(self, value):
  1164. if value is None:
  1165. return None
  1166. if value in (True, False):
  1167. return bool(value)
  1168. if value in ('None',):
  1169. return None
  1170. if value in ('t', 'True', '1'):
  1171. return True
  1172. if value in ('f', 'False', '0'):
  1173. return False
  1174. raise exceptions.ValidationError(
  1175. self.error_messages['invalid'],
  1176. code='invalid',
  1177. params={'value': value},
  1178. )
  1179. def get_prep_lookup(self, lookup_type, value):
  1180. # Special-case handling for filters coming from a Web request (e.g. the
  1181. # admin interface). Only works for scalar values (not lists). If you're
  1182. # passing in a list, you might as well make things the right type when
  1183. # constructing the list.
  1184. if value in ('1', '0'):
  1185. value = bool(int(value))
  1186. return super(NullBooleanField, self).get_prep_lookup(lookup_type,
  1187. value)
  1188. def get_prep_value(self, value):
  1189. if value is None:
  1190. return None
  1191. return bool(value)
  1192. def formfield(self, **kwargs):
  1193. defaults = {
  1194. 'form_class': forms.NullBooleanField,
  1195. 'required': not self.blank,
  1196. 'label': capfirst(self.verbose_name),
  1197. 'help_text': self.help_text}
  1198. defaults.update(kwargs)
  1199. return super(NullBooleanField, self).formfield(**defaults)
  1200. class PositiveIntegerField(IntegerField):
  1201. description = _("Positive integer")
  1202. def get_internal_type(self):
  1203. return "PositiveIntegerField"
  1204. def formfield(self, **kwargs):
  1205. defaults = {'min_value': 0}
  1206. defaults.update(kwargs)
  1207. return super(PositiveIntegerField, self).formfield(**defaults)
  1208. class PositiveSmallIntegerField(IntegerField):
  1209. description = _("Positive small integer")
  1210. def get_internal_type(self):
  1211. return "PositiveSmallIntegerField"
  1212. def formfield(self, **kwargs):
  1213. defaults = {'min_value': 0}
  1214. defaults.update(kwargs)
  1215. return super(PositiveSmallIntegerField, self).formfield(**defaults)
  1216. class SlugField(CharField):
  1217. default_validators = [validators.validate_slug]
  1218. description = _("Slug (up to %(max_length)s)")
  1219. def __init__(self, *args, **kwargs):
  1220. kwargs['max_length'] = kwargs.get('max_length', 50)
  1221. # Set db_index=True unless it's been set manually.
  1222. if 'db_index' not in kwargs:
  1223. kwargs['db_index'] = True
  1224. super(SlugField, self).__init__(*args, **kwargs)
  1225. def deconstruct(self):
  1226. name, path, args, kwargs = super(SlugField, self).deconstruct()
  1227. if kwargs.get("max_length", None) == 50:
  1228. del kwargs['max_length']
  1229. if self.db_index is False:
  1230. kwargs['db_index'] = False
  1231. else:
  1232. del kwargs['db_index']
  1233. return name, path, args, kwargs
  1234. def get_internal_type(self):
  1235. return "SlugField"
  1236. def formfield(self, **kwargs):
  1237. defaults = {'form_class': forms.SlugField}
  1238. defaults.update(kwargs)
  1239. return super(SlugField, self).formfield(**defaults)
  1240. class SmallIntegerField(IntegerField):
  1241. description = _("Small integer")
  1242. def get_internal_type(self):
  1243. return "SmallIntegerField"
  1244. class TextField(Field):
  1245. description = _("Text")
  1246. def get_internal_type(self):
  1247. return "TextField"
  1248. def get_prep_value(self, value):
  1249. if isinstance(value, six.string_types) or value is None:
  1250. return value
  1251. return smart_text(value)
  1252. def formfield(self, **kwargs):
  1253. defaults = {'widget': forms.Textarea}
  1254. defaults.update(kwargs)
  1255. return super(TextField, self).formfield(**defaults)
  1256. class TimeField(Field):
  1257. empty_strings_allowed = False
  1258. default_error_messages = {
  1259. 'invalid': _("'%(value)s' value has an invalid format. It must be in "
  1260. "HH:MM[:ss[.uuuuuu]] format."),
  1261. 'invalid_time': _("'%(value)s' value has the correct format "
  1262. "(HH:MM[:ss[.uuuuuu]]) but it is an invalid time."),
  1263. }
  1264. description = _("Time")
  1265. def __init__(self, verbose_name=None, name=None, auto_now=False,
  1266. auto_now_add=False, **kwargs):
  1267. self.auto_now, self.auto_now_add = auto_now, auto_now_add
  1268. if auto_now or auto_now_add:
  1269. kwargs['editable'] = False
  1270. kwargs['blank'] = True
  1271. Field.__init__(self, verbose_name, name, **kwargs)
  1272. def deconstruct(self):
  1273. name, path, args, kwargs = super(TimeField, self).deconstruct()
  1274. if self.auto_now is not False:
  1275. kwargs["auto_now"] = self.auto_now
  1276. if self.auto_now_add is not False:
  1277. kwargs["auto_now_add"] = self.auto_now_add
  1278. return name, path, args, kwargs
  1279. def get_internal_type(self):
  1280. return "TimeField"
  1281. def to_python(self, value):
  1282. if value is None:
  1283. return None
  1284. if isinstance(value, datetime.time):
  1285. return value
  1286. if isinstance(value, datetime.datetime):
  1287. # Not usually a good idea to pass in a datetime here (it loses
  1288. # information), but this can be a side-effect of interacting with a
  1289. # database backend (e.g. Oracle), so we'll be accommodating.
  1290. return value.time()
  1291. try:
  1292. parsed = parse_time(value)
  1293. if parsed is not None:
  1294. return parsed
  1295. except ValueError:
  1296. raise exceptions.ValidationError(
  1297. self.error_messages['invalid_time'],
  1298. code='invalid_time',
  1299. params={'value': value},
  1300. )
  1301. raise exceptions.ValidationError(
  1302. self.error_messages['invalid'],
  1303. code='invalid',
  1304. params={'value': value},
  1305. )
  1306. def pre_save(self, model_instance, add):
  1307. if self.auto_now or (self.auto_now_add and add):
  1308. value = datetime.datetime.now().time()
  1309. setattr(model_instance, self.attname, value)
  1310. return value
  1311. else:
  1312. return super(TimeField, self).pre_save(model_instance, add)
  1313. def get_prep_value(self, value):
  1314. return self.to_python(value)
  1315. def get_db_prep_value(self, value, connection, prepared=False):
  1316. # Casts times into the format expected by the backend
  1317. if not prepared:
  1318. value = self.get_prep_value(value)
  1319. return connection.ops.value_to_db_time(value)
  1320. def value_to_string(self, obj):
  1321. val = self._get_val_from_obj(obj)
  1322. return '' if val is None else val.isoformat()
  1323. def formfield(self, **kwargs):
  1324. defaults = {'form_class': forms.TimeField}
  1325. defaults.update(kwargs)
  1326. return super(TimeField, self).formfield(**defaults)
  1327. class URLField(CharField):
  1328. default_validators = [validators.URLValidator()]
  1329. description = _("URL")
  1330. def __init__(self, verbose_name=None, name=None, **kwargs):
  1331. kwargs['max_length'] = kwargs.get('max_length', 200)
  1332. CharField.__init__(self, verbose_name, name, **kwargs)
  1333. def deconstruct(self):
  1334. name, path, args, kwargs = super(URLField, self).deconstruct()
  1335. if kwargs.get("max_length", None) == 200:
  1336. del kwargs['max_length']
  1337. return name, path, args, kwargs
  1338. def formfield(self, **kwargs):
  1339. # As with CharField, this will cause URL validation to be performed
  1340. # twice.
  1341. defaults = {
  1342. 'form_class': forms.URLField,
  1343. }
  1344. defaults.update(kwargs)
  1345. return super(URLField, self).formfield(**defaults)
  1346. class BinaryField(Field):
  1347. description = _("Raw binary data")
  1348. empty_values = [None, b'']
  1349. def __init__(self, *args, **kwargs):
  1350. kwargs['editable'] = False
  1351. super(BinaryField, self).__init__(*args, **kwargs)
  1352. if self.max_length is not None:
  1353. self.validators.append(validators.MaxLengthValidator(self.max_length))
  1354. def get_internal_type(self):
  1355. return "BinaryField"
  1356. def get_default(self):
  1357. if self.has_default() and not callable(self.default):
  1358. return self.default
  1359. default = super(BinaryField, self).get_default()
  1360. if default == '':
  1361. return b''
  1362. return default
  1363. def get_db_prep_value(self, value, connection, prepared=False):
  1364. value = super(BinaryField, self
  1365. ).get_db_prep_value(value, connection, prepared)
  1366. if value is not None:
  1367. return connection.Database.Binary(value)
  1368. return value
  1369. def value_to_string(self, obj):
  1370. """Binary data is serialized as base64"""
  1371. return b64encode(force_bytes(self._get_val_from_obj(obj))).decode('ascii')
  1372. def to_python(self, value):
  1373. # If it's a string, it should be base64-encoded data
  1374. if isinstance(value, six.text_type):
  1375. return six.memoryview(b64decode(force_bytes(value)))
  1376. return value