Fix DirectoryDict getitem.

This commit is contained in:
Darko Poljak 2018-10-12 19:08:36 +02:00
parent 3b7f39cab7
commit 151edc6398
1 changed files with 11 additions and 2 deletions

View File

@ -58,7 +58,7 @@ class FileList(collections.MutableSequence):
with open(self.path) as fd:
for line in fd:
except EnvironmentError as e:
except EnvironmentError:
# error ignored
return lines
@ -127,7 +127,16 @@ class DirectoryDict(collections.MutableMapping):
def __getitem__(self, key):
with open(os.path.join(self.path, key), "r") as fd:
value =
# if there is no value/empty line then return ''
# if there is only one value then return that value
# if there are multiple lines in file then return list
if not value:
return ''
elif len(value) == 1:
return value[0]
return value
except EnvironmentError:
raise KeyError(key)