a Fb@sdZddlZddlmZddlmZddlmZddlm Z m Z m Z dd l m Z d Zd ZGd d d eZGdddeZGdddeZdS)zb An application for managing IPython history. To be invoked as the `ipython history` subcommand. N)Path) Application)BaseIPythonApplication)BoolIntDict) ask_yes_nozTrim the IPython history database to the last 1000 entries. This actually copies the last 1000 entries to a new database, and then replaces the old file with the new. Use the `--keep=` argument to specify a number other than 1000. aClear the IPython history database, deleting all entries. Because this is a destructive operation, IPython will prompt the user if they really want to do this. Passing a `-f` flag will force clearing without a prompt. This is an handy alias to `ipython history trim --keep=0` c@sjeZdZeZedddjddZedddjddZ e e dddiiej fd Z e e d d Zd d ZdS) HistoryTrimFz/Keep the old history file as history.sqlite.helpTZconfigi/Number of recent lines to keep in the database.backup)rzHistoryTrim.keep)keepc Cst|jj}|d}t|}t|d|jdf}t||jkr^t d|jt ddSt d|j| | |r|dd}t|d|f}t|d |f}| |d }d} | r| d7} |d t| }qt|} | d | d | d | |rf| 4| d|| d|| d|Wdn1s\0Y| |jrd} |d| } | r| d7} |d| } q|| t d| n|||dS)Nzhistory.sqlitez^SELECT session, line, source, source_raw FROM history ORDER BY session DESC, line DESC LIMIT ?rz=There are already at most %d entries in the history database.z>Not doing anything. Use --keep= argument to keep fewer entriesz/Trimming history to the most recent %d entries.rzCSELECT session, line, output FROM output_history WHERE session >= ?zMSELECT session, start, end, num_cmds, remark FROM sessions WHERE session >= ?zhistory.sqlite.newzCREATE 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))z'insert into sessions values (?,?,?,?,?)z$insert into history values (?,?,?,?)z)insert into output_history values (?,?,?)zhistory.sqlite.old.%dz Backed up longer history file to)r profile_dirlocationsqlite3ZconnectlistZexecuterlenprintpopreversecloseexistsstrZcommitZ executemanyrrenameunlink) selfrZ hist_fileZconZinputsZ first_sessionZoutputsZsessionsZ new_hist_fileiZnew_dbZbackup_hist_filer!6lib/python3.9/site-packages/IPython/core/historyapp.pystart6s`          ,    zHistoryTrim.startN)__name__ __module__ __qualname__trim_hist_help descriptionrtagrrrrdictr flagsaliasesr#r!r!r!r"r !s( r c@sjeZdZeZedddZedddjddZ e e dddiie j fd ddiie j fd Z e Zd d Zd S) HistoryClearrrr Fz"Don't prompt user for confirmationTrforcer )r.fcCs"|jstddddrt|dS)Nz#Really delete all ipython history? Zno)defaultZ interrupt)r.r r r#rr!r!r"r#s zHistoryClear.startN)r$r%r&clear_hist_helpr(rrrr)r.rr*r r+r,r#r!r!r!r"r-xs&  r-c@sHeZdZdZdZeeeejdfe e jdfdZ ddZ dS) HistoryAppzipython-historyz$Manage the IPython history database.r)ZtrimclearcCsL|jdur>td|jt|||dn |jSdS)Nz0No subcommand specified. Must specify one of: %sr)Zsubappr subcommandskeysZprint_descriptionZprint_subcommandsexitr#r1r!r!r"r#s  zHistoryApp.startN) r$r%r&namer(rr*r splitlinesr-r5r#r!r!r!r"r3sr3)__doc__rpathlibrZtraitlets.config.applicationrZ applicationrZ traitletsrrrZutils.ior r'r2r r-r3r!r!r!r"s     W