49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
import inspect
|
|
import types
|
|
from io import StringIO
|
|
import sys
|
|
import re
|
|
|
|
did_you_mean = re.compile(r'Did you mean (.*)\?', re.IGNORECASE)
|
|
what = 'what?'
|
|
|
|
class Yes(types.ModuleType):
|
|
def __init__(self, other):
|
|
for attr in dir(other):
|
|
setattr(self, attr, getattr(other, attr))
|
|
|
|
def __repr__(self):
|
|
try:
|
|
last_error = sys.last_value
|
|
except:
|
|
return what
|
|
|
|
if not isinstance(last_error, Exception):
|
|
return what
|
|
|
|
error_message = str(last_error)
|
|
|
|
try:
|
|
suggestion = did_you_mean.search(error_message).group(1)
|
|
except Exception as e:
|
|
return what
|
|
|
|
if not suggestion.startswith('print('):
|
|
suggestion_ = f'print(({suggestion}).__repr__())'
|
|
else:
|
|
suggestion_ = suggestion
|
|
|
|
frame = inspect.currentframe()
|
|
locals = frame.f_back.f_locals
|
|
del frame
|
|
|
|
old_stdout = sys.stdout
|
|
eval_output = sys.stdout = StringIO()
|
|
|
|
eval(suggestion_, globals(), locals)
|
|
sys.stdout = old_stdout
|
|
|
|
return(f'{sys.ps1}{suggestion} \n{eval_output.getvalue()[:-1]}')
|
|
|
|
sys.modules[__name__] = Yes(sys.modules[__name__])
|