mirror of
https://github.com/Rhinemann/IoT-Systems.git
synced 2026-03-14 20:50:39 +02:00
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from csv import reader
|
|
import config
|
|
from domain.accelerometer import Accelerometer
|
|
|
|
|
|
|
|
class FileReader:
|
|
def __init__(
|
|
self, data_filename: str,
|
|
) -> None:
|
|
self.file_path = data_filename
|
|
pass
|
|
|
|
def read(self):
|
|
return self.getNextValue()
|
|
|
|
def startReading(self, *args, **kwargs):
|
|
self.file = open(self.file_path, newline='')
|
|
self.file_reader = reader(self.file)
|
|
file_header = next(self.file_reader)
|
|
|
|
self.x_idx = file_header.index('X')
|
|
self.y_idx = file_header.index('Y')
|
|
self.z_idx = file_header.index('Z')
|
|
|
|
def getNextValue(self):
|
|
row = next(self.file_reader, None)
|
|
if row is None:
|
|
self._rewind_file()
|
|
row = next(self.file_reader)
|
|
x = int(row[self.x_idx])
|
|
y = int(row[self.y_idx])
|
|
z = int(row[self.z_idx])
|
|
return Accelerometer(x=x, y=y, z=z)
|
|
|
|
def _rewind_file(self):
|
|
self.file.seek(0)
|
|
self.file_reader = reader(self.file)
|
|
next(self.file_reader)
|
|
|
|
def stopReading(self, *args, **kwargs):
|
|
if self.file:
|
|
self.file.close()
|
|
self.file_reader = None
|
|
|
|
|