如果新文件不存在则写入新文件，如果存在则追加到文件

Question

我有一个程序将用户的highscore 写入文本文件。该文件由用户在选择playername 时命名。

如果具有该特定用户名的文件已存在，则程序应附加到该文件（以便您可以看到多个highscore）。如果不存在具有该用户名的文件（例如，如果用户是新用户），它应该创建一个新文件并写入它。

这是相关的，到目前为止还没有工作的代码：

try: 
    with open(player): #player is the varible storing the username input
        with open(player, 'a') as highscore:
            highscore.write("Username:", player)

except IOError:
    with open(player + ".txt", 'w') as highscore:
        highscore.write("Username:", player)

如果文件不存在，上面的代码会创建一个新文件，并写入它。如果它存在，则在我检查文件时没有附加任何内容，并且我没有收到任何错误。

Answer 1

你试过'a+'模式吗？

with open(filename, 'a+') as f:
    f.write(...)

但请注意，f.tell() 在 Python 2.x 中将返回 0。详情请见https://bugs.python.org/issue22651。

Answer 2

我不清楚您感兴趣的高分的确切存储位置，但下面的代码应该是您需要检查文件是否存在并在需要时附加到它的代码。我更喜欢这种方法而不是“try/except”。

import os
player = 'bob'

filename = player+'.txt'

if os.path.exists(filename):
    append_write = 'a' # append if already exists
else:
    append_write = 'w' # make a new file if not

highscore = open(filename,append_write)
highscore.write("Username: " + player + '\n')
highscore.close()

Answer 3

只需在'a'模式下打开即可：

a   开放供写作。如果文件不存在，则创建该文件。流位于文件的末尾。

with open(filename, 'a') as f:
    f.write(...)

要查看您是否正在写入新文件，请检查流位置。如果为零，则说明文件为空或为新文件。

with open('somefile.txt', 'a') as f:
    if f.tell() == 0:
        print('a new file or the file was empty')
        f.write('The header\n')
    else:
        print('file existed, appending')
    f.write('Some data\n')

如果您仍在使用 Python 2，要解决the bug，请在open 之后添加f.seek(0, os.SEEK_END) 或改用io.open。

Answer 4

请注意，如果文件的父文件夹不存在，您将收到相同的错误：

IOError: [Errno 2] 没有这样的文件或目录：

以下是处理这种情况的另一种解决方案：
(*) 我使用 sys.stdout 和 print 而不是 f.write 只是为了展示另一个用例

# Make sure the file's folder exist - Create folder if doesn't exist
folder_path = 'path/to/'+folder_name+'/'
if not os.path.exists(folder_path):
     os.makedirs(folder_path)

print_to_log_file(folder_path, "Some File" ,"Some Content")

其中内部print_to_log_file只管文件级别：

# If you're not familiar with sys.stdout - just ignore it below (just a use case example)
def print_to_log_file(folder_path ,file_name ,content_to_write):

   #1) Save a reference to the original standard output       
    original_stdout = sys.stdout   
    
    #2) Choose the mode
    write_append_mode = 'a' #Append mode
    file_path = folder_path + file_name
    if (if not os.path.exists(file_path) ):
       write_append_mode = 'w' # Write mode
     
    #3) Perform action on file
    with open(file_path, write_append_mode) as f:
        sys.stdout = f  # Change the standard output to the file we created.
        print(file_path, content_to_write)
        sys.stdout = original_stdout  # Reset the standard output to its original value

---

考虑以下状态：

'w'  --> Write to existing file
'w+' --> Write to file, Create it if doesn't exist
'a'  --> Append to file
'a+' --> Append to file, Create it if doesn't exist

在你的情况下，我会使用不同的方法，只使用 'a' 和 'a+'。

Answer 5

使用pathlib 模块（python 的object-oriented filesystem paths）

只是为了好玩，这可能是解决方案的最新 pythonic 版本。

from pathlib import Path 

path = Path(f'{player}.txt')
path.touch()  # default exists_ok=True
with path.open('a') as highscore:
   highscore.write(f'Username:{player}')