#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
데이터베이스 데이터 수정
"""

import sqlite3

def fix_database():
    conn = sqlite3.connect('database/kdrug.db')
    cursor = conn.cursor()

    print("=== 데이터베이스 수정 시작 ===")

    # 1. receipt_date 수정 (튜플 문자열을 일반 문자열로)
    print("\n1. receipt_date 수정...")
    cursor.execute("""
        UPDATE purchase_receipts
        SET receipt_date = '20260211'
        WHERE receipt_id = 6
    """)
    print(f"  receipt_date 수정 완료: {cursor.rowcount}건")

    # 2. total_amount 계산 및 수정
    print("\n2. total_amount 재계산...")
    cursor.execute("""
        UPDATE purchase_receipts
        SET total_amount = (
            SELECT SUM(line_total)
            FROM purchase_receipt_lines
            WHERE receipt_id = 6
        )
        WHERE receipt_id = 6
    """)
    print(f"  total_amount 수정 완료: {cursor.rowcount}건")

    # 변경 사항 저장
    conn.commit()

    # 수정 결과 확인
    print("\n=== 수정 후 데이터 확인 ===")
    cursor.execute("""
        SELECT receipt_date, total_amount
        FROM purchase_receipts
        WHERE receipt_id = 6
    """)
    result = cursor.fetchone()
    print(f"  receipt_date: {result[0]} (type: {type(result[0]).__name__})")
    print(f"  total_amount: {result[1]:,.0f}원 (type: {type(result[1]).__name__})")

    conn.close()
    print("\n✅ 데이터베이스 수정 완료!")

if __name__ == "__main__":
    fix_database()