from intelhex import IntelHex
import logging
import argparse


def handle_argument():
    """獲取腳本輸入?yún)?shù)

    Returns:
        _type_: Namespace(參數(shù))
    """
    parser = argparse.ArgumentParser(
        description="獲取MCU底軟版本號(hào)", usage="%(prog)s [options]"
    )
    parser.add_argument("--hex_file_path", type=str, required=True, help="hex文件路徑")
    return parser.parse_args()


# 日志初始化
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
    encoding="utf-8",
)
logger = logging.getLogger(name="GetMcuVersion")


def hex2dec(hex_number):
    """16進(jìn)制轉(zhuǎn)10進(jìn)制

    Args:
        hex_number (_type_): _description_

    Returns:
        _type_: _description_
    """
    return int(hex_number, 16)


def get_hexversion_from_hexfile(hexFilePath):
    """從hex文件中讀取十六進(jìn)制版本號(hào)

    Args:
        hexFilePath (_type_): _description_

    Returns:
        _type_: _description_
    """
    # 創(chuàng)建一個(gè)IntelHex對(duì)象
    hex_file = IntelHex()

    # 加載HEX文件
    try:
        hex_file.loadhex(hexFilePath)
        logger.info("HEX文件加載成功。")
    except FileNotFoundError:
        logger.info("未找到指定的HEX文件。")
    except Exception as e:
        logger.info(f"加載HEX文件時(shí)出錯(cuò)：{e}")

    # 讀取指定地址的字節(jié)
    address = 0x0077FC10
    if address in hex_file.addresses():
        data = hex_file.getsz(address)
        # print(f"addr的類型是：{type(data)}")
        logger.info(f"地址 0x{address:08X} 的數(shù)據(jù)是：{data.hex()}")
        logger.info(f"MCU底軟版本號(hào)對(duì)應(yīng)的十六進(jìn)制數(shù)據(jù)是：{data.hex()[0:4]}")
        return data.hex()[0:4]
    else:
        logger.info(f"地址 0x{address:08X} 未在HEX文件中定義。")


def get_decversion_from_hexversion(version_hex):
    """十六進(jìn)制版本號(hào)轉(zhuǎn)換成最終需要的版本號(hào)格式

    Args:
        version_hex (_type_): _description_

    Returns:
        _type_: _description_
    """
    version_dec = hex2dec(version_hex)
    mcu_version = (
        str(version_dec)[0:1]
        + "."
        + str(version_dec)[1:3]
        + "."
        + str(version_dec)[3:5]
    )
    logger.info(f"MCU底軟版本號(hào)是：{mcu_version}")
    return mcu_version


if __name__ == "__main__":
    # 獲取入?yún)?    args = handle_argument()
    version_hex = get_hexversion_from_hexfile(args.hex_file_path)
    mcu_version = get_decversion_from_hexversion(version_hex)
    print(mcu_version)

python intelhexDemo.py --hex_file_path="D:\hello.hex"