mam/mamweb
11
0
Fork 0
Code Issues Pull requests 7 Projects Wiki Activity Actions

Treelib: Add first_brother

Browse source
This commit is contained in:
Pavel 'LEdoian' Turinsky 2020-03-18 22:36:07 +01:00
parent 8fcb1b0871
commit 63dd0da97c
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
seminar/treelib.py
View file

@ -18,13 +18,17 @@ def safe_pred(node):
except ObjectDoesNotExist: except ObjectDoesNotExist:
return None return None
def first_brother(node):
brother = node
while safe_pred(brother) is not None:
brother = safe_pred(brother)
return brother
# A to samé pro .father_of_first # A to samé pro .father_of_first
def safe_father_of_first(node): def safe_father_of_first(node):
first_brother = node first = first_brother(node)
while safe_pred(first_brother) is not None:
first_brother = safe_pred(first_brother)
try: try:
return first_brother.father_of_first return first.father_of_first
except ObjectDoesNotExist: except ObjectDoesNotExist:
return None return None