views.py

# coding=utf-8
from django.contrib.auth import login, authenticate
from django.http import HttpResponseForbidden
from django.core.urlresolvers import reverse_lazy
from django.views.generic import TemplateView, DeleteView, DetailView, ListView, UpdateView, CreateView, FormView, View
from django.views.generic.detail import SingleObjectMixin
from ask_app.models import *
from ask_app.forms import *


# class for template
class AskAppTemplateView(TemplateView):
    template_name = 'base.html'

    # edit this func to see popular users ans tags
    def get_context_data(self, **kwargs):
        data = super(AskAppTemplateView, self).get_context_data(**kwargs)
        #  edit here
        data['best_users'] = User.objects.all()[:5]
        data['tags'] = Tags.objects.all()[:5]
        return data


# classes for working with Question models

# show all questions
class QuestionListView(ListView):
    # use CustomManeger
    queryset = Question.my.newest()
    paginate_by = 5

    def get_context_data(self, **kwargs):
        data = super(QuestionListView, self).get_context_data(**kwargs)
        data['best_users'] = User.objects.all()[:5]
        data['tags'] = Tags.objects.all()[:3]
        return data

# inclusion tags
# show hot answers
class QuestionListViewHot(ListView):
    # use CustomManeger
    queryset = Question.my.hot()
    paginate_by = 5

    def get_context_data(self, **kwargs):
        data = super(QuestionListViewHot, self).get_context_data(**kwargs)
        data['best_users'] = User.objects.all()[:5]
        data['tags'] = Tags.objects.all()[:3]
        return data


# create one question
class QuestionCreateView(CreateView):
    model = Question
    form_class = CreateQuestionForm
    template_name_suffix = '_create_form'
    success_url = reverse_lazy('home-page')

    def get_context_data(self, **kwargs):
        data = super(QuestionCreateView, self).get_context_data(**kwargs)
        #  edit here
        data['best_users'] = User.objects.all()[:5]
        data['tags'] = Tags.objects.all()[:5]
        return data

    def form_valid(self, form):
        form.instance.author = self.request.user
        return super(QuestionCreateView, self).form_valid(form)


# see one question, if we had POST - FormView, GET - DetailView
class QuestionDetailView(View):
    def get(self, request, *args, **kwargs):
        view = QuestionDetailViewShow.as_view()
        return view(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        view = QuestionFormViewAddAnswer.as_view()
        return view(request, *args, **kwargs)


class QuestionDetailViewShow(DetailView):
    model = Question
    success_url = reverse_lazy('home-page')

    # function to add form
    def get_context_data(self, **kwargs):
        context = super(QuestionDetailViewShow, self).get_context_data(**kwargs)
        context['form'] = CreateAnswerForm()
        return context


class QuestionFormViewAddAnswer(SingleObjectMixin, FormView):
    model = Question
    form_class = CreateAnswerForm
    success_url = reverse_lazy('home-page')
    template_name = 'ask_app/question_detail.html'

    # дичь - рефакторь
    def post(self, request, *args, **kwargs):
        if not request.user.is_authenticated():
            return HttpResponseForbidden()
        self.object = self.get_object()
        return super(QuestionFormViewAddAnswer, self).post(request, *args, **kwargs)

    # дичь - рефакторь
    def form_valid(self, form):
        valid = super(QuestionFormViewAddAnswer, self).form_valid(form)
        # add elements to form, then save
        form.instance.author = self.request.user
        form.instance.which_question = self.get_queryset().get(pk=self.object.id)
        form.save()
        # answer.save()
        return valid

    def get_success_url(self):
        return reverse_lazy('question-page', kwargs={'pk': self.object.pk})


# update one question
class QuestionUpdateView(UpdateView):
    model = Question
    template_name_suffix = '_update_form'


class QuestionDeleteView(DeleteView):
    model = Question


#  ------------------------------------- new shit
class QuestionUpdateRating(UpdateView):
    model = Question
    form_class = QuestionUpdateRatingForm
    template_name_suffix = '_ajax_update_form'
    success_url = reverse_lazy('home-page')

    def form_valid(self, form):
        question = self.get_queryset().get(id=self.object.id)
        form.instance.rating = question.rating + int(self.request.POST['rating'])
        form.save()
        return super(QuestionUpdateRating, self).form_valid(form)
# ------------------------------------------------

# classes for working with Answers models


# display all elements
class AnswerListView(ListView):
    model = Answer


# see one answer
class AnswerDetailView(DetailView):
    model = Answer


# change one answer
class AnswerUpdateView(UpdateView):
    model = Answer
    template_name_suffix = '_update_form'


# delete one answer
class AnswerDeleteView(DeleteView):
    model = Answer
    success_url = reverse_lazy('home-page')


# ------------------------------------------ new shit
# ajax update rating
class AnswerChangeRating(UpdateView):
    model = Answer
    form_class = ChangeAnswerRatingForm
    template_name_suffix = '_ajax_update_form'
    success_url = reverse_lazy('home-page')

    def form_valid(self, form):
        answer = self.get_queryset().get(id=self.object.id)
        answer.rating += int(self.request.POST['rating'])
        answer.save()
        return super(AnswerChangeRating, self).form_valid(form)


# ajax update flag
class AnswerChangeFlag(UpdateView):
    model = Answer
    form_class = ChangeAnswerFlagForm
    template_name_suffix = '_ajax_update_form'
    success_url = reverse_lazy('home-page')

    def form_valid(self, form):
        form.save()
        return super(AnswerChangeFlag, self).form_valid(form)
# -------------------------------------------------------

# classes for working with Users models


# display all elements
class UserListView(ListView):
    model = User


# create new user / registration
class UserCreateView(CreateView):
    model = User
    form_class = CreateUserForm
    template_name_suffix = '_create_form'
    success_url = reverse_lazy('home-page')

    # function for auto login after creating new user
    def form_valid(self, form):
        valid = super(UserCreateView, self).form_valid(form)
        username, password = form.cleaned_data.get('username'), form.cleaned_data.get('password1')
        new_user = authenticate(username=username, password=password)
        login(self.request, new_user)
        return valid

    def get_context_data(self, **kwargs):
        data = super(UserCreateView, self).get_context_data(**kwargs)
        #  edit here
        data['best_users'] = User.objects.all()[:5]
        data['tags'] = Tags.objects.all()[:5]
        return data


# see one user
class UserDetailView(DetailView):
    model = User


# change one user
class UserUpdateView(UpdateView):
    model = User
    form_class = ChangeUserForm
    template_name_suffix = '_update_form'
    success_url = reverse_lazy('home-page')


# delete one user
class UserDeleteView(DeleteView):
    model = User
    success_url = reverse_lazy('home-page')