使用python中的不同按鍵調用不同的python函數

Question

我有兩個函數：screenshot()和reader()。當按下0時，我想要調用screenshot()，按1時要調用reader()。如果按下d，它應該退出。這是我嘗試：

from PIL import Image 
import pyscreenshot as ImageGrab 
import datetime 
import time 
import os 
import pyperclip 

def screenshot(): 
    if __name__ == "__main__": 
     im=ImageGrab.grab() 
     timestr = time.strftime("%Y%m%d-%H%M%S") 
     file_name = "sample_" + timestr + ".png" 
     file_location = os.path.join('/path', file_name) 
     im.save(file_location) 

def reader(): 
    pyperclip.copy("sample") 

while True: 
    x = raw_input('Press enter for a random letter...') 
    print x 
    if x == "0": 
     screenshot() 

    if x == "1": 
     reader() 

    if x == "d": 
     break 

但預期它不工作。當我按0時，沒有任何反應。當我按1時，沒有任何反應。但是當第一次按d時，它會調用screenshot()。並且下一次d被按下時，它退出。

Answer 1

刪除該行if __name__ == "__main__":。 而reader（）正在工作，你知道它是如何工作的嗎？它複製一個sample文本，嘗試粘貼到你的文本編輯器中。