<p>在适当的地方使用一个简单的<code>letter = letter.lower()</code>应该可以实现您想要的功能:</p>
<pre><code>VOWELS = ["a", "e", "i", "o", "u"]
CONSONANTS = ["b", "c", "d", "f", "g", "h", "j", "k", "l", "m", "n", "p", "q", "r", "s", "t", "v", "w", "x", "y", "z"]
def encrypt(message: str):
encryptedMessage = ""
for letter in message:
letter = letter.lower()
if letter in VOWELS:
newIndex = VOWELS.index(letter) + 1
if newIndex >= len(VOWELS):
newIndex = 0
encryptedMessage += VOWELS[newIndex]
elif letter in CONSONANTS:
newIndex = CONSONANTS.index(letter) + 1
if newIndex >= len(CONSONANTS):
newIndex = 0
encryptedMessage += CONSONANTS[newIndex]
else:
encryptedMessage += letter
return encryptedMessage
def decrypt(message: str):
decryptedMessage = ""
for letter in message:
letter = letter.lower()
if letter in VOWELS:
newIndex = VOWELS.index(letter) - 1
if newIndex >= len(VOWELS):
newIndex = 0
decryptedMessage += VOWELS[newIndex]
elif letter in CONSONANTS:
newIndex = CONSONANTS.index(letter) - 1
if newIndex >= len(CONSONANTS):
newIndex = 0
decryptedMessage += CONSONANTS[newIndex]
else:
decryptedMessage += letter
return decryptedMessage
def better_encryption(message: str, is_encrypt: bool):
newMessage = ""
ed = 1 if is_encrypt else -1
for letter in message:
letter = letter.lower()
if letter in VOWELS:
newIndex = VOWELS.index(letter) + ed
if newIndex >= len(VOWELS):
newIndex = 0
newMessage += VOWELS[newIndex]
elif letter in CONSONANTS:
newIndex = CONSONANTS.index(letter) + ed
if newIndex >= len(CONSONANTS):
newIndex = 0
newMessage += CONSONANTS[newIndex]
else:
newMessage += letter
return newMessage
def main():
while True:
command = str(input("[E]ncrypt or [D]ecrypt? "))
if "E" in command.upper() :
print(better_encryption(input("Message: "), True))
elif "D" in command.upper() :
print(better_encryption(input("Message: "), False))
else:
print("Invalid")
</code></pre>