# -*- coding: utf-8 -*-
"""把各合同首页渲成图片，便于人工/视觉识别项目名称与金额。"""
import io
import os
import sys

import fitz

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

BASE = r"C:\Users\skygu\OneDrive\Projects\AIDOP\项目\项目招投标\相关准备材料"
CON = os.path.join(BASE, "过往业绩合同")
OUT = os.path.join(os.path.dirname(os.path.abspath(__file__)),
                   "_bid_assets", "contract_scan")
os.makedirs(OUT, exist_ok=True)

for n in sorted(os.listdir(CON)):
    d = fitz.open(os.path.join(CON, n))
    for pi in (0, 1):
        if pi >= d.page_count:
            break
        pm = d[pi].get_pixmap(matrix=fitz.Matrix(150 / 72, 150 / 72))
        p = os.path.join(OUT, f"{n[:-4]}_p{pi + 1}.png")
        pm.save(p)
        print(f"{os.path.basename(p)}  {pm.width}×{pm.height}")
    d.close()