在类中检查 Python 中的函数调用

Question

我正在尝试编写一个类来使用 Raspberry Pi 控制步进电机。

我的班级示例：

class Stepmotor:
    def __init__(...):
    ...
 
    def Rotate(..):
        while (function Stop is not called):
            'motor rotate'

    def Stop(..):
        'motor stop'

代码示例：

motor = Stepmotor(..)
motor.Rotate() ---> rotate motor for 5 second
time.sleep(5)
motor.Stop() ---> stop motor

这是我使用 Raspberry Pi 的原始代码：

import RPi.GPIO as gpio
import time

class StepMot:
    gpio.setwarnings(False)
    gpio.setmode(gpio.BCM)
    a = []
    
    def __init__(self, in1, in2, in3, in4):
        global a
        a = [in1, in2, in3, in4]
        gpio.setup(in1, gpio.OUT)
        gpio.setup(in2, gpio.OUT)
        gpio.setup(in3, gpio.OUT)
        gpio.setup(in4, gpio.OUT)

    def gpio_controll(self, x, y, z, w):
        global a
        gpio.output(a[0], x)
        gpio.output(a[1], y)
        gpio.output(a[2], z)
        gpio.output(a[3], w)
        time.sleep(0.001)

    def RotateCW(self):
        while True:
            self.gpio_controll(0,0,0,0)
            self.gpio_controll(1,0,0,0)
            self.gpio_controll(1,1,0,0)
            self.gpio_controll(0,1,0,0)
            self.gpio_controll(0,1,1,0)
            self.gpio_controll(0,0,1,0)
            self.gpio_controll(0,0,1,1)
            self.gpio_controll(0,0,0,1)
            self.gpio_controll(1,0,0,1)

   def RotateACW(self):
        while True:
           self.gpio_controll(0,0,0,0)
           self.gpio_controll(1,0,0,1)
           self.gpio_controll(0,0,0,1)
           self.gpio_controll(0,0,1,1)
           self.gpio_controll(0,0,1,0)
           self.gpio_controll(0,1,1,0)
           self.gpio_controll(0,1,0,0)
           self.gpio_controll(1,1,0,0)
           self.gpio_controll(1,0,0,0)

   def Stop(self):
       for i in a:
           gpio.output(i, 0)
 

当我“顺时针”调用RotateCW 时，电机在Stop 调用后仍继续旋转 5 秒。

我知道这是因为代码在while True 中被阻止，那是因为我直接在课堂上搜索解决方案。当在代码中调用 Stop() 时，RotateCW() 中的 while 循环停止。

Answer 1

您可以将duration 参数添加到RotateCW() 并在while 循环中检查它：

    def RotateCW(self, duration):
        stop_time = time.time() + duration
        while time.time() < stop_time:
            ...