# encoding: utf-8 import sys from six.moves import builtins def resetquit(builtins): """Redefine builtins 'quit' and 'exit' not so close stdin """ def __call__(self, code=None): raise SystemExit(code) __call__.__name__ = "FakeQuitCall" builtins.quit.__class__.__call__ = __call__ def monkeypatch_quit(): if "site" in sys.modules: resetquit(builtins)