from random import random
from typing import Iterable

from django.conf import settings
from django.db import models
from django.db.models import QuerySet
from django.utils import timezone

from .utils.doba import Doba


class Stanoviste(models.Model):
	class Meta:
		db_table = "hra_stanoviste"
		verbose_name = "Stanoviště"
		verbose_name_plural = "Stanoviště"

	user = models.OneToOneField(
		settings.AUTH_USER_MODEL, blank=False, null=False, unique=True,
		verbose_name="uživatel", on_delete=models.PROTECT,
		help_text="Stanoviště musí být přihlášena, aby se nedala přepínat stanoviště",
	)

	nazev = models.CharField(
		"název", max_length=256, blank=False, null=False,
		help_text="zobrazovaný název stanoviště",
	)

	verejny_text = models.TextField(
		"veřejný text", max_length=256, blank=False, null=False,
		help_text="zobrazovaný název stanoviště",
	)

	poradi = models.IntegerField(
		"pořadí", blank=False, null=False,
		help_text="číselné označení stanoviště (aby hráči věděli, že našli všechny)",
	)

	template = models.CharField(
		"template", max_length=256, blank=False, null=False, default="base.html",
		help_text="template tohoto stanoviště (co vidí tým)",
	)


class Tym(models.Model):
	class Meta:
		db_table = "hra_tym"
		verbose_name = "Tým"
		verbose_name_plural = "Týmy"

	user = models.OneToOneField(
		settings.AUTH_USER_MODEL, blank=False, null=False, unique=True,
		verbose_name="uživatel", on_delete=models.PROTECT,
	)

	nazev = models.CharField(
		"název týmu", max_length=256, blank=False, null=False,
	)

	hraci = models.CharField(
		"hráči", max_length=256, blank=True, null=True,
		help_text="seznam hráčů v týmu",
	)

	start = models.DateTimeField(
		"začátek hry", blank=False, null=False, default=timezone.now,
		help_text="čas, kdy tým začal",
	)

	delta = models.FloatField(
		"přidaný čas", blank=False, null=False, default=0,
		help_text="čas (v hodinách), o který se tým posunul svými akcemi",
	)

	def nastav_a_uloz(self, attr_name: str, value: object) -> "Tym":
		self.__setattr__(attr_name, value)
		self.save()
		return self

	@property
	def doba(self) -> Doba:
		return Doba((timezone.now() - self.start).total_seconds() / 60 * settings.TIME_SPEED + self.delta)

	def pricti_cas_a_uloz(self, cas: int) -> "Tym":
		self.delta += cas
		self.save()
		return self

	# Inventář
	tk_nalezena = models.FloatField(
		"čas nalezení třídní knihy", blank=True, null=True, default=None,
		help_text="čas (v hodinách), kdy tým nalezl třídní knihu (pro pořadí a rozhodnutí, zda ji tým už nalezl)",
	)

	ma_penize = models.BooleanField(
		"má tým peníze", blank=False, null=False, default=False,
		help_text="zda si tým vzal peníze a neztratil je",
	)

	ds_bad = models.FloatField(
		"čas chybného založení ds", blank=True, null=True, default=None,
		help_text="čas, kdy tým špatným způsobem požádal o datovou schránku",
	)

	ma_ds = models.BooleanField(
		"má tým ds", blank=False, null=False, default=False,
		help_text="zda si tým založil datovou schránku",
	)

	pocet_bananu = models.IntegerField(
		"počet banánů", blank=False, null=False, default=0,
		help_text="kolik vlastní tým bánánů",
	)

	vi_o_zadosti = models.BooleanField(
		"ví tým o finální žádosti", blank=False, null=False, default=False,
		help_text="ví tým o tom, že potřebuje žádost o náhled do archivů",
	)

	vi_o_mesici = models.BooleanField(
		"ví tým o svitu měsíce", blank=False, null=False, default=False,
		help_text="ví tým o tom, že žádost o náhled do archivů se může podávat pouze za svitu měsíce",
	)

	vi_o_mase = models.BooleanField(
		"ví o tom, že v jídelně už není banán", blank=False, null=False, default=False,
		help_text="už se ptali v jídelně na další banán a zjistili, že je jen MaSo",
	)

	@property
	def vyzvednute_zadosti(self) -> QuerySet["Zadost"]:
		return self.zadosti.filter(vyzvednuta=True)

	@property
	def nevyzvednute_zadosti(self) -> QuerySet["Zadost"]:
		return self.zadosti.filter(studijni__isnull=False, vyzvednuta=False)

	def ma_platnou_zadost(self, zadost_typ: str) -> bool:
		return self.vyzvednute_zadosti.filter(typ=zadost_typ, plati=True).count() > 0

	@property
	def ma_platnou_zadost_o_banan(self) -> bool:
		""" Protože django templates jsou hloupé """
		return self.vyzvednute_zadosti.filter(typ=Zadost.TYP_BANAN, plati=True).count() > 0

	@property
	def ma_platnou_zadost_o_vege(self) -> bool:
		""" Protože django templates jsou hloupé """
		return self.vyzvednute_zadosti.filter(typ=Zadost.TYP_VEGE, plati=True).count() > 0

	def zadosti_na_internatu(self, doba: Doba) -> QuerySet["Zadost"]:
		return self.zadosti.filter(vyzvednuta=False, internat__isnull=False, internat__lte=doba)

	def zadosti_na_poste(self, doba: Doba) -> QuerySet["Zadost"]:
		return self.zadosti.filter(posta__isnull=False, posta__lte=doba, internat__isnull=False, internat__gt=doba)

	@staticmethod
	def studijni_otevreno_nejdrive(doba: Doba) -> Doba:
		if not (  # copy-paste z logika.py
				doba.cas < 8 or
				doba.cas > 15 or
				doba.den_v_tydnu >= 6 or
				(doba.den_v_tydnu == 5 and doba.cas > 13)
		):
			return doba

		if doba.cas > 15 or (doba.den_v_tydnu == 5 and doba.cas > 13):
			doba += Doba.DEN - doba.cas + 8
		else:
			doba = Doba(doba.den*Doba.DEN + 8)
		if doba.den_v_tydnu == 6:
			doba += 2*Doba.DEN
		elif doba.den_v_tydnu == 7:
			doba += Doba.DEN
		return doba

	@staticmethod
	def doruceni(doba: Doba) -> Doba:
		doba += Doba.DEN - doba.cas + 8
		while True:
			if doba.den_v_tydnu == 6 or doba.den_v_tydnu == 7:
				doba += Doba.DEN
			else:
				if random() > 1/3:
					doba += Doba.DEN
				else:
					return doba

	@staticmethod
	def postovna(studijni: Doba) -> Doba:
		return (studijni.den + (settings.POSTOVNA_DEN_V_TYDNU - studijni.den_v_tydnu + 7) % 7) * Doba.DEN + 10

	def poslat_vytistene_zadosti(self, doba: Doba) -> Iterable["Zadost"]:
		if not self.ma_penize:
			return ()
		self.ma_penize = False
		self.save()
		zadosti = self.zadosti.filter(vytistena=True)
		for zadost in zadosti:
			zadost.vytistena = False

			studijni = self.doruceni(doba)
			zadost.studijni = studijni
			posta = Doba((studijni.den + (settings.POSTOVNA_DEN_V_TYDNU - studijni.den_v_tydnu + 7) % 7) * Doba.DEN + 10)
			zadost.posta = posta
			zadost.internat = self.doruceni(posta)

			if zadost.typ == Zadost.TYP_TK:
				zadost.plati = Zadost.plati_tk(doba)
			if zadost.typ == Zadost.TYP_ZADOST:
				zadost.plati = False
			zadost.save()
		return zadosti

	def vytiskni_zadost(self, typ: str) -> tuple["Zadost"]:
		return (Zadost.objects.create(tym=self, typ=typ, vytistena=True,),)

	def posli_zadost_ds(self, doba: Doba, typ: str) -> tuple["Zadost"]:
		studijni = doba
		internat = self.studijni_otevreno_nejdrive(doba)
		if typ == Zadost.TYP_TK:
			return (Zadost.objects.create(tym=self, typ=typ, studijni=studijni, internat=internat, plati=Zadost.plati_tk(doba)),)
		return (Zadost.objects.create(
			tym=self, typ=typ, studijni=studijni, internat=internat, plati=typ != Zadost.TYP_ZADOST,
		),)

	def vyzvedni_internat(self, doba: Doba) -> Iterable["Zadost"]:
		zadosti = self.zadosti_na_internatu(doba)
		for zadost in zadosti:
			zadost.vyzvednuta = True
			zadost.save()
			if zadost.typ == Zadost.TYP_TK:
				self.vi_o_mesici = True
				self.save()
		return zadosti

	def vyzvedni_postu(self, doba: Doba) -> Iterable["Zadost"]:
		zadosti = self.zadosti_na_poste(doba)
		for zadost in zadosti:
			zadost.vyzvednuta = True
			zadost.save()
			if zadost.typ == Zadost.TYP_TK:
				self.vi_o_mesici = True
				self.save()
		return zadosti

	def posli_zadost_studijni(self, doba: Doba, typ: str) -> tuple["Zadost"]:
		studijni = doba
		posta = self.postovna(studijni)
		internat = self.doruceni(doba)
		return (Zadost.objects.create(
			tym=self, typ=typ, studijni=studijni, posta=posta, internat=internat,
			plati=typ != Zadost.TYP_TK and typ != Zadost.TYP_ZADOST,
		),)

	ovesna_kase = models.IntegerField(
		blank=False, null=False, default=0,
	)


class Zadost(models.Model):
	class Meta:
		db_table = "hra_zadost"
		verbose_name = "Žádost"
		verbose_name_plural = "Žádosti"

	tym = models.ForeignKey(Tym, related_name="zadosti", blank=False, null=False, on_delete=models.CASCADE)

	TYP_ZADOST = "zadost"
	TYP_BANAN = "banan"
	TYP_VEGE = "vege"
	# TYP_BAD_TK = "bad_tk"
	TYP_TK = "tk"
	TYP_CHOICES = [
		(TYP_ZADOST, "Žádost"),
		(TYP_BANAN, "Žádost o banán"),
		(TYP_VEGE, "Žádost o více vegetariánskou stravu v jídelně"),
		# (TYP_BAD_TK, "Neplatná žádost o přístup k záznamům o třídní knize"),
		(TYP_TK, "Žádost o přístup k záznamům o třídní knize"),
	]

	typ = models.CharField(
		"toto je ", max_length=8, blank=False, null=False, choices=TYP_CHOICES,
	)

	studijni = models.FloatField(
		"žádost dorazí na studijní", blank=True, null=True, default=None,
	)

	posta = models.FloatField(
		"tým si může žádost vyzvednout na internátu od", blank=True, null=True, default=None,
		help_text="(do self.internat)",
	)

	internat = models.FloatField(
		"tým si může žádost vyzvednout na poště od", blank=True, null=True, default=None,
	)

	vyzvednuta = models.BooleanField(
		"tým si tuto žádost již vyzvedl", blank=False, null=False, default=False,
	)

	vytistena = models.BooleanField(
		"tým má u sebe tuto žádost vytištěnou",
		blank=False, null=False, default=False,
	)

	plati = models.BooleanField(
		"tato žádost platí", blank=False, null=False, default=True,
		help_text="Aktuálně má uplatnění pouze u Žádosti o přístup k záznamům o třídní knize",
	)
	@staticmethod
	def plati_tk(doba: Doba) -> bool:
		return (
			not doba.je_nov and
			doba.je_noc and
			doba.pocasi == 0
		)

	def __str__(self) -> str:
		return {
			"zadost": "Žádost",
			"banan": "Žádost o banán",
			"vege": "Žádost o více vegetariánskou stravu v jídelně",
			"tk": "Žádost o přístup k záznamům o třídní knize"
		}[self.typ]