mirror of https://github.com/conqp/rcon
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.2 KiB
50 lines
1.2 KiB
"""Wrapper for readline support."""
|
|
|
|
from logging import Logger
|
|
from os import name
|
|
from pathlib import Path
|
|
|
|
if name == 'posix':
|
|
from readline import read_history_file, write_history_file
|
|
else:
|
|
read_history_file = write_history_file = lambda _: None
|
|
|
|
|
|
__all__ = ['CommandHistory']
|
|
|
|
|
|
HIST_FILE = Path.home().joinpath('.rconshell_history')
|
|
|
|
|
|
class CommandHistory:
|
|
"""Context manager for the command line history."""
|
|
|
|
__slots__ = ('logger',)
|
|
|
|
def __init__(self, logger: Logger):
|
|
"""Sets the logger to use."""
|
|
self.logger = logger
|
|
|
|
def __enter__(self):
|
|
"""Loads the history file."""
|
|
try:
|
|
read_history_file(HIST_FILE)
|
|
except FileNotFoundError:
|
|
self.logger.warning(
|
|
'Could not find history file: %s', HIST_FILE
|
|
)
|
|
except PermissionError:
|
|
self.logger.error(
|
|
'Insufficient permissions to read: %s', HIST_FILE
|
|
)
|
|
|
|
return self
|
|
|
|
def __exit__(self, *_):
|
|
"""Writes to the history file."""
|
|
try:
|
|
write_history_file(HIST_FILE)
|
|
except PermissionError:
|
|
self.logger.error(
|
|
'Insufficient permissions to write: %s', HIST_FILE
|
|
)
|
|
|