mamweb/seminar/migrations/0059_spoj_stromy.py

# -*- coding: utf-8 -*-
# Generated by Django 1.11.20 on 2019-05-29 03:26
from __future__ import unicode_literals

from django.db import migrations

from django.db.models import Q

def pridej_potomka(rodic, potomek):
	# Daný vrchol bude posledním potomkem rodiče
	uz_ma_deti = False
	posledni = None

	# Přidávaný potomek by neměl mít následovníka -- přidáváme potomka, ne podles.
	if potomek.succ:
		raise ValueError("Potomek má následovníka, to je velmi podezřelé!")

	# Najdeme aktuálně posledního potomka:
	if rodic.first_child:
		uz_ma_deti = True
		posledni = rodic.first_child
		while posledni.succ:
			posledni = posledni.succ

	# Nastavíme kořen:
	potomek.root = rodic.root
	potomek.save()

	# Připojíme vrchol:
	if uz_ma_deti:
		posledni.succ = potomek
		posledni.save()
	else:
		rodic.first_child = potomek
		rodic.save()

def pokacej_les(apps, schema_editor):
	# Teď je potřeba všechny TreeNody příslušející k zadaným problémům připojit
	# do hlavního stromu
	# Tohle je jednoduchá verze: nejdřív témátka a seriály, pak úložky a pohádky,
	# pak články a konfery, pak vzoráky, všechno setříděné podle kódu (FIXME?)

	# Kopírování je častým zdrojem chyb!
	Cislo = apps.get_model('seminar', 'Cislo')
	Tema = apps.get_model('seminar', 'Tema')
	Konfera = apps.get_model('seminar', 'Konfera')
	Clanek = apps.get_model('seminar', 'Clanek')
	Uloha = apps.get_model('seminar', 'Uloha')
	Problem = apps.get_model('seminar', 'Problem')
	Pohadka = apps.get_model('seminar', 'Pohadka')

	for c in Cislo.objects.all().reverse():
		cnode = c.cislonode

		# Témata a seriály:
		relevantni_temata = Tema.objects.filter(Q(cislo_zadani = c) | Q(cislo_reseni = c)).order_by('kod')
		# Téma dáme do prvního čísla, kde se vyskytne
		for t in relevantni_temata:
			tnode = t.temavcislenode
			if t.cislo_zadani and t.cislo_reseni:
				assert(t.cislo_zadani <= t.cislo_reseni)
				if t.cislo_reseni == c:
					# Už by mělo být přidané do čísla zadání
					continue
			else:
				# Patří sem (buď je to jediné číslo, nebo je to číslo zadání)
				pridej_potomka(cnode, tnode)

		# Úložky (zadání) a pohádky
		for u in Uloha.objects.filter(cislo_zadani = c).order_by('kod'):
			unode = u.ulohazadaninode
			pohadky_pred = Pohadka.objects.filter(uloha_old = u.problem, pred = True)
			pohadky_po = Pohadka.objects.filter(uloha_old = u.problem, pred = False)
			for p in pohadky_pred:
				# FIXME: Chceme pohádku do stromu?
				print('!!!   !!!   !!!   !!!   !!!')
				print('FIXME: Jak uložit pohádku??')
				print('!!!   !!!   !!!   !!!   !!!')
			pridej_potomka(cnode, unode)
			for p in pohadky_po:
				# FIXME: Chceme pohádku do stromu?
				print('!!!   !!!   !!!   !!!   !!!')
				print('FIXME: Jak uložit pohádku??')
				print('!!!   !!!   !!!   !!!   !!!')

		# Články
		for cl in Clanek.objects.filter(cislo = c).order_by('kod'):
			clnode = cl.claneknode
			pridej_potomka(cnode, clnode)

		# Konfery
		for k in Konfera.objects.all():
			knode = k.konferanode
			if k.reseni and knode.root is None:
				# Takováhle konfera nejspíš neexistuje
				raise NotImplementedError("Konfery neumím zapojit do stromu")

		# Vzoráky
		for u in Uloha.objects.filter(cislo_reseni = c).order_by('kod'):
			unode = u.ulohavzoraknode
			pridej_potomka(cnode, unode)


class Migration(migrations.Migration):

	dependencies = [
		('seminar', '0058_problem_to_uloha_tema_clanek'),
	]

	operations = [
		migrations.RunPython(pokacej_les),
	]