utils.py

def print_pretty_ast(ast_text):
  i = 0
  newline = False
  for ch in str(ast_text):
    if ch == '(':
      if not newline:
        print(end='')
      #newline or print(end='')

      print(ch)
      i += 2
      newline = True
    elif ch == ')':
      if not newline:
        print()
      #newline or print()

      i -= 2
      newline = True
      print(' '*i + ch)
    else:
      if newline:
        print(' '*i, end='')
      #not newline or print(' '*i, end='')

      print(ch, end='')
      newline = False

def stringify(val):
  if isinstance(val, bool) and val == True:
    return "true"
  if isinstance(val, bool) and val == False:
    return "false"
  if isinstance(val, float) and val.is_integer():
    return str(int(val))
  return str(val)

def lexing_error(message, lineno):
  print(f'{Colors.RED}[Line {lineno}]: {message} {Colors.WHITE}')
  import sys
  sys.exit(1)

def parse_error(message, lineno):
  print(f'{Colors.RED}[Line {lineno}]: {message} {Colors.WHITE}')
  import sys
  sys.exit(1)

def runtime_error(message, lineno):
  print(f'{Colors.RED}[Line {lineno}]: {message} {Colors.WHITE}')
  import sys
  sys.exit(1)

class Colors:
  WHITE     = '\033[0m'
  BLUE      = '\033[94m'
  CYAN      = '\033[96m'
  GREEN     = '\033[92m'
  YELLOW    = '\033[93m'
  RED       = '\033[91m'