m.giller/mash-sensor-tof-pc
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases 1 Packages Wiki Activity

Small fixes

Browse source
This commit is contained in:
Maximilian Giller 2021-12-03 18:21:09 +01:00
parent 48eed72acb
commit 12602f4579
1 changed files with 4 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/sensor/people_counter.py
View file

@ -159,26 +159,19 @@ class PeopleCounter ():
th = threading.Thread(target=cb, args=(countChange, self.directionState))
th.start()
def getDirectionTime(self, direction: Directions, time: str) -> datetime:
if len(self.directionState[direction]) <= 0:
return None
return self.directionState[direction][-1][time]
def updateState(self, direction: Directions, triggered: bool) -> bool:
currentlyTriggered = False
previouslyTriggered = False
if len(self.directionState[direction]) > 0:
currentlyTriggered = self.getDirectionTime(
direction, END_TIME) is None
previouslyTriggered = self.directionState[direction][-1][END_TIME] is None
if triggered and not currentlyTriggered:
if triggered and not previouslyTriggered:
# Set as new beginning for this direction
self.directionState[direction].append({
START_TIME: datetime.now(),
END_TIME: None
})
return True
elif not triggered and currentlyTriggered:
elif not triggered and previouslyTriggered:
# Set as end for this direction
self.directionState[direction][-1][END_TIME] = datetime.now()
return True