forked from bpython/bpython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsitefix.py
More file actions
54 lines (40 loc) · 1.1 KB
/
sitefix.py
File metadata and controls
54 lines (40 loc) · 1.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import functools
import imp
import sys
from six.moves import builtins
from bpython._py3compat import py3
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)
orig_reload = None
if py3:
import importlib
if hasattr(importlib, 'reload'):
orig_reload = importlib.reload
elif hasattr(imp, 'reload'):
orig_reload = imp.reload
else:
orig_reload = builtins.reload
def reload(module):
if module is sys:
orig_stdout = sys.stdout
orig_stderr = sys.stderr
orig_stdin = sys.stdin
r = orig_reload(sys)
sys.stdout = orig_stdout
sys.stderr = orig_stderr
sys.stdin = orig_stdin
return r
else:
return orig_reload(module)
functools.update_wrapper(reload, orig_reload)
def monkeypatch_reload():
if not py3:
builtins.reload = reload