IoT-Systems/MapView/fileReader.py

from csv import reader
import config
from dataclasses import dataclass

@dataclass
class Accelerometer:
  x: int
  y: int
  z: int

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)
        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 stopReading(self, *args, **kwargs):
        if self.file:
            self.file.close()
        self.file_reader = None