KevinMidboe/python-gpiozero
1
0
Fork 0
mirror of https://github.com/KevinMidboe/python-gpiozero.git synced 2025-10-29 17:50:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add optional enable init param to motor

Browse Source
This commit is contained in:
Ben Nuttall
2016-06-08 16:03:52 +01:00
parent 8b81057f8b
commit 966ada0a0c
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
gpiozero/output_devices.py
View File

@@ -797,6 +797,10 @@ class Motor(SourceMixin, CompositeDevice):
The GPIO pin that the backward input of the motor driver chip is The GPIO pin that the backward input of the motor driver chip is
connected to. connected to.
:param int enable:
(Optional) The GPIO pin that enables the motor. Required for some motor
controller boards. Defaults to ``None``.
:param bool pwm: :param bool pwm:
If ``True`` (the default), construct :class:`PWMOutputDevice` If ``True`` (the default), construct :class:`PWMOutputDevice`
instances for the motor controller pins, allowing both direction and instances for the motor controller pins, allowing both direction and
@@ -804,13 +808,20 @@ class Motor(SourceMixin, CompositeDevice):
:class:`DigitalOutputDevice` instances, allowing only direction :class:`DigitalOutputDevice` instances, allowing only direction
control. control.
""" """
def __init__(self, forward=None, backward=None, pwm=True): def __init__(self, forward=None, backward=None, enable=None, pwm=True):
if not all([forward, backward]): if not all([forward, backward]):
raise GPIOPinMissing( raise GPIOPinMissing(
'forward and backward pins must be provided' 'forward and backward pins must be provided'
) )
PinClass = PWMOutputDevice if pwm else DigitalOutputDevice PinClass = PWMOutputDevice if pwm else DigitalOutputDevice
super(Motor, self).__init__( if enable:
super(Motor, self).__init__(
forward_device=PinClass(forward),
backward_device=PinClass(backward),
enable_device=DigitalOutputDevice(enable, initial_value=True),
_order=('forward_device', 'backward_device', 'enable_device'))
else:
super(Motor, self).__init__(
forward_device=PinClass(forward), forward_device=PinClass(forward),
backward_device=PinClass(backward), backward_device=PinClass(backward),
_order=('forward_device', 'backward_device')) _order=('forward_device', 'backward_device'))