kumi/MicroPython_PN532
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add Missing Type Annotations

Browse source
This commit is contained in:
Thomas Franks 2023-02-09 15:12:59 -05:00
parent f19d2424c5
commit 91f3385ba7
5 changed files with 112 additions and 46 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

87
adafruit_pn532/adafruit_pn532.py
View file

@ -31,6 +31,13 @@ from digitalio import Direction
from micropython import const
try:
from typing import Optional, Tuple
from typing_extensions import Literal
from digitalio import DigitalInOut
except ImportError:
pass
__version__ = "0.0.0+auto.0"
__repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_PN532.git"
@ -151,7 +158,13 @@ class BusyError(Exception):
class PN532:
"""PN532 driver base, must be extended for I2C/SPI/UART interfacing"""
def __init__(self, *, debug=False, irq=None, reset=None):
def __init__(
self,
*,
debug: bool = False,
irq: Optional[DigitalInOut] = None,
reset: Optional[DigitalInOut] = None
) -> None:
"""Create an instance of the PN532 class"""
self.low_power = True
self.debug = debug
@ -160,26 +173,26 @@ class PN532:
self.reset()
_ = self.firmware_version
def _read_data(self, count):
def _read_data(self, count: int) -> bytearray:
# Read raw data from device, not including status bytes:
# Subclasses MUST implement this!
raise NotImplementedError
def _write_data(self, framebytes):
def _write_data(self, framebytes: bytearray) -> None:
# Write raw bytestring data to device, not including status bytes:
# Subclasses MUST implement this!
raise NotImplementedError
def _wait_ready(self, timeout):
def _wait_ready(self, timeout: float) -> bool:
# Check if busy up to max length of 'timeout' seconds
# Subclasses MUST implement this!
raise NotImplementedError
def _wakeup(self):
def _wakeup(self) -> None:
# Send special command to wake up
raise NotImplementedError
def reset(self):
def reset(self) -> None:
"""Perform a hardware reset toggle and then wake up the PN532"""
if self._reset_pin:
if self.debug:
@ -191,7 +204,7 @@ class PN532:
time.sleep(0.1)
self._wakeup()
def _write_frame(self, data):
def _write_frame(self, data: bytearray) -> None:
"""Write a frame to the PN532 with the specified data bytearray."""
assert (
data is not None and 1 < len(data) < 255
@ -221,7 +234,7 @@ class PN532:
print("Write frame: ", [hex(i) for i in frame])
self._write_data(bytes(frame))
def _read_frame(self, length):
def _read_frame(self, length: int) -> bytearray:
"""Read a response frame from the PN532 of at most length bytes in size.
Returns the data inside the frame if found, otherwise raises an exception
if there is an error parsing the frame. Note that less than length bytes
@ -256,9 +269,13 @@ class PN532:
# Return frame data.
return response[offset + 2 : offset + 2 + frame_len]
def call_function(
self, command, response_length=0, params=[], timeout=1
): # pylint: disable=dangerous-default-value
def call_function( # pylint: disable=dangerous-default-value
self,
command: int,
response_length: int = 0,
params: Optional[bytearray] = [],
timeout: float = 1,
) -> bytearray:
"""Send specified command to the PN532 and expect up to response_length
bytes back in a response. Note that less than the expected bytes might
be returned! Params can optionally specify an array of bytes to send as
@ -272,12 +289,12 @@ class PN532:
command, response_length=response_length, timeout=timeout
)
def send_command(
self, command, params=[], timeout=1
): # pylint: disable=dangerous-default-value
def send_command( # pylint: disable=dangerous-default-value
self, command: int, params: Optional[bytearray] = [], timeout: float = 1
) -> bool:
"""Send specified command to the PN532 and wait for an acknowledgment.
Will wait up to timeout seconds for the acknowlegment and return True.
If no acknowlegment is received, False is returned.
If no acknowledgment is received, False is returned.
"""
if self.low_power:
self._wakeup()
@ -300,7 +317,9 @@ class PN532:
raise RuntimeError("Did not receive expected ACK from PN532!")
return True
def process_response(self, command, response_length=0, timeout=1):
def process_response(
self, command: int, response_length: int = 0, timeout: float = 1
) -> bytearray:
"""Process the response from the PN532 and expect up to response_length
bytes back in a response. Note that less than the expected bytes might
be returned! Will wait up to timeout seconds for a response and return
@ -317,7 +336,7 @@ class PN532:
# Return response data.
return response[2:]
def power_down(self):
def power_down(self) -> bool:
"""Put the PN532 into a low power state. If the reset pin is connected a
hard power down is performed, if not, a soft power down is performed
instead. Returns True if the PN532 was powered down successfully or
@ -333,7 +352,7 @@ class PN532:
return self.low_power
@property
def firmware_version(self):
def firmware_version(self) -> Tuple[int, int, int, int]:
"""Call PN532 GetFirmwareVersion function and return a tuple with the IC,
Ver, Rev, and Support values.
"""
@ -342,7 +361,7 @@ class PN532:
raise RuntimeError("Failed to detect the PN532")
return tuple(response)
def SAM_configuration(self): # pylint: disable=invalid-name
def SAM_configuration(self) -> None: # pylint: disable=invalid-name
"""Configure the PN532 to read MiFare cards."""
# Send SAM configuration command with configuration for:
# - 0x01, normal mode
@ -352,7 +371,9 @@ class PN532:
# check the command was executed as expected.
self.call_function(_COMMAND_SAMCONFIGURATION, params=[0x01, 0x14, 0x01])
def read_passive_target(self, card_baud=_MIFARE_ISO14443A, timeout=1):
def read_passive_target(
self, card_baud: int = _MIFARE_ISO14443A, timeout: float = 1
) -> bytearray:
"""Wait for a MiFare card to be available and return its UID when found.
Will wait up to timeout seconds and return None if no card is found,
otherwise a bytearray with the UID of the found card is returned.
@ -364,9 +385,11 @@ class PN532:
return None
return self.get_passive_target(timeout=timeout)
def listen_for_passive_target(self, card_baud=_MIFARE_ISO14443A, timeout=1):
def listen_for_passive_target(
self, card_baud: int = _MIFARE_ISO14443A, timeout: float = 1
):
"""Send command to PN532 to begin listening for a Mifare card. This
returns True if the command was received succesfully. Note, this does
returns True if the command was received successfully. Note, this does
not also return the UID of a card! `get_passive_target` must be called
to read the UID when a card is found. If just looking to see if a card
is currently present use `read_passive_target` instead.
@ -380,7 +403,7 @@ class PN532:
return False # _COMMAND_INLISTPASSIVETARGET failed
return response
def get_passive_target(self, timeout=1):
def get_passive_target(self, timeout: float = 1) -> bytearray:
"""Will wait up to timeout seconds and return None if no card is found,
otherwise a bytearray with the UID of the found card is returned.
`listen_for_passive_target` must have been called first in order to put
@ -404,9 +427,13 @@ class PN532:
# Return UID of card.
return response[6 : 6 + response[5]]
def mifare_classic_authenticate_block(
self, uid, block_number, key_number, key
): # pylint: disable=invalid-name
def mifare_classic_authenticate_block( # pylint: disable=invalid-name
self,
uid: bytearray,
block_number: int,
key_number: Literal[0x60, 0x61],
key: bytearray,
) -> bool:
"""Authenticate specified block number for a MiFare classic card. Uid
should be a byte array with the UID of the card, block number should be
the block to authenticate, key number should be the key type (like
@ -429,7 +456,7 @@ class PN532:
)
return response[0] == 0x00
def mifare_classic_read_block(self, block_number):
def mifare_classic_read_block(self, block_number: int) -> bytearray:
"""Read a block of data from the card. Block number should be the block
to read. If the block is successfully read a bytearray of length 16 with
data starting at the specified block will be returned. If the block is
@ -447,7 +474,7 @@ class PN532:
# Return first 4 bytes since 16 bytes are always returned.
return response[1:]
def mifare_classic_write_block(self, block_number, data):
def mifare_classic_write_block(self, block_number: int, data: bytearray) -> bool:
"""Write a block of data to the card. Block number should be the block
to write and data should be a byte array of length 16 with the data to
write. If the data is successfully written then True is returned,
@ -468,7 +495,7 @@ class PN532:
)
return response[0] == 0x0
def ntag2xx_write_block(self, block_number, data):
def ntag2xx_write_block(self, block_number: int, data: bytearray) -> bool:
"""Write a block of data to the card. Block number should be the block
to write and data should be a byte array of length 4 with the data to
write. If the data is successfully written then True is returned,
@ -487,7 +514,7 @@ class PN532:
)
return response[0] == 0x00
def ntag2xx_read_block(self, block_number):
def ntag2xx_read_block(self, block_number: int) -> bytearray:
"""Read a block of data from the card. Block number should be the block
to read. If the block is successfully read the first 4 bytes (after the
leading 0x00 byte) will be returned.

25
adafruit_pn532/i2c.py
View file

@ -23,13 +23,28 @@ from digitalio import Direction
from micropython import const
from adafruit_pn532.adafruit_pn532 import PN532, BusyError
try:
from typing import Optional
from digitalio import DigitalInOut
from busio import I2C
except ImportError:
pass
_I2C_ADDRESS = const(0x24)
class PN532_I2C(PN532):
"""Driver for the PN532 connected over I2C."""
def __init__(self, i2c, *, irq=None, reset=None, req=None, debug=False):
def __init__(
self,
i2c: I2C,
*,
irq: Optional[DigitalInOut] = None,
reset: Optional[DigitalInOut] = None,
req: Optional[DigitalInOut] = None,
debug: bool = False
) -> None:
"""Create an instance of the PN532 class using I2C. Note that PN532
uses clock stretching. Optional IRQ pin (not used),
resetp pin and debugging output.
@ -39,7 +54,7 @@ class PN532_I2C(PN532):
self._i2c = i2c_device.I2CDevice(i2c, _I2C_ADDRESS)
super().__init__(debug=debug, irq=irq, reset=reset)
def _wakeup(self):
def _wakeup(self) -> None:
"""Send any special commands/data to wake up PN532"""
if self._reset_pin:
self._reset_pin.value = True
@ -53,7 +68,7 @@ class PN532_I2C(PN532):
self.low_power = False
self.SAM_configuration() # Put the PN532 back in normal mode
def _wait_ready(self, timeout=1):
def _wait_ready(self, timeout: float = 1) -> bool:
"""Poll PN532 if status byte is ready, up to `timeout` seconds"""
status = bytearray(1)
timestamp = time.monotonic()
@ -69,7 +84,7 @@ class PN532_I2C(PN532):
# Timed out!
return False
def _read_data(self, count):
def _read_data(self, count: int) -> bytearray:
"""Read a specified count of bytes from the PN532."""
# Build a read request frame.
frame = bytearray(count + 1)
@ -82,7 +97,7 @@ class PN532_I2C(PN532):
print("Reading: ", [hex(i) for i in frame[1:]])
return frame[1:] # don't return the status byte
def _write_data(self, framebytes):
def _write_data(self, framebytes: bytearray) -> None:
"""Write a specified count of bytes to the PN532"""
with self._i2c as i2c:
i2c.write(framebytes)

27
adafruit_pn532/spi.py
View file

@ -14,6 +14,13 @@ using SPI.
"""
try:
from typing import Optional
from digitalio import DigitalInOut
from busio import SPI
except ImportError:
pass
__version__ = "0.0.0+auto.0"
__repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_PN532.git"
@ -28,7 +35,7 @@ _SPI_DATAREAD = const(0x03)
_SPI_READY = const(0x01)
def reverse_bit(num):
def reverse_bit(num: int) -> int:
"""Turn an LSB byte to an MSB byte, and vice versa. Used for SPI as
it is LSB for the PN532, but 99% of SPI implementations are MSB only!"""
result = 0
@ -44,13 +51,21 @@ class PN532_SPI(PN532):
SPI device & chip select digitalInOut pin. Optional IRQ pin (not used),
reset pin and debugging output."""
def __init__(self, spi, cs_pin, *, irq=None, reset=None, debug=False):
def __init__(
self,
spi: SPI,
cs_pin: DigitalInOut,
*,
irq: Optional[DigitalInOut] = None,
reset: Optional[DigitalInOut] = None,
debug: bool = False
) -> None:
"""Create an instance of the PN532 class using SPI"""
self.debug = debug
self._spi = spi_device.SPIDevice(spi, cs_pin)
super().__init__(debug=debug, irq=irq, reset=reset)
def _wakeup(self):
def _wakeup(self) -> None:
"""Send any special commands/data to wake up PN532"""
if self._reset_pin:
self._reset_pin.value = True
@ -61,7 +76,7 @@ class PN532_SPI(PN532):
self.low_power = False
self.SAM_configuration() # Put the PN532 back in normal mode
def _wait_ready(self, timeout=1):
def _wait_ready(self, timeout: float = 1) -> bool:
"""Poll PN532 if status byte is ready, up to `timeout` seconds"""
status_cmd = bytearray([reverse_bit(_SPI_STATREAD), 0x00])
status_response = bytearray([0x00, 0x00])
@ -77,7 +92,7 @@ class PN532_SPI(PN532):
# We timed out!
return False
def _read_data(self, count):
def _read_data(self, count: int) -> bytearray:
"""Read a specified count of bytes from the PN532."""
# Build a read request frame.
frame = bytearray(count + 1)
@ -92,7 +107,7 @@ class PN532_SPI(PN532):
print("Reading: ", [hex(i) for i in frame[1:]])
return frame[1:]
def _write_data(self, framebytes):
def _write_data(self, framebytes: bytearray) -> None:
"""Write a specified count of bytes to the PN532"""
# start by making a frame with data write in front,
# then rest of bytes, and LSBify it

18
adafruit_pn532/uart.py
View file

@ -17,6 +17,12 @@ using UART.
__version__ = "0.0.0+auto.0"
__repo__ = "https://github.com/adafruit/Adafruit_CircuitPython_PN532.git"
try:
from typing import Optional
from digitalio import DigitalInOut
from busio import UART
except ImportError:
pass
import time
from adafruit_pn532.adafruit_pn532 import PN532, BusyError
@ -25,7 +31,9 @@ from adafruit_pn532.adafruit_pn532 import PN532, BusyError
class PN532_UART(PN532):
"""Driver for the PN532 connected over Serial UART"""
def __init__(self, uart, *, reset=None, debug=False):
def __init__(
self, uart: UART, *, reset: Optional[DigitalInOut] = None, debug: bool = False
) -> None:
"""Create an instance of the PN532 class using Serial connection.
Optional reset pin and debugging output.
"""
@ -33,7 +41,7 @@ class PN532_UART(PN532):
self._uart = uart
super().__init__(debug=debug, reset=reset)
def _wakeup(self):
def _wakeup(self) -> None:
"""Send any special commands/data to wake up PN532"""
if self._reset_pin:
self._reset_pin.value = True
@ -44,7 +52,7 @@ class PN532_UART(PN532):
) # wake up!
self.SAM_configuration()
def _wait_ready(self, timeout=1):
def _wait_ready(self, timeout: float = 1) -> bool:
"""Wait `timeout` seconds"""
timestamp = time.monotonic()
while (time.monotonic() - timestamp) < timeout:
@ -54,7 +62,7 @@ class PN532_UART(PN532):
# Timed out!
return False
def _read_data(self, count):
def _read_data(self, count: int) -> bytearray:
"""Read a specified count of bytes from the PN532."""
frame = self._uart.read(count)
if not frame:
@ -63,7 +71,7 @@ class PN532_UART(PN532):
print("Reading: ", [hex(i) for i in frame])
return frame
def _write_data(self, framebytes):
def _write_data(self, framebytes: bytearray) -> None:
"""Write a specified count of bytes to the PN532"""
self._uart.reset_input_buffer()
self._uart.write(framebytes)

1
requirements.txt
View file

@ -5,3 +5,4 @@
Adafruit-Blinka
adafruit-circuitpython-busdevice
pyserial
typing-extensions~=4.0