U pu f@sdZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z ddl mZdddZd d Zd d Zd dZe GdddeZddZdS)z %store magic for lightweight persistence. Stores variables, aliases and macros in IPython's database. To automatically restore stored variables at startup, add this to your :file:`ipython_config.py` file:: c.StoreMagics.autorestore = True N) UsageError)Magics magics_class line_magic) skip_doctest)BoolcCsN|jdi}|dkr8|D]\}}|j||qn|j|||dS)Nstored_aliases)dbgetitems alias_managerZ define_alias)ipalias staliaseskvrA/tmp/pip-unpacked-wheel-bxwx4lqe/IPython/extensions/storemagic.pyrestore_aliasess rc Csp|j}|dD]Z}tj|}z ||}Wn2tk r^td|tdtdYqX||j |<qdS)N autorestore/*zEUnable to restore variable '%s', ignoring (use %%store -d to forget!)zThe error was:r) r keysospathbasenameKeyErrorprintsysexc_infouser_ns)r r keyjustkeyobjrrrrefresh_variables#s   r"cCs|jdg|jd<dS)NZdhistZ_dh)r r rr rrr restore_dhist2sr$cCst|t|t|dS)N)r"rr$r#rrr restore_data6sr%csFeZdZdZedddjddZfddZee d d d Z Z S) StoreMagicszMLightweight persistence for python variables. Provides the %store magic.FzdIf True, any %store-d variables will be automatically restored when IPython starts. )helpT)configcs4tt|j|d|jj||jr0t|jdS)N)shell)superr&__init__r)Z configurablesappend autorestorer%)selfr) __class__rrr+HszStoreMagics.__init__cCs|j|ddd\}}|}|j}|j}d|krz |d}Wn,tk rh}ztd|W5d}~XYnBXz|d|=Wn0tk r}ztd ||W5d}~XYnXn(d |kr|d D] } || =qnd |kr^|rR|D]j} z|d| } WnJtk rBzt || d Wn"tk r<t d| YnXYqX| |j | <qnt |nx|s|d } | | rttt| } nd} t ddt| d}|j}| D]4}tj|}t ||t||dddfqnt|dkr|ddrtj|dd}|ddrDt|ddd}nt|ddd}|r||d} t d|d| jj|ft| tsddlm}|| |n | | | !ds| dW5QRXdS|D]} z|j | } Wntk rx| }z|j"#|}Wn2t$k rB}ztd||W5d}~XYnX|di}|||<||d<t d ||fYdSXt%t&'| d!d"}|d#krt t()d$| | fdS| |d| <t d%| | jjfqdS)&aLightweight persistence for python variables. Example:: In [1]: l = ['hello',10,'world'] In [2]: %store l Stored 'l' (list) In [3]: exit (IPython session is closed and started again...) ville@badger:~$ ipython In [1]: l NameError: name 'l' is not defined In [2]: %store -r In [3]: l Out[3]: ['hello', 10, 'world'] Usage: * ``%store`` - Show list of all variables and their current values * ``%store spam bar`` - Store the *current* value of the variables spam and bar to disk * ``%store -d spam`` - Remove the variable and its value from storage * ``%store -z`` - Remove all variables from storage * ``%store -r`` - Refresh all variables, aliases and directory history from store (overwrite current vals) * ``%store -r spam bar`` - Refresh specified variables and aliases from store (delete current val) * ``%store foo >a.txt`` - Store value of foo to new file a.txt * ``%store foo >>a.txt`` - Append value of foo to file a.txt It should be noted that if you change the value of a variable, you need to %store it again if you want to persist the new value. Note also that the variables will need to be pickleable; most basic python types can be safely %store'd. Also aliases can be %store'd across sessions. To remove an alias from the storage, use the %unalias magic. Zdrzstring)modedrz'You must provide the variable to forgetNz autorestore/zCan't delete variable '%s'zrr)rzno stored variable or alias %sz(Stored variables and their in-db values:z%-zs -> %sz 2>z>>azutf-8)encodingwzWriting '%s' (%s) to file '%s'.)pprint zUnknown variable '%s'rzAlias stored: %s (%s)__name__r1__main__a6 Warning:%s is %s Proper storage of interactively declared classes (or instances of those classes) is not possible! Only instances of classes in real modules on file system can be %%store'd. zStored '%s' (%s))*Z parse_optionssplitr)r IndexErrorr BaseExceptionrrrrrr%sortmaxmaplenstrr rrrrepr startswith expanduserlstripopenZevr0r? isinstancer=writeendswithr Zretrieve_alias ValueErrorgetattrinspect getmoduletextwrapdedent)r.Z parameter_soptsZargslargsr r Ztodelerargr!varssizefmtr varr ZfnamZfilr=namecmdrmodnamerrrstoreNs. $     *            zStoreMagics.store)r1) r? __module__ __qualname____doc__rtagr-r+rrrb __classcell__rrr/rr&<s r&cCs|tdS)zLoad the extension in IPython.N)Zregister_magicsr&r#rrrload_ipython_extensionsrh)N)rerSrrrUZIPython.core.errorrZIPython.core.magicrrrZIPython.testing.skipdoctestrZ traitletsrrr"r$r%r&rhrrrrs     -