204 lines
8.2 KiB
Python
204 lines
8.2 KiB
Python
from reportlab.lib.colors import HexColor
|
||
from reportlab.lib.enums import TA_CENTER, TA_LEFT, TA_RIGHT
|
||
from reportlab.pdfbase import pdfmetrics
|
||
from reportlab.pdfbase.ttfonts import TTFont
|
||
from reportlab.lib.styles import ParagraphStyle as PS, ParagraphStyle
|
||
from reportlab.lib.styles import getSampleStyleSheet
|
||
from reportlab.platypus import TableStyle, Paragraph
|
||
from reportlab.lib import colors
|
||
|
||
import pandas as pd
|
||
import numpy as np
|
||
|
||
from Report.scripts.path_tool import get_font_path
|
||
|
||
# rgb
|
||
darkGolden = '#C8A063'
|
||
lightGrey = '#F4F2EF'
|
||
darkGrey = '#A1A8AB'
|
||
Seashell4 = '#8B8682'
|
||
|
||
# font
|
||
pdfmetrics.registerFont(TTFont('pingbold', get_font_path(font='PingBold.ttf')))
|
||
pdfmetrics.registerFont(TTFont('SimHei', get_font_path(font='simhei.ttf')))
|
||
pdfmetrics.registerFont(TTFont('SIMSUN', get_font_path(font='SIMSUN.ttf')))
|
||
pdfmetrics.registerFont(TTFont('SourceHanSerifCN-Bold', get_font_path(font='SourceHanSerifCN-Bold.ttf')))
|
||
|
||
# toc
|
||
toc_style_1 = PS(name='TOCHeading1', fontName='pingbold', fontSize=16, leftIndent=20, firstLineIndent=-20,
|
||
spaceBefore=10, leading=16)
|
||
toc_style_2 = PS(name='TOCHeading2', fontName='pingbold', fontSize=12, leftIndent=40, firstLineIndent=-20,
|
||
spaceBefore=5, leading=12)
|
||
|
||
# table of content
|
||
table_content_style = PS(name="table_content_style", fontName="pingbold", fontSize=18, leading=40, alignment=TA_CENTER)
|
||
|
||
# cover
|
||
cover_space = PS(name='cover_space', fontSize=0, leading=75, alignment=TA_CENTER)
|
||
cover_company_style = PS(name="cover_company_style", fontName="pingbold", fontSize=30, leading=60, alignment=TA_CENTER)
|
||
cover_report_style = PS(name="cover_report_style", fontName="pingbold", fontSize=18, leading=400, alignment=TA_CENTER)
|
||
cover_fecr_style = PS(name="cover_fecr_style", fontName="pingbold", fontSize=18, leading=20, alignment=TA_CENTER)
|
||
cover_time_style = PS(name="cover_time_style", fontName="pingbold", fontSize=16, leading=40, alignment=TA_CENTER,
|
||
spaceBefore=20)
|
||
|
||
# chapter & section
|
||
chapter_style = PS(name="chapter_style", fontName="pingbold", fontSize=18, leading=40, alignment=TA_LEFT, spaceBefore=6)
|
||
section_style = PS(name="section_style", fontName="pingbold", fontSize=12, leading=30, alignment=TA_LEFT, spaceBefore=6,
|
||
textColor=HexColor(darkGolden))
|
||
|
||
# table
|
||
table_name = PS(name="table_name", fontName="pingbold", fontSize=8, leading=16, alignment=TA_CENTER, spaceBefore=2,
|
||
textColor=HexColor(darkGrey))
|
||
table_unit = PS(name="table_unit", fontName="SimHei", fontSize=6, leading=8, alignment=TA_RIGHT, spaceBefore=2)
|
||
table_mark = PS(name="table_mark", fontName="SimHei", fontSize=6, leading=16, alignment=TA_LEFT, spaceBefore=2)
|
||
table_style = getSampleStyleSheet()
|
||
table_style.add(
|
||
ParagraphStyle(fontName='SimHei', name='Song', leading=12, fontSize=8, spaceBefore=2, alignment=TA_CENTER))
|
||
table_style.add(
|
||
ParagraphStyle(fontName='SimHei', name='Long', leading=12, fontSize=8, spaceBefore=2, alignment=TA_LEFT))
|
||
table_style.add(
|
||
ParagraphStyle(fontName='SimHei', name='Longs', leading=9, fontSize=6, spaceBefore=2, alignment=TA_LEFT))
|
||
table_style.add(
|
||
ParagraphStyle(fontName='SimHei', name='Songs', leading=9, fontSize=6, spaceBefore=2, alignment=TA_CENTER))
|
||
table_style.add(
|
||
ParagraphStyle(fontName='SimHei', name='Song_small', leading=7, fontSize=5, spaceBefore=2, alignment=TA_CENTER))
|
||
# para
|
||
para_style_single = PS(name="para_style_single", fontName="SimHei", fontSize=8, leading=18, alignment=TA_LEFT,
|
||
spaceBefore=6)
|
||
para_style_special = PS(name="para_style_single", fontName="SimHei", fontSize=8, leading=18, alignment=TA_LEFT,
|
||
spaceBefore=6, textColor='red')
|
||
para_style_normal = PS(name="para_style_normal", fontName="SimHei", fontSize=8, leading=18, alignment=TA_LEFT,
|
||
spaceBefore=6, firstLineIndent=16)
|
||
para_style_esg = PS(name="para_style_esg", fontName="SimHei", fontSize=10, leading=18, alignment=TA_LEFT,
|
||
spaceBefore=6)
|
||
para_style_bold = PS(name="para_style_bold", fontName="pingbold", fontSize=10, leading=18, alignment=TA_LEFT,
|
||
spaceBefore=6)
|
||
para_bold_style = PS(name="para_style", fontName="pingbold", fontSize=9, leading=18, alignment=TA_LEFT, spaceBefore=6)
|
||
|
||
|
||
# table
|
||
def adjust_table_widths(list_):
|
||
"""
|
||
计算表格每列宽度
|
||
Parameters:
|
||
list_: list 表格数据
|
||
Returns:
|
||
result: list 表格宽度列表
|
||
"""
|
||
# 表格默认宽度
|
||
total_width = 410
|
||
df = pd.DataFrame()
|
||
|
||
for list__ in list_:
|
||
df = df.append(dict(zip(range(len(list__)), list__)), ignore_index=True)
|
||
|
||
# 将列表中的数据替换成数据的长度，并将0置为NaN
|
||
df = df.apply(lambda x: x.str.len())
|
||
df = df.replace(0, np.NaN)
|
||
|
||
# 计算每列数据长度平均值占总数的多少，并转为列表
|
||
width_rate = (df.mean() / df.mean().sum()).values.tolist()
|
||
# 列数
|
||
cols_num = len(width_rate)
|
||
|
||
# 判断首列是否小于默认宽度
|
||
if width_rate[0] < 0.33:
|
||
# 小于默认宽度，用当前数据重新计算各列宽度
|
||
widths = [width_rate[0] * total_width] + [((1 - width_rate[0]) / (cols_num - 1)) * total_width] * (cols_num - 1)
|
||
else:
|
||
# 大于默认宽度，则继续使用默认宽度继续各列宽度
|
||
widths = [0.33 * total_width] + [(0.67 / (cols_num - 1)) * total_width] * (cols_num - 1)
|
||
|
||
return widths
|
||
|
||
|
||
def adjust_table_data(list_):
|
||
"""
|
||
调整表格数据样式
|
||
Parameters:
|
||
list_: list 表格数据
|
||
Returns:
|
||
result: list 处理好的表格数据
|
||
"""
|
||
for item_ in list_:
|
||
if len(item_) > 6:
|
||
for index in range(len(item_)):
|
||
if item_[index] != '报告期' and index == 0:
|
||
item_[index] = Paragraph(item_[index], table_style['Longs'])
|
||
else:
|
||
if index == 5 and len(item_[index]) > 10:
|
||
item_[index] = Paragraph(item_[index], table_style['Song_small'])
|
||
else:
|
||
item_[index] = Paragraph(item_[index], table_style['Songs'])
|
||
else:
|
||
for index in range(len(item_)):
|
||
if item_[index] != '报告期' and index == 0:
|
||
item_[index] = Paragraph(item_[index], table_style['Long'])
|
||
else:
|
||
item_[index] = Paragraph(item_[index], table_style['Song'])
|
||
|
||
return list_
|
||
|
||
|
||
def adjust_table_style(list_):
|
||
return TableStyle([
|
||
# 边框
|
||
('INNERGRID', (0, 0), (-1, -1), 0.25, colors.grey),
|
||
('BOX', (0, 0), (-1, -1), 1, colors.black),
|
||
# 文字
|
||
('FONTNAME', (0, 0), (-1, -1), 'SimHei'),
|
||
('FONTSIZE', (0, 0), (-1, -1), 8),
|
||
|
||
('TEXTCOLOR', (0, 0), (-1, 0), colors.white),
|
||
# 排版
|
||
('ALIGN', (1, 0), (-1, -1), 'CENTER'),
|
||
# 背景
|
||
# ('BACKGROUND', (0, 0), (0, -1), HexColor(lightGrey)),
|
||
('BACKGROUND', (0, 0), (-1, 0), HexColor(darkGolden)),
|
||
])
|
||
|
||
|
||
def adjust_line_width(str_, font_size):
|
||
mark_length = 400
|
||
trans_length = mark_length / font_size - 1
|
||
|
||
if len(str_) < trans_length:
|
||
return str_
|
||
|
||
n = 0
|
||
s1 = ''
|
||
s2 = ''
|
||
for i in range(len(str_)):
|
||
|
||
if n == 0 and str_[i] in ['，', '。', '、', '；']:
|
||
continue
|
||
|
||
if u'\u4e00' <= str_[i] <= u'\u9fa5':
|
||
s2 += str_[i]
|
||
n = n + 1
|
||
elif str_[i] in ['！', '？', '｡', '＂', '＃', '＄', '％', '＆', '＇', '（', '）', '＊', '＋', '，', '－', '-', '／', '：', '；',
|
||
'＜', '＝', '＞', '＠', '［', '＼', '］', '＾', '＿', '｀', '｛', '｜', '｝', '～', '、', '〃', '》', '《', '「',
|
||
'」', '『', '』', '【', '】', '〔', '〕', '〖', '〗', '〝', '〞', '“', '”']:
|
||
s2 += str_[i]
|
||
n = n + 1
|
||
else:
|
||
s2 += str_[i]
|
||
n = n + 0.5
|
||
|
||
if (n > trans_length) and (i != len(str_) - 1):
|
||
if str_[i + 1] in ['，', '。', '、', '；']:
|
||
s2 += str_[i + 1]
|
||
s2 += '<br/>'
|
||
n = 0
|
||
s1 += s2
|
||
s2 = ''
|
||
else:
|
||
s2 += '<br/>'
|
||
n = 0
|
||
s1 += s2
|
||
s2 = ''
|
||
elif i == len(str_) - 1:
|
||
s1 += s2
|
||
|
||
return s1
|