在python中使用不同的按键调用不同的python函数

Question

我有两个功能：screenshot()和reader()。我希望在按下screenshot()时调用0，并在按下reader()时调用1。如果按下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__":。 读者（）正在工作，你知道它是如何工作的吗？复制sample文本，尝试粘贴到文本编辑器中。