Python使用进程间共享变量来控制两个进程（监听键盘和相机录制）的交互

我有个简单的应用需求：

1. 该应用随时会监听键盘的输入；

2. 当输入指定键时会控制相机录制的启动和关闭。

监听键盘是一个事件循环，相机录制也是一个循环录制的过程。我试着用 Python 启动两个进程，并用两个进程共享变量的更新来控制两个进程的交互。

监听键盘输入

首先我找到python 监听键盘输入的方案可以满足我监听键盘的需求。

 import sys, select, tty, termios

 old_attr = termios.tcgetattr(sys.stdin)

 tty.setcbreak(sys.stdin.fileno())

 print('Please input keys, press Ctrl + C to quit')

 while(1):

     if select.select([sys.stdin], [], [], 0)[0] == [sys.stdin]:

         print(sys.stdin.read(1))

 termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_attr)

这段代码可以进行键盘监听，但是却有一个潜在 Bug：第 11 行的代码永远无法执行，即使加上跳出循环的判断，当使用 Ctrl+C 时程序会终止。

而第 11 行的作用是恢复当前终端的原始属性，没执行到该语句，终端会变得不正常：你会发现程序结束后在该终端输入字符不会有显示。因此，程序应该设法程序终止时必然执行该语句。

我建议改为类似的代码来修复这个 Bug：

 import sys

 import select

 import tty

 import termios

 old_attr = termios.tcgetattr(sys.stdin)

 tty.setcbreak(sys.stdin.fileno())

 print('Please input keys, press Ctrl + C to quit')

 try:

     while True:

         if select.select([sys.stdin], [], [], 0)[0] == [sys.stdin]:

             key = sys.stdin.read(1)

             if key == 'q':

                 # to stop the infinite loop

                 break

             elif key == 's':

                 # do something

                 print('I will do something')

             else:

                 # do other things

                 print(key)

 except Exception as e:

     print(e)

 finally:

     termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_attr)

相机录制

相机录制的路径当然因不同的相机而异，因此这里我只写 Demo 级别的抽象代码：

 import time

 class Recorder:

     def __init__(self):

         print('Recorder initialization.')

     def run(self):

         time.sleep(0.3)

         print('Recording a frame.')

     def close(self):

         print('Recording ended.')

 recorder = Recorder()

 t0 = time.time()

 while time.time() - t0 < 10:

     recorder.run()

 recorder.close()

可以看到这段逻辑也有一个条件循环。

假设应用的需求是按键 s，进行录制；再次按键 s，终止录制。

由于这两个逻辑需要完全同时进行，而 Python 的线程又是伪并行，因此这里考虑用多进程共享变量的通信。

使用进程间共享变量

使用 multiprocessing，直接看完整的代码是比较直观的：

 import time

 import sys

 import select

 import tty

 import termios

 import multiprocessing

 class Recorder:

     def __init__(self):

         print('Recorder initialization.')

     def run(self):

         time.sleep(0.3)

         print('Recording a frame.')

     def close(self):

         print('Recording ended.')

 def recording_process(shared):

     recorder = Recorder()

     while shared['is_recording']:

         recorder.run()

     recorder.close()

 if __name__ == '__main__':

     # set shared variable

     manager = multiprocessing.Manager()

     shared = manager.dict()

     shared['is_recording'] = False

     old_attr = termios.tcgetattr(sys.stdin)

     tty.setcbreak(sys.stdin.fileno())

     print('Please input keys, press Ctrl + C to quit')

     # check the keyborad

     p = None

     try:

         while True:

             if select.select([sys.stdin], [], [], 0)[0] == [sys.stdin]:

                 key = sys.stdin.read(1)

                 if key == 'q':

                     # to stop the infinite loop

                     break

                 elif key == 's':

                     # toggle

                     shared['is_recording'] = not shared['is_recording']

                     # resume or stop recording

                     if shared['is_recording']:

                         p = multiprocessing.Process(target=recording_process, args=(shared,))

                         p.start()

                     else:

                         p and p.join()

                 else:

                     # do other things

                     print(key)

     except Exception as e:

         print(e)

     finally:

         termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_attr)

原文作者：雨先生
原文链接：https://www.cnblogs.com/noluye/p/11704982.html
许可协议：知识共享署名-非商业性使用 4.0 国际许可协议

参考