Python修改DBF文件指定列

更新時間：2020年12月19日 10:49:02 作者：Peanut_C

這篇文章主要介紹了Python修改DBF文件指定列的實現(xiàn)，幫助大家更好的理解和使用python，感興趣的朋友可以了解下

一、需求：

　　某公司每日收到一批DBF文件，A系統(tǒng)實時處理后將其中dealstat字段置為1（已處理）?，F(xiàn)在每日晚間B系統(tǒng)也需要處理該文件，因此需將文件中dealstat字段修改為空（未處理）。

二、分析：

　　1、應(yīng)創(chuàng)建副本進行修改

　　解答：使用shutil.copy

　　2、修改DBF

　　解答：使用dbf模塊。此模塊能找到的文檔比較舊，需要結(jié)合代碼進行理解。

三、代碼實現(xiàn)：

#!/usr/bin/env python
# _*_ coding:utf-8 _*_


"""
@Time  : 2020-01-15 10:43
@Author : Peanut_C
@FileName: DBF_Modifier.py
"""

import os
import time
import shutil
import dbf


source_dir = r'D:\'
destination_dir = r'E:\'

"""刪除目標文件夾內(nèi)舊文件"""
os.chdir(destination_dir)
for file in os.listdir(destination_dir):
  os.remove(file)
print('INFO ===>>> 歷史文件刪除完成！\n')

"""文件拷貝任務(wù)"""
os.chdir(source_dir)
for file in os.listdir(source_dir):
  shutil.copy(file, destination_dir)
print('INFO ===>>> 今日文件拷貝完成！\n')

"""DBF修改任務(wù)"""
os.chdir(destination_dir)
for file in os.listdir(destination_dir):
  tb = dbf.Table(file) # 創(chuàng)建tb實例
  # print(tb) # 打印tb信息
  titles = dbf.get_fields(file) # 將表頭以列表形式打印出來
  # print(titles)
  if 'dealstat' in titles:
    flag = 0 # 文件修改標記
    tb.open(mode=dbf.READ_WRITE) # 讀寫方式打開tb
    for record in tb:
      with record as r:
        if r.dealstat is not None:
          # print(r.dealstat)
          r.dealstat = ''
          flag = 1 # 修改后將標記改為1
          # print(r.dealstat)
        else:
          continue
      # print(record.dealstat)
    tb.close()
    if flag == 0:
      print(file + "===>>>沒有要修改的數(shù)據(jù)！\n")
    else:
      print(file + "===>>>DealStat字段修改完畢！\n")
  else:
    print(file + "===>>>沒有DealStat字段！\n")

print('INFO ===>>> 今日文件修改完畢！\n')