# 經(jīng)典的浮點(diǎn)數(shù)問題
print(0.1 + 0.2) 
# 輸出: 0.30000000000000004

from decimal import Decimal, getcontext

# 設(shè)置精度（可選，視業(yè)務(wù)需求而定）
getcontext().prec = 4

# 模擬業(yè)務(wù)數(shù)據(jù)
value_a = Decimal('0.1')
value_b = Decimal('0.2')
result = value_a + value_b  # 結(jié)果精確為 Decimal('0.3')

# 在寫入 openpyxl 時(shí)，可以直接寫入 Decimal 對(duì)象
# openpyxl 會(huì)自動(dòng)將其轉(zhuǎn)換為浮點(diǎn)數(shù)，但為了保險(xiǎn)，建議轉(zhuǎn)為 float
ws.cell(row=1, column=1, value=float(result))

# ? 性能殺手：在循環(huán)中反復(fù)保存或頻繁操作單元格對(duì)象
for row in range(1, 10000):
    for col in range(1, 10):
        # 每次調(diào)用 ws.cell 都有一定開銷
        ws.cell(row=row, column=col, value=row * col)
wb.save('slow_file.xlsx')

# ? 推薦：按行追加數(shù)據(jù)
import time
from decimal import Decimal

data_source = [
    [Decimal('100.50'), Decimal('200.30')],
    [Decimal('101.00'), Decimal('202.00')],
    # ... 假設(shè)這里有成千上萬(wàn)行
]

for row_data in data_source:
    # 將 Decimal 轉(zhuǎn)換為 float 或直接寫入
    ws.append([float(x) for x in row_data])

# 假設(shè) data_generator 是一個(gè)生成器，源源不斷地產(chǎn)生數(shù)據(jù)
def data_generator():
    for i in range(1, 100000):
        yield [Decimal(i) * Decimal('1.05'), Decimal(i) * Decimal('0.95')]

# 流式寫入
for row_idx, row_data in enumerate(data_generator(), 1):
    # 這里的邏輯比較復(fù)雜，因?yàn)?openpyxl 的 append 是最快的
    # 如果必須使用 cell 賦值，請(qǐng)注意減少屬性訪問次數(shù)
    ws.cell(row=row_idx, column=1, value=float(row_data[0]))
    ws.cell(row=row_idx, column=2, value=float(row_data[1]))

# ? 慢
for row in range(1, ws.max_row + 1):
    val = ws.cell(row=row, column=1).value
    ws.cell(row=row, column=2, value=val * 2)

# ? 快 (先批量讀取到內(nèi)存，再批量計(jì)算，最后寫入)
# 但對(duì)于超大文件，這會(huì)撐爆內(nèi)存，所以折中方案是：
# 1. 將 ws.max_row 分段處理
# 2. 或者使用 openpyxl 的 read_only 模式讀取，計(jì)算，然后用 write_only 模式寫入新文件。

from openpyxl import load_workbook, Workbook

# 1. 以只讀模式加載源文件（極低內(nèi)存占用）
wb_read = load_workbook('big_data.xlsx', read_only=True)
ws_read = wb_read.active

# 2. 創(chuàng)建新工作簿（或以 write_only 模式保存）
wb_write = Workbook(write_only=True)
ws_write = wb_write.create_sheet()

# 3. 循環(huán)處理
# read_only 模式下，只能使用 ws.iter_rows() 遍歷
for row in ws_read.iter_rows(values_only=True):
    # row 是一個(gè)元組，包含該行的所有值
    # 這里進(jìn)行 Decimal 計(jì)算
    if row[0] is not None:
        val_a = Decimal(str(row[0])) # 轉(zhuǎn)換為 Decimal
        val_b = val_a * Decimal('1.1')
        
        # write_only 模式下，只能使用 append 寫入
        ws_write.append([float(val_b)])

# 4. 保存
wb_write.save('processed_big_data.xlsx')

import csv
from decimal import Decimal, ROUND_HALF_UP
from openpyxl import Workbook
from openpyxl.styles import Font, Alignment

# 模擬生成一個(gè)大 CSV 文件（實(shí)際中可能是讀取外部文件）
def generate_mock_csv(filename, rows=50000):
    with open(filename, 'w', newline='') as f:
        writer = csv.writer(f)
        writer.writerow(['ID', 'Amount', 'TaxRate'])
        for i in range(1, rows + 1):
            writer.writerow([i, f"{(i % 100) + 100}.50", "0.08"])

def process_financial_report(input_csv, output_xlsx):
    # 1. 初始化工作簿
    wb = Workbook(write_only=True)
    ws = wb.create_sheet()
    
    # 2. 寫入表頭
    headers = ['ID', '原始金額', '稅率', '稅額', '總金額']
    ws.append(headers)
    
    # 3. 設(shè)置 Decimal 上下文
    # ROUND_HALF_UP: 四舍五入，0.5 向上進(jìn)位
    Decimal('0.01').quantize(Decimal('0.00'), rounding=ROUND_HALF_UP)

    # 4. 讀取 CSV 并計(jì)算 (流式處理)
    with open(input_csv, 'r') as f:
        reader = csv.reader(f)
        next(reader)  # 跳過表頭
        
        batch_data = [] # 緩沖區(qū)，批量寫入可略微提升性能，但 write_only 模式下 append 已經(jīng)很快
        
        for row in reader:
            if not row: continue
            
            raw_id = int(row[0])
            raw_amount = Decimal(row[1])
            tax_rate = Decimal(row[2])
            
            # 計(jì)算邏輯 (Decimal 精度)
            tax_amount = (raw_amount * tax_rate).quantize(Decimal('0.00'), rounding=ROUND_HALF_UP)
            total_amount = raw_amount + tax_amount
            
            # 準(zhǔn)備寫入數(shù)據(jù) (轉(zhuǎn)換為 float 或保留 Decimal)
            # openpyxl 支持寫入 Decimal，但為了顯式控制，我們轉(zhuǎn)為 float
            # 注意：如果 Excel 僅用于展示，float 足夠；若需二次計(jì)算，建議轉(zhuǎn)為字符串或保留 Decimal
            
            # 這里我們轉(zhuǎn)為 float 展示
            row_data = [
                raw_id,
                float(raw_amount),
                float(tax_rate),
                float(tax_amount),
                float(total_amount)
            ]
            
            ws.append(row_data)

            # 簡(jiǎn)單的進(jìn)度提示（實(shí)際生產(chǎn)中可使用 tqdm）
            if raw_id % 10000 == 0:
                print(f"已處理 {raw_id} 行數(shù)據(jù)...")

    # 5. 保存文件
    print(f"正在保存文件: {output_xlsx}")
    wb.save(output_xlsx)
    print("完成！")

# 執(zhí)行演示
if __name__ == "__main__":
    input_csv = 'mock_transactions.csv'
    output_xlsx = 'financial_report.xlsx'
    
    # 生成測(cè)試數(shù)據(jù)
    print("正在生成模擬數(shù)據(jù)...")
    generate_mock_csv(input_csv, rows=50000)
    
    # 處理數(shù)據(jù)
    process_financial_report(input_csv, output_xlsx)