139 lines
5.5 KiB
Python
139 lines
5.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""보고서 시트 모형 → 엑셀(.xlsx) 쓰기 (report_xlsx.py).
|
|
|
|
시트 모형 = report_*.js 가 만든 { name, title, rows:[{A:값,…,_k:종류,_link:시트}] } (rows[i] = i+1 행).
|
|
근거: 지식DB STmate/old/20_분석 19 §3(목차·안내 시트) · 20 §3(SET_REP 서식) · 06 §3(출력별 반올림).
|
|
쓰는 것: 값 · 글꼴(제목·머리·묶음 줄 굵게) · 테두리 · 숫자 서식(정수 #,##0 · 소수 자리 그대로) ·
|
|
열 폭 · 인쇄 설정(A4 · 폭 맞춤 · 머리 행 반복 · 쪽 넘김) · 목차 바로가기 하이퍼링크.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from io import BytesIO
|
|
|
|
import openpyxl
|
|
from openpyxl.styles import Alignment, Border, Font, PatternFill, Side
|
|
from openpyxl.utils import column_index_from_string, get_column_letter
|
|
from openpyxl.worksheet.pagebreak import Break
|
|
|
|
COL_RE = re.compile(r"^[A-Z]{1,2}$")
|
|
BAD_TITLE = re.compile(r"[\[\]:*?/\\]")
|
|
THIN = Side(style="thin", color="808080")
|
|
BOX = Border(left=THIN, right=THIN, top=THIN, bottom=THIN)
|
|
HEAD_FILL = PatternFill("solid", start_color="E8EEF5", end_color="E8EEF5")
|
|
FONT = "맑은 고딕"
|
|
BOLD_KINDS = {"head", "T1", "T2", "T3", "T4", "T5", "sub", "total", "final", "hopyo", "parent"}
|
|
|
|
|
|
def _cells(row: dict) -> dict:
|
|
return {k: v for k, v in row.items() if COL_RE.match(k) and v not in (None, "")}
|
|
|
|
|
|
def _num_format(v: float) -> str:
|
|
if float(v).is_integer():
|
|
return "#,##0"
|
|
digits = len(repr(float(v)).split(".")[1].rstrip("0"))
|
|
return "#,##0." + "0" * min(max(digits, 1), 6)
|
|
|
|
|
|
def sheet_title(name: str, used: set[str]) -> str:
|
|
"""엑셀 시트 이름 — 금지 글자 걷고 31자 · 겹치면 번호."""
|
|
base = BAD_TITLE.sub("", str(name or "Sheet"))[:31] or "Sheet"
|
|
title, i = base, 2
|
|
while title in used:
|
|
title = f"{base[:28]}({i})"
|
|
i += 1
|
|
used.add(title)
|
|
return title
|
|
|
|
|
|
def write_sheet(ws, sheet: dict) -> None:
|
|
rows = sheet.get("rows") or []
|
|
width: dict[int, int] = {}
|
|
max_col = 1
|
|
head_rows = 0
|
|
for r_idx, row in enumerate(rows, start=1):
|
|
kind = row.get("_k", "")
|
|
if kind == "head":
|
|
head_rows = r_idx
|
|
if kind == "page" and r_idx > 1:
|
|
ws.row_breaks.append(Break(id=r_idx - 1))
|
|
for col, val in _cells(row).items():
|
|
c_idx = column_index_from_string(col)
|
|
max_col = max(max_col, c_idx)
|
|
cell = ws.cell(row=r_idx, column=c_idx, value=val)
|
|
bold = kind in BOLD_KINDS or kind == "title"
|
|
size = 16 if kind == "title" else 10
|
|
cell.font = Font(name=FONT, size=size, bold=bold)
|
|
if isinstance(val, (int, float)) and not isinstance(val, bool):
|
|
cell.number_format = _num_format(val)
|
|
cell.alignment = Alignment(horizontal="right", vertical="center")
|
|
else:
|
|
text = str(val)
|
|
cell.alignment = Alignment(
|
|
horizontal="center" if kind == "head" else "left",
|
|
vertical="center",
|
|
wrap_text="\n" in text,
|
|
)
|
|
if kind == "head":
|
|
cell.fill = HEAD_FILL
|
|
if kind not in ("title", "pname") and r_idx > 2:
|
|
cell.border = BOX
|
|
n = max(len(s) for s in str(val).split("\n")) if not isinstance(val, float) else 12
|
|
width[c_idx] = max(width.get(c_idx, 0), n)
|
|
link = row.get("_link")
|
|
if link:
|
|
target = next((c for c in ("D", "A") if c in row), None)
|
|
if target:
|
|
cell = ws[f"{target}{r_idx}"]
|
|
cell.hyperlink = f"#'{link}'!A1"
|
|
cell.font = Font(name=FONT, size=10, color="1D4ED8", underline="single")
|
|
if rows and rows[0].get("_k") == "title" and max_col > 1:
|
|
ws.merge_cells(start_row=1, start_column=1, end_row=1, end_column=max_col)
|
|
ws.cell(row=1, column=1).alignment = Alignment(horizontal="center", vertical="center")
|
|
for c_idx in range(1, max_col + 1):
|
|
w = width.get(c_idx, 4)
|
|
ws.column_dimensions[get_column_letter(c_idx)].width = min(max(w * 1.6 + 2, 6), 48)
|
|
ws.page_setup.paperSize = ws.PAPERSIZE_A4
|
|
ws.page_setup.orientation = "landscape" if max_col > 8 else "portrait"
|
|
ws.page_setup.fitToWidth = 1
|
|
ws.page_setup.fitToHeight = 0
|
|
ws.sheet_properties.pageSetUpPr.fitToPage = True
|
|
if head_rows:
|
|
ws.print_title_rows = f"1:{head_rows}"
|
|
ws.freeze_panes = ws.cell(row=head_rows + 1, column=1) if head_rows else None
|
|
|
|
|
|
def build_report_xlsx(sheets: list[dict]) -> bytes:
|
|
"""시트 모형 여럿 → 한 통 엑셀."""
|
|
wb = openpyxl.Workbook()
|
|
wb.remove(wb.active)
|
|
used: set[str] = set()
|
|
for sheet in sheets:
|
|
ws = wb.create_sheet(title=sheet_title(sheet.get("name", ""), used))
|
|
write_sheet(ws, sheet)
|
|
if not wb.worksheets:
|
|
wb.create_sheet("빈 보고서")
|
|
buf = BytesIO()
|
|
wb.save(buf)
|
|
return buf.getvalue()
|
|
|
|
|
|
def build_csv(sheet: dict) -> str:
|
|
"""시트 하나 → CSV(쉼표 · 큰따옴표 감쌈 · 엑셀 한글용 BOM 은 호출 쪽)."""
|
|
rows = sheet.get("rows") or []
|
|
max_col = max(
|
|
(column_index_from_string(k) for r in rows for k in _cells(r)),
|
|
default=1,
|
|
)
|
|
out = []
|
|
for row in rows:
|
|
vals = []
|
|
for c_idx in range(1, max_col + 1):
|
|
v = row.get(get_column_letter(c_idx))
|
|
s = "" if v is None else str(v)
|
|
vals.append('"' + s.replace('"', '""') + '"' if re.search(r'[",\n]', s) else s)
|
|
out.append(",".join(vals))
|
|
return "\r\n".join(out) + "\r\n"
|