pretty-printer for twoninetyex tries

pprint_twoninetyex_trie.py

from example.twoninetyex import SPEC297_TRIE

def pprint(t, lvl=0, remaining_siblings=()):
  indent = lvl * [' ', ' ']

  for i in range(lvl):
      if remaining_siblings[i] > 1:
          indent[i * 2] = '┃'
      if i == lvl - 1:
          if remaining_siblings[i] == 1:
              indent[i * 2] = '┗'
              indent[i * 2 + 1] = '━'
          else:
              indent[i * 2] = '┣'
              indent[i * 2 + 1] = '━'

  if t.record:
      r = t.record[0]
      lbl = ': ' + r['description']
  else:
      lbl = ''
  if t.character == '':
      char = '<START>'
  elif t.character == ' ':
      char = '<END>'
  else:
      char = t.character

  print(f"{''.join(indent)}{char}{lbl}")

  for i, s in enumerate(t.suffixes.values()):
    pprint(s, lvl+1, (*remaining_siblings, len(t.suffixes) - i))
 
if __name__ == '__main__':
    pprint(SPEC297_TRIE)