refactor: Add Python type annotations wherever appropriate (#269)

* Add Python type annotations wherever appropriate

* Might as well annotate this too

Author: mikomikotaishi

head-tracking/colors.py

@@ -0,0 +1,29 @@
+import logging
+from logging import Formatter, LogRecord
+from typing import Dict
+
+class Colors:
+    RESET: str = "\033[0m"
+    BOLD: str = "\033[1m"
+    RED: str = "\033[91m"
+    GREEN: str = "\033[92m"
+    YELLOW: str = "\033[93m"
+    BLUE: str = "\033[94m"
+    MAGENTA: str = "\033[95m"
+    CYAN: str = "\033[96m"
+    WHITE: str = "\033[97m"
+    BG_BLACK: str = "\033[40m"
+
+class ColorFormatter(Formatter):
+    FORMATS: Dict[int, str] = {
+        logging.DEBUG: f"{Colors.BLUE}[%(levelname)s] %(message)s{Colors.RESET}",
+        logging.INFO: f"{Colors.GREEN}%(message)s{Colors.RESET}",
+        logging.WARNING: f"{Colors.YELLOW}%(message)s{Colors.RESET}",
+        logging.ERROR: f"{Colors.RED}[%(levelname)s] %(message)s{Colors.RESET}",
+        logging.CRITICAL: f"{Colors.RED}{Colors.BOLD}[%(levelname)s] %(message)s{Colors.RESET}"
+    }
+
+    def format(self, record: LogRecord) -> str:
+        log_fmt: str = self.FORMATS.get(record.levelno)
+        formatter: Formatter = Formatter(log_fmt, datefmt="%H:%M:%S")
+        return formatter.format(record)

head-tracking/connection_manager.py

@@ -1,23 +1,25 @@
 import bluetooth
 import logging
+from bluetooth import BluetoothSocket
+from logging import Logger
 
 class ConnectionManager:
-    INIT_CMD  = "00 00 04 00 01 00 02 00 00 00 00 00 00 00 00 00"
-    START_CMD = "04 00 04 00 17 00 00 00 10 00 10 00 08 A1 02 42 0B 08 0E 10 02 1A 05 01 40 9C 00 00"
-    STOP_CMD  = "04 00 04 00 17 00 00 00 10 00 11 00 08 7E 10 02 42 0B 08 4E 10 02 1A 05 01 00 00 00 00"
+    INIT_CMD: str = "00 00 04 00 01 00 02 00 00 00 00 00 00 00 00 00"
+    START_CMD: str  = "04 00 04 00 17 00 00 00 10 00 10 00 08 A1 02 42 0B 08 0E 10 02 1A 05 01 40 9C 00 00"
+    STOP_CMD: str = "04 00 04 00 17 00 00 00 10 00 11 00 08 7E 10 02 42 0B 08 4E 10 02 1A 05 01 00 00 00 00"
 
-    def __init__(self, bt_addr="28:2D:7F:C2:05:5B", psm=0x1001, logger=None):
-        self.bt_addr = bt_addr
-        self.psm = psm
-        self.logger = logger if logger else logging.getLogger(__name__)
-        self.sock = None
-        self.connected = False
-        self.started = False
+    def __init__(self, bt_addr: str = "28:2D:7F:C2:05:5B", psm: int = 0x1001, logger: Logger = None) -> None:
+        self.bt_addr: str = bt_addr
+        self.psm: int = psm
+        self.logger: Logger = logger if logger else logging.getLogger(__name__)
+        self.sock: BluetoothSocket = None
+        self.connected: bool = False
+        self.started: bool = False
 
-    def connect(self):
+    def connect(self) -> bool:
         self.logger.info(f"Connecting to {self.bt_addr} on PSM {self.psm:#04x}...")
         try:
-            self.sock = bluetooth.BluetoothSocket(bluetooth.L2CAP)
+            self.sock = BluetoothSocket(bluetooth.L2CAP)
             self.sock.connect((self.bt_addr, self.psm))
             self.connected = True
             self.logger.info("Connected to AirPods.")
@@ -28,7 +30,7 @@ def connect(self):
             self.connected = False
         return self.connected
 
-    def send_start(self):
+    def send_start(self) -> bool:
         if not self.connected:
             self.logger.error("Not connected. Cannot send START command.")
             return False
@@ -40,7 +42,7 @@ def send_start(self):
             self.logger.info("START command has already been sent.")
         return True
 
-    def send_stop(self):
+    def send_stop(self) -> None:
         if self.connected and self.started:
             try:
                 self.sock.send(bytes.fromhex(self.STOP_CMD))
@@ -51,12 +53,12 @@ def send_stop(self):
         else:
             self.logger.info("Cannot send STOP; not started or not connected.")
 
-    def disconnect(self):
+    def disconnect(self) -> None:
         if self.sock:
             try:
                 self.sock.close()
                 self.logger.info("Disconnected from AirPods.")
             except Exception as e:
                 self.logger.error(f"Error during disconnect: {e}")
         self.connected = False
-        self.started = False
+        self.started = False