使Python從目錄顯示.txt文件

Question

print "Which category would you like to view? Savory, Dessert, Cake, Soup or Drink?    "
category = raw_input()
if category == "Savory": #opens directory according to their answer.
    os.listdir("savory")

elif category == "Drink":
    os.listdir("drink")

elif category == "Cake":
    os.listdir("cake")

elif category == "Dessert":
    os.listdir("dessert")

elif category == "savory":
    os.listdir("savory")

elif category == "drink":
    os.listdir("drink")

elif category == "cake":
    os.listdir("cake")

elif category == "dessert":
    os.listdir("dessert")

我正在嘗試創建一個代碼，以顯示已保存在某些目錄下的配方。 運行代碼時，即使我在目錄中輸入了配方，也不會顯示任何內容。

Answer 1

首先，您什么都不打印。 例如

os.listdir("savory")

應該

print(os.listdir("savory"))

其次，您可以大大縮短代碼：

print("Which category would you like to view? Savory, Dessert, Cake, Soup or Drink?")
print(os.listdir(raw_input().lower()))

如果要限制用戶訪問其他目錄：

safe = ['savory', 'dessert', 'cake', 'soup', 'drink']

category = raw_input().lower()
if category in safe:
    print(os.listdir(category))
else:
    print('You have typed an invalid category')