Python實作按任意鍵繼續/退出的功能

前言

要實現該功能，需要的就是暫停程式、等待並捕捉使用者的一個鍵盤輸入，然後繼續執行。 Python 有內建的函式庫能幫我們實作該功能，不過要區別對待 Windows 和 Linux。

msvcrt 中的 getch() 方法 能夠幫助在 Windows 下實現，其作用是取得一個按鍵回應並傳回對應的字元。它並不在命令列中回顯。有如下程式段：

import msvcrt
print ord(msvcrt.getch())

這裡利用ord 將獲得的字元轉換為ASCII 數值，例如捕獲按鍵“d”（注意是小寫）將得到數值100。

Linux 下呢？嗯，相對複雜一點點，不過先理清思路的話就好辦了。

首先要知道Linux 終端機的三種模式，分別為規範模式、非規範模式與 raw 模式：

規範模式

規格模式，也被成為cooked 模式，是使用者常見的模式。驅動程式輸入的字元保存在緩衝區，並且僅在接收到回車鍵時才將這些緩衝的字元傳送到程式。緩衝資料使驅動程式可以實現最基本的編輯功能， 被指派這些功能的特定鍵在驅動程式裡設置，可以透過命令 stty 或系統呼叫 tcsetattr 來修改。

非規範模式

當緩衝和編輯功能被關閉時，連線被變成非規範模式。終端處理器仍舊進行特定的字元處理，例如處理 Ctrl-C 及換行符之間的轉換，但編輯鍵將沒有意義，因此對應的輸入被視為常規的資料輸入，程式需要自行實現編輯功能。

raw 模式

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

這裡我們需要藉助於非規範模式，那麼要實作剛才Windows 上的相似行為，需要以下程式碼：

##

import os
import termios
 
# 获取标准输入的描述符
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
 
# 使设置生效
termios.tcsetattr(fd, termios.TCSANOW, new_ttyinfo)
 
# 从终端读取
print ord(os.read(fd, 7))

import msvcrt
 
print("Press 'D' to exit...")
 
while True:
 if ord(msvcrt.getch()) in [68, 100]:
  break