Add type annotations

Author: tekktrik

adafruit_gizmo/eink_gizmo.py

@@ -21,11 +21,28 @@
 from adafruit_il0373 import IL0373
 from adafruit_ssd1681 import SSD1681
 
+try:
+    from typing import Optional
+    from busio import SPI
+    from microcontroller import Pin
+except ImportError:
+    pass
+
 # pylint: disable=invalid-name, too-few-public-methods
 class EInk_Gizmo(IL0373):
-    """Class representing a 152x152 Tri-Color EInk Gizmo."""
+    """Class representing a 152x152 Tri-Color EInk Gizmo.
+
+    :param"""
 
-    def __init__(self, *, spi=None, cs=None, dc=None, reset=None, busy=None):
+    def __init__(
+        self,
+        *,
+        spi: Optional[SPI] = None,
+        cs: Optional[Pin] = None,
+        dc: Optional[Pin] = None,
+        reset: Optional[Pin] = None,
+        busy: Optional[Pin] = None
+    ) -> None:
         displayio.release_displays()
         if spi is None:
             import busio  # pylint: disable=import-outside-toplevel
@@ -55,7 +72,15 @@ def __init__(self, *, spi=None, cs=None, dc=None, reset=None, busy=None):
 class EInk_HD_Gizmo(SSD1681):
     """Class representing a 200x200 Tri-Color EInk HD Gizmo."""
 
-    def __init__(self, *, spi=None, cs=None, dc=None, reset=None, busy=None):
+    def __init__(
+        self,
+        *,
+        spi: Optional[SPI] = None,
+        cs: Optional[Pin] = None,
+        dc: Optional[Pin] = None,
+        reset: Optional[Pin] = None,
+        busy: Optional[Pin] = None
+    ) -> None:
         displayio.release_displays()
         if spi is None:
             import busio  # pylint: disable=import-outside-toplevel

adafruit_gizmo/tft_gizmo.py

@@ -19,13 +19,26 @@
 import displayio
 from adafruit_st7789 import ST7789
 
+try:
+    from typing import Optional
+    from busio import SPI
+    from microcontroller import Pin
+except ImportError:
+    pass
+
 # pylint: disable=invalid-name, too-few-public-methods
 class TFT_Gizmo(ST7789):
     """Class representing a TFT Gizmo."""
 
     def __init__(
-        self, *, spi=None, cs=board.RX, dc=board.TX, backlight=board.A3, rotation=180
-    ):
+        self,
+        *,
+        spi: Optional[SPI] = None,
+        cs: Pin = board.RX,
+        dc: Pin = board.TX,
+        backlight: Pin = board.A3,
+        rotation: int = 180
+    ) -> None:
         displayio.release_displays()
         if spi is None:
             import busio  # pylint: disable=import-outside-toplevel