import cv2
import time


def test_video():
    cv2.namedWindow('video', cv2.WINDOW_AUTOSIZE)
    cv2.resizeWindow('video', 640, 480)  # 设置窗口大小
    # 获取视频设备
    cap = cv2.VideoCapture(0)  # 将0换成文件名，可进行文件播放，注意路径
    while True:
        ret, fame = cap.read()
        cv2.imshow('video', fame)
        # 等待键盘事件
        key = cv2.waitKey(1)  # 为0则只显示一帧
        # 若播放过快，造成卡顿
        # key=cv2.waitKey(40)#25帧-1000/25
        if (key & 0xff == ord('q')):
            break
        if (key & 0xff == ord("s")):
            cv2.imwrite("/home/caiquan/work/save/{}.jpg".format(int(time.time() * 1000)), fame)
    # 释放viedcapture
    cap.release()
    # 销毁所有窗口
    cv2.destroyAllWindows()


if __name__ == '__main__':
    test_video()