common: move ElevatorData and FloorData to common module

pokgak · Feb 14, 2020 · 1c67317 · 1c67317
1 parent 4906954
commit 1c67317
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/common/cps_common/data.py b/common/cps_common/data.py
@@ -4,6 +4,29 @@
 
 import logging
 from datetime import datetime
+from collections import deque
+from typing import Deque
+
+
+class ElevatorData:
+    def __init__(self, id: int):
+        self.id = id
+        self.floor = 0
+        self.door = "open"
+        self.status = "online"
+        self.actual_capacity = 0
+        self.max_capacity = 0
+
+        # used by controller only
+        self.queue: Deque[int] = deque()
+
+
+class FloorData:
+    def __init__(self, id: int):
+        self.id = id
+        self.waiting_count = 0
+        self.up_pressed = False
+        self.down_pressed = False
 
 
 class Passenger: