From 137f5f3f78e190a4ce62e896188e9b818734deca Mon Sep 17 00:00:00 2001
From: kienle <kienle@passcal.nmt.edu>
Date: Mon, 25 Sep 2023 22:31:26 -0600
Subject: [PATCH] Implement log file iterator
---
.../reftek_data/reftek_reader/log_file_reader.py | 16 ++++++++++++++--
1 file changed, 14 insertions(+), 2 deletions(-)
diff --git a/sohstationviewer/model/reftek_data/reftek_reader/log_file_reader.py b/sohstationviewer/model/reftek_data/reftek_reader/log_file_reader.py
index e0c1e6a99..83aaf4ab7 100644
--- a/sohstationviewer/model/reftek_data/reftek_reader/log_file_reader.py
+++ b/sohstationviewer/model/reftek_data/reftek_reader/log_file_reader.py
@@ -25,8 +25,20 @@ class LogFile:
def __iter__(self):
return self
- def __next__(self):
- return self.file.readline()
+ def __next__(self) -> List[str]:
+ line = self.file.readline()
+ if line == '':
+ raise StopIteration
+ while line == '\n':
+ line = self.file.readline()
+ packet = []
+ # We have to check that we are not at the end of the file as well.
+ while line != '\n' and line != '':
+ packet.append(line)
+ line = self.file.readline()
+ if line == '':
+ break
+ return packet
def __del__(self):
self.file.close()
--
GitLab