python - 如何在换行符处从文本文件中分割文本？

Question

我需要加密一条消息。消息如下，它保存在名为 assignmenttest.txt 的文件中

Hi my name is Allie
I am a Junior
I like to play volleyball

我需要程序加密每一行并保持其格式，以便因此，我编写了以下程序:

fileInputName = input("Enter the file you want to encrypt: ")
key = int(input("Enter your shift key: "))
outputFileName = input("Enter the file name to write to: ")

fileInputOpen = open(fileInputName, "r")
message = fileInputOpen.read()


alphabet = " ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
shiftedStart = alphabet[len(alphabet) - key:]
shiftedEnd = alphabet[:len(alphabet) - key]

shiftedAlphabet = shiftedStart + shiftedEnd

encryptedMessage = ""

for character in message:
    letterIndex = message.split("\n")
    letterIndex = alphabet.find(character)
    encryptedCharacter = shiftedAlphabet[letterIndex]
    #print( "{0} -> {1}".format(character, encryptedCharacter))
    encryptedMessage += encryptedCharacter

print("The encrypted message is: {0}".format(encryptedMessage))


outputFile = open( outputFileName, "w")
print(encryptedMessage, file=outputFile)
outputFile.close()

print("Done writing encrypted message to file {0}".format(outputFileName))

我尝试在\n 处使用分割，但输出并未格式化为三行，而是一长串加密字母。

关于如何在正确的位置分割加密消息并使其显示出来，有什么想法吗？我尝试了多种分割方法，但没有一个有效。非常感谢。

Answer 1

正如其他答案所说，您可以替换

fileInputOpen = open(fileInputName, "r")
message = fileInputOpen.read()

与

with open(fileInputName, "r") as f:
    messages = f.readlines()

这样，messages 将是一个字符串列表，其中每个字符串都是输入文件中一行的文本。然后，通过对消息中每个字符的循环进行一些轻微修改，您可以加密 messages 列表中的每个字符串。在这里，我将您的 encryptedMessage 替换为 currentEncryptedMessage 并添加了 encryptedMessages，这是一个跟踪 消息中每个字符串的加密版本的列表.

encryptedMessages = []
currentEncryptedMessage = ""

for message in messages:
    for character in message:
        ... # same as code provided
        currentEncryptedMessage += encryptedCharacter
    encryptedMessages.append(currentEncryptedMessage)

写入文件时，您可以迭代 encryptedMessages 中的每个元素以逐行打印。

with open( outputFileName, "w") as outputFile:
    for message in encryptedMessages:
        print(message, file=outputFile)

因此您的输出文本文件将保留输入文件中的换行符。