Added requested type annotations to uc8151d.py

Author: sdomoszlai13

adafruit_epd/uc8151d.py

@@ -14,6 +14,15 @@
 import adafruit_framebuf
 from adafruit_epd.epd import Adafruit_EPD
 
+try:
+ """Needed for type annotations"""
+ from typing import Union, Any
+ from busio import SPI
+ from digitalio import DigitalInOut
+
+except ImportError:
+ pass
+
 __version__ = "0.0.0+auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_EPD.git"
 
@@ -63,8 +72,17 @@ class Adafruit_UC8151D(Adafruit_EPD):
 
  # pylint: disable=too-many-arguments
  def __init__(
- self, width, height, spi, *, cs_pin, dc_pin, sramcs_pin, rst_pin, busy_pin
- ):
+ self,
+ width: int,
+ height: int,
+ spi: SPI,
+ *,
+ cs_pin: DigitalInOut,
+ dc_pin: DigitalInOut,
+ sramcs_pin: DigitalInOut,
+ rst_pin: DigitalInOut,
+ busy_pin: DigitalInOut
+ ) -> None:
  super().__init__(
  width, height, spi, cs_pin, dc_pin, sramcs_pin, rst_pin, busy_pin
  )
@@ -90,13 +108,13 @@ def __init__(
  self.set_color_buffer(1, True)
  # pylint: enable=too-many-arguments
 
- def begin(self, reset=True):
+ def begin(self, reset: bool = True) -> None:
  """Begin communication with the display and set basic settings"""
  if reset:
  self.hardware_reset()
  self.power_down()
 
- def busy_wait(self):
+ def busy_wait(self) -> None:
  """Wait for display to be done with current task, either by polling the
  busy pin, or pausing"""
  if self._busy:
@@ -105,7 +123,7 @@ def busy_wait(self):
  else:
  time.sleep(0.5)
 
- def power_up(self):
+ def power_up(self) -> None:
  """Power up the display in preparation for writing RAM and updating"""
  self.hardware_reset()
  self.busy_wait()
@@ -119,22 +137,22 @@ def power_up(self):
  self.command(_UC8151D_CDI, bytearray([0x97]))
  time.sleep(0.05)
 
- def power_down(self):
+ def power_down(self) -> None:
  """Power down the display - required when not actively displaying!"""
  self.command(_UC8151D_CDI, bytearray([0xF7]))
  self.command(_UC8151D_POWER_OFF)
  self.busy_wait()
  self.command(_UC8151D_DEEP_SLEEP, bytearray([0xA5]))
 
- def update(self):
+ def update(self) -> None:
  """Update the display from internal memory"""
  self.command(_UC8151D_DISPLAY_REFRESH)
  time.sleep(0.1)
  self.busy_wait()
  if not self._busy:
  time.sleep(15) # wait 15 seconds
 
- def write_ram(self, index):
+ def write_ram(self, index: Union[0, 1]) -> Any:
  """Send the one byte command for starting the RAM write process. Returns
  the byte read at the same time over SPI. index is the RAM buffer, can be
  0 or 1 for tri-color displays."""
@@ -144,7 +162,9 @@ def write_ram(self, index):
  return self.command(_UC8151D_DTM2, end=False)
  raise RuntimeError("RAM index must be 0 or 1")
 
- def set_ram_address(self, x, y): # pylint: disable=unused-argument, no-self-use
+ def set_ram_address(
+ self, x: int, y: int
+ ) -> None: # pylint: disable=unused-argument, no-self-use
  """Set the RAM address location, not used on this chipset but required by
  the superclass"""
  return # on this chip it does nothing