Python - 开源库 ReportLab 库合并 CVS 和图像生成 PDF 文档 - ToB企服应用市场:ToB评测及商务社交产业平台

import csv
import os.path
from reportlab.lib import colors
from reportlab.lib.pagesizes import letter
from reportlab.lib.styles import getSampleStyleSheet
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Image, Spacer, Paragraph
from root_dir import DATA_DIR
from utils.project_utils import traverse_dir_files
def create_pdf(csv_files, image_files, headline, output_pdf):
"""
Merge the data from each CSV file into a single PDF file.
"""
# 创建 PDF 文档
doc = SimpleDocTemplate(output_pdf, pagesize=letter)
elements = []
line1 = headline
styles = getSampleStyleSheet()
style_normal = styles['Heading1']
elements.append(Paragraph(line1, style_normal))
# 读取并添加每个 CSV 文件的数据
for csv_file in csv_files:
data = []
with open(csv_file, 'r', newline='') as file:
reader = csv.reader(file)
headers = next(reader) # 读取标题行
data += [headers]
data += [row for row in reader] # 读取数据行
# 创建表格
table = Table(data)
table.setStyle(TableStyle([
('BACKGROUND', (0, 0), (-1, 0), colors.white),
('TEXTCOLOR', (0, 0), (-1, 0), colors.black),
('ALIGN', (0, 0), (-1, -1), 'CENTER'),
('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
('BOTTOMPADDING', (0, 0), (-1, 0), 12),
('BACKGROUND', (0, 1), (-1, -1), colors.beige),
]))
# 将表格添加到文档元素中
elements.append(table)
elements.append(Spacer(1, 20))
for image_file in image_files:
img = Image(image_file)
# img.hAlign = 'CENTER'
# img.vAlign = 'TOP'
ratio = img.imageWidth / 400
img.drawHeight = img.imageHeight / ratio
img.drawWidth = img.imageWidth / ratio
# img.scaleToFit(200, 200) # 调整图像大小以适应页面
elements.append(img)
# 构建文档
doc.build(elements)
def main():
input_csv_path = os.path.join(DATA_DIR, "abag-summary.csv")
input_img_path = os.path.join(DATA_DIR, "plots")
path_list = traverse_dir_files(input_img_path, ext="png")
output_path = os.path.join(DATA_DIR, "output.pdf")
create_pdf([input_csv_path], path_list, output_path)
if __name__ == '__main__':
main()

复制代码