feat: handle empty rows

This commit is contained in:
ІМ-24 Владислав Коваленко
2026-03-03 17:05:53 +00:00
parent a52da042ef
commit 4a81434cb6
+12 -8
View File
@@ -24,14 +24,18 @@ class FileReader:
self.z_idx = file_header.index('Z') self.z_idx = file_header.index('Z')
def getNextValue(self): def getNextValue(self):
row = next(self.file_reader, None) while True:
if row is None: row = next(self.file_reader, None)
self._rewind_file() if row is None:
row = next(self.file_reader) self._rewind_file()
x = int(row[self.x_idx]) continue
y = int(row[self.y_idx]) try:
z = int(row[self.z_idx]) x = int(row[self.x_idx])
return Accelerometer(x=x, y=y, z=z) y = int(row[self.y_idx])
z = int(row[self.z_idx])
return Accelerometer(x=x, y=y, z=z)
except Exception as e:
continue
def _rewind_file(self): def _rewind_file(self):
self.file.seek(0) self.file.seek(0)