shelfzilla/shelfzilla/apps/account/views.py

from itertools import chain
from django.views.generic import View
from django.template import RequestContext
from django.shortcuts import render_to_response, get_object_or_404
from django.contrib.auth import logout, authenticate, login
from django.contrib.auth.decorators import login_required
from django.utils.decorators import method_decorator
from django.utils.translation import ugettext as _
from django.http import HttpResponseRedirect
from django.contrib import messages
from django.contrib.auth import login
from django.core.urlresolvers import reverse

from .forms import LoginForm, PasswordChangeForm, RegistrationForm
from .models import User
from .signals import user_registered
from shelfzilla.apps.manga.models import (
    UserReadVolume, UserHaveVolume, UserWishlistVolume
)


class LoginView(View):
    template = 'users/login.html'

    def get(self, request):
        if request.user.is_authenticated():
            return HttpResponseRedirect('/')

        context = {
            'login_form': LoginForm()
        }

        ctx = RequestContext(request, context)
        return render_to_response(self.template, context_instance=ctx)

    def post(self, request):
        if request.user.is_authenticated():
            return HttpResponseRedirect('/')

        login_form = LoginForm(request.POST)

        if login_form.is_valid():
            user = login_form.authenticate()
            login(request, user)

            messages.success(
                request,
                _('Logged in successfully.')
            )

            return HttpResponseRedirect('/')

        context = {
            'login_form': login_form,
        }

        ctx = RequestContext(request, context)
        return render_to_response(self.template, context_instance=ctx)


class LogoutView(View):
    def get(self, request):
        logout(request)

        messages.success(
            request,
            _('Logged out successfully')
        )

        return HttpResponseRedirect('/')


class UserProfileView(View):
    tempalte = 'users/profile.html'
    template_section = 'users/profile/{}.html'

    def get(self, request, username, section='summary'):
        user = get_object_or_404(User, username=username)
        data = {
            'item': user,
            'section': section
        }
        if section != 'summary':
            template = self.template_section.format(section)
        else:
            template = self.tempalte
        data = self.get_context_from_section(request, section, data, user)

        ctx = RequestContext(request, data)
        return render_to_response(template, context_instance=ctx)

    def get_summary(self, request, context, user):
        owned_list = UserHaveVolume.objects.filter(user=user)
        wishlisted_list = UserWishlistVolume.objects.filter(user=user)
        read_list = UserReadVolume.objects.filter(user=user)

        timeline = sorted(
            chain(owned_list, wishlisted_list, read_list),
            key=lambda model: model.date,
            reverse=True
        )[:20]
        context['timeline'] = timeline
        return context

    def get_context_from_section(self, request, section, context, user):
        method = getattr(self, 'get_{}'.format(section), None)

        if method:
            context = method(request, context, user)

        return context


class AccountView(View):
    template = 'account/main.html'

    @method_decorator(login_required)
    def get(self, request):
        data = {
            'item': request.user,
            'form_password': PasswordChangeForm(request.user)
        }

        ctx = RequestContext(request, data)
        return render_to_response(self.template, context_instance=ctx)

    @method_decorator(login_required)
    def post(self, request):
        form_password = PasswordChangeForm(request.user, data=request.POST)

        if form_password.is_valid():
            # Do stuff
            form_password.save()
            messages.success(request, _('Password changed.'))
            return HttpResponseRedirect(reverse('account'))

        data = {
            'item': request.user,
            'form_password': form_password
        }

        ctx = RequestContext(request, data)
        return render_to_response(self.template, context_instance=ctx)


class RegisterView(View):
    template = 'account/register.html'
    form_class = RegistrationForm

    def get(self, request):
        form_data = {}
        
        if 'code' in request.GET:
            form_data['access_code'] = request.GET['code']

        data = {
            'form': self.form_class(initial=form_data)
        }

        ctx = RequestContext(request, data)
        return render_to_response(self.template, context_instance=ctx)

    def post(self, request):
        form = self.form_class(request.POST)
        if form.is_valid():
            form.save()
            user = authenticate(username=request.POST.get('username'),
                                password=request.POST.get('password1'))
            login(request, user)
            messages.success(request, _('Welcome to the community! :)'))
            user_registered.send(sender=self.__class__, user=user)
            return HttpResponseRedirect(reverse('homepage'))

        ctx = RequestContext(request, {'form': form})
        return render_to_response(self.template, context_instance=ctx)