__init__.py 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  1. from django import forms
  2. from django.utils.translation import ugettext_lazy as _
  3. from misago.forms.layouts import *
  4. from recaptcha.client.captcha import submit as recaptcha_submit
  5. class Form(forms.Form):
  6. """
  7. Custom form implementation extending Django Forms functionality
  8. """
  9. validate_repeats = []
  10. repeats_errors = []
  11. dont_strip = []
  12. allow_nl = []
  13. error_source = None
  14. def __init__(self, data=None, file=None, request=None, *args, **kwargs):
  15. self.request = request
  16. # Extract request from first argument
  17. if data != None:
  18. super(Form, self).__init__(data, file, *args, **kwargs)
  19. else:
  20. super(Form, self).__init__(*args, **kwargs)
  21. # Kill captcha fields
  22. try:
  23. if self.request.settings['bots_registration'] != 'recaptcha' or self.request.session.get('captcha_passed'):
  24. del self.fields['recaptcha']
  25. except KeyError:
  26. pass
  27. try:
  28. if self.request.settings['bots_registration'] != 'qa' or self.request.session.get('captcha_passed'):
  29. del self.fields['captcha_qa']
  30. else:
  31. # Make sure we have any questions loaded
  32. self.fields['captcha_qa'].label = self.request.settings['qa_test']
  33. self.fields['captcha_qa'].help_text = self.request.session['qa_test_help']
  34. except KeyError:
  35. pass
  36. # Let forms do mumbo-jumbo with fields removing
  37. self.finalize_form()
  38. def finalize_form(self):
  39. pass
  40. def full_clean(self):
  41. """
  42. Trim inputs and strip newlines
  43. """
  44. for key, field in self.base_fields.iteritems():
  45. try:
  46. if field.__class__.__name__ in ['ModelChoiceField', 'TreeForeignKey'] and self.data[key]:
  47. self.data[key] = int(self.data[key])
  48. elif field.__class__.__name__ == 'ModelMultipleChoiceField':
  49. self.data.setlist(key, [int(x) for x in self.data.getlist(key, [])])
  50. elif field.__class__.__name__ not in ['DateField', 'DateTimeField']:
  51. if not key in self.dont_strip:
  52. if field.__class__.__name__ in ['MultipleChoiceField', 'TypedMultipleChoiceField']:
  53. self.data.setlist(key, [x.strip() for x in self.data.getlist(key, [])])
  54. else:
  55. self.data[key] = self.data[key].strip()
  56. if not key in self.allow_nl:
  57. if field.__class__.__name__ in ['MultipleChoiceField', 'TypedMultipleChoiceField']:
  58. self.data.setlist(key, [x.replace("\n", '') for x in self.data.getlist(key, [])])
  59. else:
  60. self.data[key] = self.data[key].replace("\n", '')
  61. except (KeyError, AttributeError):
  62. pass
  63. super(Form, self).full_clean()
  64. def clean(self):
  65. """
  66. Clean data, do magic checks and stuff
  67. """
  68. cleaned_data = super(Form, self).clean()
  69. self._check_all()
  70. return cleaned_data
  71. def clean_recaptcha(self):
  72. """
  73. Test reCaptcha, scream if it went wrong
  74. """
  75. response = recaptcha_submit(
  76. self.request.POST.get('recaptcha_challenge_field'),
  77. self.request.POST.get('recaptcha_response_field'),
  78. self.request.settings['recaptcha_private'],
  79. self.request.session.get_ip(self.request)
  80. ).is_valid
  81. if not response:
  82. raise forms.ValidationError(_("Entered words are incorrect. Please try again."))
  83. self.request.session['captcha_passed'] = True
  84. return ''
  85. def clean_captcha_qa(self):
  86. """
  87. Test QA Captcha, scream if it went wrong
  88. """
  89. if not unicode(self.cleaned_data['captcha_qa']).lower() in (name.lower() for name in unicode(self.request.settings['qa_test_answers']).splitlines()):
  90. raise forms.ValidationError(_("The answer you entered is incorrect."))
  91. self.request.session['captcha_passed'] = True
  92. return self.cleaned_data['captcha_qa']
  93. def _check_all(self):
  94. # Check repeated fields
  95. self._check_repeats()
  96. # Check CSRF, we dont allow un-csrf'd forms in Misago
  97. self._check_csrf()
  98. # Check if we have any errors from fields, if we do, we will set fancy form-wide error message
  99. self._check_fields_errors()
  100. def _check_repeats(self):
  101. for index, repeat in enumerate(self.validate_repeats):
  102. # Check empty fields
  103. for field in repeat:
  104. if not self.data[field]:
  105. try:
  106. if len(repeat) == 2:
  107. self.errors['_'.join(repeat)] = [self.repeats_errors[index]['fill_both']]
  108. else:
  109. self.errors['_'.join(repeat)] = [self.repeats_errors[index]['fill_all']]
  110. except (IndexError, KeyError):
  111. if len(repeat) == 2:
  112. self.errors['_'.join(repeat)] = [_("You have to fill in both fields.")]
  113. else:
  114. self.errors['_'.join(repeat)] = [_("You have to fill in all fields.")]
  115. break
  116. else:
  117. # Check different fields
  118. past_field = self.data[repeat[0]]
  119. for field in repeat:
  120. if self.data[field] != past_field:
  121. try:
  122. self.errors['_'.join(repeat)] = [self.repeats_errors[index]['different']]
  123. except (IndexError, KeyError):
  124. self.errors['_'.join(repeat)] = [_("Entered values differ from each other.")]
  125. break
  126. past_field = self.data[field]
  127. def _check_csrf(self):
  128. if not self.request.csrf.request_secure(self.request):
  129. raise forms.ValidationError(_("Request authorization is invalid. Please resubmit your form."))
  130. def _check_fields_errors(self):
  131. if self.errors:
  132. if self.error_source and self.error_source in self.errors:
  133. field_error, self.errors[self.error_source] = self.errors[self.error_source][0], []
  134. raise forms.ValidationError(field_error)
  135. raise forms.ValidationError(_("Form contains errors."))
  136. class YesNoSwitch(forms.CheckboxInput):
  137. """
  138. Custom Yes-No switch as fancier alternative to checkboxes
  139. """
  140. pass