eabfb6ad88
remove unnecessary variable return an empty set if the file doesn't exist xrange is range in python3
37 lines
894 B
Python
37 lines
894 B
Python
import os
|
|
|
|
|
|
__all__ = ['load_history', 'save_history']
|
|
|
|
|
|
def history_path():
|
|
"""
|
|
Create the path to the history log
|
|
"""
|
|
HOME = os.path.expanduser('~')
|
|
XDG_CONFIG_HOME = os.getenv('XDG_CACHE_HOME', os.path.join(HOME, '.config'))
|
|
path = os.path.join(XDG_CONFIG_HOME, 'rtv', 'history.log')
|
|
return path
|
|
|
|
|
|
def load_history():
|
|
"""
|
|
Load the history file into memory if it exists
|
|
"""
|
|
path = history_path()
|
|
if os.path.exists(path):
|
|
with open(path) as history_file:
|
|
return set([line.replace('\n', '') for line in history_file])
|
|
return set()
|
|
|
|
|
|
def save_history(history):
|
|
"""
|
|
Save the visited links to the history log
|
|
"""
|
|
path = history_path()
|
|
with open(path, 'w+') as history_file:
|
|
for i in range(200):
|
|
if not history:
|
|
break
|
|
history_file.write(history.pop() + '\n') |