Compare commits
11 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 4a81434cb6 | |||
| a52da042ef | |||
| 11c590cf25 | |||
| 550d29c48c | |||
| 8a1327b10a | |||
| db1b7cc6fc | |||
| a899ef6a6e | |||
| 95176ea467 | |||
| 081a2d4240 | |||
| 92c20ef612 | |||
| c31363aa57 |
+1422
-1
File diff suppressed because one or more lines are too long
@@ -0,0 +1,7 @@
|
|||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Accelerometer:
|
||||||
|
x: int
|
||||||
|
y: int
|
||||||
|
z: int
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
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, skipinitialspace=True)
|
||||||
|
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):
|
||||||
|
while True:
|
||||||
|
row = next(self.file_reader, None)
|
||||||
|
if row is None:
|
||||||
|
self._rewind_file()
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
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)
|
||||||
|
except Exception as e:
|
||||||
|
continue
|
||||||
|
|
||||||
|
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
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user