forms.py 7.49 KB
Newer Older
1
from django import forms
2
from go.models import URL, RegisteredUser
3
from django.core.exceptions import ValidationError
4
from django.utils.safestring import mark_safe
David Haynes's avatar
David Haynes committed
5
from crispy_forms.helper import FormHelper
6
7
from crispy_forms.layout import Layout, Fieldset, Submit, HTML, Div, Field
from crispy_forms.bootstrap import StrictButton, PrependedText, Accordion, AccordionGroup
8
9

class URLForm(forms.ModelForm):
10

11
12
13
    # Custom target URL field
    target = forms.URLField(
        required=True,
14
        label='Long URL (Required)',
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
        max_length=1000,
        widget=forms.URLInput(attrs={
        })
    )

    # Check to make sure the short url has not been used
    def unique_short(value):
        try:
            URL.objects.get(short__iexact=value)
        except URL.DoesNotExist:
            return
        raise ValidationError('Short url already exists.')

    # Custom short-url field with validators.
    short = forms.SlugField(
        required=False,
        label='Short URL (Optional)',
        widget=forms.TextInput(),
        validators=[unique_short],
        max_length=20,
        min_length=3,
    )

    DAY = '1 Day'
    WEEK = '1 Week'
    MONTH = '1 Month'
    NEVER = 'Never'

    EXPIRATION_CHOICES = (
        (DAY, DAY),
        (WEEK, WEEK),
        (MONTH, MONTH),
        (NEVER, NEVER),
    )

    # Add a custom expiration choice field.
    expires = forms.ChoiceField(
        required=True,
53
        label='Expiration (Required)',
54
55
56
57
58
59
        choices=EXPIRATION_CHOICES,
        initial=NEVER,
        widget=forms.RadioSelect(),
    )

    def __init__(self, *args, **kwargs):
60
        super(URLForm, self).__init__(*args, **kwargs)
61
        self.helper = FormHelper()
62
        self.helper.form_method = 'POST'
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

        self.helper.form_class = 'form-horizontal'
        self.helper.label_class = 'col-md-1'
        self.helper.field_class = 'col-md-6'


        self.helper.layout = Layout(
            Fieldset(
                '',
                Accordion(
                    AccordionGroup('Step 1: Long URL',
                        Div(
                            HTML("""
                                <h4>Paste the URL you would like to shorten:</h4>
                                <br />
                            """),
                            PrependedText('target',
80
                            'https://',
81
82
83
84
85
                            ),
                            style="background: rgb(#F6F6F6);",
                            title="target_url",
                            css_class="first_group",
                        ),
86
87
88
                        css_id='firstCollapse',
                        active=True,
                        template='crispy/accordian-group.html',
89
90
91
92
93
                    ),
                    AccordionGroup('Step 2: Short URL',
                        Div(
                            HTML("""
                                <h4>Create a custom Go address:</h4>
94
                                <br />
95
96
                            """),
                            PrependedText('short',
97
                            'https://go.gmu.edu/',
98
99
100
101
                            ),
                            style="background: rgb(#F6F6F6);",
                            title="short_url",
                            css_class="second_group",
102
103
                        ),
                        css_id='secondCollapse',
104
                        active=True,
105
                        template='crispy/accordian-group.html',
106
107
108
109
110
                    ),
                    AccordionGroup('Step 3: URL Expiration',
                        Div(
                            HTML("""
                                <h4>Set when you would like your Go address to expire:</h4>
111
                                <br />
112
113
114
115
116
                            """),
                            'expires',
                            style="background: rgb(#F6F6F6);",
                            title="expires_url",
                            css_class="third_group",
117
118
                        ),
                        css_id='thirdCollapse',
119
                        active=True,
120
121
122
123
                        template='crispy/accordian-group.html',
                    ),
                    css_id='accordian',
                    template='crispy/accordian.html'
124
125
126
127
                ),
            HTML("""
                <br />
            """),
128
            StrictButton('Shorten', css_class="btn-success", type='submit'),
129
130
131
        )
    )

132

133
134
135
136
137
138
    class Meta:
        model = URL
        fields = ('target',)
        exclude = ('owner', 'short', 'date_created', 'clicks', 'expires')


139
class SignupForm(forms.ModelForm):
140
141
142
143
144
145
146
147

    def validate_username(username):
        try:
            registered = RegisteredUser.objects.get(username=username)
            raise ValidationError('Username "%s" is already in use.' % username)
        except RegisteredUser.DoesNotExist:
            return

148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
    def __init__(self, request, *args, **kwargs):
        # Necessary to call request in forms.py, is otherwise restricted to views.py and models.py
        self.request = request
        super(SignupForm, self).__init__(*args, **kwargs)
        self.helper = FormHelper(form=self)
        self.helper.form_method = 'POST'
        
        self.helper.form_class = 'form-horizontal'
        self.helper.label_class = 'col-md-4'
        self.helper.field_class = 'col-md-6' 
    
        self.helper.layout = Layout(
            Fieldset(
                '',
                HTML("""<p>In order to succesfully provide this service, users must be manually approved. This prevents misuse of the URL shortener. Please indicate below if you are interested.</p><br>"""),
                Div(
                    Div(
                        'username',
                        'full_name',
                        'organization',
                        'description',
                        'tos_box',
                        css_class='well',
                    ),      
                    StrictButton('Submit',css_class='btn btn-primary btn-md col-md-6', type='submit'),
                    css_class='col-md-6',
                )

            )
            ,
        )

180
    username = forms.CharField(
181
        required=True,
182
        label='Mason NetID (Required)',
183
        max_length=30,
184
        validators=[validate_username],
185
        widget=forms.TextInput(attrs={
186
187
188
        }),
    )
    full_name = forms.CharField(
189
        required=True,
190
        label='Full Name (Required)',
191
192
        max_length=100,
        widget=forms.TextInput(attrs={
193
194
        }),
    )
195
196
197
198
199
200
201
    organization = forms.CharField(
        required=True,
        label='Organization (Required)',
        max_length=100,
        widget=forms.TextInput(attrs={
        })
    )
202
    description = forms.CharField(
203
204
205
206
        required=False,
        label='Description (Optional)',
        max_length=200,
        widget=forms.Textarea(attrs={
207
208
        }),
    )
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
    tos_box = forms.BooleanField(
        required=True,
        #Need to add a Terms of Service Page and replace the href below
        label = mark_safe('Do you accept the <a href="#" target="_blank">Terms of Service</a>?'),
    )

    def clean_username(self):
        # Prevent hax: (non-staff) Users cannot signup for other users
        cleaned_data = super(SignupForm, self).clean()
        data_username = cleaned_data.get("username")

        if not self.request.user.is_staff:
            if self.request.user.username not in data_username:
                self.add_error('username', "This is not your NetID!")
        return data_username
224
225
226

    class Meta:
        model = RegisteredUser
227
        fields = '__all__'