fdZddlZddlZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZe Gd d eZe d Zd Ze d ZGdde Z Gdde Z!Gdde!Z"GddejFZ$ejJdejLZ'dZ(dZ)y)z* History related magics and functionality N)Path)LoggingConfigurable) decorator)undoc)locate_profile) AnyBoolDictInstanceIntegerListUnicodeUnion TraitErrordefaultobservec(eZdZdZdZdZdZdZy)DummyDBz[Dummy DB that will act as a black hole for history. Only used in the absence of sqlitecgSN)argskwargss O/var/www/cvtools/html/venv/lib/python3.12/site-packages/IPython/core/history.pyexecutezDummyDB.execute)s cyrrselfrrs rcommitzDummyDB.commit, rcyrrrs r __enter__zDummyDB.__enter__/r!rcyrrrs r__exit__zDummyDB.__exit__2r!rN)__name__ __module__ __qualname____doc__rr r#r%rrrrr$s*   rrc6|jsgS||g|i|S)z9Decorator: return an empty list in the absence of sqlite.)enabled)frakws ronly_when_enabledr/6s% <<   R  ri@cd ||g|i|S#tjtjf$r}|xjdz c_|jj d|j ||j dk7r|j|jkDr$d|_|jj dnW|j jr>Z '''$*@*@@!+bc'')4>>004>>3F3FFGnn++~~**,44=("++//1;;=EEc3OC"[036&1*&MPS&SG T#Z/#5G%%g.UW^_ LLNI A s! $H/G1H*#H/)H**H/c4eZdZdZddZ ddZd dZd dZy) HistoryAccessorBasez(An abstract class for History Accessors ctrNotImplementedError)rnrawoutputinclude_latests rget_tailzHistoryAccessorBase.get_tailt!!rNctrrX)rpatternr[ search_rawr\rZuniques rsearchzHistoryAccessorBase.searchws!!rctrrX)rsessionstartstopr[r\s r get_rangezHistoryAccessorBase.get_range{r_rctrrX)rrangestrr[r\s rget_range_by_strz$HistoryAccessorBase.get_range_by_str~r_r TFF*TTFNFr1NTFTF)r&r'r(r)r^rdrirlrrrrVrVqs!2"8<,1"""rrVceZdZdZdZdZeeee gdjdZ e ddjdZ ed jdZeZed d Zdfd Zdd ZedZdZddZeedZedZeddZe ddZeddZddZ xZ!S)HistoryAccessorzAccess the history database without adding to it. This is intended for use by standalone history tools. IPython shells use HistoryManager, below, which is a subclass of this.raPath to file to use for SQLite history database. By default, IPython will put the history database in the IPython profile directory. If you would rather share one history among profiles, you can set this value in each, so that they are consistent. Due to an issue with fcntl, SQLite is known to misbehave on some NFS mounts. If you see IPython hanging, try setting this to something on a local disk, e.g:: ipython --HistoryManager.hist_file=/tmp/ipython_hist.sqlite you can also use the specific value `:memory:` (including the colon at both end but not the back ticks), to avoid creating an history file. helpTconfiga*enable the SQLite history set enabled=False to disable the SQLite history, in which case there will be no stored history, no SQLite connection, and no background saving thread. This may be necessary in some threaded environments where IPython is embedded. zOptions for configuring the SQLite connection These options are passed as keyword args to sqlite3.connect when establishing database connections. dbc|d}ttjf}t||s&|jj d|}t |y)zCvalidate the db, since it can be an Instance of two different typesnewz/.db must be sqlite3 Connection or DummyDB, not N)rr6 Connection isinstance __class__r&r)rchanger|connection_typesmsgs r _db_changedzHistoryAccessor._db_changedsNUm#W%7%78#/0^^,,c3CS/ !1rc tt| di||r||_ |j|j y#t$r|j ||_Y2wxYw)aCreate a new history accessor. Parameters ---------- profile : str The name of the profile from which to open history. hist_file : str Path to an SQLite history database stored by IPython. If specified, hist_file overrides profile. config : :class:`~traitlets.config.loader.Config` Config object. hist_file can also be set through this. Nr)superrt__init__r<r_get_hist_file_namerM)rprofiler<traitsrs rrzHistoryAccessor.__init__s] ot-77 &DN ? NN   ?!55g>DN ?s <AAc0tt|dz S)aFind the history file for the given profile name. This is overridden by the HistoryManager subclass, to use the shell's active profile. Parameters ---------- profile : str The name of a profile which has a history file. history.sqlite)rr)rrs rrz#HistoryAccessor._get_hist_file_namesN7+,/???rc$|jst|_yttj tj z}|j|jt jt|jfi||_|j5|jjd|jjd|jjddddd|_ y#1swYd|_ yxYw)z8Connect to the database, and create tables if necessary.N) detect_typeszCREATE TABLE IF NOT EXISTS sessions (session integer primary key autoincrement, start timestamp, end timestamp, num_cmds integer, remark text)zCREATE TABLE IF NOT EXISTS history (session integer, line integer, source text, source_raw text, PRIMARY KEY (session, line))zCREATE TABLE IF NOT EXISTS output_history (session integer, line integer, output text, PRIMARY KEY (session, line))r)r+rrzdictr6PARSE_DECLTYPESPARSE_COLNAMESupdateconnection_optionsconnectr?r<rr9)rrs rrMzHistoryAccessor.init_dbs||iDG 7#:#:7;Q;Q#QR d--.//#dnn"5@@ WW  GGOOM  GGOO4  GGOO<  &$% ' &$% s AC??Dcy)zWOverridden by HistoryManager to dump the cache before certain database lookups.Nrrs rwriteout_cachezHistoryAccessor.writeout_cache s rc|rdnd}d}|rd}d|z}|r|dz }d|d|d |z}|jj||} |r d | D} |r d | DS| S) aPrepares and runs an SQL query for the history database. Parameters ---------- sql : str Any filtering expressions to go after SELECT ... FROM ... params : tuple Parameters passed to the SQL query (to replace "?") raw, output : bool See :meth:`get_range` latest : bool Select rows with max (session, line) Returns ------- Tuples as :meth:`get_range` source_rawsourcehistoryz6history LEFT JOIN output_history USING (session, line)z!history.%s, output_history.outputz", MAX(session * 128 * 1024 + line)zSELECT session, line, z FROM  c3&K|] }|dd yw)Nr).0rows r z+HistoryAccessor._run_sql..-s+3s8+sc34K|]\}}}}||||ffywrr)rseslininpouts rrz+HistoryAccessor._run_sql../s$I/AsCcS#Sz*Is)rzr) rsqlparamsr[r\latesttogetsqlfrom this_querrycurs r_run_sqlzHistoryAccessor._run_sqlsv$!$  NG7%?E  9 9E=BGLsR ggook62 +s+C ISI I rc\d}|jj||fjS)a)Get info about a session. Parameters ---------- session : int Session number to retrieve. Returns ------- session_id : int Session ID number start : datetime Timestamp for the start of the session. end : datetime Timestamp for the end of the session, or None if IPython crashed. num_cmds : int Number of commands run, or None if IPython crashed. remark : unicode A manually set description. z)SELECT * from sessions where session == ?)rzrfetchone)rrfquerys rget_session_infoz HistoryAccessor.get_session_info2s).<wwuwj1::<= ? AND line < ?zline>=?zWHERE session==? AND %sr)r)rrfrgrhr[r\ lineclausers rrizHistoryAccessor.get_rangesN8 1Jud+F"Ju%F}}6C$*FD Drc#rKt|D]%\}}}|j|||||D]}|'yw)aGet lines of history from a string of ranges, as used by magic commands %hist, %save, %macro, etc. Parameters ---------- rangestr : str A string specifying ranges, e.g. "5 ~2/1-4". If empty string is used, this will return everything from current session's history. See the documentation of :func:`%history` for the full details. raw, output : bool As :meth:`get_range` Returns ------- Tuples as :meth:`get_range` rN)extract_hist_rangesri)rrkr[r\sesssrNlines rrlz HistoryAccessor.get_range_by_strsI&.h7 JD!QtQs6J    s57)r)r)TFFrmrorqrr)"r&r'r(r)r9r=rr rrtagr<r r+r rrrzrrrrrTrMrrr/rrr^rdrirl __classcell__rs@rrtrtsP;  $# $ cc%(4  cc    cc  B T]""8 @%%> B==0##87;,1''R#D#DJrrtc`eZdZdZeddZedgZedgZeZ e ddZ e Z e ZeZedd j%d Zed d j%d ZeZeZeddZeej2dZedZedZedZedZe jBdZ"dfd Z#d dZ$e%d dZ&dZ'dZ(d!dZ)d"fd Z*e+d#dZ,d$dZ-d%fd Z.d dZ/dZ0dZ1dZ2e%d dZ3xZ4S)&HistoryManagerzHA class to organize all history-related functionality in one place. z1IPython.core.interactiveshell.InteractiveShellABCT) allow_nonerdir_histcP tjgS#t$rgcYSwxYwr)rcwdOSErrorrs r_dir_hist_defaultz HistoryManager._dir_hist_defaults) HHJ<  I s  %%Fz9Should the history database include output? (default: no)rvrxrz}Write to database every x commands (higher values save disk access & power). Values of 1 or less effectively disable caching.z(IPython.core.history.HistorySavingThreadz(exit|quit)(\s*\(.*\))?$c tt| d||d|tj|_tj |_tj |_ |j|jr;|jdk7r+t!||_|j"j%yyy#tj$r2|jjd|jdd|_YwxYw)zGCreate a new history manager associated with a shell instance. )shellryzBFailed to create history session in %s. History will not be saved.T)exc_infor2Nr)rrr threadingEvent save_flagLockdb_input_cache_lockdb_output_cache_lock new_sessionr6r8r:r;r<r+HistorySavingThread save_threadrg)rrryrrs rrzHistoryManager.__init__s nd, 5  "*#,>>#3 $-NN$4! (     <~?\d+)/)? (?P\d+)? ((?P[\-:]) ((?P~?\d+)/)? (?P\d+))? $c#K|dk(rdy|jD]/}tj|}|s|jd}|r/t |}|jd}|r t |n|dz}n|jdspd}d}|jdd k(r|dz }|jdxsd }|jd xs|}t |j d d }t |j d d }||k\sJd ||k(r |||f||dft |dz|D] }|ddf |d|f2yw)a!Turn a string of history ranges into 3-tuples of (session, start, stop). Empty string results in a `[(0, 1, None)]`, i.e. "everything from current session". Examples -------- >>> list(extract_hist_ranges("~8/5-~7/4 2")) [(-8, 5, None), (-7, 1, 5), (0, 2, 3)] r)rr1NNrgendr1 startsesssep-0endsess~z.start session must be earlier than end session)splitrange_rergroupintrIrJ) ranges_str range_strrmatchrgr.r/r3rs rrrsoR%%'  *  W% JE,,u%C!#c(uqyC<< ,EC << # % 1HCLL-4 ,,y)6Y ))#c23 gooc#./)#U%UU# i eS) ) %&&)A+w/ "DD/ ! "3? sEEc0|dk(r t|S|d|S)z0Helper function to format line numbers properly.r#)r?)rfrs r_format_linenor>s!|4yt $$r)*r)r!rFpathlibrrr6rtraitlets.config.configurablerrIPython.utils.decoratorsr IPython.pathsr traitletsrr r r r r rrrrrobjectrr/rErTrVrtrThreadrrVERBOSEr6rr>rrrrGs0  =*(    $ f  " ! !  * *Z"-""V)Vr }*_}*@ +)**+^ 2::  jj  . b%r