Linux 下 Python 實現按任意鍵退出

某天在群內有同學問到，在python下我用input或者raw_input都得輸入完後回車才能獲取到輸入的值，那如何實現任意鍵退出暫停等功能呢，我當時也沒有多想，因為接觸python時間也不算長，主要還是Linux下的。

當然，Windows系統下會稍微簡單一些，Windows系統下如果你安裝了python的環境，默認自帶的一個模塊叫做msvcrt，import msvcrt，然後調用msvcrt.getch()即可。接下來即Linux下實現python版本的按任意鍵退出。

初學Python時在總想實現一個按任意鍵繼續/退出的程序(受.bat毒害), 奈何一直寫不出來, 最近學習Unix C時發現可以通過termios.h庫來實現, 嘗試一下發現Python也有這個庫, 所以終於寫出一個這樣的程序. 下面是代碼:

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import os
import sys
import termios


def press_any_key_exit(msg):
    # 獲取標准輸入的描述符
    fd = sys.stdin.fileno()

    # 獲取標准輸入(終端)的設置
    old_ttyinfo = termios.tcgetattr(fd)

    # 配置終端
    new_ttyinfo = old_ttyinfo[:]

    # 使用非規范模式(索引3是c_lflag 也就是本地模式)
    new_ttyinfo[3] &= ~termios.ICANON
    # 關閉回顯(輸入不會被顯示)
    new_ttyinfo[3] &= ~termios.ECHO

    # 輸出信息
    sys.stdout.write(msg)
    sys.stdout.flush()
    # 使設置生效
    termios.tcsetattr(fd, termios.TCSANOW, new_ttyinfo)
    # 從終端讀取
    os.read(fd, 7)

    # 還原終端設置
    termios.tcsetattr(fd, termios.TCSANOW, old_ttyinfo)


if __name__ == "__main__":
    press_any_key_exit("按任意鍵繼續...")
    press_any_key_exit("按任意鍵退出...")

其他關於termios的信息可以參考Linux手冊:

man 3 termios

另補充一下*nix終端的三種模式(摘自<Unix-Linux編程實踐教程>)

規范模式

規范模式, 也被成為cooked模式, 是用戶常見的模式.驅動程序輸入的字符保存在緩沖區, 並且僅在接收到回車鍵時才將這些緩沖的字符發送到程序.緩沖數據使驅動程序可以實現最基本的編輯功能, 被指派這些功能的特定鍵在驅動程序裡設置, 可以通過命令stty或系統調用tcsetattr來修改

非規范模式

當緩沖和編輯功能被關閉時, 連接被成為非規范模式.終端處理器仍舊進行特定的字符處理, 例如處理Ctrl-C及換行符之間的轉換, 但是編輯鍵將沒有意義, 因此相應的輸入被視為常規的數據輸入 程序需要自己實現編輯功能

raw模式

當所有處理都被關閉後, 驅動程序將輸入直接傳遞給程序, 連接被成為raw模式.