event_rsvp.forms: 64 total statements, 100.0% covered

Generated: Tue 2013-04-16 10:12 CEST

Source file: /home/tobi/Projects/event-rsvp/src/event_rsvp/forms.py

Stats: 61 executed, 0 missed, 3 excluded, 56 ignored

1. """Forms for the ``event_rsvp`` app."""
2. from django import forms
3. from django.utils.translation import ugettext_lazy as _
5. from event_rsvp.models import Event, Guest
8. class EventForm(forms.ModelForm):
9. """Form to handle specific validations of the Event model."""
10. required_css_class = 'requiredField'
12. def __init__(self, created_by, create_from_template=False, *args,
13. **kwargs):
14. """
15. :param created_by: Django-Auth User model, which is calling this view
16. :param create_from_template: Boolean to know, if someone is using a
17. template as draft.
19. """
20. self.instance = kwargs.get('instance')
21. self.create_from_template = create_from_template
22. if hasattr(self.instance, 'id') and not self.create_from_template:
23. # Ignore the current user if it's the update view or if the
24. # instance is a template
25. self.created_by = self.instance.created_by
26. else:
27. # Add the current user as owner
28. self.created_by = created_by
29. if self.create_from_template:
30. # If we use a template, the template_name is the only attribute,
31. # which should not be preset
32. self.instance.template_name = ''
33. super(EventForm, self).__init__(*args, **kwargs)
35. if self.instance.id and self.instance.template_name:
36. self.this_is_a_template = True
37. else:
38. self.this_is_a_template = False
40. def save(self, commit=True):
41. """Saves twice if a template should be created."""
42. if self.create_from_template:
43. # Save a new event
44. return forms.models.save_instance(
45. self, Event(created_by=self.created_by), self._meta.fields,
46. 'created', commit, False)
47. self.instance.created_by = self.created_by
48. self.instance = super(EventForm, self).save(commit=True)
49. if self.instance.template_name and not self.this_is_a_template:
50. """
51. If someone wants to save this instance as a template, create a
52. second event as the template and keep the current instance in a
53. 'normal' condition. This is necessary, 'cause otherwise the
54. template would keep the 'good' slug like 'foo-bar', while the new
55. event got a slug like '_foo-bar'.
57. """
58. template = Event(template_name=self.instance.template_name,
59. created_by=self.created_by)
60. self.instance.template_name = ''
61. self.instance.save()
62. forms.models.save_instance(
63. self, template, self._meta.fields, 'created', commit, False)
64. return self.instance
66. class Meta:
67. model = Event
68. exclude = ('created_by', 'slug')
71. class GuestForm(forms.ModelForm):
72. """Form to handle specific validations of the Guest model."""
73. required_css_class = 'requiredField'
75. def __init__(self, event, user, *args, **kwargs):
76. """
77. :param event: Event to participate
78. :param user: Current user or anonymous.
80. """
81. self.event = event
82. if user and user.is_authenticated():
83. self.user = user
84. else:
85. self.user = None
86. super(GuestForm, self).__init__(*args, **kwargs)
87. if self.event.id:
88. for field in self.event.required_fields:
89. if field:
90. self.fields[field].required = True
92. def clean_number_of_seats(self):
93. data = self.cleaned_data['number_of_seats'] or 1
94. free_seats = self.event.get_free_seats()
95. if self.event.available_seats and free_seats < data:
96. if free_seats == 1:
97. msg = _('Sorry. There is only 1 seat left.')
98. else:
99. msg = _('Sorry. There are only %(amount)s seats left.') % {
100. 'amount': free_seats}
101. raise forms.ValidationError(msg)
102. if (self.event.max_seats_per_guest > 0
103. and data > self.event.max_seats_per_guest):
104. if self.event.max_seats_per_guest == 1:
105. msg = _('Pardon. There is only 1 seat per person reservable.')
106. else:
107. msg = _('Pardon. There are only %(amount)s seats per person'
108. ' reservable.') % {
109. 'amount': self.event.max_seats_per_guest}
110. raise forms.ValidationError(msg)
111. return data
113. def save(self, *args, **kwargs):
114. self.instance.user = self.user
115. self.instance.event = self.event
116. super(GuestForm, self).save(*args, **kwargs)
118. class Meta:
119. model = Guest
120. fields = ('name', 'email', 'phone', 'number_of_seats', 'message')