widgets.py 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601
  1. """
  2. Form Widget classes specific to the Django admin site.
  3. """
  4. import copy
  5. import json
  6. from django import forms
  7. from django.conf import settings
  8. from django.core.exceptions import ValidationError
  9. from django.core.validators import URLValidator
  10. from django.db.models import CASCADE, UUIDField
  11. from django.urls import reverse
  12. from django.urls.exceptions import NoReverseMatch
  13. from django.utils.html import smart_urlquote
  14. from django.utils.http import urlencode
  15. from django.utils.text import Truncator
  16. from django.utils.translation import get_language
  17. from django.utils.translation import gettext as _
  18. class FilteredSelectMultiple(forms.SelectMultiple):
  19. """
  20. A SelectMultiple with a JavaScript filter interface.
  21. Note that the resulting JavaScript assumes that the jsi18n
  22. catalog has been loaded in the page
  23. """
  24. class Media:
  25. js = [
  26. "admin/js/core.js",
  27. "admin/js/SelectBox.js",
  28. "admin/js/SelectFilter2.js",
  29. ]
  30. def __init__(self, verbose_name, is_stacked, attrs=None, choices=()):
  31. self.verbose_name = verbose_name
  32. self.is_stacked = is_stacked
  33. super().__init__(attrs, choices)
  34. def get_context(self, name, value, attrs):
  35. context = super().get_context(name, value, attrs)
  36. context["widget"]["attrs"]["class"] = "selectfilter"
  37. if self.is_stacked:
  38. context["widget"]["attrs"]["class"] += "stacked"
  39. context["widget"]["attrs"]["data-field-name"] = self.verbose_name
  40. context["widget"]["attrs"]["data-is-stacked"] = int(self.is_stacked)
  41. return context
  42. class BaseAdminDateWidget(forms.DateInput):
  43. class Media:
  44. js = [
  45. "admin/js/calendar.js",
  46. "admin/js/admin/DateTimeShortcuts.js",
  47. ]
  48. def __init__(self, attrs=None, format=None):
  49. attrs = {"class": "vDateField", "size": "10", **(attrs or {})}
  50. super().__init__(attrs=attrs, format=format)
  51. class AdminDateWidget(BaseAdminDateWidget):
  52. template_name = "admin/widgets/date.html"
  53. class BaseAdminTimeWidget(forms.TimeInput):
  54. class Media:
  55. js = [
  56. "admin/js/calendar.js",
  57. "admin/js/admin/DateTimeShortcuts.js",
  58. ]
  59. def __init__(self, attrs=None, format=None):
  60. attrs = {"class": "vTimeField", "size": "8", **(attrs or {})}
  61. super().__init__(attrs=attrs, format=format)
  62. class AdminTimeWidget(BaseAdminTimeWidget):
  63. template_name = "admin/widgets/time.html"
  64. class AdminSplitDateTime(forms.SplitDateTimeWidget):
  65. """
  66. A SplitDateTime Widget that has some admin-specific styling.
  67. """
  68. template_name = "admin/widgets/split_datetime.html"
  69. def __init__(self, attrs=None):
  70. widgets = [BaseAdminDateWidget, BaseAdminTimeWidget]
  71. # Note that we're calling MultiWidget, not SplitDateTimeWidget, because
  72. # we want to define widgets.
  73. forms.MultiWidget.__init__(self, widgets, attrs)
  74. def get_context(self, name, value, attrs):
  75. context = super().get_context(name, value, attrs)
  76. context["date_label"] = _("Date:")
  77. context["time_label"] = _("Time:")
  78. return context
  79. class AdminRadioSelect(forms.RadioSelect):
  80. template_name = "admin/widgets/radio.html"
  81. class AdminFileWidget(forms.ClearableFileInput):
  82. template_name = "admin/widgets/clearable_file_input.html"
  83. def url_params_from_lookup_dict(lookups):
  84. """
  85. Convert the type of lookups specified in a ForeignKey limit_choices_to
  86. attribute to a dictionary of query parameters
  87. """
  88. params = {}
  89. if lookups and hasattr(lookups, "items"):
  90. for k, v in lookups.items():
  91. if callable(v):
  92. v = v()
  93. if isinstance(v, (tuple, list)):
  94. v = ",".join(str(x) for x in v)
  95. elif isinstance(v, bool):
  96. v = ("0", "1")[v]
  97. else:
  98. v = str(v)
  99. params[k] = v
  100. return params
  101. class ForeignKeyRawIdWidget(forms.TextInput):
  102. """
  103. A Widget for displaying ForeignKeys in the "raw_id" interface rather than
  104. in a <select> box.
  105. """
  106. template_name = "admin/widgets/foreign_key_raw_id.html"
  107. def __init__(self, rel, admin_site, attrs=None, using=None):
  108. self.rel = rel
  109. self.admin_site = admin_site
  110. self.db = using
  111. super().__init__(attrs)
  112. def get_context(self, name, value, attrs):
  113. context = super().get_context(name, value, attrs)
  114. rel_to = self.rel.model
  115. if self.admin_site.is_registered(rel_to):
  116. # The related object is registered with the same AdminSite
  117. related_url = reverse(
  118. "admin:%s_%s_changelist"
  119. % (
  120. rel_to._meta.app_label,
  121. rel_to._meta.model_name,
  122. ),
  123. current_app=self.admin_site.name,
  124. )
  125. params = self.url_parameters()
  126. if params:
  127. related_url += "?" + urlencode(params)
  128. context["related_url"] = related_url
  129. context["link_title"] = _("Lookup")
  130. # The JavaScript code looks for this class.
  131. css_class = "vForeignKeyRawIdAdminField"
  132. if isinstance(self.rel.get_related_field(), UUIDField):
  133. css_class += " vUUIDField"
  134. context["widget"]["attrs"].setdefault("class", css_class)
  135. else:
  136. context["related_url"] = None
  137. if context["widget"]["value"]:
  138. context["link_label"], context["link_url"] = self.label_and_url_for_value(
  139. value
  140. )
  141. else:
  142. context["link_label"] = None
  143. return context
  144. def base_url_parameters(self):
  145. limit_choices_to = self.rel.limit_choices_to
  146. if callable(limit_choices_to):
  147. limit_choices_to = limit_choices_to()
  148. return url_params_from_lookup_dict(limit_choices_to)
  149. def url_parameters(self):
  150. from django.contrib.admin.views.main import TO_FIELD_VAR
  151. params = self.base_url_parameters()
  152. params.update({TO_FIELD_VAR: self.rel.get_related_field().name})
  153. return params
  154. def label_and_url_for_value(self, value):
  155. key = self.rel.get_related_field().name
  156. try:
  157. obj = self.rel.model._default_manager.using(self.db).get(**{key: value})
  158. except (ValueError, self.rel.model.DoesNotExist, ValidationError):
  159. return "", ""
  160. try:
  161. url = reverse(
  162. "%s:%s_%s_change"
  163. % (
  164. self.admin_site.name,
  165. obj._meta.app_label,
  166. obj._meta.object_name.lower(),
  167. ),
  168. args=(obj.pk,),
  169. )
  170. except NoReverseMatch:
  171. url = "" # Admin not registered for target model.
  172. return Truncator(obj).words(14), url
  173. class ManyToManyRawIdWidget(ForeignKeyRawIdWidget):
  174. """
  175. A Widget for displaying ManyToMany ids in the "raw_id" interface rather than
  176. in a <select multiple> box.
  177. """
  178. template_name = "admin/widgets/many_to_many_raw_id.html"
  179. def get_context(self, name, value, attrs):
  180. context = super().get_context(name, value, attrs)
  181. if self.admin_site.is_registered(self.rel.model):
  182. # The related object is registered with the same AdminSite
  183. context["widget"]["attrs"]["class"] = "vManyToManyRawIdAdminField"
  184. return context
  185. def url_parameters(self):
  186. return self.base_url_parameters()
  187. def label_and_url_for_value(self, value):
  188. return "", ""
  189. def value_from_datadict(self, data, files, name):
  190. value = data.get(name)
  191. if value:
  192. return value.split(",")
  193. def format_value(self, value):
  194. return ",".join(str(v) for v in value) if value else ""
  195. class RelatedFieldWidgetWrapper(forms.Widget):
  196. """
  197. This class is a wrapper to a given widget to add the add icon for the
  198. admin interface.
  199. """
  200. template_name = "admin/widgets/related_widget_wrapper.html"
  201. def __init__(
  202. self,
  203. widget,
  204. rel,
  205. admin_site,
  206. can_add_related=None,
  207. can_change_related=False,
  208. can_delete_related=False,
  209. can_view_related=False,
  210. ):
  211. self.needs_multipart_form = widget.needs_multipart_form
  212. self.attrs = widget.attrs
  213. self.widget = widget
  214. self.rel = rel
  215. # Backwards compatible check for whether a user can add related
  216. # objects.
  217. if can_add_related is None:
  218. can_add_related = admin_site.is_registered(rel.model)
  219. self.can_add_related = can_add_related
  220. # XXX: The UX does not support multiple selected values.
  221. multiple = getattr(widget, "allow_multiple_selected", False)
  222. self.can_change_related = not multiple and can_change_related
  223. # XXX: The deletion UX can be confusing when dealing with cascading deletion.
  224. cascade = getattr(rel, "on_delete", None) is CASCADE
  225. self.can_delete_related = not multiple and not cascade and can_delete_related
  226. self.can_view_related = not multiple and can_view_related
  227. # so we can check if the related object is registered with this AdminSite
  228. self.admin_site = admin_site
  229. def __deepcopy__(self, memo):
  230. obj = copy.copy(self)
  231. obj.widget = copy.deepcopy(self.widget, memo)
  232. obj.attrs = self.widget.attrs
  233. memo[id(self)] = obj
  234. return obj
  235. @property
  236. def is_hidden(self):
  237. return self.widget.is_hidden
  238. @property
  239. def media(self):
  240. return self.widget.media
  241. @property
  242. def choices(self):
  243. return self.widget.choices
  244. @choices.setter
  245. def choices(self, value):
  246. self.widget.choices = value
  247. def get_related_url(self, info, action, *args):
  248. return reverse(
  249. "admin:%s_%s_%s" % (info + (action,)),
  250. current_app=self.admin_site.name,
  251. args=args,
  252. )
  253. def get_context(self, name, value, attrs):
  254. from django.contrib.admin.views.main import IS_POPUP_VAR, TO_FIELD_VAR
  255. rel_opts = self.rel.model._meta
  256. info = (rel_opts.app_label, rel_opts.model_name)
  257. related_field_name = self.rel.get_related_field().name
  258. url_params = "&".join(
  259. "%s=%s" % param
  260. for param in [
  261. (TO_FIELD_VAR, related_field_name),
  262. (IS_POPUP_VAR, 1),
  263. ]
  264. )
  265. context = {
  266. "rendered_widget": self.widget.render(name, value, attrs),
  267. "is_hidden": self.is_hidden,
  268. "name": name,
  269. "url_params": url_params,
  270. "model": rel_opts.verbose_name,
  271. "can_add_related": self.can_add_related,
  272. "can_change_related": self.can_change_related,
  273. "can_delete_related": self.can_delete_related,
  274. "can_view_related": self.can_view_related,
  275. "model_has_limit_choices_to": self.rel.limit_choices_to,
  276. }
  277. if self.can_add_related:
  278. context["add_related_url"] = self.get_related_url(info, "add")
  279. if self.can_delete_related:
  280. context["delete_related_template_url"] = self.get_related_url(
  281. info, "delete", "__fk__"
  282. )
  283. if self.can_view_related or self.can_change_related:
  284. context["view_related_url_params"] = f"{TO_FIELD_VAR}={related_field_name}"
  285. context["change_related_template_url"] = self.get_related_url(
  286. info, "change", "__fk__"
  287. )
  288. return context
  289. def value_from_datadict(self, data, files, name):
  290. return self.widget.value_from_datadict(data, files, name)
  291. def value_omitted_from_data(self, data, files, name):
  292. return self.widget.value_omitted_from_data(data, files, name)
  293. def id_for_label(self, id_):
  294. return self.widget.id_for_label(id_)
  295. class AdminTextareaWidget(forms.Textarea):
  296. def __init__(self, attrs=None):
  297. super().__init__(attrs={"class": "vLargeTextField", **(attrs or {})})
  298. class AdminTextInputWidget(forms.TextInput):
  299. def __init__(self, attrs=None):
  300. super().__init__(attrs={"class": "vTextField", **(attrs or {})})
  301. class AdminEmailInputWidget(forms.EmailInput):
  302. def __init__(self, attrs=None):
  303. super().__init__(attrs={"class": "vTextField", **(attrs or {})})
  304. class AdminURLFieldWidget(forms.URLInput):
  305. template_name = "admin/widgets/url.html"
  306. def __init__(self, attrs=None, validator_class=URLValidator):
  307. super().__init__(attrs={"class": "vURLField", **(attrs or {})})
  308. self.validator = validator_class()
  309. def get_context(self, name, value, attrs):
  310. try:
  311. self.validator(value if value else "")
  312. url_valid = True
  313. except ValidationError:
  314. url_valid = False
  315. context = super().get_context(name, value, attrs)
  316. context["current_label"] = _("Currently:")
  317. context["change_label"] = _("Change:")
  318. context["widget"]["href"] = (
  319. smart_urlquote(context["widget"]["value"]) if value else ""
  320. )
  321. context["url_valid"] = url_valid
  322. return context
  323. class AdminIntegerFieldWidget(forms.NumberInput):
  324. class_name = "vIntegerField"
  325. def __init__(self, attrs=None):
  326. super().__init__(attrs={"class": self.class_name, **(attrs or {})})
  327. class AdminBigIntegerFieldWidget(AdminIntegerFieldWidget):
  328. class_name = "vBigIntegerField"
  329. class AdminUUIDInputWidget(forms.TextInput):
  330. def __init__(self, attrs=None):
  331. super().__init__(attrs={"class": "vUUIDField", **(attrs or {})})
  332. # Mapping of lowercase language codes [returned by Django's get_language()] to
  333. # language codes supported by select2.
  334. # See django/contrib/admin/static/admin/js/vendor/select2/i18n/*
  335. SELECT2_TRANSLATIONS = {
  336. x.lower(): x
  337. for x in [
  338. "ar",
  339. "az",
  340. "bg",
  341. "ca",
  342. "cs",
  343. "da",
  344. "de",
  345. "el",
  346. "en",
  347. "es",
  348. "et",
  349. "eu",
  350. "fa",
  351. "fi",
  352. "fr",
  353. "gl",
  354. "he",
  355. "hi",
  356. "hr",
  357. "hu",
  358. "id",
  359. "is",
  360. "it",
  361. "ja",
  362. "km",
  363. "ko",
  364. "lt",
  365. "lv",
  366. "mk",
  367. "ms",
  368. "nb",
  369. "nl",
  370. "pl",
  371. "pt-BR",
  372. "pt",
  373. "ro",
  374. "ru",
  375. "sk",
  376. "sr-Cyrl",
  377. "sr",
  378. "sv",
  379. "th",
  380. "tr",
  381. "uk",
  382. "vi",
  383. ]
  384. }
  385. SELECT2_TRANSLATIONS.update({"zh-hans": "zh-CN", "zh-hant": "zh-TW"})
  386. def get_select2_language():
  387. lang_code = get_language()
  388. supported_code = SELECT2_TRANSLATIONS.get(lang_code)
  389. if supported_code is None and lang_code is not None:
  390. # If 'zh-hant-tw' is not supported, try subsequent language codes i.e.
  391. # 'zh-hant' and 'zh'.
  392. i = None
  393. while (i := lang_code.rfind("-", 0, i)) > -1:
  394. if supported_code := SELECT2_TRANSLATIONS.get(lang_code[:i]):
  395. return supported_code
  396. return supported_code
  397. class AutocompleteMixin:
  398. """
  399. Select widget mixin that loads options from AutocompleteJsonView via AJAX.
  400. Renders the necessary data attributes for select2 and adds the static form
  401. media.
  402. """
  403. url_name = "%s:autocomplete"
  404. def __init__(self, field, admin_site, attrs=None, choices=(), using=None):
  405. self.field = field
  406. self.admin_site = admin_site
  407. self.db = using
  408. self.choices = choices
  409. self.attrs = {} if attrs is None else attrs.copy()
  410. self.i18n_name = get_select2_language()
  411. def get_url(self):
  412. return reverse(self.url_name % self.admin_site.name)
  413. def build_attrs(self, base_attrs, extra_attrs=None):
  414. """
  415. Set select2's AJAX attributes.
  416. Attributes can be set using the html5 data attribute.
  417. Nested attributes require a double dash as per
  418. https://select2.org/configuration/data-attributes#nested-subkey-options
  419. """
  420. attrs = super().build_attrs(base_attrs, extra_attrs=extra_attrs)
  421. attrs.setdefault("class", "")
  422. attrs.update(
  423. {
  424. "data-ajax--cache": "true",
  425. "data-ajax--delay": 250,
  426. "data-ajax--type": "GET",
  427. "data-ajax--url": self.get_url(),
  428. "data-app-label": self.field.model._meta.app_label,
  429. "data-model-name": self.field.model._meta.model_name,
  430. "data-field-name": self.field.name,
  431. "data-theme": "admin-autocomplete",
  432. "data-allow-clear": json.dumps(not self.is_required),
  433. "data-placeholder": "", # Allows clearing of the input.
  434. "lang": self.i18n_name,
  435. "class": attrs["class"]
  436. + (" " if attrs["class"] else "")
  437. + "admin-autocomplete",
  438. }
  439. )
  440. return attrs
  441. def optgroups(self, name, value, attr=None):
  442. """Return selected options based on the ModelChoiceIterator."""
  443. default = (None, [], 0)
  444. groups = [default]
  445. has_selected = False
  446. selected_choices = {
  447. str(v) for v in value if str(v) not in self.choices.field.empty_values
  448. }
  449. if not self.is_required and not self.allow_multiple_selected:
  450. default[1].append(self.create_option(name, "", "", False, 0))
  451. remote_model_opts = self.field.remote_field.model._meta
  452. to_field_name = getattr(
  453. self.field.remote_field, "field_name", remote_model_opts.pk.attname
  454. )
  455. to_field_name = remote_model_opts.get_field(to_field_name).attname
  456. choices = (
  457. (getattr(obj, to_field_name), self.choices.field.label_from_instance(obj))
  458. for obj in self.choices.queryset.using(self.db).filter(
  459. **{"%s__in" % to_field_name: selected_choices}
  460. )
  461. )
  462. for option_value, option_label in choices:
  463. selected = str(option_value) in value and (
  464. has_selected is False or self.allow_multiple_selected
  465. )
  466. has_selected |= selected
  467. index = len(default[1])
  468. subgroup = default[1]
  469. subgroup.append(
  470. self.create_option(
  471. name, option_value, option_label, selected_choices, index
  472. )
  473. )
  474. return groups
  475. @property
  476. def media(self):
  477. extra = "" if settings.DEBUG else ".min"
  478. i18n_file = (
  479. ("admin/js/vendor/select2/i18n/%s.js" % self.i18n_name,)
  480. if self.i18n_name
  481. else ()
  482. )
  483. return forms.Media(
  484. js=(
  485. "admin/js/vendor/jquery/jquery%s.js" % extra,
  486. "admin/js/vendor/select2/select2.full%s.js" % extra,
  487. )
  488. + i18n_file
  489. + (
  490. "admin/js/jquery.init.js",
  491. "admin/js/autocomplete.js",
  492. ),
  493. css={
  494. "screen": (
  495. "admin/css/vendor/select2/select2%s.css" % extra,
  496. "admin/css/autocomplete.css",
  497. ),
  498. },
  499. )
  500. class AutocompleteSelect(AutocompleteMixin, forms.Select):
  501. pass
  502. class AutocompleteSelectMultiple(AutocompleteMixin, forms.SelectMultiple):
  503. pass