o h@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)BoolcCsP|jdi}|dur|D] \}}|j||qdS|j|||dS)Nstored_aliases)dbgetitems alias_managerZ define_alias)ipalias staliaseskvrSc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/extensions/storemagic.pyrestore_aliasess rc Csn|j}|dD],}tj|}z||}Wnty.td|tdtdYqw||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|dSN)r"rr$r#rrr restore_data6s r&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)configcs8tt|j|d|jj||jrt|jdSdS)N)shell)superr'__init__r*Z configurablesappend autorestorer&)selfr* __class__rrr,Hs zStoreMagics.__init__cCs|j|ddd\}}|}|j}|j}d|vrLz|d}Wnty/}ztd|d}~wwz|d|=WdStyK}ztd ||d}~wwd |vr]|d D]} || =qUdSd |vr|r|D]1} z|d| } Wn!tyzt || d Wntyt d| YnwYqew| |j | <qedSt |dS|s|d } | | rttt| } nd} t ddt| d}|j}| D]}tj|}t ||t||dddfqdSt|dkrr|ddrrtj|dd}|ddrt|ddd}nt|ddd}|O||d} t d|d| jj|ft| ts?ddlm}|| |n | | | !dsW| dWddSWddSWddS1skwYdS|D]x} z|j | } WnAty| }z|j"#|}Wnt$y}ztd||d}~ww|di}|||<||d<t d ||fYdSwt%t&'| d!d"}|d#krt t()d$| | fdS| |d| <t d%| | jjfqtdS)&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__r2__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 expanduserlstripopenZevr1r@ isinstancer>writeendswithr Zretrieve_alias ValueErrorgetattrinspect getmoduletextwrapdedent)r/Z parameter_soptsZargslargsr r Ztodelerargr!varssizefmtr varr ZfnamZfilr>namecmdrmodnamerrrstoreNs.        $                 zStoreMagics.store)r2) r@ __module__ __qualname____doc__rtagr.r,rrrc __classcell__rrr0rr'<s r'cCs|tdS)zLoad the extension in IPython.N)Zregister_magicsr'r#rrrload_ipython_extensionsrir%)rfrTrrrVZIPython.core.errorrZIPython.core.magicrrrZIPython.testing.skipdoctestrZ traitletsrrr"r$r&r'rirrrrs       -