mamweb/odevzdavatko/forms.py

from django import forms
from dal import autocomplete
from django.forms import formset_factory
from django.forms.models import inlineformset_factory
from django.utils import timezone

from seminar.models import Resitel
import seminar.models as m

import logging

# pro přidání políčka do formuláře je potřeba
# - mít v modelu tu položku, kterou chci upravovat
# - přidat do views (prihlaskaView, resitelEditView)
# - přidat do forms
# - includovat do html

class DateInput(forms.DateInput):
    # aby se datum dalo vybírat z kalendáře
    input_type = 'date' 


class PosliReseniForm(forms.Form):
	#FIXME jen podproblémy daného problému
	problem = forms.ModelChoiceField(label='Problém',queryset=m.Problem.objects.all())
	# to_field_name
	#problem = models.ManyToManyField(Problem, verbose_name='problém', help_text='Problém',
	#	through='Hodnoceni')

	# FIXME pridat vice resitelu
	resitel = forms.ModelChoiceField(label="Řešitel",
		queryset=Resitel.objects.all(),
		widget=autocomplete.ModelSelect2(
			url='autocomplete_resitel',
			attrs = {'data-placeholder--id': '-1',
				'data-placeholder--text' : '---',
				'data-allow-clear': 'true'})
    		)


	#resitele = models.ManyToManyField(Resitel, verbose_name='autoři řešení',
	#	help_text='Seznam autorů řešení', through='Reseni_Resitele')
	
	cas_doruceni = forms.DateField(widget=DateInput(),label="Čas doručení")

	#cas_doruceni = models.DateTimeField('čas_doručení', default=timezone.now, blank=True)

	forma = forms.ChoiceField(label="Forma řešení",choices = m.Reseni.FORMA_CHOICES)
	#forma = models.CharField('forma řešení', max_length=16, choices=FORMA_CHOICES, blank=False,
	#	 default=FORMA_EMAIL)

	poznamka = forms.CharField(label='Neveřejná poznámka', required=False)
	#poznamka = models.TextField('neveřejná poznámka', blank=True,
	#	help_text='Neveřejná poznámka k řešení (plain text)')

	#TODO body do cisla
	#TODO prilohy

	##def __init__(self, *args, **kwargs):
	##	super().__init__(*args, **kwargs)
	##	#self.fields['favorite_color'] = forms.ChoiceField(choices=[(color.id, color.name) for color in Resitel.objects.all()])

class NahrajReseniForm(forms.ModelForm):
	class Meta:
		model = m.Reseni
		fields = ('problem',)
		help_texts = {'problem':''} # Nezobrazovat help text ve formuláři
		
		widgets = {'problem':
				autocomplete.ModelSelect2Multiple(
					url='autocomplete_problem_odevzdatelny',
					attrs = {'data-placeholder--id': '-1',
						'data-placeholder--text' : '---',
						'data-allow-clear': 'true'},
				)
		}

ReseniSPrilohamiFormSet = inlineformset_factory(m.Reseni,m.PrilohaReseni, 
		form = NahrajReseniForm,
		fields = ('soubor','res_poznamka'),
		widgets = {'res_poznamka':forms.TextInput()},
		extra = 1,
		can_delete = False,

		)


class JednoHodnoceniForm(forms.ModelForm):
	class Meta:
		model = m.Hodnoceni
		fields = ('problem', 'body', 'deadline_body')
		widgets = {
			'problem': autocomplete.ModelSelect2(
				url='autocomplete_problem_odevzdatelny',   # FIXME: Dovolit i starší?
				)
			}

OhodnoceniReseniFormSet = formset_factory(JednoHodnoceniForm,
		extra = 0,
		)

class PoznamkaReseniForm(forms.ModelForm):
	class Meta:
		model = m.Reseni
		fields = ('poznamka',)

# FIXME: Ideálně by mělo být součástí třídy níž, ale neumím to udělat
DATE_FORMAT = '%Y-%m-%d'

class OdevzdavatkoTabulkaFiltrForm(forms.Form):
	"""Form pro filtrování přehledové odevzdávátkové tabulky

	Inspirováno https://kam.mff.cuni.cz/mffzoom/"""

	# Věci definované níž se importují i ve views pro odevzdávátko (Inspirováno https://docs.djangoproject.com/en/3.1/ref/models/fields/#field-choices)

	RESITELE_RELEVANTNI = 'relevantni'
	RESITELE_NEODMATUROVAVSI = 'neodmaturovavsi'
	RESITELE_CHOICES = [
		(RESITELE_RELEVANTNI, 'Relevantní řešitelé'), # I.e. nezobrazovat prázdné řádky tabulky
		(RESITELE_NEODMATUROVAVSI, 'Všichni bez maturity'),
		# Možná: všechny vč. historických?
		]

	PROBLEMY_MOJE = 'moje'
	PROBLEMY_LETOSNI = 'letosni'
	PROBLEMY_CHOICES = [
		(PROBLEMY_MOJE, 'Moje problémy'), # Letošní problémy, které mají v sobě nebo v nadproblémech přiřazeného daného orga
		(PROBLEMY_LETOSNI, 'Všechny letošní'),
		# TODO: *hlavní problémy, možná všechny...
		# XXX: Chtělo by to i "aktuálně zadané...
		]

	# TODO: Typy problémů (problémy, úlohy, ostatní, všechny)? Jen některá řešení (obodovaná/neobodovaná, víc řešitelů, ...)?


	@classmethod
	def gen_terminy(cls, rocnik=None):
		import datetime
		from time import strftime
		
		from django.db.utils import OperationalError
		try:
			aktualni_rocnik = m.Nastaveni.get_solo().aktualni_rocnik
		except OperationalError:
			# django.db.utils.OperationalError: no such table: seminar_nastaveni
			# Nemáme databázi, takže to selhalo. Pro jistotu vrátíme aspoň dvě možnosti, ať to nepadá dál
			logger = logging.getLogger(__name__)
			logger.error("Rozbitá databáze (před počátečními migracemi?)")
			return [('broken', 'Je to rozbitý'), ('fubar', 'Nefunguje to')]

		# FIXME: Tohle je hnusný monkey patch, mělo by to být nějak zahrnuto výš.
		if rocnik is not None:
			aktualni_rocnik = rocnik

		result = []

		date_str = strftime(DATE_FORMAT, datetime.date.min.timetuple())
		if date_str[0] == '1':  # Někde očividně vrací strftime rok bez nul.
			date_str = "000" + date_str
		result.append((date_str, f"Odjakživa"))

		for deadline in m.Deadline.objects.filter(
				deadline__lte=timezone.now(),
				cislo__rocnik=aktualni_rocnik
				).order_by("deadline"):

			result.append((
				strftime(DATE_FORMAT, deadline.deadline.timetuple()),
				str(deadline)))

		result.append((
			strftime(DATE_FORMAT, datetime.date.today().timetuple()), f"Dnes"))

		return result

	@classmethod
	def gen_initial(cls, rocnik=None):
		terminy = cls.gen_terminy(rocnik)
		initial = {
			'resitele': cls.RESITELE_RELEVANTNI,
			'problemy': cls.PROBLEMY_MOJE,
			# Pokud chceme neaktuální ročník, tak nás nejspíš zajímají všechna řešení…
			'reseni_od': terminy[-2] if rocnik is None else terminy[0],
			'reseni_do': terminy[-1],
			'neobodovane': False,
		}
		return initial

	def __init__(self, *args, rocnik=None, **kwargs):
		if 'initial' not in kwargs:
			super().__init__(initial=self.gen_initial(rocnik), *args, **kwargs)
		else:
			super().__init__(*args, **kwargs)
		# choices jako parametr Select widgetu neumí brát callable, jen iterable, takže si pro jednoduchost můžu rovnou uložit výsledek sem...
		# A "sem" znamená do libovolné metody, protože jinak se jedná o kód, který django spustí při inicializaci a protože potřebujeme databázi, tak by spadnul při vyrábění testdat...
		self.terminy = self.gen_terminy(rocnik)
		self.fields['reseni_od'].widget = forms.Select(choices=self.gen_terminy(rocnik))
		# Pokud chceme neaktuální ročník, tak nás nejspíš zajímají všechna řešení…
		self.fields['reseni_od'].initial = self.terminy[-2] if rocnik is None else self.terminy[0]
		self.fields['reseni_do'].widget = forms.Select(choices=self.gen_terminy(rocnik))
		self.fields['reseni_do'].initial = self.terminy[-1]

	# NOTE: Initial definuji pro jednotlivé fieldy, aby to bylo tady a nebylo potřeba to řešit ve views...
	resitele = forms.ChoiceField(choices=RESITELE_CHOICES)
	problemy = forms.ChoiceField(choices=PROBLEMY_CHOICES)
	
	reseni_od = forms.DateField(input_formats=[DATE_FORMAT])
	reseni_do = forms.DateField(input_formats=[DATE_FORMAT])
	neobodovane = forms.BooleanField(required=False)