用openpyxl和docx写一个简单的数据比对程序，提高工作准确率

allearn_Python

工作原因需要经常核对word中一些数据是否准确。有时候工作量一大很容易出现遗漏。学习完 python操作excel后，发现可以利用openpyxl 和 docx 来帮助自己核对word中的内容，于是就写了一个很简单的小程序，基本实现了自动提取数据，自动分类比对，错误提示的功能。

思路是：
1. 从excle中读取需要的数据，编写了独立程序去实现。
2. 从word中读取需要对比的数据，编写独立程序。
3. 按类别分类对比。

希望得到大家指点 ，

另外我想对比的数据每一个都有 “型号”，“公司”，“证书号”等若干信息，我就把源数据和待对比数据分别封装成一个独立的类对象，再把这些类对象储存到一个列表中，然后去依次对比。但感觉这样不是很方便，请问有什么更好的办法吗？

效果GIF如下图：


代码如下，请指教： 程序代码.zip (91.66 KB, 下载次数: 2)

2020-4-2 10:14 上传

点击文件名下载附件

1. '''
   
2. 名称：安全部件信息比对程序 V1.0
   
3. 功能：提取Word中安全部件信息与Excel中的正确信息进行对比。
   
4. 最后更新：2020/3/28
   
5. 
   
6. '''
   
7. import openpyxl
   
8. from docx import Document
   
9. from ComparisonData import *
   
10. from OriginalData import *
    
11. 
    
12. #打开目标文件
    
13. wb = openpyxl.load_workbook('Excel file.xlsx')
    
14. ws = wb.active
    
15. document = Document('Word file.docx')
    
16. 
    
17. class DateError():
    
18.     '定义错误类，用于存放错误型号'
    
19.     def __init__(self):
    
20.         self.type = 'None'
    
21.         self.model = 'None'
    
22.         
    
23. 
    
24. def check(original_data_list, comparison_data_list, name):
    
25.     '检查两个数据类的每个属性值是否相同'
    
26.     error_list = []
    
27.     print('正在对比', name)
    
28.     for B in comparison_data_list:
    
29.         date_error = DateError()
    
30.         model_ok = False
    
31.         company_ok = False
    
32.         cert_ok = False
    
33.         nb_ok = False
    
34.         
    
35.         #依次比较
    
36.         for A in original_data_list:
    
37.             if B.model == A.model:
    
38.                 model_ok = True
    
39.                 #有同一型号，不同制造商的情况，单独判断
    
40.                 if B.model  in ['AQ1', 'XS1']:
    
41.                         for A in original_data_list:
    
42.                                 if B.model == A.model and B.company == A.company:
    
43.                                         model_ok = True
    
44.                                         company_ok = True
    
45.                                         if B.cert == A.cert:
    
46.                                                 cert_ok = True
    
47.                                                 if B.nb == A.nb:
    
48.                                                         nb_ok = True
    
49.                                                         print('%s | %s   -->  OK' % (B.model, B.company))                                
    
50.                                                         break
    
51.                                                 else:
    
52.                                                         print('%s该型号公告机构错误！' % B.model)
    
53.                                                         date_error.type = '公告机构错误'
    
54.                                                         date_error.model = B.model
    
55.                                                         error_list.append(date_error)
    
56.                                                         break
    
57.                                         else:
    
58.                                                 print('%s该型号证书号错误！' % B.model)
    
59.                                                 date_error.type = '证书号错误'
    
60.                                                 date_error.model = B.model
    
61.                                                 error_list.append(date_error)
    
62.                                                 break
    
63.                         if not company_ok:
    
64.                             print('%s该型号制造商错误！' % B.model)
    
65.                             print(B.company)
    
66.                             date_error.type = '制造商错误'
    
67.                             date_error.model = B.model
    
68.                             error_list.append(date_error)
    
69.                             break
    
70.                 if B.company == A.company:
    
71.                     company_ok = True
    
72.                     if B.cert == A.cert:
    
73.                         cert_ok = True
    
74.                         if B.nb == A.nb:
    
75.                             nb_ok = True
    
76.                             print('%s   -->  OK' % B.model)
    
77.                             break
    
78.                         else:
    
79.                             print('%s该型号公告机构错误！' % B.model)
    
80.                             print(A.nb, B.nb)
    
81.                             date_error.type = '公告机构错误'
    
82.                             date_error.model = B.model
    
83.                             error_list.append(date_error)
    
84.                             break
    
85.                     else:
    
86.                         print('%s该型号证书号错误！' % B.model)
    
87.                         date_error.type = '证书号错误'
    
88.                         date_error.model = B.model
    
89.                         error_list.append(date_error)
    
90.                         break
    
91.                 else:
    
92.                     print('%s该型号制造商错误！' % B.model)
    
93.                     print(A.company, B.company, sep='\n')
    
94.                     date_error.type = '制造商错误'
    
95.                     date_error.model = B.model
    
96.                     error_list.append(date_error)
    
97.                     break
    
98.         if not model_ok:
    
99.             #遍历完列表找不到该型号
    
100.             print('%s没有对应型号！' % B.model)
     
101.             date_error.type = '没有此型号'
     
102.             date_error.model = B.model
     
103.             error_list.append(date_error)
     
104.     return error_list
     
105.                            
     
106.             
     
107.         
     
108. 
     
109. 
     
110. def sort(original_data, comparison_data):
     
111.     '按照安全部件的名称分类'
     
112.     for name in comparison_data.__dict__:
     
113.         if name == 'SG':
     
114.             original_data_list = original_data.SG
     
115.             comparison_data_list = comparison_data.SG
     
116.         if name == 'OG':
     
117.             original_data_list = original_data.OG
     
118.             comparison_data_list = comparison_data.OG
     
119.         if name == 'BRAKE':
     
120.             original_data_list = original_data.BRAKE
     
121.             comparison_data_list = comparison_data.BRAKE
     
122.         if name == 'BF':
     
123.             original_data_list = original_data.BF
     
124.             comparison_data_list = comparison_data.BF
     
125.         if name == 'DL':
     
126.             original_data_list = original_data.DL
     
127.             comparison_data_list = comparison_data.DL
     
128.         if name == 'UCMP':
     
129.             original_data_list = original_data.UCMP
     
130.             comparison_data_list = comparison_data.UCMP
     
131. 
     
132.         #检查
     
133.         error = check(original_data_list, comparison_data_list, name)
     
134. 
     
135.         #集中打印错误
     
136.         if error:
     
137.             print('\n%s的错误有：' % name)
     
138.             for each in error:
     
139.                 print(each.model, each.type)
     
140.         else:
     
141.             print('\n%s无错误^_^' % name)
     
142. 
     
143.         input('\n按任意键继续：')
     
144. 
     
145. 
     
146. def main():
     
147.     comparison_data = Comparison()     
     
148.     original_data = Original()
     
149.    
     
150.     #从Word获取数据
     
151.     get_data_from_word(document, comparison_data)
     
152.     #从Excel获取数据        
     
153.     get_data_from_excel(original_data, ws)
     
154. 
     
155.     #开始对比
     
156.     sort(original_data, comparison_data)
     
157.            
     
158.         
     
159.             
     
160.    
     
161. if __name__ == '__main__':
     
162.     main()
     
163.    
     
164.         
     
165. 
     
166. 
     
167.    
     
168. 
     
169. 
     
170. 
     
171. 
     
172. 
     
173. 
     
174. 
     

复制代码

1. '从word获得待对比数据'
   
2. from docx import Document
   
3. 
   
4. #定义安全部件的类
   
5. class Component:
   
6.     def __init__(self):
   
7.         self.infos = []
   
8.         self.model = 'None'
   
9.         self.cert = 'None'
   
10.         self.company = 'None'
    
11.         self.nb = 'None'
    
12.         self.nb_name = 'None'
    
13.         self.type_ = 'None'
    
14. 
    
15.     def info(self):
    
16.         self.infos.append(self.model)
    
17.         self.infos.append(self.company)
    
18.         self.infos.append(self.cert)
    
19.         self.infos.append(self.nb)
    
20.         self.infos.append(self.nb_name)
    
21.         return self.infos
    
22. 
    
23.     def print(self):
    
24.         self.info(self)
    
25.         for each in self.infos:
    
26.             print(each, sep='   |   ')
    
27. 
    
28. #定义Comparison类
    
29. class Comparison:
    
30.     def __init__(self):
    
31.         self.SG = []
    
32.         self.BF = []
    
33.         self.OG = []
    
34.         self.BRAKE = []
    
35.         self.UCMP = []
    
36.         self.DL = []
    
37. 
    
38. #定义添加到列表的函数
    
39. def add2comparison_list(table, component, comparison_data):
    
40.     'table是哪一个安全部件的表格，component是相应的类对象'
    
41.     if table == sg_table:
    
42.         comparison_data.SG.append(component)
    
43.     if table == og_table:
    
44.         comparison_data.OG.append(component)
    
45.     if table == brake_table:
    
46.         comparison_data.BRAKE.append(component)
    
47.     if table == bf_table:
    
48.         comparison_data.BF.append(component)
    
49.     if table == dl_table:
    
50.         comparison_data.DL.append(component)
    
51.     if table == ucmp_table:
    
52.         comparison_data.UCMP.append(component)
    
53. 
    
54. #定义从表格获取数据的函数
    
55. def get_data_from_table(table, comparison_data):
    
56.     '从表格中获取数据，每一行的数据储存在实例化的component对象中，每个component对象再分类储存在comparison_data类中'
    
57.     for each in range(1, len(table.rows)): #第0行和第1行非正文，默认从第1行开始读取数据
    
58.         if 'Type' not in table.row_cells(each)[0].text:
    
59.             #每个部件实例化一个对象
    
60.             component = Component()
    
61.             #model
    
62.             component.model = table.row_cells(each)[0].text
    
63.             #Manufacturer Name
    
64.             component.company = table.row_cells(each)[1].text
    
65.             #Cert. No
    
66.             component.cert = table.row_cells(each)[2].text
    
67.             #Notified body
    
68.             component.nb = table.row_cells(each)[3].text
    
69. 
    
70.             #component.print()
    
71.             #component 以列表形式添加进comparison类中
    
72.             add2comparison_list(table, component, comparison_data)
    
73. 
    
74. def get_data_from_word(document, comparison_data):
    
75.     '定位各个安全部件所在表格位置\
    
76.     document: Word文件对象; comparison_data: 要对比的数据类'
    
77.     table_list = []
    
78.     global sg_table, og_table, brake_table, bf_table, dl_table, ucmp_table
    
79. 
    
80.     #判断word中每一个表格
    
81.     for each_table in document.tables:
    
82.         #以每个表格左上角内容是否匹配关键字的方式来确定是否属于目标表格，是的话命名并存入列表
    
83.         if 'gear' in each_table.cell(0, 0).text:
    
84.             sg_table = each_table
    
85.             table_list.append(each_table)
    
86.         if 'governor' in each_table.cell(0, 0).text:
    
87.             og_table = each_table
    
88.             table_list.append(each_table)
    
89.         if 'overspeed' in each_table.cell(0, 0).text:
    
90.             brake_table = each_table
    
91.             table_list.append(each_table)
    
92.         if 'Buffer' in each_table.cell(0, 0).text:
    
93.             bf_table = each_table
    
94.             table_list.append(each_table)
    
95.         if 'Locking' in each_table.cell(0, 0).text:
    
96.             dl_table = each_table
    
97.             table_list.append(each_table)
    
98.         if 'UCMP' in each_table.cell(0, 0).text:
    
99.             ucmp_table = each_table
    
100.             table_list.append(each_table)
     
101.         
     
102.     #从各个表格获取数据
     
103.     for each_table in table_list:
     
104.         get_data_from_table(each_table, comparison_data)
     
105. 
     
106. 
     
107. def main():
     
108.     '测试函数'
     
109.     document = Document('LF_A_C_18897_19-FINAL.docx')
     
110.     comparison_data = Comparison()
     
111.     get_data_from_word(document, comparison_data)
     
112. 
     
113.     print('共统计得到：')
     
114.     print('安全钳：%d个' % len(comparison_data.SG))
     
115.     print('限速器：%d个' % len(comparison_data.OG))
     
116.     print('制动器：%d个' % len(comparison_data.BRAKE))
     
117.     print('缓冲器：%d个' % len(comparison_data.BF))
     
118.     print('门锁：%d个' % len(comparison_data.DL))
     
119.     print('UCMP：%d个' % len(comparison_data.UCMP))
     
120. 
     
121. if __name__ == '__main__':
     
122.     main()
     
123. 
     
124.    
     
125. 
     
126. 
     
127. 
     

复制代码

1. ''' 从Excel中获取原始的数据信息，存储在Original类中以便于对比 '''
   
2. import openpyxl
   
3. from ComparisonData import Component
   
4. 
   
5. #定义Original类
   
6. class Original:
   
7.     def __init__(self):
   
8.         self.SG = []
   
9.         self.BF = []
   
10.         self.OG = []
    
11.         self.BRAKE = []
    
12.         self.UCMP = []
    
13.         self.DL = []
    
14. 
    
15. def get_data_from_excel(original_data, worksheet):
    
16.     #识别安全部件类型, 默认从39行开始
    
17.     #detect_component('B39', 'D200')
    
18. 
    
19.     #安全钳
    
20.     for each in worksheet['B42':'E69']:
    
21.         component = Component()
    
22.         
    
23.         component.model = each[0].value
    
24.         component.cert = each[1].value
    
25.         component.nb = each[2].value
    
26.         component.company = each[3].value
    
27. 
    
28.         original_data.SG.append(component)
    
29. 
    
30.     #限速器
    
31.     for each in worksheet['B73':'E84']:
    
32.         component = Component()
    
33.         
    
34.         component.model = each[0].value
    
35.         component.cert = each[1].value
    
36.         component.nb = each[2].value
    
37.         component.company = each[3].value
    
38. 
    
39.         original_data.OG.append(component)
    
40. 
    
41.     #制动器
    
42.     for each in worksheet['B88':'E108']:
    
43.         component = Component()
    
44.         
    
45.         component.model = each[0].value
    
46.         component.cert = each[1].value
    
47.         component.nb = each[2].value
    
48.         component.company = each[3].value
    
49. 
    
50.         original_data.BRAKE.append(component)
    
51. 
    
52.     #缓冲器
    
53.     for each in worksheet['B112':'E126']:
    
54.         component = Component()
    
55.         
    
56.         component.model = each[0].value
    
57.         component.cert = each[1].value
    
58.         component.nb = each[2].value
    
59.         component.company = each[3].value
    
60. 
    
61.         original_data.BF.append(component)
    
62. 
    
63.     #门锁
    
64.     for each in worksheet['B130':'E159']:
    
65.         component = Component()
    
66.         
    
67.         component.model = each[0].value
    
68.         component.cert = each[1].value
    
69.         component.nb = each[2].value
    
70.         component.company = each[3].value
    
71. 
    
72.         original_data.DL.append(component)
    
73. 
    
74.     #UCMP
    
75.     for each in worksheet['B163':'E188']:
    
76.         component = Component()
    
77.         
    
78.         component.model = each[0].value
    
79.         component.cert = each[1].value
    
80.         component.nb = each[2].value
    
81.         component.company = each[3].value
    
82. 
    
83.         original_data.UCMP.append(component)
    
84. 
    
85. 
    
86. def main():
    
87.     '测试函数'
    
88.     wb = openpyxl.load_workbook('A3-TECHNICAL SPECIFICATION SHEET-FUJI-MR.xlsx')
    
89.     ws = wb.active
    
90.     original_data = Original()
    
91.     get_data_from_excel(original_data, ws)
    
92. 
    
93.     print('共统计得到：')
    
94.     print('安全钳：%d个' % len(original_data.SG))
    
95.     print('限速器：%d个' % len(original_data.OG))
    
96.     print('制动器：%d个' % len(original_data.BRAKE))
    
97.     print('缓冲器：%d个' % len(original_data.BF))
    
98.     print('门锁：%d个' % len(original_data.DL))
    
99.     print('UCMP：%d个' % len(original_data.UCMP))
    
100. 
     
101.    
     
102. if __name__ == '__main__':
     
103.     main()
     
104.    
     
105.    
     
106.         
     
107.         
     
108.    
     
109.    
     
110.    
     
111.    
     

复制代码

allearn_Python

弱弱的问一句：如此简单的功能，用这么多行代码，正常吗

超哥超级爱学习

allearn_Python 发表于 2020-4-2 10:34
弱弱的问一句：如此简单的功能，用这么多行代码，正常吗

你好 我也遇到了一个和你类似的问题  可我实在不会openpyxl  方便时帮我下吧  太感谢了

allearn_Python

超哥超级爱学习 发表于 2020-8-24 21:25
你好 我也遇到了一个和你类似的问题  可我实在不会openpyxl  方便时帮我下吧  太感谢了

我对openpyxl也是初学者，也就是对excel进行读写操作等基本功能熟悉，帮人解决问题怕能力还不够

超哥超级爱学习

allearn_Python 发表于 2020-8-25 13:06
我对openpyxl也是初学者，也就是对excel进行读写操作等基本功能熟悉，帮人解决问题怕能力还不够{:10_266: ...