Python – 實現文件名自動更改，避免同名文件被覆蓋的兩個解決方法

• 2020 年 4 月 21 日
• 筆記
• Python

【原創】轉載請註明作者Johnthegreat和本文鏈接。

在一些不多的數據下載和生成的時候，我們傾向於直接保存為文件，當我們修改某些參數後再一次運行時，之前運行時生成的文件就被覆蓋了。為了解決這個問題，這裡提供幾個解決方案。

方案一思路：

1. 判斷文件是否存在；

2. 判斷是否帶有」0)「這種數字帶括弧的格式；

3. 文件名添加」(0), (1), (2)….「之類的編號。

以下是方案一程式碼：

import os
import re


def auto_save_file(path):
    directory, file_name = os.path.split(path)
    while os.path.isfile(path):
        if re.search('(\d+)\)\.', file_name) is None:
            file_name = file_name.replace('.', '(0).')
        else:
            current_number = int(re.findall('(\d+)\)\.', file_name)[-1])
            new_number = current_number + 1
            file_name = file_name.replace(f'({current_number}).', f'({new_number}).')
        path = os.path.join(directory + os.sep + file_name)
    return path

如果使用如下創建文件的程式碼測試：

path = r'D:\test.txt'
for i in range(10):
    with open(auto_save_file(path), 'w') as f:
        f.write('This is a test!')

結果如下：

方案二思路：

來自（//blog.csdn.net/yyly2012/article/details/79106554）

1. 使用os.path.isfile判斷文件是否存在
2. 使用遞歸函數判斷
3. 添加編號時, 使用可變list 傳值n=[0]

import os

def check_filename_available(filename):
    n=[0]
    def check_meta(file_name):
        file_name_new=file_name
        if os.path.isfile(file_name):
            file_name_new=file_name[:file_name.rfind('.')]+'_'+str(n[0])+file_name[file_name.rfind('.'):]
            n[0]+=1
        if os.path.isfile(file_name_new):
            file_name_new=check_meta(file_name)
        return file_name_new
    return_name=check_meta(filename)
    return return_name
with open(check_filename_available('t.txt'),'w') as f:
    f.write('Checking func!')

方案二使用了閉包函數來實現，也很巧妙。

以上都是通用型程式碼，直接在程式碼裡面添加這個函數就可以用了，有問題歡迎聯繫我。