python向pdf添加文字和图片

gaohaikuo1

1. import fitz  # PyMuPDF
   
2. import PyPDF2  
   
3. from PIL import Image
   
4. 
   
5. class pdfEditor():
   
6.     def __init__(self,pdf_path):
   
7.         self.pdf_document = fitz.open(pdf_path)
   
8.         
   
9.     def find_text_positions(self, search_text):
   
10.         positions = []
    
11. 
    
12.         # 遍历每一页
    
13.         for page_num in range(self.pdf_document.page_count):
    
14.             page = self.pdf_document.load_page(page_num)
    
15.             # 获取页面的文本块和位置信息
    
16.             text_blocks = page.get_text_blocks()
    
17.             
    
18.             # 解析文本块并查找特定文本的位置
    
19.             for block in text_blocks:
    
20.                 word = block[4]
    
21.                 if search_text in word:
    
22.                     # 获取单词在页面上的位置（矩形框）
    
23.                     rect = block[0:4]
    
24.                     # 将位置信息添加到列表中
    
25.                     positions.append((page_num, rect))
    
26.         return positions
    
27. 
    
28.     def insert_pic_in_pdf(self,image_path,positions = [(0,(0,0,0,0))],relative_pos = [-20,-20]):
    
29.         
    
30.         for position in positions:
    
31.             # 选择要插入图片的页面（例如第一页）
    
32.             page_number = position[0]
    
33.             page = self.pdf_document[page_number]
    
34. 
    
35. 
    
36.             # 定义图片插入的位置和大小
    
37.             
    
38.             rect = fitz.Rect(int(position[1][2])+relative_pos[0],int(position[1][1])+relative_pos[1], int(position[1][2])+200+relative_pos[0], int(position[1][1])+200+relative_pos[1])  # (x0, y0, x1, y1)
    
39. 
    
40.             # 插入图片
    
41.             page.insert_image(rect, filename=image_path)
    
42. 
    
43.         
    
44.     #插入文字,文字和位置一一对应，如果文字只有一个，就各处相同
    
45.     def add_text_at_positions(self,texts=['text'],font_size = 24,skip = 0,positions = [(0,(0,0,0,0))],relative_pos = [-20,-20]):
    
46.          # 打开现有的PDF文件
    
47.         while positions[0][0]<skip:
    
48.             positions = positions[1:];
    
49.         if len(texts) == 1:
    
50.             texts = [texts[0]] * len(positions)
    
51.             
    
52.         for text,position in zip(texts,positions):
    
53.             page = self.pdf_document.load_page(position[0])
    
54.             # 定义文本、字体、大小和位置
    
55.             font_path = "C:/Windows/Fonts/simsun.ttc"  # 替换为你的字体文件路径
    
56.             font_name = "SimSun"  # 使用内置的Helvetica字体
    
57.             font_size = 24
    
58.             rect = [int(position[1][2])+relative_pos[0],int(position[1][1])+relative_pos[1]]  # 文本矩形框的左下角坐标（x, y），单位：点（1点 = 1/72英寸）
    
59. 
    
60. 
    
61.             # 加载字体
    
62.             page.insert_font(fontname=font_name, fontfile=font_path)
    
63.             page.insert_text(rect, text, fontsize=font_size, fontname=font_name, color=(1, 0, 0))  # 重新插入文本框
    
64. 
    
65.             # 保存修改后的PDF文件
    
66.         
    
67.     def save_pdf(self,name_pdf_output):
    
68.         self.pdf_document.save(name_pdf_output,incremental=False);
    
69. 
    
70.     def add_text_in_pdf_where_text(self,texts=['2025'],font_size = 24,pdf_path = "ah.pdf",search_text = "年",skip = 0, relative_pos = [-20,-20]):
    
71.         positions = self.find_text_positions(search_text);
    
72.         self.add_text_at_positions(texts,font_size ,skip,positions ,relative_pos);
    
73.         
    
74. 
    
75.     def add_pic_in_pdf_where_text(self,stamp = 'sign(1).png',pdf_path = "ah.pdf",search_text = "指导教师签名",relative_pos = [-20,-100]):
    
76.         positions = self.find_text_positions(search_text)
    
77.         #self.add_stamp(pdf_path, stamp,positions,name_pdf_output)
    
78.         self.insert_pic_in_pdf(stamp,positions,relative_pos)
    
79.         
    
80. 
    
81. if __name__ == '__main__':
    
82.     pe = pdfEditor( "ah.pdf");
    
83.     #pe.add_stamp_in_pdf_where_text(stamp = 'sign(1).png',pdf_path = "ah.pdf",search_text = "指导教师签名",name_pdf_output='output1.pdf');
    
84.     #pe.add_text(pdf_path,texts=['text'],positions = [(0,(0,0,0,0))],relative_pos = [-20,-20],name_pdf_output='o.pdf'):
    
85.     pe.add_text_in_pdf_where_text();
    
86.     pe.add_pic_in_pdf_where_text();
    
87.     pe.save_pdf('output667.pdf')
    

复制代码

gaohaikuo1

1. from openpyxl import load_workbook
   
2. from pdfEditor import pdfEditor
   
3. 
   
4. 
   
5. 
   
6. # 加载 Excel 文件
   
7. file_path = "00名单_22光信本1 -实验成绩.xlsx"  # Excel 文件路径
   
8. wb = load_workbook(file_path)  # 加载工作簿
   
9. ws = wb['实验成绩']  # 获取当前活动的工作表
   
10. 
    
11. flag_1st = True
    
12. pe = pdfEditor();
    
13. 
    
14. root = r"D:\\0.山东航空学院\\课程：数字图像处理\\2023-2024-2 22光信\";
    
15. root_temp = root+r'datatemp\\'
    
16. for row in ws.iter_rows(values_only=True):  # 逐行读取数据
    
17.     print(row)
    
18.     if flag_1st:
    
19.         flag_1st = False;
    
20.         continue
    
21.     a_pdfname = root_temp +str(row[0])+"2023-2024-2《数字图像处理》上机考核材料（22光信本1) .pdf"
    
22.     b_pdfname = root_temp +str(row[0]+1)+"2023-2024-2《数字图像处理》上机考核材料（22光信本1) .pdf"
    
23.     pe.add_text_in_pdf_where_text(texts=['错误数'+str(row[4])+'操作分'+str(row[7])+'用时'+str(row[5])+'分钟，熟练分：'+str(row[6])+'总分'+str(row[8])],\
    
24.                                   font_size = 18,pdf_path =a_pdfname ,\
    
25.                                   search_text = row[1],skip =2,relative_pos = [-480,-60],\
    
26.                                   name_pdf_output= b_pdfname );

复制代码

gaohaikuo1

import fitz  # PyMuPDF

def find_text_positions(pdf_path, search_text):
    # 打开PDF文件
    doc = fitz.open(pdf_path)
    positions = []

    # 遍历每一页
    for page_num in range(doc.page_count):
        page = doc.load_page(page_num)
        # 获取页面的文本块和位置信息
        text_blocks = page.get_text_blocks()
        
        # 解析文本块并查找特定文本的位置
        for block in text_blocks:
            word = block[4]
            if search_text in word:
                # 获取单词在页面上的位置（矩形框）
                rect = block[0:4]
                # 将位置信息添加到列表中
                positions.append((page_num, rect))

    return positions

# 示例使用
pdf_path = "ah.pdf"
search_text = "指导教师签名"
positions = find_text_positions(pdf_path, search_text)
print(positions)