【原創】轉載請註明做者Johnthegreat和本文連接。html
在一些很少的數據下載和生成的時候,咱們傾向於直接保存爲文件,當咱們修改某些參數後再一次運行時,以前運行時生成的文件就被覆蓋了。爲了解決這個問題,這裏提供幾個解決方案。閉包
方案一思路:函數
1. 判斷文件是否存在;測試
2. 判斷是否帶有」0)「這種數字帶括號的格式;ui
3. 文件名添加」(0), (1), (2)….「之類的編號。.net
如下是方案一代碼:code
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
若是使用以下建立文件的代碼測試:htm
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
方案二思路:遞歸
來自(https://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!')
方案二使用了閉包函數來實現,也很巧妙。
以上都是通用型代碼,直接在代碼裏面添加這個函數就能夠用了,有問題歡迎聯繫我。