From d22f8ec13687efe22ed208a0780527bfbb3f8381 Mon Sep 17 00:00:00 2001
From: "Pavel Turinsky (DebianVM @ Zr)" <ledoian@matfyz.cz>
Date: Wed, 30 Jun 2021 01:04:48 +0000
Subject: [PATCH] m0060: likvidace potomku bez blbosti kolem

---
 seminar/migrations/0060_spoj_stromy.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/seminar/migrations/0060_spoj_stromy.py b/seminar/migrations/0060_spoj_stromy.py
index 59b2943d..d0826ca1 100644
--- a/seminar/migrations/0060_spoj_stromy.py
+++ b/seminar/migrations/0060_spoj_stromy.py
@@ -6,6 +6,15 @@ from django.db import migrations
 
 from django.db.models import Q
 
+def nastav_koren(koren, node):
+	node.root = koren
+	node.save()
+	
+	if node.succ:
+		nastav_koren(koren, node.succ)
+	if node.first_child:
+		nastav_koren(koren, node.first_child)
+
 def pridej_potomka(rodic, potomek):
 	# Daný vrchol bude posledním potomkem rodiče
 	uz_ma_deti = False
@@ -23,8 +32,7 @@ def pridej_potomka(rodic, potomek):
 			posledni = posledni.succ
 
 	# Nastavíme kořen:
-	potomek.root = rodic.root
-	potomek.save()
+	nastav_koren(rodic.root, potomek)
 
 	# Připojíme vrchol:
 	if uz_ma_deti: