m0060: likvidace potomku bez blbosti kolem

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: