Context Manager

ContextManagerExample.py

# custom context manager

import contextlib, sys

p = sys.stdout.write # shorthand for print (without spaces)

@contextlib.contextmanager
def printcolour(colour):
  p('\033[{0}m'.format(colour))
  yield
  p('\033[m')


p('Hello would you like some ')
with printcolour(93):
  p('differently coloured ')
  p('text ')
p('or just normal text?\n')