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