import sys class Context: def __init__(self): self._vars = {} def get(self, key, default=None): return self._vars.get(key, default) def __getitem__(self, key): return self._vars[key] def __setitem__(self, key, value): self._vars[key] = value class Run: def __init__(self): self._name = None self._context = None def get_context(self): return self._context def start(self, name): self._context = Context() self._name = name def end(self): sys.stdout.write('OK: {}\n'.format(self._name))