Guest User

Untitled

a guest
Dec 10th, 2019
52
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.68 KB | None | 0 0
  1. class Node:
  2.  
  3.     def __init__(self, name):
  4.         self.parent = None
  5.         # self.children = set()
  6.         self.name = name
  7.         self.orbits = -1
  8.  
  9.     def __str__(self):
  10.         if self.parent:
  11.             r = "( {} : {} ".format(self.name, self.parent.name)
  12.         else:
  13.             r = "( {} : {} ".format(self.name, "-")
  14.  
  15.         if self.orbits != -1:
  16.             r += "@ {} ".format(self.orbits)
  17.         r += ")"
  18.         return r
  19.  
  20.     def set_parent(self, n):
  21.         self.parent = n
  22.  
  23.     def get_orbits_to_root(self):
  24.         if self.parent:
  25.             return self.parent.get_orbits_to_root() + 1
  26.         else:
  27.             return 0
  28.  
  29.     __repr__ = __str__
Advertisement
Add Comment
Please, Sign In to add comment