【Python办公自动化】使用reportlab制作pdf报告
原文作者:我輩李想
版權聲明:文章原創,轉載時請務必加上原文超鏈接、作者信息和本聲明。
Python使用folium制作地圖并生成png圖片
第一章 folium的方法和類的介紹(思維導圖)
第二章 使用folium制作地圖
第三章 folium實用功能進階
第三章 使用Html2Image生成png圖片
第四章 使用reportlab制作pdf報告
文章目錄
- Python使用folium制作地圖并生成png圖片
- 前言
- 一、reportlab是什么
- 二、
- 1.安裝和導入庫
- 2.將畫圖、畫表格、編輯文字抽象為類
- 3.pdf插入圖片
- 3.1 以文件路徑寫入pdf
- 3.2 以流文件寫入pdf
- 4.pdf分頁
- 5.以生成pdf流文件為例
- 總結
前言
提示:這里可以添加本文要記錄的大概內容:
本博客重點內容:reportlab生成流文件格式、reportlab分頁和圖片流文件寫入reportlab等。
我講一下我這個需求的來源,做的項目是一個地理空間查詢和使用的系統,通過在前端調用高德地圖api創建了一個查詢區域,獲取區域內的地理數據(數據庫)。具體的需求就是,將查詢區域和地理數據制作成一個覆蓋率分析報告,報告中的其他內容都已完成,但報告中需要展示高德地圖、查詢區域、地理數據的完整圖片這個功能卡了2個星期,主要原因是我對地理空間數據不熟悉,很多python相關庫也不清楚,在構建圖形的過程中走了很多彎路。
現在將整個實現過程梳理完成,希望對各位同道有幫助,跟其他文章和官網不同,本博客是以使用的優先級來講解這個庫。<我們靠所得來謀生,但靠給予來創造生活>
一、reportlab是什么
reportlab是Python的一個標準庫,可以畫圖、畫表格、編輯文字,最后可以輸出PDF格式。它的邏輯和編輯一個word文檔或者PPT很像。有兩種方法:
1)建立一個空白文檔,然后在上面寫文字、畫圖等;
2)建立一個空白list,以填充表格的形式插入各種文本框、圖片等,最后生成PDF文檔。
因為需要產生一份給用戶看的報告,里面需要插入圖片、表格等,所以采用的是第二種方法。
**由于我們這個功能是django網站項目上,我們并不想生成一個文件,第二種方法還可以生成io流文件,進而傳遞給前端生成pdf,這樣后端就不存在文件的存寫讀的操作了。**本篇文章在方法2的基礎上,加入了reportlab生成流文件格式、reportlab分頁和圖片流文件寫入reportlab等功能。
二、
1.安裝和導入庫
官網鏈接:https://pypi.org/project/reportlab/
官方文檔:https://docs.reportlab.com/reportlab/userguide/ch1_intro/
代碼如下(示例):
from reportlab.pdfbase import pdfmetrics # 注冊字體 from reportlab.pdfbase.ttfonts import TTFont # 字體類 from reportlab.platypus import Table, SimpleDocTemplate, Paragraph, Image # 報告內容相關類 from reportlab.lib.pagesizes import letter # 頁面的標志尺寸(8.5*inch, 11*inch) from reportlab.lib.styles import getSampleStyleSheet # 文本樣式 from reportlab.lib import colors # 顏色模塊 from reportlab.graphics.charts.barcharts import VerticalBarChart # 圖表類 from reportlab.graphics.charts.legends import Legend # 圖例類 from reportlab.graphics.shapes import Drawing # 繪圖工具 from reportlab.lib.units import cm # 單位:cm# 注冊字體(提前準備好字體文件, 如果同一個文件需要多種字體可以注冊多個) pdfmetrics.registerFont(TTFont('SimSun', 'SimSun.ttf'))2.將畫圖、畫表格、編輯文字抽象為類
抽象為類后,可實現多次調用,方便處理。這里我將頁眉和頁腳的方法也放在了抽象類中。在生成文件時,實現pdf報告的自動分頁。
class Graphs(object):def __init__(self):# 獲取所有樣式表self.style = getSampleStyleSheet()# 繪制標題def draw_1_title(self, title1: str):# 拿到標題樣式ct = self.style['Heading1']# 單獨設置樣式相關屬性ct.fontName = 'hei' # 字體名ct.fontSize = 20 # 字體大小ct.leading = 1.5 * ct.fontSize # 行間距ct.textColor = colors.black # 字體顏色ct.alignment = 1 # 居中return Paragraph(title1, ct)# 繪制小標題(黑色)def draw_2_title(self, title: str, content: str):# 拿到標題樣式ct = self.style['Normal']# 單獨設置樣式相關屬性ct.fontName = 'song' # 字體名ct.fontSize = 15 # 字體大小ct.leading = 1.5 * ct.fontSize # 行間距ct.textColor = colors.black # 字體顏色return Paragraph(("<font name='Dengb'>%s</font>" % title) + content, ct)# 繪制普通段落內容def draw_text(self, text: str):# 獲取普通樣式ct = self.style['Normal']ct.fontName = 'song'ct.fontSize = 12ct.leading = 1.5 * ct.fontSize # 行間距ct.wordWrap = 'CJK' # 設置自動換行ct.alignment = 0 # 左對齊ct.firstLineIndent = 32 # 第一行開頭空格ct.leading = 25return Paragraph(text, ct)# 繪制表格標題def draw_table_title(self, title: str):# 拿到標題樣式ct = self.style['Heading1']# 單獨設置樣式相關屬性ct.fontName = 'Dengb' # 字體名ct.fontSize = 14 # 字體大小ct.leading = 15 # 行間距ct.textColor = colors.black # 字體顏色ct.alignment = 1 # 居中ct.bold = True# 創建標題對應的段落,并且返回return Paragraph(title, ct)# 繪制表格@staticmethoddef draw_table(*args):# 列寬度col_width = [50, 300, 50, 50]style = TableStyle([('FONTNAME', (0, 0), (-1, -1), 'song'), # 全表字體('FONTNAME', (0, 0), (-1, 0), 'Dengb'), # 首行字體('FONTSIZE', (0, 0), (-1, 0), 12), # 第一行的字體大小('FONTSIZE', (0, 1), (-1, -1), 10), # 第二行到最后一行的字體大小('ALIGN', (0, 0), (-1, -1), 'CENTER'), # 第一行水平居中('VALIGN', (0, 0), (-1, -1), 'MIDDLE'), # 所有表格上下居中對齊('TEXTCOLOR', (0, 0), (-1, -1), colors.black), # 設置表格內文字顏色(不包含第一行)('GRID', (0, 0), (-1, -1), 0.5, colors.black), # 設置表格框線為grey色,線寬為0.5# ('SPAN', (0, 1), (0, 2)), # 合并第一列二三行# ('SPAN', (0, 3), (0, 4)), # 合并第一列三四行# ('SPAN', (0, 5), (0, 6)), # 合并第一列五六行# ('SPAN', (0, 7), (0, 8)), # 合并第一列五六行])table = Table(args, colWidths=col_width, style=style)return table# 創建圖表@staticmethoddef draw_bar(bar_data: list, ax: list, items: list):drawing = Drawing(500, 250)bc = VerticalBarChart()bc.x = 45 # 整個圖表的x坐標bc.y = 45 # 整個圖表的y坐標bc.height = 200 # 圖表的高度bc.width = 350 # 圖表的寬度bc.data = bar_databc.strokeColor = colors.black # 頂部和右邊軸線的顏色bc.valueAxis.valueMin = 5000 # 設置y坐標的最小值bc.valueAxis.valueMax = 26000 # 設置y坐標的最大值bc.valueAxis.valueStep = 2000 # 設置y坐標的步長bc.categoryAxis.labels.dx = 2bc.categoryAxis.labels.dy = -8bc.categoryAxis.labels.angle = 20bc.categoryAxis.categoryNames = ax# 圖示leg = Legend()leg.fontName = 'song'leg.alignment = 'right'leg.boxAnchor = 'ne'leg.x = 475 # 圖例的x坐標leg.y = 240leg.dxTextSpace = 10leg.columnMaximum = 3leg.colorNamePairs = itemsdrawing.add(leg)drawing.add(bc)return drawing# 繪制圖片@staticmethoddef draw_img(path):img = Image(path, width=500, height=500, kind='bound') # 讀取指定路徑下的圖片return img# 設置頁腳def footer(self, canvas, doc):"""設置頁腳--頁碼:param canvas:Canvas類型 pdf畫布:param doc:doc類型 整個pdf文件""" # 拿到標題樣式ct = self.style['Heading1']# 單獨設置樣式相關屬性ct.fontName = 'song' # 字體名ct.fontSize = 14 # 字體大小ct.leading = 15 # 行間距ct.textColor = colors.black # 字體顏色ct.alignment = 1 # 居中canvas.saveState() # 先保存當前的畫布狀態pageNumber = ("%s" % canvas.getPageNumber()) # 獲取當前的頁碼p = Paragraph(pageNumber, ct)p.wrap(1 * cm, 1 * cm) # 申請一塊1cm大小的空間,返回值是實際使用的空間p.drawOn(canvas, 520, 50) # 將頁碼放在指示坐標處canvas.restoreState()# 設置頁眉def header(self, canvas, doc):"""設置頁眉:param canvas:Canvas類型 pdf畫布:param doc:doc類型 整個pdf文件https://cxybb.com/article/liyadian/81253117"""canvas.saveState()im = Image("utils/map_png/image/logo.png", height=20, width=98)im.drawOn(canvas, 430, 780)# canvas.setStrokeColorRGB(0.8, 0.8, 0.8)# canvas.setFillColorRGB(0, 0, 0)# canvas.line(15, 28, doc.width - 15, 28)# canvas.setFont('regular', 8)# str = f"XXX銀行 | Page {doc.page}"# textLen = stringWidth(str, 'regular', 8)# canvas.drawCentredString(int((doc.width - textLen) / 2), 0.15 * inch, str)canvas.restoreState()3.pdf插入圖片
pdf中插入圖片會有2中方式,網絡上常用的方式都是文件路徑,本片文章的圖片是通過Html2Image生成的,無法保存成流文件格式,只能保存至本地文件。但是如果圖片是通過其他途徑生成的,可以直接以流文件的方式寫入pdf,這樣可以節省文件的創建、寫入、讀取和刪除等操作,節省內存和磁盤空間。
3.1 以文件路徑寫入pdf
# base 是當前文件所在路徑的上一級路徑,此種方法適用Windows和linux等多種系統,其他方法無法保證,多系統路徑是否一致。base = Path(__file__).resolve().parent# out_file就是文件名字(即Html2Image生成的圖片),img_path 是拼接后的文件路徑content.append(graphs.draw_table_title('圖1 影像覆蓋情況'))img_path = os.path.join(base, 'map_png', out_file)content.append(graphs.draw_img(img_path))content.append(Spacer(0, 10))3.2 以流文件寫入pdf
from pillow import Image as pilImagewith open('test.png',mode='rb') as f:# 讀取到圖片content.append(graphs.draw_img(f))content.append(Spacer(0, 10))4.pdf分頁
在第二部分我們已經寫入了pdf頁眉和頁腳的方法和參數,具體如何加載到pdf中可以參考下列方法。
doc = SimpleDocTemplate('report.pdf', pagesize=letter)# 聲明一塊Frame,存放頁碼frame_footer = Frame(doc.leftMargin, doc.bottomMargin, doc.width, doc.height, id='normal')# 設置頁面模板,在加載頁面時先運行herder函數,在加載完頁面后運行footer函數template = PageTemplate(id='test', frames=frame_footer, onPage=graphs.header, onPageEnd=graphs.footer)doc.addPageTemplates([template])doc.build(content)5.以生成pdf流文件為例
def pics_2_pdf(info, table, out_file):"""info:基本信息table:表格數據"""# 獲取Graphs類graphs = Graphs()# 創建內容對應的空列表content = list()# 1.添加標題content.append(graphs.draw_1_title('數據中心'))# 隔行的高度content.append(Spacer(0, 10))# 2.添加小標題content.append(graphs.draw_2_title("報告形成日期: ", str(datetime.date.today().strftime("%Y/%m/%d"))))content.append(Spacer(0, 10))# 3.添加圖片base = Path(__file__).resolve().parentcontent.append(graphs.draw_table_title('圖1 影像覆蓋情況'))img_path = os.path.join(base, 'map_png', out_file)content.append(graphs.draw_img(img_path))content.append(Spacer(0, 10))# 4.添加表格標題content.append(graphs.draw_table_title('表1 框選區域覆蓋數據清單'))# 5.添加表格數據# data = [# ['職位名稱', '平均薪資', '較上年增長率'],# ['數據分析師', '18.5K', '25%'],# ('高級數據分析師', '25.5K', '14%'),# ('資深數據分析師', '29.3K', '10%')# ]content.append(graphs.draw_table(*table))# 生成圖表# content.append(Graphs.draw_title(''))# content.append(Graphs.draw_little_title('熱門城市的就業情況'))# b_data = [(25400, 12900, 20100, 20300, 20300, 17400), (15800, 9700, 12982, 9283, 13900, 7623)]# ax_data = ['BeiJing', 'ChengDu', 'ShenZhen', 'ShangHai', 'HangZhou', 'NanJing']# leg_items = [(colors.red, '平均薪資'), (colors.green, '招聘量')]# content.append(Graphs.draw_bar(b_data, ax_data, leg_items))# 添加文字# content.append(Graphs.draw_text(# '眾所周知,大數據分析師崗位是香餑餑,近幾年數據分析熱席卷了整個互聯網行業,與數據分析的相關的崗位招聘、培訓數不勝數。很多人前赴后繼,想要參與到這波紅利當中。那么數據分析師就業前景到底怎么樣呢?'))# 生成pdf文件# 方式一:# doc = SimpleDocTemplate('report.pdf', pagesize=letter)# 方式二-存儲文件:# doc = BaseDocTemplate('report.pdf')# 方式二-流文件格式buffer = io.BytesIO() # 重點 起一個 iodoc = BaseDocTemplate(buffer)# 聲明一塊Frame,存放頁碼frame_footer = Frame(doc.leftMargin, doc.bottomMargin, doc.width, doc.height, id='normal')# 設置頁面模板,在加載頁面時先運行herder函數,在加載完頁面后運行footer函數template = PageTemplate(id='test', frames=frame_footer, onPage=graphs.header, onPageEnd=graphs.footer)doc.addPageTemplates([template])doc.build(content)return buffer總結
提示:這里對文章進行總結:
本博客是以使用的優先級來講解這個庫。<我們靠所得來謀生,但靠給予來創造生活>
總結
以上是生活随笔為你收集整理的【Python办公自动化】使用reportlab制作pdf报告的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 爬虫如何爬取猫眼电影TOP榜数据
- 下一篇: pyppeteer 模拟登录淘宝H5,获