Add type annotations

Author: tekktrik

adafruit_nunchuk.py

@@ -29,6 +29,12 @@
 from collections import namedtuple
 from adafruit_bus_device.i2c_device import I2CDevice
 
+try:
+    from typing import Type
+    from busio import I2C
+except ImportError:
+    pass
+
 __version__ = "0.0.0-auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_Nunchuk.git"
 
@@ -54,7 +60,9 @@ class Nunchuk:
     _Buttons = namedtuple("Buttons", ("C", "Z"))
     _Acceleration = namedtuple("Acceleration", ("x", "y", "z"))
 
-    def __init__(self, i2c, address=0x52, i2c_read_delay=0.002):
+    def __init__(
+        self, i2c: I2C, address: int = 0x52, i2c_read_delay: float = 0.002
+    ) -> None:
         self.buffer = bytearray(8)
         self.i2c_device = I2CDevice(i2c, address)
         self._i2c_read_delay = i2c_read_delay
@@ -67,7 +75,7 @@ def __init__(self, i2c, address=0x52, i2c_read_delay=0.002):
             i2c_dev.write(b"\xFB\x00")
 
     @property
-    def values(self):
+    def values(self) -> Type[tuple]:
         """The current state of all values."""
         self._read_data()
         return self._Values(
@@ -77,33 +85,33 @@ def values(self):
         )
 
     @property
-    def joystick(self):
+    def joystick(self) -> Type[tuple]:
         """The current joystick position."""
         return self._joystick()
 
     @property
-    def buttons(self):  # pylint: disable=invalid-name
-        """The current pressed state of button Z."""
+    def buttons(self) -> Type[tuple]:  # pylint: disable=invalid-name
+        """The current pressed state of buttons C and Z."""
         return self._buttons()
 
     @property
-    def acceleration(self):
+    def acceleration(self) -> Type[tuple]:
         """The current accelerometer reading."""
         return self._acceleration()
 
-    def _joystick(self, do_read=True):
+    def _joystick(self, do_read: bool = True) -> Type[tuple]:
         if do_read:
             self._read_data()
         return self._Joystick(self.buffer[0], self.buffer[1])  # x, y
 
-    def _buttons(self, do_read=True):
+    def _buttons(self, do_read: bool = True) -> Type[tuple]:
         if do_read:
             self._read_data()
         return self._Buttons(
             not bool(self.buffer[5] & 0x02), not bool(self.buffer[5] & 0x01)  # C  # Z
         )
 
-    def _acceleration(self, do_read=True):
+    def _acceleration(self, do_read: bool = True) -> Type[tuple]:
         if do_read:
             self._read_data()
         return self._Acceleration(
@@ -112,10 +120,10 @@ def _acceleration(self, do_read=True):
             ((self.buffer[5] & 0x0C) >> 2) | (self.buffer[4] << 2),  # az
         )
 
-    def _read_data(self):
+    def _read_data(self) -> bytearray:
         return self._read_register(b"\x00")
 
-    def _read_register(self, address):
+    def _read_register(self, address) -> bytearray:
         with self.i2c_device as i2c:
             i2c.write(address)
             time.sleep(self._i2c_read_delay)  # at least 200us