path: root/project/scipycon/user/views.py

from urlparse import urlparse

import simplejson as json
import os

from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.contrib.auth.forms import AuthenticationForm
from django.contrib.auth.forms import PasswordChangeForm
from django.contrib.auth.models import User
from django.core.exceptions import ObjectDoesNotExist
from django.core.urlresolvers import reverse
from django.db.models import Q
from django.http import HttpResponse
from django.shortcuts import render_to_response
from django.template import RequestContext

from PIL import Image

from project.scipycon.base.models import Event
from project.scipycon.registration.models import Registration
from project.scipycon.registration.models import Wifi
from project.scipycon.registration.forms import WifiForm
from project.scipycon.talk.models import Talk
from project.scipycon.user.forms import EditProfileForm
from project.scipycon.user.forms import RegisterForm
from project.scipycon.user.forms import UsernameForm
from project.scipycon.user.utils import handle_uploaded_photo
from project.scipycon.user.utils import scipycon_createuser
from project.scipycon.utils import set_message_cookie

#User_dump Http404 Error
from django.http import Http404
#for user_dump creation
from project.scipycon.registration.models import Accommodation

#Pdf badge generation
from reportlab.pdfgen import canvas
from reportlab.lib.units import cm
from reportlab.platypus import  Image as reportlabImage
from django.core.exceptions import ObjectDoesNotExist


@login_required
def account(request, scope, template_name="user/account.html"):
    """Displays the main screen of the current user's account.
    """

    user = request.user
    profile = user.get_profile()

    talks = Talk.objects.filter(speaker=user)

    try:
        registration = Registration.objects.get(registrant=user)
    except ObjectDoesNotExist:
        registration = None

    try:
        wifiobj = Wifi.objects.get(user=user)
    except ObjectDoesNotExist:
        wifiobj = None

    event = Event.objects.get(scope=scope)

    if profile.photo:
        photo = os.path.join(settings.USER_MEDIA_URL, profile.photo)
    else:
        photo = '/img/user-default.png'

    return render_to_response(template_name, RequestContext(request, {
        'params': {'scope': scope},
        'user' : user,
        'profile' : profile,
        'photo' : photo,
        'talks' : talks,
        'registration' : registration,
        'event': event}))

@login_required
def edit_profile(request, scope, template_name="user/editprofile.html"):
    """Allows user to edit profile
    """

    user = request.user
    profile = user.get_profile()

    if request.method == "POST":
        form = EditProfileForm(data=request.POST,
                               files=request.FILES)

        if form.is_valid():
            photo = request.FILES.get('photo', None)
            filename= None
            if photo:
                filename = handle_uploaded_photo(user, request.FILES['photo'])
            if filename:
                profile.photo = filename

            user.email = form.data.get("email")
            user.first_name = form.data.get("first_name")
            user.last_name = form.data.get("last_name")
            user.save()

            profile.url = form.data.get("url")
            profile.about = form.data.get("about")
            profile.save()

            redirect_to = reverse('scipycon_account',
                                  kwargs={'scope': scope})
            return set_message_cookie(redirect_to,
                    msg = u'Your profile has been changed.')

    else:
        form = EditProfileForm(
            initial={
                'email' : user.email,
                'email2' : user.email, # hidden field
                'first_name' : user.first_name,
                'last_name' : user.last_name,
                'url' : profile.url,
                'about' : profile.about,
            })

    return render_to_response(template_name, RequestContext(request, {
        'params': {'scope': scope},
        'form': form
    }))

def login(request, scope, template_name="user/login.html"):
    """Custom view to login or register/login a user.
       Integration of register and login form
       It uses Django's standard AuthenticationForm, though.
    """

    user = request.user
    if user.is_authenticated():
        redirect_to = reverse("scipycon_account", kwargs={'scope': scope})
        return set_message_cookie(redirect_to,
                msg = u"Redirected to account from login form.")

    # Using Djangos default AuthenticationForm
    login_form = AuthenticationForm()
    register_form = RegisterForm()

    if request.POST.get("action") == "login":
        login_form = AuthenticationForm(data=request.POST)

        if login_form.is_valid():
            redirect_to = request.POST.get("next")
            # Light security check -- make sure redirect_to isn't garbage.
            if not redirect_to or '//' in redirect_to or ' ' in redirect_to:
                redirect_to = reverse('scipycon_account',
                                      kwargs={'scope': scope})

            from django.contrib.auth import login
            login(request, login_form.get_user())

            return set_message_cookie(redirect_to, msg = u"You have been logged in.")

    elif request.POST.get("action") == "register":
        register_form = RegisterForm(data=request.POST)
        if register_form.is_valid():

            user = scipycon_createuser(request, register_form.data, scope)

            redirect_to = request.POST.get("next")
            if not redirect_to or '//' in redirect_to or ' ' in redirect_to:
                redirect_to = reverse('scipycon_account',
                                      kwargs={'scope': scope})

            return set_message_cookie(
                redirect_to, msg = u"You have been registered and logged in.")

    # Get next_url
    next_url = request.REQUEST.get("next")
    if next_url is None:
        next_url = request.META.get("HTTP_REFERER")
    if next_url is None:
        next_url = reverse('scipycon_account', kwargs={'scope': scope})

    # Get just the path of the url.
    # See django.contrib.auth.views.login for more
    next_url = urlparse(next_url)
    next_url = next_url[2]

    try:
        login_form_errors = login_form.errors["__all__"]
    except KeyError:
        login_form_errors = None

    return render_to_response(template_name, RequestContext(request, {
        'params': {'scope': scope},
        'login_form' : login_form,
        'login_form_errors' : login_form_errors,
        'register_form' : register_form,
        'next_url' : next_url,
    }))

def logout(request, scope):
    """Custom method to logout a user.

    The reason to use a custom logout method is just to provide a login and a
    logoutmethod on one place.
    """

    from django.contrib.auth import logout
    logout(request)

    redirect_to = '/%s' % (scope)
    return set_message_cookie(redirect_to, msg = u"You have been logged out.")

@login_required
def password(request, scope, template_name='user/password.html'):
    """Changes the password of current user.
    """

    if request.method == 'POST':
        form = PasswordChangeForm(request.user, request.POST)
        if form.is_valid():
            form.save()
            redirect_to = reverse('scipycon_account', kwargs={'scope': scope})
            return set_message_cookie(redirect_to,
                    msg = u'Your password has been changed.')
    else:
        form = PasswordChangeForm(request.user)

    return render_to_response(template_name, RequestContext(request, {
        'params': {'scope': scope},
        'form' : form
    }))

@login_required
def username(request, scope, template_name='user/username.html'):
    """Saves the username from the data form.
    """
    if request.method == 'POST':
        username_form = UsernameForm(
            initial={'username' : request.user.username},
            data=request.POST)
        if username_form.is_valid():
            request.user.username = username_form.cleaned_data.get("username")
            request.user.save()
            redirect_to = reverse('scipycon_account',
                                  kwargs={'scope': scope})
            return set_message_cookie(redirect_to,
                    msg = u"Your username has been changed.")
    else:        
        username_form = UsernameForm(initial={"username" : request.user.username})

    return render_to_response(template_name, RequestContext(request, {
        'params': {'scope': scope},
        'form': username_form
    }))


def get_usernames(request, scope):
    """Returns in json the list of ten possible usernames
    starting with the last pattern in the comma separated string
    """

    get_params = request.GET
    authors_str = get_params.get('input')

    if not authors_str:
        return HttpResponse(json.dumps(''))

    authors = authors_str.split(',')
    search_author = authors[-1].strip()

    users = User.objects.filter(
        Q(username__istartswith=search_author) | Q(
        first_name__istartswith=search_author) | Q(
        last_name__istartswith=search_author))

    results = [{'id': '',
                'info': 'plugin_header',
                'value': 'User Names'
              }]
    
    for user in users:
        results.append(
            {'id': 'author_name',
             'info': str(user.get_full_name()),
             'value': str(user.username)
            })

    json_response = {'results': results}

    return HttpResponse(json.dumps(json_response))


@login_required
def get_user_dump(request, scope,template_name='user/dump.html'):
    """ Gets a general dump of user related info
    """
    print request.user.is_staff
    if request.user.is_staff:
        qs=Registration.objects.all()
        rows=[]    
        for obj in qs:
            row = {}
            row['first_name'] = obj.registrant.first_name
            row['last_name'] = obj.registrant.last_name
            try:
                accomodation_require = Accommodation.objects.filter(user__username=obj.registrant.username)[0]
                row['sex'] = accomodation_require.sex
            except:
                row['sex'] = '-'
            row['city'] = obj.city
            row['organization'] = obj.organisation
            row['occupation'] = obj.occupation
            row['conference'] = obj.conference 
            row['sprint'] = obj.sprint
            row['tutorial'] = obj.tutorial
            try:
                wifi_require = Wifi.objects.filter(user__username=obj.registrant.username)[0]
                row['wifi'] = wifi_require.wifi
            except:
                row['wifi']='Wifi Unspecified'
            rows.append(row)
        return render_to_response(template_name, RequestContext(request, {
                    'rows':  rows}))
                    

    else:
            raise Http404


@login_required
def badge(request,scope):

    from django.conf import settings

    # Create the HttpResponse object with the appropriate PDF headers.
    response = HttpResponse(mimetype='application/pdf')
    response['Content-Disposition'] = 'attachment; filename=scipybadge.pdf'

    # Create the PDF object, using the response object as its "file."
    c = canvas.Canvas(response)

    ref=5*cm
    # Draw things on the PDF. Here's where the PDF generation happens.
    # See the ReportLab documentation for the full list of functionality.
    c.rect(ref,ref,9*cm,6*cm)

    img_path = os.path.join(settings.STATIC_ROOT, 'img', 'scipyshiny_small.png')
    im = reportlabImage(img_path, width=1.75*cm, height=1.75*cm)
    im.drawOn(c,(ref+0.8*cm),(ref+4.3*cm))
    c.setFont('Helvetica', 6)
    c.drawString((ref+1.0*cm),(ref+4.2*cm),'scipy.in 2010') 
    c.drawString((ref+1.1*cm),(ref+4.0*cm),'Hyderabad') 

    c.setFont('Helvetica', 14)
    print request.user.id
    reg_obj=Registration.objects.get(registrant=request.user.id)
    c.drawString((ref+3.4*cm),(ref+4.9*cm),str(reg_obj.slug)) 
    
    c.setFont('Helvetica-Bold', 15)
    c.drawString((ref+0.6*cm),(ref+3.4*cm),str(request.user.get_full_name()))
    c.setFont('Helvetica', 11)
    c.drawString((ref+2.8*cm),(ref+2.7*cm),reg_obj.organisation)
    c.setFont('Helvetica', 11)
    try:
        c.drawString((ref+2.8*cm),(ref+2.2*cm),reg_obj.occupation.split(':')[1])
    except IndexError:
        c.drawString((ref+2.8*cm),(ref+2.3*cm),reg_obj.occupation)
        
    c.setFont('Helvetica', 10)
    c.drawString((ref+2.8*cm),(ref+1.7*cm),reg_obj.city)
    c.setFont('Helvetica', 10)
    c.drawString((ref+2.8*cm),(ref+1*cm),'Participant')
    
   
    try:
        wifi_obj=Wifi.objects.get(user=request.user.id)
        c.setFont('Helvetica', 10)
        c.drawString((ref+5.6*cm),(ref+0.5*cm),wifi_obj.registration_id)
    except :
        pass

    
    # Close the PDF object cleanly, and we're done.
    c.showPage()
    c.save()
    return response