views.py 7.23 KB
Newer Older
1
from go.models import URL, RegisteredUser
Jean Michel Rouly's avatar
Jean Michel Rouly committed
2
from go.forms import URLForm, SignupForm
3
from datetime import timedelta
4
from django.conf import settings
5
from django.http import Http404, HttpResponseServerError
6
from django.utils import timezone
Jean Michel Rouly's avatar
Jean Michel Rouly committed
7
from django.contrib.auth.models import User
8
from django.core.exceptions import PermissionDenied, ValidationError
9
from django.core.mail import send_mail
10
from django.contrib.auth.decorators import login_required
Jean Michel Rouly's avatar
Jean Michel Rouly committed
11
from django.shortcuts import render, get_object_or_404, redirect
12
import os
13

Jean Michel Rouly's avatar
Jean Michel Rouly committed
14 15 16 17 18 19 20

##############################################################################
"""
Define useful helper methods here.
"""


21
def is_registered( user ):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
22 23 24 25 26
    """
    This function checks if a user account has a corresponding RegisteredUser,
    thus checking if the user is registered.
    """

27
    try:
28
        registered = RegisteredUser.objects.get( username=user.username )
29
        return registered.approved
30 31 32
    except RegisteredUser.DoesNotExist:
        return False

Jean Michel Rouly's avatar
Jean Michel Rouly committed
33 34 35 36 37 38 39

##############################################################################
"""
Define error page handling here.
"""


40
def error_404(request):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
41 42 43 44
    """
    Error 404 view, in case a url is not found.
    """

45 46 47 48 49
    return render(request, '404.html', {
    },
    )

def error_500(request):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
50 51 52 53
    """
    Error 500 view, in case a server error occurs.
    """

54 55 56 57
    return render(request, '500.html', {
    },
    )

Jean Michel Rouly's avatar
Jean Michel Rouly committed
58 59 60 61 62 63
##############################################################################
"""
Define user views here.
"""


64
@login_required
Jean Michel Rouly's avatar
Jean Michel Rouly committed
65
def index(request):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
66 67 68 69 70 71 72
    """
    This view handles the homepage that the user is presented with when
    they request '/'. If they're not logged in, they're redirected to
    login. If they're logged in but not registered, they're given the
    not_registered error page. If they are logged in AND registered, they
    get the URL registration form.
    """
Jean Michel Rouly's avatar
Jean Michel Rouly committed
73

Jean Michel Rouly's avatar
Jean Michel Rouly committed
74
    # If the user isn't registered, don't give them any leeway.
75 76 77
    if not is_registered(request.user):
        return render(request, 'not_registered.html')

78 79 80 81 82 83
    url_form = URLForm() # unbound form

    if request.method == 'POST':
        url_form = URLForm( request.POST ) # bind dat form
        if url_form.is_valid():

Jean Michel Rouly's avatar
Jean Michel Rouly committed
84 85
            # We don't commit the url object yet because we need to add its
            # owner, and parse its date field.
86 87 88
            url = url_form.save(commit=False)
            url.owner = request.user

Jean Michel Rouly's avatar
Jean Michel Rouly committed
89
            # If the user entered a short url, it's already been validated,
Jean Michel Rouly's avatar
Jean Michel Rouly committed
90 91
            # so accept it. If they did not, however, then generate a
            # random one and use that instead.
92
            short = url_form.cleaned_data.get('short').strip()
93 94 95
            if len(short) > 0:
                url.short = short
            else:
96 97 98 99 100 101 102 103 104 105
                # If the user didn't enter a short url, generate a random
                # one. However, if a random one can't be generated, return
                # a 500 server error.
                random_short = URL.generate_valid_short()
                if random_short is None:
                    return HttpResponseServerError(
                        render(request, '500.html', {})
                    )
                else:
                    url.short = random_short
106

Jean Michel Rouly's avatar
Jean Michel Rouly committed
107 108 109
            # Grab the expiration field value. It's currently an unsable
            # string value, so we need to parse it into a datetime object
            # relative to right now.
110
            expires = url_form.cleaned_data.get('expires')
111 112 113 114 115 116 117 118 119 120

            if expires == URLForm.DAY:
                url.expires = timezone.now() + timedelta(days=1)
            elif expires == URLForm.WEEK:
                url.expires = timezone.now() + timedelta(weeks=1)
            elif expires == URLForm.MONTH:
                url.expires = timezone.now() + timedelta(weeks=3)
            else:
                pass # leave the field NULL

Jean Michel Rouly's avatar
Jean Michel Rouly committed
121 122
            # Make sure that our new URL object is clean, then save it and
            # let's redirect to view this baby.
123 124
            url.full_clean()
            url.save()
Jean Michel Rouly's avatar
Jean Michel Rouly committed
125
            return redirect('view', url.short)
126 127 128

    return render(request, 'index.html', {
        'form': url_form,
Jean Michel Rouly's avatar
Jean Michel Rouly committed
129 130 131
    },
    )

Jean Michel Rouly's avatar
Jean Michel Rouly committed
132

Jean Michel Rouly's avatar
Jean Michel Rouly committed
133
def view(request, short):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
134 135 136 137 138
    """
    This view allows the user to view details about a URL. Note that they
    do not need to be logged in to view info.
    """

Jean Michel Rouly's avatar
Jean Michel Rouly committed
139
    url = get_object_or_404(URL, short__iexact = short)
140

Jean Michel Rouly's avatar
Jean Michel Rouly committed
141
    return render(request, 'view.html', {
142 143 144 145
        'url': url,
    },
    )

Jean Michel Rouly's avatar
Jean Michel Rouly committed
146

147
@login_required
148
def my_links(request):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
149 150 151 152 153
    """
    This view displays all the information about all of your URLs. You
    obviously need to be logged in to view your URLs.
    """

154 155
    if not is_registered(request.user):
        return render(request, 'not_registered.html')
Jean Michel Rouly's avatar
Jean Michel Rouly committed
156

157
    urls = URL.objects.filter( owner = request.user )
158
    return render(request, 'my_links.html', {
159
        'urls' : urls,
160 161 162
    },
    )

Jean Michel Rouly's avatar
Jean Michel Rouly committed
163

164
@login_required
165
def delete(request, short):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
166 167 168 169 170
    """
    This view deletes a URL if you have the permission to. User must be
    logged in and registered, and must also be the owner of the URL.
    """

171 172
    if not is_registered(request.user):
        return render(request, 'not_registered.html')
Jean Michel Rouly's avatar
Jean Michel Rouly committed
173

Jean Michel Rouly's avatar
Jean Michel Rouly committed
174
    url = get_object_or_404(URL, short__iexact = short )
175 176
    if url.owner == request.user:
        url.delete()
177 178
        return redirect('my_links')
    else:
179
        raise PermissionDenied()
180

Jean Michel Rouly's avatar
Jean Michel Rouly committed
181

182
@login_required
Jean Michel Rouly's avatar
Jean Michel Rouly committed
183
def signup(request):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
184 185 186 187 188 189
    """
    This view presents the user with a registration form. You can register
    yourself, or another person.

    """

190
    signup_form = SignupForm()
Jean Michel Rouly's avatar
Jean Michel Rouly committed
191 192

    if request.method == 'POST':
193 194 195 196 197
        signup_form = SignupForm(request.POST, initial={'approved': False})
        if signup_form.is_valid():
            username = signup_form.cleaned_data.get('username')
            full_name = signup_form.cleaned_data.get('full_name')
            description = signup_form.cleaned_data.get('description')
198

199 200 201 202 203 204
            send_mail('Signup from %s' % (username), '%s signed up at %s\n'
                'Username: %s\nMessage: %s\nPlease attend to this request at '
                'your earliest convenience.' % (str(full_name),
                str(timezone.now()).strip(), str(username), str(description)),
                settings.EMAIL_FROM, [settings.EMAIL_TO])

205 206
            signup_form.save()

207
            return redirect('registered')
Jean Michel Rouly's avatar
Jean Michel Rouly committed
208

Jean Michel Rouly's avatar
Jean Michel Rouly committed
209
    return render(request, 'signup.html', {
210
        'form': signup_form,
Jean Michel Rouly's avatar
Jean Michel Rouly committed
211 212
    },
    )
Jean Michel Rouly's avatar
Jean Michel Rouly committed
213

Jean Michel Rouly's avatar
Jean Michel Rouly committed
214

Jean Michel Rouly's avatar
Jean Michel Rouly committed
215
def redirection(request, short):
Jean Michel Rouly's avatar
Jean Michel Rouly committed
216 217 218
    """
    This view redirects a user based on the short URL they requested.
    """
Jean Michel Rouly's avatar
Jean Michel Rouly committed
219

Jean Michel Rouly's avatar
Jean Michel Rouly committed
220
    url = get_object_or_404( URL, short__iexact = short )
Jean Michel Rouly's avatar
Jean Michel Rouly committed
221 222
    url.clicks = url.clicks + 1
    url.save()
223

Jean Michel Rouly's avatar
Jean Michel Rouly committed
224 225 226 227 228
    """
    Include server-side tracking because there is no template displayed to
    the user which would include javascript tracking.
    """

229 230 231 232 233 234 235
    from piwikapi.tracking import PiwikTracker
    from django.conf import settings
    piwiktracker = PiwikTracker(settings.PIWIK_SITE_ID, request)
    piwiktracker.set_api_url(settings.PIWIK_URL)
    piwiktracker.do_track_page_view('Redirect to %s' % url.target)

    return redirect( url.target )
Jean Michel Rouly's avatar
Jean Michel Rouly committed
236 237 238 239 240 241 242 243 244 245 246 247 248


##############################################################################
"""
Define static user views here.
"""


def about(request):
    return render(request, 'about.html', {
    },
    )

249 250 251 252
def registered(request):
    return render(request, 'registered.html', {
    },
    )