From d45180d2611c8580705610c13a9f09364351322f Mon Sep 17 00:00:00 2001 From: Matt Goff Date: Mon, 14 Feb 2022 09:07:16 -0700 Subject: [PATCH 1/2] Added Type Annotations --- adafruit_neokey/neokey1x4.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/adafruit_neokey/neokey1x4.py b/adafruit_neokey/neokey1x4.py index a13e5c7..d961a46 100644 --- a/adafruit_neokey/neokey1x4.py +++ b/adafruit_neokey/neokey1x4.py @@ -33,6 +33,12 @@ from adafruit_seesaw.neopixel import NeoPixel from adafruit_seesaw.seesaw import Seesaw +try: + import typing # pylint: disable=unused-import + from busio import I2C +except ImportError: + pass + _NEOKEY1X4_ADDR = const(0x30) _NEOKEY1X4_NEOPIX_PIN = const(3) @@ -45,7 +51,9 @@ class NeoKey1x4(Seesaw): """Driver for the Adafruit NeoKey 1x4.""" - def __init__(self, i2c_bus, interrupt=False, addr=_NEOKEY1X4_ADDR): + def __init__( + self, i2c_bus: I2C, interrupt: bool = False, addr: int = _NEOKEY1X4_ADDR + ): super().__init__(i2c_bus, addr) self.interrupt_enabled = interrupt self.pixels = NeoPixel( @@ -55,7 +63,7 @@ def __init__(self, i2c_bus, interrupt=False, addr=_NEOKEY1X4_ADDR): for b in range(4, 8): self.pin_mode(b, self.INPUT_PULLUP) - def __getitem__(self, index): + def __getitem__(self, index: int) -> bool: if not isinstance(index, int) or (index < 0) or (index > 3): raise RuntimeError("Index must be 0 thru 3") return not self.digital_read(index + 4) From 1b091f67dd2336b2ac398cc5193c7e293c8a815d Mon Sep 17 00:00:00 2001 From: Matt Goff Date: Mon, 14 Feb 2022 09:12:22 -0700 Subject: [PATCH 2/2] Added Type Annotations --- adafruit_neokey/neokey1x4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/adafruit_neokey/neokey1x4.py b/adafruit_neokey/neokey1x4.py index d961a46..327f3bd 100644 --- a/adafruit_neokey/neokey1x4.py +++ b/adafruit_neokey/neokey1x4.py @@ -53,7 +53,7 @@ class NeoKey1x4(Seesaw): def __init__( self, i2c_bus: I2C, interrupt: bool = False, addr: int = _NEOKEY1X4_ADDR - ): + ) -> None: super().__init__(i2c_bus, addr) self.interrupt_enabled = interrupt self.pixels = NeoPixel(