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 国际许可协议

参考

• python 监听键盘输入
• 进程间共享变量
• multiprocessing — Process-based parallelism