import calendar
from openpyxl import Workbook
from openpyxl.styles import Alignment, Font
from openpyxl.utils import get_column_letter

def generate_aligned_calendar(target_year):
    """
    生成標題與日期嚴格對齊的橫向年度日歷
    核心修復：每個月標題合并的列數(shù) = 該月實際天數(shù)，無空白單元格
    """
    wb = Workbook()
    ws = wb.active
    ws.title = f"{target_year}年日歷"

    # 定義樣式
    title_font = Font(bold=True, size=12)
    title_align = Alignment(horizontal='center', vertical='center')
    date_align = Alignment(horizontal='right', vertical='center')

    current_col = 1  # 記錄當前起始列，動態(tài)累加

    for month in range(1, 13):
        # 獲取當前月份的實際天數(shù)（關鍵：用實際天數(shù)代替固定31）
        days_in_month = calendar.monthrange(target_year, month)[1]
        end_col = current_col + days_in_month - 1

        # 1. 寫入并合并月份標題（合并列數(shù) = 該月天數(shù)）
        month_title = f"{month}月"
        ws.merge_cells(start_row=1, start_column=current_col,
                       end_row=1, end_column=end_col)
        title_cell = ws.cell(row=1, column=current_col, value=month_title)
        title_cell.font = title_font
        title_cell.alignment = title_align

        # 2. 寫入日期（從當前列開始，連續(xù)寫入1-月末）
        for day in range(1, days_in_month + 1):
            date_cell = ws.cell(row=2, column=current_col + day - 1, value=day)
            date_cell.alignment = date_align

        # 更新下一個月的起始列
        current_col = end_col + 1

    # 統(tǒng)一調整列寬
    for col in range(1, current_col):
        ws.column_dimensions[get_column_letter(col)].width = 2.5

    # 保存文件
    file_name = f"{target_year}年對齊版日歷.xlsx"
    wb.save(file_name)
    print(f"? {target_year}年日歷生成成功！文件：{file_name}")
    print(f"?? 修復效果：標題與日期嚴格對齊，無空白單元格")

# 調用示例：生成2026年日歷
if __name__ == "__main__":
    generate_aligned_calendar(2026)