edit.py 8.5 KB

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