比較兩個文件的異同Python3 標準庫difflib 實現

  • 2019 年 10 月 21 日
  • 筆記

比較兩個文件的異同Python3 標準庫difflib 實現

對於要比較兩個文件特別是配置文件的差異,這種需求很常見,如果用眼睛看,真是眼睛疼。
可以使用linux命令行工具diff a_file b_file,但是輸出的結果讀起來不是很友好。這時候使用python的標準庫difflib就能滿足我們的需求。

下面這個腳本使用了difflib和argparse,argparse用於解析我們給此腳本傳入的兩個參數(即兩份待比較的文件),由difflib執行比較,比較的結果放到了一個html裡面,只要找個瀏覽器打開此html文件,就能直觀地看到比較結果,兩份文件有差異的地方會高亮顯示出來。

廢話少說,直接上程式碼:
『』『

_*_ coding: utf-8 _*_

import difflib
import sys
import argparse

讀取建表語句或配置文件

def read_file(file_name):
try:
file_desc = open(file_name, ‘r’)
# 讀取後按行分割
text = file_desc.read().splitlines()
file_desc.close()
return text
except IOError as error:
print(‘Read input file Error: {0}’.format(error))
sys.exit()

比較兩個文件並把結果生成一份html文本

def compare_file(file1, file2):
if file1 == "" or file2 == "":
print(‘文件路徑不能為空:第一個文件的路徑:{0}, 第二個文件的路徑:{1} .’.format(file1, file2))
sys.exit()
else:
print("正在比較文件{0} 和 {1}".format(file1, file2))
text1_lines = read_file(file1)
text2_lines = read_file(file2)
diff = difflib.HtmlDiff() # 創建HtmlDiff 對象
result = diff.make_file(text1_lines, text2_lines) # 通過make_file 方法輸出 html 格式的對比結果
# 將結果寫入到result_comparation.html文件中
try:
with open(‘result_comparation.html’, ‘w’) as result_file:
result_file.write(result)
print("0============> Successfully Finishedn")
except IOError as error:
print(‘寫入html文件錯誤:{0}’.format(error))

if name == "main":
# To define two arguments should be passed in, and usage: -f1 fname1 -f2 fname2
my_parser = argparse.ArgumentParser(description="傳入兩個文件參數")
my_parser.add_argument(‘-f1′, action=’store’, dest=’fname1′, required=True)
my_parser.add_argument(‘-f2′, action=’store’, dest=’fname2′, required=True)
# retrieve all input arguments
given_args = my_parser.parse_args()
file1 = given_args.fname1
file2 = given_args.fname2
compare_file(file1, file2)

』『』