B 't\O@sdZddlmZddlmZddlmZmZmZm Z m Z ddl m Z m Z ddlmZmZddlZddlZd d d d d gZGdd d ee eZGdd d eZGdd d eZGdd d eZGdd d eZdS)a] Implementations for the history of a `Buffer`. NOTE: Notice that there is no `DynamicHistory`. This doesn't work well, because the `Buffer` needs to be able to attach an event handler to the event when a history entry is loaded. This loading can be done asynchronously and making the history swappable would probably break this. )unicode_literals)Event)AsyncGeneratorItemFrom ensure_futureconsume_async_generatorgenerator_to_async_generator)ABCMetaabstractmethod)with_metaclass text_typeNHistoryThreadedHistory DummyHistory FileHistoryInMemoryHistoryc@s`eZdZdZddZddZddZdd Zd d Zd d Z e ddZ ddZ e ddZ dS)rzg Base ``History`` class. This also includes abstract methods for loading/storing history. cCsd|_g|_t||_dS)NF)_loading_loaded_stringsr _item_loaded)selfr5lib/python3.7/site-packages/prompt_toolkit/history.py__init__#szHistory.__init__c#s,fdd}ttdd|dVdS)z Consume the asynchronous generator: `load_history_strings_async`. This is only called once, because once the history is loaded, we don't have to load it again. csjd|jdS)z9 Got one string from the asynchronous history generator. rN)rinsertrZfire)string)rrr add_string0sz*History._start_loading..add_stringcSsdS)NFrrrrr7sz(History._start_loading..)ZcancelZ item_callbackN)rrload_history_strings_async)rrr)rr_start_loading)s  zHistory._start_loadingcCs|jsd|_t|dS)z Start loading the history. TN)rrr)rrrr start_loading?szHistory.start_loadingcCs|jS)z5 Event which is triggered when a new item is loaded. )r)rrrrget_item_loaded_eventEszHistory.get_item_loaded_eventcCs|jS)zJ Get the strings from the history that are loaded so far. )r)rrrr get_stringsIszHistory.get_stringscCs|j|||dS)z Add string to the history. N)rappend store_string)rrrrr append_stringOs zHistory.append_stringccsdS)z This should be a generator that yields `str` instances. It should yield the most recent items first, because they are the most important. (The history can already be used, even when it's only partially loaded.) Nr)rrrrload_history_stringsXs zHistory.load_history_stringsccs.x(|D]}t|tstt|Vq WdS)a Asynchronous generator for history strings. (Probably, you won't have to override this.) This should return an iterable that can yield both `str` and `Future` objects. The `str` objects have to be wrapped in a `AsyncGeneratorItem` object. If we drop Python 2 support in the future, this could become a true asynchronous generator. N)r& isinstancer AssertionErrorr)ritemrrrrds z"History.load_history_strings_asynccCsdS)z9 Store the string in persistent storage. Nr)rrrrrr$tszHistory.store_stringN)__name__ __module__ __qualname____doc__rrr r!r"r%r r&rr$rrrrrs csBeZdZdZd fdd ZddZddZd d Zd d ZZ S)ra Wrapper that runs the `load_history_strings` generator in a thread. Use this to increase the start-up time of prompt_toolkit applications. History entries are available as soon as they are loaded. We don't have to wait for everything to be loaded. Ncs0t|tstd|f||_tt|dS)NzGot %r)r'rr(historysuperrr)rr.) __class__rrrszThreadedHistory.__init__cCs t|jjS)zp Asynchronous generator of completions. This yields both Future and Completion objects. )r r.r&)rrrrrsz*ThreadedHistory.load_history_strings_asynccCs |jS)N)r.r&)rrrrr&sz$ThreadedHistory.load_history_stringscCs|j|dS)N)r.r$)rrrrrr$szThreadedHistory.store_stringcCs d|jfS)NzThreadedHistory(%r))r.)rrrr__repr__szThreadedHistory.__repr__)N) r*r+r,r-rrr&r$r1 __classcell__rr)r0rr{s  c@s eZdZdZddZddZdS)rzM :class:`.History` class that keeps a list of all strings in memory. cCsgS)Nr)rrrrr&sz$InMemoryHistory.load_history_stringscCsdS)Nr)rrrrrr$szInMemoryHistory.store_stringN)r*r+r,r-r&r$rrrrrsc@s(eZdZdZddZddZddZdS) rzB :class:`.History` object that doesn't remember anything. cCsgS)Nr)rrrrr&sz!DummyHistory.load_history_stringscCsdS)Nr)rrrrrr$szDummyHistory.store_stringcCsdS)Nr)rrrrrr%szDummyHistory.append_stringN)r*r+r,r-r&r$r%rrrrrscs0eZdZdZfddZddZddZZS)rzD :class:`.History` class that stores all strings in a file. cs||_tt|dS)N)filenamer/rr)rr3)r0rrrszFileHistory.__init__c sggfdd}tj|jrt|jdL}x>|D]6}|d}|drd|ddq8|gq8W|WdQRXtS)Ncs$r ddd}|dS)N)joinr#)r)linesstringsrraddsz-FileHistory.load_history_strings..addrbzutf-8+r) ospathexistsr3opendecode startswithr#reversed)rr9fliner)r7r8rr&s   z FileHistory.load_history_stringsc sZt|jdDfdd}|dtjx|dD]}|d|q8WWdQRXdS)NZabcs|ddS)Nzutf-8)writeencode)t)rCrrrEsz'FileHistory.store_string..writez # %s  z+%s )r?r3datetimeZnowsplit)rrrErDr)rCrr$s  zFileHistory.store_string)r*r+r,r-rr&r$r2rr)r0rrs )r-Z __future__rZutilsrZ eventlooprrrrr abcr r Zsixr r rIr<__all__objectrrrrrrrrrs"  ^!