# python生成pdf报告

from pyecharts.charts import Bar
from pyecharts.render import make_snapshot

# 使用 snapshot-selenium 渲染图片
from snapshot_selenium import snapshot

bar = (
    Bar()
    .add_xaxis(["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"])
    .add_yaxis("商家A", [5, 20, 36, 10, 75, 90])
)
make_snapshot(snapshot, bar.render(), "bar.png")

# index.py
import json
import io
import sys
from utils import timer, get_chrome_driver
from generate import Generator
import os

sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')


@timer
def generate_report(driver, data):
    Generator().generatePdf(data, driver)


if __name__ == "__main__":
    chrome_driver = get_chrome_driver()

    # analysis
    # f = open('data/analysis.json', encoding='utf-8')
    # peertest
    # f = open('data/peertest.json', encoding='utf-8')
    # dashboard
    f = open('data/dashboard.json', encoding='utf-8')

    d = json.load(f)['data']
    # print(chrome_driver.session_id)
    generate_report(chrome_driver, d)
    chrome_driver.close()
    chrome_driver.quit()

# snapshot.py
import base64
import codecs
import logging
import os
from io import BytesIO
from PIL import Image

logger = logging.getLogger(__name__)

PNG_FORMAT = "png"
JPG_FORMAT = "jpeg"
GIF_FORMAT = "gif"
PDF_FORMAT = "pdf"
SVG_FORMAT = "svg"
EPS_FORMAT = "eps"
B64_FORMAT = "base64"


def make_snapshot(
        engine,
        file_name: str,
        output_name: str,
        delay: float = 2,
        pixel_ratio: int = 2,
        is_remove_html: bool = False,
        **kwargs,
):
    logger.info("Generating file ...")
    file_type = output_name.split(".")[-1]

    content = engine.make_snapshot(
        html_path=file_name,
        file_type=file_type,
        delay=delay,
        pixel_ratio=pixel_ratio,
        **kwargs,
    )

    if file_type in [SVG_FORMAT, B64_FORMAT]:
        save_as_text(content, output_name)
    else:
        # pdf, gif, png, jpeg
        content_array = content.split(",")
        if len(content_array) != 2:
            raise OSError(content_array)

        image_data = decode_base64(content_array[1])

        if file_type in [PDF_FORMAT, GIF_FORMAT, EPS_FORMAT]:
            save_as(image_data, output_name, file_type)
        elif file_type in [PNG_FORMAT, JPG_FORMAT]:
            save_as_png(image_data, output_name)
        else:
            raise TypeError("Not supported file type '%s'".format(file_type))

    if "/" not in output_name:
        output_name = os.path.join(os.getcwd(), output_name)

    if is_remove_html and not file_name.startswith("http"):
        os.unlink(file_name)
    logger.info("File saved in %s" % output_name)


def make_snapshot_in_memory(
        engine,
        file_name: str,
        delay: float = 2,
        pixel_ratio: int = 2,
        is_remove_html: bool = False,
        **kwargs,
):
    logger.info("Generating file ...")
    file_type = "png"

    content = engine.make_snapshot(
        html_path=file_name,
        file_type=file_type,
        delay=delay,
        pixel_ratio=pixel_ratio,
        **kwargs,
    )

    content_array = content.split(",")
    if len(content_array) != 2:
        raise OSError(content_array)

    image_data = decode_base64(content_array[1])

    if is_remove_html and not file_name.startswith("http"):
        os.unlink(file_name)

    m = BytesIO(image_data)
    m.seek(0)
    return m


def decode_base64(data: str) -> bytes:
    """Decode base64, padding being optional.

    :param data: Base64 data as an ASCII byte string
    :returns: The decoded byte string.
    """
    missing_padding = len(data) % 4
    if missing_padding != 0:
        data += "=" * (4 - missing_padding)
    return base64.decodebytes(data.encode("utf-8"))


def save_as_png(image_data: bytes, output_name: str):
    with open(output_name, "wb") as f:
        f.write(image_data)


def save_as_text(image_data: str, output_name: str):
    with codecs.open(output_name, "w", encoding="utf-8") as f:
        f.write(image_data)


def save_as(image_data: bytes, output_name: str, file_type: str):
    try:
        from PIL import Image

        m = Image.open(BytesIO(image_data))
        m.load()
        color = (255, 255, 255)
        b = Image.new("RGB", m.size, color)
        b.paste(m, mask=m.split()[3])
        b.save(output_name, file_type, quality=100)
    except ModuleNotFoundError:
        raise Exception("Please install PIL for {} image type".format(file_type))

# generate.py
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.pagesizes import A4, landscape, portrait
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image, Table, TableStyle, Frame, ListFlowable, ListItem, PageBreak
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.lib.enums import TA_JUSTIFY, TA_CENTER
from reportlab.lib import colors
from reportlab.lib.colors import CMYKColor
from reportlab.lib.units import inch, mm
import shutil
import os
from pages import summary, detail, miss, testInfo, peertest, overall, dashboard
from utils import timer

# 设置中文字体
pdfmetrics.registerFont(TTFont('SimSun', './Fonts/SimSun.ttf'))

# 设置样式
styles = getSampleStyleSheet()

# header
styles.add(ParagraphStyle(name='header', leftIndent=8, rightIndent=8, 
leading=18, alignment=TA_JUSTIFY, fontName="SimSun", fontWeight="bold", lineHeight=1.5, fontSize=12, textColor='#003153', bulletFontSize=12, bulletIndent=8, bulletAnchor='start', bulletFontName='Symbol'))
# 大写标题
styles.add(ParagraphStyle(name='bigTitle', leftIndent=8, rightIndent=8, 
leading=18, alignment=TA_JUSTIFY, fontName="SimSun", fontWeight="bold", lineHeight=1.5, fontSize=20, textColor='#003153', bulletFontSize=12, bulletIndent=8, bulletAnchor='start', bulletFontName='Symbol'))
# 正文
styles.add(ParagraphStyle(name='txt', leftIndent=8, rightIndent=8, alignment=TA_JUSTIFY, fontName="SimSun", fontSize=12, lineHeight=1.5, textColor='#003153', bulletFontSize=12, bulletIndent=8, bulletAnchor='start', bulletFontName='Symbol'))

class Generator:
  story = []
  totalData = None
  pages = []
  # a4纸的尺寸[210cm / 8.27 inch, 297cm / 11.69inch]
  PAGE_WIDTH = 297 * mm
  PAGE_HEIGHT = 210 * mm
  # 公用
  logo = './data/logo_big.png'
  company_logo = ''

  # 外部调用，根据数据渲染生成pdf文件
  @timer
  def generatePdf(self, pagesData, driver):
      # 初始化
      self.init()
      # 格式化
      self.totalData = pagesData
      self.pages = self.formatPages(pagesData['pages'])
      self.company_logo = pagesData['company']['logo']

      # 遍历页面，根据页面类型插入内容
      # for i in range(20):
      for page in self.pages:
          self.renderPage(page, driver)

      # 已存在文件先删除
      reportName = self.totalData['name'] + '.pdf'
      if os.path.exists(os.path.join(reportName)):
          os.remove(os.path.join(reportName))

      # 生成文件，自动按照横向A4大小分页
      doc = SimpleDocTemplate(reportName, pagesize=(self.PAGE_WIDTH, self.PAGE_HEIGHT), leftMargin=0.5*inch, rightMargin=0.5*inch)
      doc.multiBuild(self.story, onFirstPage=self.myFirstPage, onLaterPages=self.myLaterPages)

  # 初始化数据
  def init(self):
      self.story = []
      self.pages = []
      self.totalData = None
      # 清理img文件夹
      imgPath = os.path.join('img/')
      if not os.path.exists(imgPath):
        os.mkdir(imgPath)
      else:
        shutil.rmtree(imgPath)
        os.mkdir(imgPath)

  # 处理个别页面是个数组而非对象
  def formatPages(self, rawPages):
      list = []
      for page in rawPages:
          if type(page).__name__ == 'list':
              for item in page:
                  list.append(item)
          else:
              list.append(page)
      return list

  # 首页钩子，插入页眉页脚
  def myFirstPage(self, canvas, doc):
      canvas.saveState()
      self.renderHeader(canvas, doc)
      self.renderFooter(canvas, doc)
      canvas.restoreState()

  # 其他页面钩子，插入页眉页脚
  def myLaterPages(self, canvas, doc):
      canvas.saveState()
      self.renderHeader(canvas, doc)
      self.renderFooter(canvas, doc)
      canvas.restoreState()

  # 渲染页眉
  def renderHeader(self, canvas, doc):
      padding = 10
      # Logo
      logo_width = 127
      logo_height = 50
      canvas.drawImage(self.logo, padding, self.PAGE_HEIGHT - logo_height - padding, width=logo_width, height=logo_height, mask='auto')
      # Title
      canvas.setFont('Times-Bold', 16)
      canvas.drawCentredString(self.PAGE_WIDTH / 2, self.PAGE_HEIGHT - padding - 24, self.totalData['title'])
      if self.totalData['subtitle']:
        canvas.drawCentredString(self.PAGE_WIDTH / 2, self.PAGE_HEIGHT - padding - 24 - 24, self.totalData['subtitle'])
      # Company Logo
      company_logo_width = 50
      company_logo_height = 50
      canvas.drawImage(self.company_logo, self.PAGE_WIDTH - company_logo_width - padding, self.PAGE_HEIGHT - company_logo_height - padding, width=company_logo_width, height=company_logo_height, mask='auto')

  # 渲染页脚
  def renderFooter(self, canvas, doc):
      padding = 20
      canvas.setFont('Times-Roman', 9)
      canvas.drawCentredString(self.PAGE_WIDTH / 2, padding, "%s" % doc.page)

  # 渲染页面内容
  # @timer
  def renderPage(self, page, driver):
      pageType = page['type']

      # analysis
      if pageType == 'summary':
          summary.draw(self.story, self.totalData, page, styles, driver)
      elif pageType == 'detail':
          testInfo.draw(self.story, self.totalData, page, styles, driver)
          detail.draw(self.story, self.totalData, page, styles, driver)
      elif pageType == 'miss':
          testInfo.draw(self.story, self.totalData, page, styles, driver)
          miss.draw(self.story, self.totalData, page, styles, driver)

      # peertest
      elif pageType == 'peertest':
          peertest.draw(self.story, self.totalData, page, styles, driver)

      # dashboard
      elif pageType == 'overall':
        overall.draw(self.story, self.totalData, page, styles, driver)
      elif pageType == 'dashboard':
        dashboard.draw(self.story, self.totalData, page, styles, driver)

      # others
      else:
          pass

      # 分页
      self.story.append(PageBreak())

myBar.py
from reportlab.platypus import Image
from reportlab.lib.units import cm
import pyecharts
from pyecharts.charts import Bar
from pyecharts import options as opts
from pyecharts.globals import ThemeType
import uuid
from utils import timer, snap_shot

def draw(driver, x, y, data):
  # 默认900*500
    bar = Bar(init_opts=opts.InitOpts(theme=ThemeType.MACARONS,animation_opts=opts.AnimationOpts(animation=False)))
    x_list = []
    for item in data:
        x_list.append(item[x])
    bar.add_xaxis(x_list)

    for y_key in y:
        y_item = []
        for item in data:
            y_item.append(item[y_key])
        bar.add_yaxis(y_key, y_item)
    
    return bar

def drawImg(driver, x, y, data):
    bar = draw(driver, x, y, data)

    name = uuid.uuid4().hex
    img = snap_shot(bar, name, driver)
    img.drawHeight = 7*cm  # 设置读取后图片的高
    img.drawWidth = 13*cm  # 设置读取后图片的宽

    return img

# detail.py
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image, Table, TableStyle, Frame, ListFlowable, ListItem
from reportlab.lib import colors
from components import myTable, myBar
from decimal import *


def draw(story, totalData, page, styles, driver):
    # 图表标题
    t1 = Paragraph(page['title'], styles['header'])
    story.append(t1)
    t2 = Paragraph('By:' + page['by'], styles['header'])
    story.append(t2)

    story.append(Spacer(200, 20))

    # 柱形图
    if page['chart'] and page['chart']['show']:
      bar_x = page['chart']['x']
      bar_y = page['chart']['y']
      bar_data = page['data']
      story.append(myBar.drawImg(driver, bar_x, bar_y, bar_data))

    # 数据表格
    table_cols = page['th']
    table_data = []
    for row in page['data']:
        rowData = []
        for col in table_cols:
            rowData.append(row[col])
        table_data.append(rowData)
    story.append(myTable.draw(table_cols, table_data))