一、需求:
某公司每日收到一批DBF文件,A系統實時處理后將其中dealstat字段置為1(已處理)。現在每日晚間B系統也需要處理該文件,因此需將文件中dealstat字段修改為空(未處理)。
二、分析:
1、應創建副本進行修改
解答:使用shutil.copy
2、修改DBF
解答:使用dbf模塊。此模塊能找到的文檔比較舊,需要結合代碼進行理解。
三、代碼實現:
1 #!/usr/bin/env python 2 # _*_ coding:utf-8 _*_ 3 4 5 """ 6 @Time : 2020-01-15 10:43 7 @Author : Peanut_C 8 @FileName: DBF_Modifier.py 9 """ 10 11 import os 12 import time 13 import shutil 14 import dbf 15 16 17 source_dir = r'D:\' 18 destination_dir = r'E:\' 19 20 """刪除目標文件夾內舊文件""" 21 os.chdir(destination_dir) 22 for file in os.listdir(destination_dir): 23 os.remove(file) 24 print('INFO ===>>> 歷史文件刪除完成!\n') 25 26 """文件拷貝任務""" 27 os.chdir(source_dir) 28 for file in os.listdir(source_dir): 29 shutil.copy(file, destination_dir) 30 print('INFO ===>>> 今日文件拷貝完成!\n') 31 32 """DBF修改任務""" 33 os.chdir(destination_dir) 34 for file in os.listdir(destination_dir): 35 tb = dbf.Table(file) # 創建tb實例 36 # print(tb) # 打印tb信息 37 titles = dbf.get_fields(file) # 將表頭以列表形式打印出來 38 # print(titles) 39 if 'dealstat' in titles: 40 flag = 0 # 文件修改標記 41 tb.open(mode=dbf.READ_WRITE) # 讀寫方式打開tb 42 for record in tb: 43 with record as r: 44 if r.dealstat is not None: 45 # print(r.dealstat) 46 r.dealstat = '' 47 flag = 1 # 修改后將標記改為1 48 # print(r.dealstat) 49 else: 50 continue 51 # print(record.dealstat) 52 tb.close() 53 if flag == 0: 54 print(file + "===>>>沒有要修改的數據!\n") 55 else: 56 print(file + "===>>>DealStat字段修改完畢!\n") 57 else: 58 print(file + "===>>>沒有DealStat字段!\n") 59 60 print('INFO ===>>> 今日文件修改完畢!\n')
四、運行情況:
運行程序,首先清空目標目錄,然后創建文件副本,最后依次處理目錄中的DBF文件。
雖是個簡單的功能,但可節省不少時間,依此類推可對DBF進行其他修改。
希望能幫到有需要的朋友。
多多指教!