首頁  >  文章  >  後端開發  >  Python實作按任意鍵繼續/退出的功能

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

高洛峰
高洛峰原創
2017-02-25 10:41:2310287瀏覽

前言

要實現該功能,需要的就是暫停程式、等待並捕捉使用者的一個鍵盤輸入,然後繼續執行。 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

這樣,當用戶按下“D”或“d”時,則程式退出。

更多Python實現按任意鍵繼續/退出的功能相關文章請關注PHP中文網!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn