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__'