Das deutsche Python-Forum

Code: Alles auswählen

pythondatei.py <- Python Datei die du bekommen hast (Name vor dem ".py" habe ich mir gerade ausgedacht)
datei.xml <- Datei die du erstellt hast. 

Code: Alles auswählen

python3 pythondatei.py datei.xml

Code: Alles auswählen

python3 pythondatei.py

Code: Alles auswählen

#!/usr/bin/env python3
import sys
import xml.etree.ElementTree as ET

UMLAUTS_TABLE = {
    ord(character): replacement
    for character, replacement in [
        ("ä", "ae"),
        ("ö", "oe"),
        ("ü", "ue"),
        ("Ä", "Ae"),
        ("Ö", "Oe"),
        ("Ü", "Ue"),
        ("ß", "ss"),
    ]
}

CHANNEL_DEF_TEMPLATE = (
    "readingsBulkUpdate($hash, {id_name_string}, {get_value_code})"
    " if(ReadingsVal($name, {id_name_string}, '') ne {get_value_code});"
)


def remove_umlauts(string):
    """
    Removes umlauts from strings and replaces them with the letter+e convention.

    :param string: string to remove umlauts from
    :return: unumlauted string
    """
    return string.translate(UMLAUTS_TABLE)


def build_analog_get_value_code(channel, channel_id):
    unit = channel.get("unit")
    return f"$array[{channel_id}].' {unit}'"


def build_digital_get_value_code(channel, channel_id):
    bit = int(channel.get("bit"))
    return f"query_bit($array[{channel_id}], {bit})"


def process_channels(
    xml_document, channel_type, id_offset, build_get_value_code
):
    channel_id = 0
    for channel in xml_document.findall(f".//{channel_type}/CHANNEL"):
        name = remove_umlauts(channel.get("name")).replace(" ", "_")
        channel_id = int(channel.get("id")) + id_offset
        print(
            CHANNEL_DEF_TEMPLATE.format(
                id_name_string=f"'{channel_id:03d}_{name}'",
                get_value_code=build_get_value_code(channel, channel_id),
            )
        )
    return channel_id + 1


def main():
    filename = sys.argv[1] if len(sys.argv) > 1 else "hargassner_channels.xml"
    xml_document = ET.parse(filename)
    id_offset = 1
    for channel_type, build_get_value_code in [
        ("ANALOG", build_analog_get_value_code),
        ("DIGITAL", build_digital_get_value_code),
    ]:
        id_offset = process_channels(
            xml_document,
            channel_type,
            id_offset,
            build_get_value_code,
        )


if __name__ == "__main__":
    main()