mirror of
https://github.com/KevinMidboe/python-gpiozero.git
synced 2025-12-08 20:39:01 +00:00
More tidying up
Ensure LEDCollection cleans up upon construction failure, rename some internals to be a bit more obvious, rename PinGPIOUnsupported to PinUnsupported, and some other stuff I've forgotten!
This commit is contained in:
@@ -12,6 +12,7 @@ from ..exc import (
|
||||
PinInvalidFunction,
|
||||
PinSetInput,
|
||||
PinFixedPull,
|
||||
PinUnsupported,
|
||||
PinSPIUnsupported,
|
||||
PinPWMUnsupported,
|
||||
PinEdgeDetectUnsupported,
|
||||
@@ -58,7 +59,7 @@ class Factory(object):
|
||||
:meth:`pin` for the same pin specification must return the same
|
||||
object.
|
||||
"""
|
||||
raise PinGPIOUnsupported("GPIO not supported by this pin factory")
|
||||
raise PinUnsupported("Individual pins are not supported by this pin factory")
|
||||
|
||||
def pin_address(self, spec):
|
||||
"""
|
||||
|
||||
@@ -32,10 +32,12 @@ class LocalPiFactory(PiFactory):
|
||||
|
||||
def __init__(self):
|
||||
super(LocalPiFactory, self).__init__()
|
||||
self.spi_hardware_class = LocalPiHardwareSPI
|
||||
self.spi_software_class = LocalPiSoftwareSPI
|
||||
self.shared_spi_hardware_class = LocalPiHardwareSPIShared
|
||||
self.shared_spi_software_class = LocalPiSoftwareSPIShared
|
||||
self.spi_classes = {
|
||||
('hardware', 'exclusive'): LocalPiHardwareSPI,
|
||||
('hardware', 'shared'): LocalPiHardwareSPIShared,
|
||||
('software', 'exclusive'): LocalPiSoftwareSPI,
|
||||
('software', 'shared'): LocalPiSoftwareSPIShared,
|
||||
}
|
||||
# Override the pins dict to be this class' pins dict. This is a bit of
|
||||
# a dirty hack, but ensures that anyone evil enough to mix pin
|
||||
# implementations doesn't try and control the same pin with different
|
||||
@@ -74,7 +76,7 @@ class LocalPiHardwareSPI(SPI, Device):
|
||||
raise ImportError('failed to import spidev')
|
||||
self._port = port
|
||||
self._device = device
|
||||
self._intf = None
|
||||
self._interface = None
|
||||
self._address = factory.address + ('SPI(port=%d, device=%d)' % (port, device),)
|
||||
super(LocalPiHardwareSPI, self).__init__()
|
||||
self._reserve_pins(
|
||||
@@ -83,9 +85,9 @@ class LocalPiHardwareSPI(SPI, Device):
|
||||
factory.pin_address(9),
|
||||
factory.pin_address((8, 7)[device])
|
||||
)
|
||||
self._intf = SpiDev()
|
||||
self._intf.open(port, device)
|
||||
self._intf.max_speed_hz = 500000
|
||||
self._interface = SpiDev()
|
||||
self._interface.open(port, device)
|
||||
self._interface.max_speed_hz = 500000
|
||||
|
||||
def _conflicts_with(self, other):
|
||||
return not (
|
||||
@@ -94,17 +96,17 @@ class LocalPiHardwareSPI(SPI, Device):
|
||||
)
|
||||
|
||||
def close(self):
|
||||
if self._intf:
|
||||
if self._interface:
|
||||
try:
|
||||
self._intf.close()
|
||||
self._interface.close()
|
||||
finally:
|
||||
self._intf = None
|
||||
self._interface = None
|
||||
self._release_all()
|
||||
super(LocalPiHardwareSPI, self).close()
|
||||
|
||||
@property
|
||||
def closed(self):
|
||||
return self._intf is None
|
||||
return self._interface is None
|
||||
|
||||
def __repr__(self):
|
||||
try:
|
||||
@@ -119,31 +121,31 @@ class LocalPiHardwareSPI(SPI, Device):
|
||||
:attr:`bits_per_word` bits or less) to the SPI interface, and reads an
|
||||
equivalent number of words, returning them as a list of integers.
|
||||
"""
|
||||
return self._intf.xfer2(data)
|
||||
return self._interface.xfer2(data)
|
||||
|
||||
def _get_clock_mode(self):
|
||||
return self._intf.mode
|
||||
return self._interface.mode
|
||||
|
||||
def _set_clock_mode(self, value):
|
||||
self._intf.mode = value
|
||||
self._interface.mode = value
|
||||
|
||||
def _get_lsb_first(self):
|
||||
return self._intf.lsbfirst
|
||||
return self._interface.lsbfirst
|
||||
|
||||
def _set_lsb_first(self, value):
|
||||
self._intf.lsbfirst = bool(value)
|
||||
self._interface.lsbfirst = bool(value)
|
||||
|
||||
def _get_select_high(self):
|
||||
return self._intf.cshigh
|
||||
return self._interface.cshigh
|
||||
|
||||
def _set_select_high(self, value):
|
||||
self._intf.cshigh = bool(value)
|
||||
self._interface.cshigh = bool(value)
|
||||
|
||||
def _get_bits_per_word(self):
|
||||
return self._intf.bits_per_word
|
||||
return self._interface.bits_per_word
|
||||
|
||||
def _set_bits_per_word(self, value):
|
||||
self._intf.bits_per_word = value
|
||||
self._interface.bits_per_word = value
|
||||
|
||||
|
||||
class LocalPiSoftwareSPI(SPI, OutputDevice):
|
||||
|
||||
@@ -35,10 +35,12 @@ class PiFactory(Factory):
|
||||
self._info = None
|
||||
self.pins = {}
|
||||
self.pin_class = None
|
||||
self.spi_hardware_class = None
|
||||
self.spi_software_class = None
|
||||
self.shared_spi_hardware_class = None
|
||||
self.shared_spi_software_class = None
|
||||
self.spi_classes = {
|
||||
('hardware', 'exclusive'): None,
|
||||
('hardware', 'shared'): None,
|
||||
('software', 'exclusive'): None,
|
||||
('software', 'shared'): None,
|
||||
}
|
||||
|
||||
def close(self):
|
||||
for pin in self.pins.values():
|
||||
@@ -93,7 +95,7 @@ class PiFactory(Factory):
|
||||
doesn't matter).
|
||||
"""
|
||||
spi_args, kwargs = self._extract_spi_args(**spi_args)
|
||||
shared = kwargs.pop('shared', False)
|
||||
shared = 'shared' if kwargs.pop('shared', False) else 'exclusive'
|
||||
if kwargs:
|
||||
raise SPIBadArgs(
|
||||
'unrecognized keyword argument %s' % kwargs.popitem()[0])
|
||||
@@ -108,10 +110,7 @@ class PiFactory(Factory):
|
||||
'port': 0,
|
||||
'device': {8: 0, 7: 1}[spi_args['select_pin']],
|
||||
}
|
||||
if shared:
|
||||
return self.shared_spi_hardware_class(self, **hardware_spi_args)
|
||||
else:
|
||||
return self.spi_hardware_class(self, **hardware_spi_args)
|
||||
return self.spi_classes[('hardware', shared)](self, **hardware_spi_args)
|
||||
except Exception as e:
|
||||
warnings.warn(
|
||||
SPISoftwareFallback(
|
||||
@@ -125,10 +124,7 @@ class PiFactory(Factory):
|
||||
key: pin.number if isinstance(pin, Pin) else pin
|
||||
for key, pin in spi_args.items()
|
||||
}
|
||||
if shared:
|
||||
return self.shared_spi_software_class(self, **spi_args)
|
||||
else:
|
||||
return self.spi_software_class(self, **spi_args)
|
||||
return self.spi_classes[('software', shared)](self, **spi_args)
|
||||
|
||||
def _extract_spi_args(self, **kwargs):
|
||||
"""
|
||||
|
||||
@@ -76,10 +76,12 @@ class PiGPIOFactory(PiFactory):
|
||||
port=int(os.getenv('PIGPIO_PORT', 8888))):
|
||||
super(PiGPIOFactory, self).__init__()
|
||||
self.pin_class = PiGPIOPin
|
||||
self.spi_hardware_class = PiGPIOHardwareSPI
|
||||
self.spi_software_class = PiGPIOSoftwareSPI
|
||||
self.shared_spi_hardware_class = PiGPIOHardwareSPIShared
|
||||
self.shared_spi_software_class = PiGPIOSoftwareSPIShared
|
||||
self.spi_classes = {
|
||||
('hardware', 'exclusive'): PiGPIOHardwareSPI,
|
||||
('hardware', 'shared'): PiGPIOHardwareSPIShared,
|
||||
('software', 'exclusive'): PiGPIOSoftwareSPI,
|
||||
('software', 'shared'): PiGPIOSoftwareSPIShared,
|
||||
}
|
||||
self._connection = pigpio.pi(host, port)
|
||||
self._host = host
|
||||
self._port = port
|
||||
|
||||
Reference in New Issue
Block a user