diff options
-rw-r--r-- | LegoJAM.py | 18 |
1 files changed, 6 insertions, 12 deletions
@@ -146,20 +146,14 @@ class Robot(gobject.GObject): gobject.timeout_add(1000, self.handle) def run_motor(self, motor): - if motor == "A": - self.motorA.run(power=127, regulated=False) - elif motor == "B": - self.motorB.run(power=127, regulated=False) - elif motor == "C": - self.motorC.run(power=127, regulated=False) + method_name = 'motor%s' % motor + method = getattr(self, method_name) + method.run(power=127, regulated=False) def stop_motor(self, motor): - if motor == "A": - self.motorA.brake() - elif motor == "B": - self.motorB.brake() - elif motor == "C": - self.motorC.brake() + method_name = 'motor%s' % motor + method = getattr(self, method_name) + method.brake() def handle(self): valor_boton = self.presion.get_sample() |