Skip to content

Commit a06d894

Browse files
add file existence check in __BaseFile._to_read, add read_lines in JsonLineFile
1 parent ecf5673 commit a06d894

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

pysenal/io/file.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,8 @@ def __change_mode(self, mode):
320320
self._file = open(self.filename, mode, encoding=self.encoding)
321321

322322
def _to_read(self):
323+
if not os.path.exists(self.filename):
324+
raise FileNotFoundError(self.filename)
323325
self.__change_mode('r')
324326

325327
def _to_write(self):
@@ -403,6 +405,13 @@ def read_line(self):
403405
for line in self._file:
404406
yield json.loads(line)
405407

408+
def read_lines(self, skip_empty=False, *args, **kwargs):
409+
self._to_read()
410+
items = []
411+
for line in self._file:
412+
items.append(json.loads(line))
413+
return items
414+
406415
def write(self, data):
407416
super().write(data)
408417

0 commit comments

Comments
 (0)