from contextlib import contextmanager
from sys import getrecursionlimit, setrecursionlimit

@contextmanager
def recursion_limit(limit):
  """ Temporarily manage recursion limit. """
  oldlimit = getrecursionlimit()
  setrecursionlimit(limit)
  try:
      yield
  finally:
      setrecursionlimit(oldlimit)