mirror of
https://github.com/ItsDrike/dotfiles.git
synced 2024-12-27 21:54:34 +00:00
26 lines
753 B
Python
26 lines
753 B
Python
import atexit
|
|
import os
|
|
import readline
|
|
from functools import partial
|
|
from pathlib import Path
|
|
from types import ModuleType
|
|
|
|
cache_xdg_dir = Path(os.environ.get("XDG_CACHE_HOME", str(Path("~/.cache"))))
|
|
cache_xdg_dir.mkdir(exist_ok=True, parents=True)
|
|
|
|
history_file = cache_xdg_dir.joinpath("python_history")
|
|
|
|
readline.read_history_file(history_file)
|
|
|
|
|
|
def write_history(readline: ModuleType, history_file: Path) -> None:
|
|
"""
|
|
We need to get ``readline`` and ``history_file`` as arguments, as it
|
|
seems they get garbage collected when the function is registered and
|
|
the program ends, even though we refer to them here.
|
|
"""
|
|
readline.write_history_file(history_file)
|
|
|
|
|
|
atexit.register(partial(write_history, readline, history_file))
|