136 lines
5.3 KiB
Python
136 lines
5.3 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
|
||
from reportlab.platypus import TableStyle
|
||
from reportlab.lib import colors
|
||
|
||
import pandas as pd
|
||
import numpy as np
|
||
|
||
from Report.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')))
|
||
|
||
# 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)
|
||
|
||
# para
|
||
para_style_single = PS(name="para_style_single", fontName="SimHei", fontSize=8, leading=18, alignment=TA_LEFT, spaceBefore=6)
|
||
para_style_normal = PS(name="para_style_normal", fontName="SimHei", fontSize=8, leading=18, alignment=TA_LEFT, spaceBefore=6, firstLineIndent=16)
|
||
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_):
|
||
total_width = 410
|
||
df = pd.DataFrame()
|
||
|
||
for list__ in list_:
|
||
df = df.append(dict(zip(range(len(list__)), list__)), ignore_index=True)
|
||
|
||
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_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
|