GetTogether/get_together/views/user.py

from django.utils.translation import ugettext_lazy as _

from django.contrib import messages
from django.contrib.auth import logout as logout_user
from django.contrib.auth import login as login_user, authenticate
from django.contrib.auth.forms import UserCreationForm, AuthenticationForm
from django.shortcuts import render, redirect, get_object_or_404
from django.http import HttpResponse, JsonResponse
from django.core.exceptions import ObjectDoesNotExist

from events.models.profiles import Team, UserProfile, Member
from events.forms import UserForm, UserProfileForm

from events.models.events import Event, Place, Attendee

import datetime
import simplejson

from .teams import *
from .events import *

def logout(request):
    if request.user.is_authenticated:
        logout_user(request)
    return redirect('home')

def login(request):
    if request.user.is_authenticated:
        messages.add_message(request, messages.INFO, message=_('You are already logged in.'))
        return redirect('home')

    context = {
        'login_form': AuthenticationForm(),
        'signup_form': UserCreationForm(),
    }
    if request.method == 'POST':
        if request.POST.get('action') == 'login':
            login_form = AuthenticationForm(data=request.POST)
            if login_form.is_valid():
                username = login_form.cleaned_data.get('username')
                raw_password = login_form.cleaned_data.get('password')
                user = authenticate(username=username, password=raw_password)
                login_user(request, user, backend=user.backend)
                return redirect('home')
            else:
                context['login_form'] = login_form
                context['action'] = 'login'
        elif request.POST.get('action') == 'signup':
            signup_form = UserCreationForm(data=request.POST)
            if signup_form.is_valid():
                signup_form.save()
                username = signup_form.cleaned_data.get('username')
                raw_password = signup_form.cleaned_data.get('password1')
                user = authenticate(username=username, password=raw_password)
                login_user(request, user, backend=user.backend)
                return redirect('home')
            else:
                context['signup_form'] = signup_form
                context['action'] = 'signup'

    return render(request, 'get_together/users/login.html', context)

def show_profile(request, user_id):
    user = get_object_or_404(UserProfile, id=user_id)

    teams = user.memberships.all()
    talks = Talk.objects.filter(speaker__user=user)
    badges = user.user.account.badges.all()
    upcoming_events = Event.objects.filter(team=user.personal_team, end_time__gt=datetime.datetime.now()).order_by('start_time')
    recent_events = None#Event.objects.filter(team=user.personal_team, end_time__lte=datetime.datetime.now()).order_by('-start_time')[:3]

    context = {
            'user': user,
            'teams': teams,
            'talks': talks,
            'badges': badges,
            'upcoming_events': upcoming_events,
            'recent_events': recent_events,
    }

    return render(request, 'get_together/users/show_profile.html', context)


def edit_profile(request):
    if not request.user.is_authenticated:
        messages.add_message(request, messages.WARNING, message=_('You must be logged in to edit your profile.'))
        return redirect('login')

    user = request.user
    profile = request.user.profile
    account = request.user.account

    if request.method == 'GET':
        user_form = UserForm(instance=user)
        profile_form = UserProfileForm(instance=profile)

        context = {
            'user': user,
            'profile': profile,
            'user_form': user_form,
            'profile_form': profile_form,
        }
        return render(request, 'get_together/users/edit_profile.html', context)
    elif request.method == 'POST':
        old_email = request.user.email
        user_form = UserForm(request.POST, instance=user)
        profile_form = UserProfileForm(request.POST, request.FILES, instance=profile)
        if user_form.is_valid() and profile_form.is_valid():
            user = user_form.save()
            profile = profile_form.save()
            if user.email != old_email:
                if user.email is None or user.email == "":
                    messages.add_message(request, messages.ERROR, message=_('Your email address has been removed.'))
                    account.is_email_confirmed = False
                    account.save()
                else:
                    messages.add_message(request, messages.WARNING, message=_('Your email address has changed, please confirm your new address.'))
                    return redirect('send-confirm-email')
            return redirect('show-profile', profile.id)
        else:
            context = {
            'user_form': user_form,
            'profile_form': profile_form,
            }
            return render(request, 'get_together/users/edit_profile.html', context)
    else:
     return redirect('home')