thestudentlife/Manhattan-Project

from django.core.urlresolvers import reverse_lazy
from django.views.generic.edit import CreateView, DeleteView
from django.views.generic.detail import DetailView
from mainsite.models import Article, Issue, ArticleForm, Album
from workflow.models import WArticle, Revision
from workflow.views import group_required
from django.utils import timezone
from django.shortcuts import render, redirect
from django.core.urlresolvers import reverse

class ArticleDetailView(DetailView):
    model = Article
    template_name = "articles/warticle.html"

class ArticleCreateView(CreateView):
    model = Article
    fields = ['title', 'content', 'section']
    template_name = 'articles/new_article.html'
    success_url = reverse_lazy('latest_article')

    def form_valid(self, form):
        article = form.save(commit=False)
        article.issue = Issue.objects.get(pk=self.kwargs['issue_id']);
        article.save()
        article.authors.add(self.request.user.profile)
        album = Album(article=article)
        album.save()
        workflowArticle = WArticle(article=article, status='')
        workflowArticle.save()
        return super(ArticleCreateView, self).form_valid(form)

@group_required('silver')
def article_edit(request, issue_id, pk):
    article = Article.objects.get(pk=pk)
    locked_by = request.user in article.edited_by.all()
    if request.method == 'GET':
        original_time = article.updated_date.second
        form = ArticleForm(instance=article)
        return render(request, 'articles/edit_article.html',
                      {'form': form, 'article': article, 'time': original_time,'locked_by':locked_by})
    else:
        original_time = request.POST['time']
        form = ArticleForm(request.POST, instance=article)
        if form.is_valid():
            currently_saved_time = article.updated_date.second
            if original_time == str(currently_saved_time):
                article = form.save(commit=False)
                article.updated_date = timezone.now()
                article.save()
            else:
                overwrite_message = "This article was edited while you were making changes. " \
                                    "Please click edit again if you indeed want to overwrite the recent edit. " \
                                    "Otherwise, save your edits elsewhere, exit this page, and try editing again."
                return render(request, 'articles/edit_article.html',
                              {'form': form, 'article': article,
                               'time': currently_saved_time, 'overwrite': overwrite_message,'locked_by':locked_by})
            revision = Revision(article=article,
                                editor=request.user.profile, body=article.content)
            revision.save()
            form.save_m2m()
            return render(request, 'articles/edit_article.html',
                          {'form': form, 'article': article, 'time': original_time, 'locked_by': locked_by})
        else:
            return render(request, 'articles/edit_article.html', {'form': form, 'article': article,'locked_by':locked_by})

class ArticleDeleteView(DeleteView):
    model = Article
    template_name = "articles/article_confirm_delete.html"
    success_url = reverse_lazy('whome')

group_required('plastic')
def latest_article(request):
    article = Article.objects.order_by('-updated_date')[0]
    return redirect(reverse('warticle', args=[article.issue.pk, article.pk]))