# -*- coding: utf-8 -*-
import win32com.client
import csv
import os

# 配置區(qū)（按需修改）=====================================
CSV_PATH = r'E:\psd\jobs.csv'  # 簡歷數(shù)據(jù)路徑
PSD_TEMPLATE = r'E:\psd\zhaopin.psd'  # PSD模板路徑
OUTPUT_FOLDER = r'E:\psd\output'  # 成品輸出目錄
TEXT_LAYER_MAP = {  # 圖層名稱與CSV列名對應關(guān)系
    'company': 'company',
    'position': 'position',
    'salary': 'salary'
}


# =====================================================

def main():
    # 創(chuàng)建Photoshop實例
    ps = win32com.client.Dispatch("Photoshop.Application")
    if ps:
        print("成功創(chuàng)建 Photoshop 實例")
    else:
        print("創(chuàng)建 Photoshop 實例失敗")

    # 讀取CSV數(shù)據(jù)
    with open(CSV_PATH, 'r', encoding='utf-8') as f:
        reader = csv.DictReader(f)
        data = [row for row in reader]

    # 創(chuàng)建輸出目錄
    os.makedirs(OUTPUT_FOLDER, exist_ok=True)

    # 處理每條數(shù)據(jù)
    for index, item in enumerate(data, 1):
        # 復制模板文件
        output_psd = os.path.join(OUTPUT_FOLDER, f'招聘_{index}.psd')
        open(PSD_TEMPLATE, 'r')  # 檢查模板是否存在
        doc = ps.Open(PSD_TEMPLATE)

        try:
            # 遍歷所有圖層
            for layer in doc.Layers:
                # 僅處理文本圖層
                
                if layer.Kind == 2:  # 2表示文本圖層
                    layer_name = layer.Name
                    if layer_name in TEXT_LAYER_MAP:
                        csv_column = TEXT_LAYER_MAP[layer_name]
                        new_text = str(item.get(csv_column, ''))

                        # 中文換行處理（每10個字符換行）
                        if len(new_text) > 10:
                            new_text = '\n'.join([new_text[i:i + 10] for i in range(0, len(new_text), 10)])

                        layer.TextItem.Contents = new_text

            # 導出為PNG
            png_path = output_psd.replace('.psd', '.png')
            export_options = win32com.client.Dispatch('Photoshop.ExportOptionsSaveForWeb')
            export_options.Format = 13  # PNG格式
            export_options.PNG8 = False  # 使用24位PNG
            doc.Export(ExportIn=png_path, ExportAs=2, Options=export_options)

            # 保存PSD
            doc.SaveAs(output_psd)
            print(f'已生成：{output_psd}')

        finally:
            doc.Close(2)  # 關(guān)閉文檔不保存修改（因為已另存）


if __name__ == '__main__':
    main()