Created
July 15, 2020 18:08
-
-
Save switowski/301320fb6388822a76644461219e4386 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import os | |
import sys | |
from contextlib import contextmanager | |
from copy import deepcopy | |
@contextmanager | |
def no_output(): | |
original_stdout = sys.stdout | |
sys.stdout = open(os.devnull, "w") | |
yield | |
sys.stdout = original_stdout | |
def copy_namespace(local_ns): | |
copy_ns = {} | |
for key, value in local_ns.items(): | |
try: | |
copy_ns[key] = deepcopy(value) | |
except TypeError: | |
# TypeError is raised when an object can't be pickled | |
copy_ns[key] = value | |
return copy_ns | |
def combine_multiline(lines): | |
new_lines = [] | |
for line in lines: | |
if line.startswith(" ") and new_lines: | |
new_lines[-1] += line | |
else: | |
new_lines.append(line) | |
return new_lines | |
def curry_clean(ipython): | |
def clean_input(lines): | |
new_lines = [] | |
lines = combine_multiline(lines) | |
local_ns = copy_namespace(ipython.user_ns) | |
for line in lines: | |
with no_output(): | |
try: | |
exec(line, None, local_ns) | |
new_lines.append(line) | |
except Exception: | |
pass | |
return new_lines | |
return clean_input | |
def load_ipython_extension(ipython): | |
ipython.input_transformers_post.append(curry_clean(ipython)) | |
def unload_ipython_extension(ipython): | |
ipython.input_transformers_post = [ | |
f for f in ipython.input_transformers_post if f.__name__ != "clean_input" | |
] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from IPython.core.magic import register_line_magic | |
@register_line_magic | |
def rerunplus(line): | |
get_ipython().run_line_magic("load_ext", "ignore_exceptions") | |
get_ipython().run_line_magic("rerun", line) | |
get_ipython().run_line_magic("unload_ext", "ignore_exceptions") | |
def load_ipython_extension(ipython): | |
ipython.register_magics(rerunplus) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment