Added changes requested by @FoamyGuy.

This commit is contained in:
sdomoszlai13 2023-05-18 21:13:17 +02:00
parent 251322690f
commit 95758b357f

View file

@ -11,14 +11,15 @@ CircuitPython driver for Adafruit ePaper display breakouts
import time import time
from micropython import const from micropython import const
from digitalio import Direction, DigitalInOut
from adafruit_epd import mcp_sram from adafruit_epd import mcp_sram
try: try:
"""Needed for type annotations""" """Needed for type annotations"""
from typing import Any, Union, Callable from typing import Any, Union, Callable, Optional
from busio import SPI from busio import SPI
from digitalio import Direction, DigitalInOut
from PIL.Image import Image from PIL.Image import Image
except ImportError: except ImportError:
pass pass
@ -26,7 +27,7 @@ __version__ = "0.0.0+auto.0"
__repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_EPD.git" __repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_EPD.git"
class Adafruit_EPD: # pylint: disable=too-many-instance-attributes, too-many-public-methods class Adafruit_EPD: # pylint: disable=too-many-instance-attributes, too-many-public-methods, too-many-arguments
"""Base class for EPD displays""" """Base class for EPD displays"""
BLACK = const(0) BLACK = const(0)
@ -173,7 +174,9 @@ class Adafruit_EPD: # pylint: disable=too-many-instance-attributes, too-many-pu
self._rst.value = True self._rst.value = True
time.sleep(0.1) time.sleep(0.1)
def command(self, cmd: Any, data: Any = None, end: bool = True) -> Any: def command(
self, cmd: int, data: Optional[bytearray] = None, end: bool = True
) -> int:
"""Send command byte to display.""" """Send command byte to display."""
self._cs.value = True self._cs.value = True
self._dc.value = False self._dc.value = False
@ -192,7 +195,7 @@ class Adafruit_EPD: # pylint: disable=too-many-instance-attributes, too-many-pu
return ret return ret
def _spi_transfer(self, data: Any) -> Any: def _spi_transfer(self, data: Union[int, bytearray]) -> int:
"""Transfer one byte or bytearray, toggling the cs pin if required by the EPD chipset""" """Transfer one byte or bytearray, toggling the cs pin if required by the EPD chipset"""
if isinstance(data, int): # single byte! if isinstance(data, int): # single byte!
self._spibuf[0] = data self._spibuf[0] = data