mamweb/korektury/views.py

from django.shortcuts import get_object_or_404, render
from django.views import generic
from django.conf import settings
from django.http import HttpResponseForbidden
from django.core.mail import EmailMessage
from django.db.models import Count,Q

from .models import Oprava,Komentar,KorekturovanePDF, Organizator

class KorekturyListView(generic.ListView):
	model = KorekturovanePDF
	queryset = KorekturovanePDF.objects.annotate(
		k_oprave_cnt=Count('oprava',distinct=True,filter=Q(oprava__status=Oprava.STATUS.K_OPRAVE)),
		opraveno_cnt=Count('oprava',distinct=True,filter=Q(oprava__status=Oprava.STATUS.OPRAVENO)),
		neni_chyba_cnt=Count('oprava',distinct=True,filter=Q(oprava__status=Oprava.STATUS.NENI_CHYBA)),
		k_zaneseni_cnt=Count('oprava',distinct=True,filter=Q(oprava__status=Oprava.STATUS.K_ZANESENI)),
		)
	template_name = 'korektury/seznam.html'

class KorekturyAktualniListView(KorekturyListView):
	def get_queryset(self, *args, **kwargs):
		queryset=super().get_queryset()
		queryset=queryset.exclude(status="zastarale")
		return queryset

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)
		context['selected'] = 'aktualni'
		return context



class KorekturyZastaraleListView(KorekturyListView):
	def get_queryset(self, *args, **kwargs):
		queryset=super().get_queryset()
		queryset=queryset.filter(status="zastarale").order_by("-cas")
		return queryset

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)
		context['selected'] = 'zastarale'
		return context

class KorekturySeskupeneListView(KorekturyAktualniListView):
	template_name = 'korektury/seskupeny_seznam.html'

	# {% regroup %} potřebuje dostat správně setříděné názvy, aby nedělal duplikáty
	def get_queryset(self, *args, **kwargs):
		qs = super().get_queryset(**kwargs)
		return reversed(sorted(qs, key=lambda it: it.cislo_a_tema))

### Korektury
class KorekturyView(generic.TemplateView):
	model = Oprava
	template_name = 'korektury/opraf.html'

	def setup(self, request, *args, **kwargs):
		super().setup(request, *args, **kwargs)
		self.pdf_id = self.kwargs["pdf"]
		self.pdf = get_object_or_404(KorekturovanePDF, id=self.pdf_id)

	def post(self, request, *args, **kwargs):
		q = request.POST
		scroll = q.get('scroll')

		# prirazeni autora podle prihlaseni
		autor_user = request.user
		# pokud existuje ucet (user), ale neni to organizator = 403
		autor = Organizator.objects.filter(osoba__user=autor_user).first()
		if not autor:
			return HttpResponseForbidden()

		if not scroll:
			scroll = 0

		action = q.get('action')
		if (action == ''):  # Přidej
			x = int(q.get('x'))
			y = int(q.get('y'))
			text = q.get('txt')
			strana = int(q.get('img-id')[4:])

			op = Oprava(x=x,y=y, autor=autor, text=text, strana=strana, pdf=self.pdf)
			op.save()
			self.send_email_notification_komentar(op,autor)
		elif (action == 'del'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			op.delete()
		elif (action == 'update'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			text = q.get('txt')
			op.autor = autor
			op.text = text
			op.save()
		elif (action == 'undone'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			op.status = op.STATUS.K_OPRAVE
			op.save()
		elif (action == 'done'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			op.status = op.STATUS.OPRAVENO
			op.save()
		elif (action == 'ready'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			op.status = op.STATUS.K_ZANESENI
			op.save()
		elif (action == 'wontfix'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			op.status = op.STATUS.NENI_CHYBA
			op.save()
		elif (action == 'comment'):
			id = int(q.get('id'))
			op = Oprava.objects.get(id=id)
			text = q.get('txt')
			kom = Komentar(oprava=op,autor=autor,text=text)
			kom.save()
			self.send_email_notification_komentar(op,autor)
		elif (action == 'update-comment'):
			id = int(q.get('id'))
			kom = Komentar.objects.get(id=id)
			text = q.get('txt')
			kom.text = text
			kom.autor = autor
			kom.save()
		elif (action == 'del-comment'):
			id = int(q.get('id'))
			kom = Komentar.objects.get(id=id)
			kom.delete()
		elif (action == 'set-state'):
			status = q.get('state')
			assert status in KorekturovanePDF.STATUS.values
			self.pdf.status = status
			self.pdf.save()
		context = self.get_context_data()
		context['scroll'] = scroll
		context['autor'] = autor
		return render(request, 'korektury/opraf.html',context)

	def send_email_notification_komentar(self, oprava, autor):
		''' Rozesle e-mail pri pridani komentare / opravy,
			ktery obsahuje text vlakna opravy.
		'''

		# parametry e-mailu
		#odkaz = "https://mam.mff.cuni.cz/korektury/{}/".format(oprava.pdf.pk)
		from django.urls import reverse
		odkaz = self.request.build_absolute_uri(reverse('korektury', kwargs={'pdf': oprava.pdf.pk}))
		odkaz = f"{odkaz}#op{oprava.id}-pointer"
		from_email = 'korekturovatko@mam.mff.cuni.cz'
		subject = 'Nová korektura od {} v {}'.format(autor, oprava.pdf.nazev)
		texty = [(oprava.autor.osoba.plne_jmeno(),oprava.text)]
		for kom in Komentar.objects.filter(oprava=oprava):
			texty.append((kom.autor.osoba.plne_jmeno(),kom.text))
		optext = "\n\n\n".join([": ".join(t) for t in texty])
		text = u"Text komentáře:\n\n{}\n\n=== Konec textu komentáře ===\n\
				\nodkaz do korekturovátka: {}\n\
				\nVaše korekturovátko\n".format(optext, odkaz)

		# Prijemci e-mailu
		emails = set()
		# e-mail autora korektury
		email = oprava.autor.osoba.email
		if email:
			emails.add(email)

		# nalezeni e-mailu na autory komentaru
		for komentar in oprava.komentar_set.all():
			email_komentujiciho = komentar.autor.osoba.email
			if email_komentujiciho:
				emails.add(email_komentujiciho)

		# zodpovedni orgove
		for org in oprava.pdf.orgove.all():
			email_zobpovedny = org.osoba.email
			if email_zobpovedny:
				emails.add(email_zobpovedny)

		# odstran e-mail autora opravy
		email = autor.osoba.email
		if email:
			emails.discard(email)

		EmailMessage(
			subject=subject,
			body=text,
			from_email=from_email,
			to=list(emails),
		).send()

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)
		context['pdf'] = self.pdf
		context['img_prefix'] = self.pdf.get_prefix()
		context['img_path'] = settings.KOREKTURY_IMG_DIR
		context['img_indexes'] = range(self.pdf.stran)
		opravy = Oprava.objects.filter(pdf=self.pdf_id)
		zasluhy = {}
		for o in opravy:
			if o.autor in zasluhy:
				zasluhy[o.autor]+=1
			else:
				zasluhy[o.autor]=1
			o.komentare = o.komentar_set.all()
			for k in o.komentare:
				if k.autor in zasluhy:
					zasluhy[k.autor] += 1
				else:
					zasluhy[k.autor] = 1
		zasluhy = [
			{'autor': jmeno, 'pocet': pocet}
			for (jmeno, pocet) in zasluhy.items()
		]
		zasluhy.sort(key=lambda z: z['pocet'], reverse=True)

		strany = set(o.strana for o in opravy)
		opravy_na_stranu = [{'strana': s, 'op_id': opravy.filter(strana=s)} for s in strany]
		context['opravy_strany'] = opravy_na_stranu

		context['k_oprave_cnt'] = opravy.filter(status='k_oprave').count() 
		context['opraveno_cnt'] = opravy.filter(status='opraveno').count() 
		context['neni_chyba_cnt'] = opravy.filter(status='neni_chyba').count() 
		context['k_zaneseni_cnt'] = opravy.filter(status='k_zaneseni').count() 
		
		context['opravy'] = opravy
		context['zasluhy'] = zasluhy
		return context