25 lines
554 B
Python
25 lines
554 B
Python
|
import os
|
||
|
|
||
|
def open_all():
|
||
|
"""open_all() → [file(rb)]
|
||
|
Open the default sshwot files"""
|
||
|
|
||
|
try:
|
||
|
homedir = os.environ['HOME']
|
||
|
except KeyError as err:
|
||
|
raise KeyError('$HOME is not set') from err
|
||
|
|
||
|
# If the directory doesn't exist, just return empty
|
||
|
try:
|
||
|
sshwot_dir = os.listdir(os.path.join(homedir, '.sshwot'))
|
||
|
except FileNotFoundError:
|
||
|
return []
|
||
|
|
||
|
# Read all the .sshwot files from /.sshwot by default
|
||
|
files = []
|
||
|
for dir_entry in sshwot_dir:
|
||
|
if dir_entry.split('.')[-1] == 'sshwot':
|
||
|
files.append(open(dir_entry, 'rb'))
|
||
|
|
||
|
return files
|