import re

from django.http import HttpResponseRedirect


class DiggBarMiddleware(object):
    diggbar_regex = re.compile("http://digg.com/\w{1,8}/?")
    rickroll = "http://www.youtube.com/watch?v=oHg5SJYRHA0"

    def process_request(self, request):
        if self.diggbar_regex.match(request.META.get('HTTP_REFERER', '')):
            return HttpResponseRedirect(self.rickroll)