python/python_aida64/__init__.py

import mmap
from xml.etree import ElementTree as ET
import sys


def _readRawData(length):
    with mmap.mmap(
            -1, length,  # anonymous file
            tagname='AIDA64_SensorValues',
            access=mmap.ACCESS_READ) as mm:
        return mm.read()


def _decode(b):
    for encoding in (sys.getdefaultencoding(), 'utf-8', 'gbk'):
        try:
            return b.decode(encoding=encoding)
        except UnicodeDecodeError:
            continue
    return b.decode()


def getXmlRawData() -> str:
    options = [100 * i for i in range(20, 100)]  # ranges in [2k, 10k]
    low = 0
    high = len(options) - 1
    while low < high:  # [low, high], stops at [low, low]
        mid = (low + high) // 2  # legit
        try:
            length = options[mid]
            raw = _readRawData(length)
            if raw[-1] == 0:  # legit ending
                decoded = _decode(raw.rstrip(b'\x00'))
                return '<root>{}</root>'.format(decoded)
            else:  # not long enough
                low = mid
                continue
        except PermissionError:  # illegal length (too long)
            high = mid
            continue


def getData() -> dict:
    data = {}
    tree = ET.fromstring(getXmlRawData())

    for item in tree:
        if item.tag not in data:
            data[item.tag] = []
        data[item.tag].append({
            key: item.find(key).text
            for key in ('id', 'label', 'value')
        })
    return data


__all__ = [
    'getXmlRawData',
    'getData'
]