python 批阅 考核 打分

gaohaikuo1

1. import PyPDF2  
   
2. from PyPDF2 import PdfWriter, PdfReader  
   
3. from PIL import Image, ImageDraw, ImageFont
   
4. import fitz  # PyMuPDF  
   
5. import pandas as pd
   
6. 
   
7. 
   
8. COL_NAME = 3
   
9. COL_VALUE = 5
   
10. 
    
11. 
    
12. def add_stamp(content_page,x1,y1,text, image_path=''):      
    
13.     image = Image.open(image_path)  
    
14.     pdf = Image.new('RGBA',(595,841),color=(255,255,255,255))
    
15.    
    
16.     draw = ImageDraw.Draw(pdf)
    
17.     font = ImageFont.truetype('simsun.ttc', 10)      
    
18.     draw.text((100,y1-30), text, font=font, fill=(255, 0, 0))  
    
19.     if len(image_path)>0:        
    
20.         pdf.paste(image, (x1,y1))  
    
21.     pdf.save('temp.pdf', 'PDF')   
    
22.     image_input = PdfReader('temp.pdf')  
    
23.     image_page = image_input.pages[0]
    
24.     image_page.merge_page(content_page)
    
25.     return image_page
    
26. 
    
27. 
    
28. 
    
29. 
    
30. #add_stamp('ah.pdf', 'sign.png')
    
31. 
    
32. 
    
33. def foundTextInPDF(doc,page_number_start,texts,texts_pre =[],mode='first'):
    
34.     #print(texts)
    
35.    # print(texts_pre)
    
36.     page_number = page_number_start
    
37.     flag_running = True
    
38.     flag_pre = False
    
39.     while page_number<len(doc) and flag_running:
    
40.         page = doc[page_number]
    
41. 
    
42.         #前提页已经找到
    
43.         flag_pre_found_all = True
    
44.         if len(texts_pre)>0:
    
45.             for text in texts_pre:
    
46.                
    
47.                 instances = page.search_for(text)
    
48.                 if len(instances) == 0:
    
49.                     flag_pre_found_all = False
    
50.         else:
    
51.             flag_pre = True  #如果没有前提条件直接判定满足
    
52.         if flag_pre_found_all:
    
53.             flag_pre = True
    
54.             print(texts_pre)
    
55.             print('上述前提页已经在'+str(page_number)+'页找到')
    
56.             
    
57.         
    
58. 
    
59.         if flag_pre:
    
60.             #搜索当前页是否为目标页
    
61.             flag_found_all = True
    
62.             for text in texts:
    
63.                 instances = page.search_for(text)
    
64.                 if len(instances) == 0:
    
65.                     flag_found_all = False
    
66. 
    
67.             if  flag_found_all:
    
68.                 #记录目标物位置
    
69. 
    
70.                 page_number_target = page_number
    
71.                 text_instances = page.search_for(texts[-1])  # 你可以替换为你要搜索的文本  
    
72.                 for inst in text_instances:  
    
73.                     x1, y1, x2, y2 = inst#.rect
    
74.                     x1 = int(x1)
    
75.                     y1 = int(y1)
    
76.                     x2 = int(x2)
    
77.                     y2 = int(y2)
    
78.                     if mode =='first':
    
79.                         flag_running = False
    
80.         page_number = page_number+1
    
81.         
    
82.     if 'page_number_target' not in locals():#没有找到目标赋值-1
    
83.         page_number_target = -1
    
84.         x2 = 10000
    
85.         y2 = 10000   
    
86.     return page_number_target,x2,y2
    
87. #1打开excel
    
88. #定位到第一个姓名
    
89. def get_record(pdf_path = '' , excel_name ='aaa.xlsx', sheet_name='Sheet1',
    
90.                 COL_NAME=3 ,ROW_NAME_RANGE = range(1,33),COL_VALUES=3,text_pre_later = '' ):
    
91.     df = pd.read_excel(excel_name, sheet_name)
    
92.     print(df)
    
93.     doc = fitz.open(pdf_path)
    
94.     record = {}
    
95.     page_number_target = 0;
    
96.     for row in ROW_NAME_RANGE:
    
97.         #print(row)
    
98.         name = df.iloc[row,COL_NAME]  
    
99.         #2遍历PDF页码，尝试找定位姓名
    
100.         #print(name)
     
101.         page_number_target,x,y = foundTextInPDF(doc,page_number_target,
     
102.                                                 ['评分'],
     
103.                                                 [name,text_pre_later])
     
104. 
     
105.         if x != 10000:
     
106.             values = '';
     
107.             for col in COL_VALUES:
     
108.                 values+= str(df.iloc[0, col])+':'+str(df.iloc[row, col])+';'
     
109.             record[page_number_target] = {'x':x,'y':y, 'value':values}
     
110.             print(name+' 已经批阅')
     
111.         else:
     
112.             print(name+' not found')
     
113.     doc.close()
     
114.     return record
     
115.    
     
116. pdf_path = '1.pdf'
     
117. pdf_path_out = '期末考核记录单_电子电路CAD_22电信本4_评阅版.pdf'
     
118. COL_VALUES_SET=[[3,4,5,6,8,12,15]]
     
119. texts_pre_set = ['记录单']
     
120. for i in range(1):#看多少轮，实验报告等材料需要设定
     
121.     print(texts_pre_set[i])
     
122.     record = get_record(pdf_path = pdf_path,
     
123.                         excel_name ='【成绩总表】电子电路CAD_22电信_成绩.xlsx',
     
124.                         sheet_name='期末考核_22电信本4',
     
125.                         COL_NAME = 2 ,
     
126.                         ROW_NAME_RANGE = range(2,29),
     
127.                         COL_VALUES = COL_VALUES_SET[i],
     
128.                         text_pre_later = texts_pre_set[i]
     
129.                         )
     
130.     print(record)
     
131.     pdf_input = PdfReader(pdf_path)
     
132.     pdf_output = PdfWriter()
     
133.     for page_number in range(len(pdf_input.pages)):
     
134.         content_page = pdf_input.pages[page_number]
     
135.         if page_number in record:
     
136.             print(page_number)
     
137.             content_page = add_stamp(content_page,
     
138.                                      record[page_number]['x'],
     
139.                                      record[page_number]['y']+40,
     
140.                                      record[page_number]['value'], 'sign1.png')
     
141.         pdf_output.add_page(content_page)
     
142. 
     
143.     with open(pdf_path_out, 'wb') as output_file:  
     
144.           pdf_output.write(output_file)
     
145. 
     

复制代码