edit.py 8.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274
  1. from django.core.exceptions import ImproperlyConfigured
  2. from django.forms import Form
  3. from django.forms import models as model_forms
  4. from django.http import HttpResponseRedirect
  5. from django.views.generic.base import ContextMixin, TemplateResponseMixin, View
  6. from django.views.generic.detail import (
  7. BaseDetailView,
  8. SingleObjectMixin,
  9. SingleObjectTemplateResponseMixin,
  10. )
  11. class FormMixin(ContextMixin):
  12. """Provide a way to show and handle a form in a request."""
  13. initial = {}
  14. form_class = None
  15. success_url = None
  16. prefix = None
  17. def get_initial(self):
  18. """Return the initial data to use for forms on this view."""
  19. return self.initial.copy()
  20. def get_prefix(self):
  21. """Return the prefix to use for forms."""
  22. return self.prefix
  23. def get_form_class(self):
  24. """Return the form class to use."""
  25. return self.form_class
  26. def get_form(self, form_class=None):
  27. """Return an instance of the form to be used in this view."""
  28. if form_class is None:
  29. form_class = self.get_form_class()
  30. return form_class(**self.get_form_kwargs())
  31. def get_form_kwargs(self):
  32. """Return the keyword arguments for instantiating the form."""
  33. kwargs = {
  34. "initial": self.get_initial(),
  35. "prefix": self.get_prefix(),
  36. }
  37. if self.request.method in ("POST", "PUT"):
  38. kwargs.update(
  39. {
  40. "data": self.request.POST,
  41. "files": self.request.FILES,
  42. }
  43. )
  44. return kwargs
  45. def get_success_url(self):
  46. """Return the URL to redirect to after processing a valid form."""
  47. if not self.success_url:
  48. raise ImproperlyConfigured("No URL to redirect to. Provide a success_url.")
  49. return str(self.success_url) # success_url may be lazy
  50. def form_valid(self, form):
  51. """If the form is valid, redirect to the supplied URL."""
  52. return HttpResponseRedirect(self.get_success_url())
  53. def form_invalid(self, form):
  54. """If the form is invalid, render the invalid form."""
  55. return self.render_to_response(self.get_context_data(form=form))
  56. def get_context_data(self, **kwargs):
  57. """Insert the form into the context dict."""
  58. if "form" not in kwargs:
  59. kwargs["form"] = self.get_form()
  60. return super().get_context_data(**kwargs)
  61. class ModelFormMixin(FormMixin, SingleObjectMixin):
  62. """Provide a way to show and handle a ModelForm in a request."""
  63. fields = None
  64. def get_form_class(self):
  65. """Return the form class to use in this view."""
  66. if self.fields is not None and self.form_class:
  67. raise ImproperlyConfigured(
  68. "Specifying both 'fields' and 'form_class' is not permitted."
  69. )
  70. if self.form_class:
  71. return self.form_class
  72. else:
  73. if self.model is not None:
  74. # If a model has been explicitly provided, use it
  75. model = self.model
  76. elif getattr(self, "object", None) is not None:
  77. # If this view is operating on a single object, use
  78. # the class of that object
  79. model = self.object.__class__
  80. else:
  81. # Try to get a queryset and extract the model class
  82. # from that
  83. model = self.get_queryset().model
  84. if self.fields is None:
  85. raise ImproperlyConfigured(
  86. "Using ModelFormMixin (base class of %s) without "
  87. "the 'fields' attribute is prohibited." % self.__class__.__name__
  88. )
  89. return model_forms.modelform_factory(model, fields=self.fields)
  90. def get_form_kwargs(self):
  91. """Return the keyword arguments for instantiating the form."""
  92. kwargs = super().get_form_kwargs()
  93. if hasattr(self, "object"):
  94. kwargs.update({"instance": self.object})
  95. return kwargs
  96. def get_success_url(self):
  97. """Return the URL to redirect to after processing a valid form."""
  98. if self.success_url:
  99. url = self.success_url.format(**self.object.__dict__)
  100. else:
  101. try:
  102. url = self.object.get_absolute_url()
  103. except AttributeError:
  104. raise ImproperlyConfigured(
  105. "No URL to redirect to. Either provide a url or define"
  106. " a get_absolute_url method on the Model."
  107. )
  108. return url
  109. def form_valid(self, form):
  110. """If the form is valid, save the associated model."""
  111. self.object = form.save()
  112. return super().form_valid(form)
  113. class ProcessFormView(View):
  114. """Render a form on GET and processes it on POST."""
  115. def get(self, request, *args, **kwargs):
  116. """Handle GET requests: instantiate a blank version of the form."""
  117. return self.render_to_response(self.get_context_data())
  118. def post(self, request, *args, **kwargs):
  119. """
  120. Handle POST requests: instantiate a form instance with the passed
  121. POST variables and then check if it's valid.
  122. """
  123. form = self.get_form()
  124. if form.is_valid():
  125. return self.form_valid(form)
  126. else:
  127. return self.form_invalid(form)
  128. # PUT is a valid HTTP verb for creating (with a known URL) or editing an
  129. # object, note that browsers only support POST for now.
  130. def put(self, *args, **kwargs):
  131. return self.post(*args, **kwargs)
  132. class BaseFormView(FormMixin, ProcessFormView):
  133. """A base view for displaying a form."""
  134. class FormView(TemplateResponseMixin, BaseFormView):
  135. """A view for displaying a form and rendering a template response."""
  136. class BaseCreateView(ModelFormMixin, ProcessFormView):
  137. """
  138. Base view for creating a new object instance.
  139. Using this base class requires subclassing to provide a response mixin.
  140. """
  141. def get(self, request, *args, **kwargs):
  142. self.object = None
  143. return super().get(request, *args, **kwargs)
  144. def post(self, request, *args, **kwargs):
  145. self.object = None
  146. return super().post(request, *args, **kwargs)
  147. class CreateView(SingleObjectTemplateResponseMixin, BaseCreateView):
  148. """
  149. View for creating a new object, with a response rendered by a template.
  150. """
  151. template_name_suffix = "_form"
  152. class BaseUpdateView(ModelFormMixin, ProcessFormView):
  153. """
  154. Base view for updating an existing object.
  155. Using this base class requires subclassing to provide a response mixin.
  156. """
  157. def get(self, request, *args, **kwargs):
  158. self.object = self.get_object()
  159. return super().get(request, *args, **kwargs)
  160. def post(self, request, *args, **kwargs):
  161. self.object = self.get_object()
  162. return super().post(request, *args, **kwargs)
  163. class UpdateView(SingleObjectTemplateResponseMixin, BaseUpdateView):
  164. """View for updating an object, with a response rendered by a template."""
  165. template_name_suffix = "_form"
  166. class DeletionMixin:
  167. """Provide the ability to delete objects."""
  168. success_url = None
  169. def delete(self, request, *args, **kwargs):
  170. """
  171. Call the delete() method on the fetched object and then redirect to the
  172. success URL.
  173. """
  174. self.object = self.get_object()
  175. success_url = self.get_success_url()
  176. self.object.delete()
  177. return HttpResponseRedirect(success_url)
  178. # Add support for browsers which only accept GET and POST for now.
  179. def post(self, request, *args, **kwargs):
  180. return self.delete(request, *args, **kwargs)
  181. def get_success_url(self):
  182. if self.success_url:
  183. return self.success_url.format(**self.object.__dict__)
  184. else:
  185. raise ImproperlyConfigured("No URL to redirect to. Provide a success_url.")
  186. class BaseDeleteView(DeletionMixin, FormMixin, BaseDetailView):
  187. """
  188. Base view for deleting an object.
  189. Using this base class requires subclassing to provide a response mixin.
  190. """
  191. form_class = Form
  192. def post(self, request, *args, **kwargs):
  193. # Set self.object before the usual form processing flow.
  194. # Inlined because having DeletionMixin as the first base, for
  195. # get_success_url(), makes leveraging super() with ProcessFormView
  196. # overly complex.
  197. self.object = self.get_object()
  198. form = self.get_form()
  199. if form.is_valid():
  200. return self.form_valid(form)
  201. else:
  202. return self.form_invalid(form)
  203. def form_valid(self, form):
  204. success_url = self.get_success_url()
  205. self.object.delete()
  206. return HttpResponseRedirect(success_url)
  207. class DeleteView(SingleObjectTemplateResponseMixin, BaseDeleteView):
  208. """
  209. View for deleting an object retrieved with self.get_object(), with a
  210. response rendered by a template.
  211. """
  212. template_name_suffix = "_confirm_delete"