while文の中のreturnの戻り値を、まわすたびに返したい

Question

前提・実現したいこと

ラズパイでモーターを制御し、ボタンを押すとリフトが上下動するプログラムを組んでいます。その際、measurementボタンを押して、func1の3つ目のwhileの中のwhile num <=100:がまわるたびにreturnでnumを返し、他の関数でnumで受け取れるようにしたいと考えています。（1,2,...99,100を逐次func1に返すイメージ）いろいろ試してはいるのですが、そもそもreturnを記述するだけで逐次変化するnumをfunc1に返すことが可能なのか、また他の関数でnumを受け取るにはfunc1の呼び出しだけでいいのかどうか、どうもいまひとつわかりません。なにかよい方法があればアドバイスいただけますとありがたいです。よろしくお願いいたします。

該当のソースコード

#improtなどの前提部分は省略

def func1():
    while True:
        while (GPIO.input(up_pin) == 1):
            GPIO.output(dircPin, 0)
            time.sleep(0.001)
            GPIO.output(stepPin, 1)
            time.sleep(0.001)
            GPIO.output(stepPin, 0)
            time.sleep(0.001)
            #print("up")
        
        while (GPIO.input(down_pin) == 1):
            GPIO.output(dircPin, 1)
            time.sleep(0.001)
            GPIO.output(stepPin, 1)
            time.sleep(0.001)
            GPIO.output(stepPin, 0)
            time.sleep(0.001)
            #print("down")

        while (GPIO.input(measurement_pin) == 1):
            #print("measurement")
            num = 0
            while num <=100:
                GPIO.output(dircPin, 0)
                time.sleep(0.001)
                GPIO.output(stepPin, 1)
                time.sleep(0.001)
                GPIO.output(stepPin, 0)
                x=analog_read(0)    
                num += 1
                time.sleep(0.05)
                return num
           
def main():
    func1()    
    print()



if __name__=="__main__":
    main()

試したこと

上記のプログラムで、main()関数を用意し、その中でfunc1を呼び出してnumをprintできるかどうか試してみましたが、measurementボタンを押してもリフトは反応せず、printも何もされませんでした。(returnなしのプログラムではリフト動作することは確認できています）

Answer 1

Yield https://docs.python.org/ja/3/reference/expressions.html#yieldexpr
を利用すると良いのではないでしょうか。

        while (GPIO.input(measurement_pin) == 1):
          ...
          yield num
          ...


def main():
  for n in func1():
    print(n)

Comments

[Uka]

@t_obaraさん、
ご回答ありがとうございます。ご教示いただいたyieldを使った方法で解決できました！以前自分で試したときにはyieldでもうまくいかなかったのですが、、、ありがとうございました！