一、需求:
某公司每日收到一批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进行其他修改。
希望能帮到有需要的朋友。
多多指教!