Created proper sensor package

src/sensors/__init__.py

@@ -1,2 +1 @@
 from sensors.tof_sensor import ToFSensor, Directions
-from sensors.people_counter import PeopleCounter

src/sensors/people_counter.py

@@ -1,4 +1,3 @@
-from typing import Dict
 from sensors import ToFSensor, Directions
 from datetime import datetime
 import threading
@@ -13,7 +12,7 @@ TRIGGER_DISTANCES = "trigger_distances"
 END_DISTANCE = "end_distance"
 
 
-class PeopleCounter ():
+class PeopleCounter:
     def __init__(self, sensor: ToFSensor) -> None:
         self.sensor = sensor
         self.callbacks = {COUNTING_CB: [], TRIGGER_CB: [], CHANGE_CB: []}
@@ -37,11 +36,8 @@ class PeopleCounter ():
     def unhookChange(self, cb) -> None:
         self.callbacks[CHANGE_CB].remove(cb)
 
-    def getInitialDirectionState(self) -> Dict:
-        return {
-            Directions.INSIDE: [],
-            Directions.OUTSIDE: []
-        }
+    def getInitialDirectionState(self) -> dict[Directions, list]:
+        return {Directions.INSIDE: [], Directions.OUTSIDE: []}
 
     def run(self) -> None:
         self.keepRunning = True
@@ -66,7 +62,9 @@ class PeopleCounter ():
                 th.start()
 
                 # Reset state if state is finalised
-                if not self.isDirectionTriggered(Directions.INSIDE) and not self.isDirectionTriggered(Directions.OUTSIDE):
+                if not self.isDirectionTriggered(
+                    Directions.INSIDE
+                ) and not self.isDirectionTriggered(Directions.OUTSIDE):
                     self.directionState = self.getInitialDirectionState()
 
         self.sensor.close()
@@ -79,7 +77,10 @@ class PeopleCounter ():
                 return 0
 
             # Did every record start and end?
-            if directionState[direction][0][START_TIME] is None or directionState[direction][-1][END_TIME] is None:
+            if (
+                directionState[direction][0][START_TIME] is None
+                or directionState[direction][-1][END_TIME] is None
+            ):
                 return 0  # Return no change if not valid
 
         # Get times into variables
@@ -148,7 +149,7 @@ class PeopleCounter ():
     def handleTriggerCallbacks(self) -> None:
         triggerState = {
             Directions.INSIDE: self.isDirectionTriggered(Directions.INSIDE),
-            Directions.OUTSIDE: self.isDirectionTriggered(Directions.OUTSIDE)
+            Directions.OUTSIDE: self.isDirectionTriggered(Directions.OUTSIDE),
         }
 
         for cb in self.callbacks[TRIGGER_CB]:
@@ -159,7 +160,10 @@ class PeopleCounter ():
             cb(countChange, self.directionState)
 
     def isDirectionTriggered(self, direction: Directions) -> bool:
-        return len(self.directionState[direction]) > 0 and self.directionState[direction][-1][END_TIME] is None
+        return (
+            len(self.directionState[direction]) > 0
+            and self.directionState[direction][-1][END_TIME] is None
+        )
 
     def updateState(self, direction: Directions, distance: float) -> bool:
         triggered: bool = self.isTriggerDistance(distance)
@@ -170,12 +174,14 @@ class PeopleCounter ():
 
         if triggered and not previouslyTriggered:
             # Set as new beginning for this direction
-            self.directionState[direction].append({
+            self.directionState[direction].append(
+                {
                     START_TIME: datetime.now(),
                     END_TIME: None,
                     TRIGGER_DISTANCES: [distance],
-                END_DISTANCE: None
-            })
+                    END_DISTANCE: None,
+                }
+            )
             return True
         elif not triggered and previouslyTriggered:
             # Set as end for this direction

src/sensors/tof_sensor.py

@@ -5,7 +5,7 @@ class Directions(str, Enum):
     INSIDE = "indoor"
     OUTSIDE = "outdoor"
 
-    def other(direction: 'Direction') -> 'Direction':
+    def other(direction: "Directions") -> "Directions":
         if direction is Directions.INSIDE:
             return Directions.OUTSIDE
         else:
@@ -20,13 +20,11 @@ class ToFSensor:
         raise NotImplementedError()
 
     def setDirection(self, direction: Directions) -> None:
-        """Configure sensor to pick up the distance in a specific direction.
-        """
+        """Configure sensor to pick up the distance in a specific direction."""
         raise NotImplementedError()
 
     def getDistance(self) -> float:
-        """Returns new distance in cm.
-        """
+        """Returns new distance in cm."""
         raise NotImplementedError()
 
     def close(self) -> None:

src/sensors/vl53l1x_sensor.py

@@ -1,4 +1,4 @@
-from sensor.tof_sensor import Directions, ToFSensor
+from sensors import Directions, ToFSensor
 import VL53L1X
 
 # Reference: https://github.com/pimoroni/vl53l1x-python
@@ -39,11 +39,10 @@ class VL53L1XSensor (ToFSensor):
         # 3 = Long Range
 
     def setDirection(self, direction: Directions) -> None:
-        """Configure sensor to pick up the distance in a specific direction.
-        """
+        """Configure sensor to pick up the distance in a specific direction."""
         direction_roi = {
             Directions.INSIDE: VL53L1X.VL53L1xUserRoi(6, 3, 9, 0),
-            Directions.OUTSIDE: VL53L1X.VL53L1xUserRoi(6, 15, 9, 12)
+            Directions.OUTSIDE: VL53L1X.VL53L1xUserRoi(6, 15, 9, 12),
         }
 
         roi = direction_roi[direction]
@@ -53,8 +52,7 @@ class VL53L1XSensor (ToFSensor):
         self.sensor.start_ranging(self.ranging)
 
     def getDistance(self) -> float:
-        """Returns new distance in cm.
-        """
+        """Returns new distance in cm."""
         distance = self.sensor.get_distance()
 
         return distance / 10