| | """Support for interactive macros in IPython""" |
| |
|
| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| | import re |
| |
|
| | from IPython.utils.encoding import DEFAULT_ENCODING |
| |
|
| | coding_declaration = re.compile(r"#\s*coding[:=]\s*([-\w.]+)") |
| |
|
| | class Macro(object): |
| | """Simple class to store the value of macros as strings. |
| | |
| | Macro is just a callable that executes a string of IPython |
| | input when called. |
| | """ |
| |
|
| | def __init__(self,code): |
| | """store the macro value, as a single string which can be executed""" |
| | lines = [] |
| | enc = None |
| | for line in code.splitlines(): |
| | coding_match = coding_declaration.match(line) |
| | if coding_match: |
| | enc = coding_match.group(1) |
| | else: |
| | lines.append(line) |
| | code = "\n".join(lines) |
| | if isinstance(code, bytes): |
| | code = code.decode(enc or DEFAULT_ENCODING) |
| | self.value = code + '\n' |
| | |
| | def __str__(self): |
| | return self.value |
| |
|
| | def __repr__(self): |
| | return 'IPython.macro.Macro(%s)' % repr(self.value) |
| | |
| | def __getstate__(self): |
| | """ needed for safe pickling via %store """ |
| | return {'value': self.value} |
| | |
| | def __add__(self, other): |
| | if isinstance(other, Macro): |
| | return Macro(self.value + other.value) |
| | elif isinstance(other, str): |
| | return Macro(self.value + other) |
| | raise TypeError |
| |
|
