在Python 中自訂輸入的預設值

在Python 中，預設輸入指令(input()) 可讓您擷取使用者輸入。但是，您可能希望提供一個使用者可以編輯或接受的預設值。

令人驚訝的是，標準輸入函數 input() 和 raw_input() 缺乏此功能。這是一個利用 readline 模組的解決方案。

readline 模組

如果您在 Linux 環境中工作，您可以利用 readline 模組建立自訂輸入功能。 readline 模組提供行編輯並允許您預先填入輸入欄位。

定義自訂輸入函數

以下是如何定義自訂輸入函數，該函數採用提示和可選的預先填充值：

import readline

def rlinput(prompt, prefill=''):
   readline.set_startup_hook(lambda: readline.insert_text(prefill))
   try:
      return input(prompt)  # or raw_input in Python 2
   finally:
      readline.set_startup_hook()

用法

要使用這個自訂輸入函數，只要像這樣呼叫它：

folder_name = rlinput('Folder name: ', 'Downloads')

此程式碼將顯示一個提示，要求輸入資料夾名稱。它最初會在輸入欄位中顯示“下載”，使用者可以對其進行編輯或按 Enter 接受預設值。