homeassistant-config/config_creation/main.py

import os
import argparse
from util import DeviceInfo, add_to_group, name_to_id
from ruamel.yaml import YAML
import knx_conf as knx

script_path = os.path.dirname(os.path.realpath(__file__))
yaml = YAML()
yaml.preserve_quotes = True


sensor_exclude_list = [
    'sensor.fritz_box_7490_bytes_received',
    'sensor.fritz_box_7490_bytes_sent',
    #'sensor.fritz_box_7490_kbyte_sec_received',
    #'sensor.fritz_box_7490_kbyte_sec_sent',
    'sensor.fritz_box_7490_packets_received',
    'sensor.fritz_box_7490_packets_sec_received',
    'sensor.fritz_box_7490_packets_sec_sent',
    'sensor.fritz_box_7490_packets_sent',

    'sensor.martin_handy_availability_sensor',
    'sensor.martin_handy_light_sensor',
    'sensor.martin_handy_step_counter_sensor',

    'sensor.date_time',
    'sensor.date_time_iso',
    'sensor.time',
    'sensor.time_date',
]


def add_knx_devices(devices, groups):
    imported_csv = knx.import_ets5_csv_file(os.path.join(script_path, 'knx_data/export_project1.csv'))
    imported_csv.update(knx.import_ets5_csv_file(os.path.join(script_path, 'knx_data/export_project2.csv')))

    lights = [
        # Dimmers
        DeviceInfo('Wohnzimmerlampe', 'Wohnzimmer Deckenlampe', 'living_area'),
        DeviceInfo('EsszimmerlampeWest', 'Esszimmer Deckenlampe West', 'living_area'),
        DeviceInfo('EsszimmerlampeMitte', 'Esszimmer Deckenlampe Mitte', 'living_area'),
        DeviceInfo("EsszimmerWandlampe", 'Esszimmer Schrankleuchte', 'living_area'),
        DeviceInfo("Küchenlampe", "Küche Deckenlampe", 'living_area'),
        DeviceInfo("AussenleuchteUntenSO", "Aussen Terassenlicht", 'outside'),
        DeviceInfo("Gang", "Gang Licht", 'hallway'),
        DeviceInfo("Bad", "Bad Licht", 'bathroom'),
        DeviceInfo("GangWindfang", "Gang Einganglicht", 'hallway'),
        DeviceInfo("LichtWaschküche", "Waschküche Licht", 'hallway'),
        # Normal lights
        DeviceInfo('AussenleuchteHaustüren', 'Haustür Licht', 'outside'),
        DeviceInfo('AussenleuchteObenNW', 'Haustür Licht NW', 'outside'),
        DeviceInfo('TreppenhausLicht', "Treppenhaus Licht", 'first_floor'),
        DeviceInfo('WCLicht', "WC Licht", 'other'),
        DeviceInfo('LampeVorratsraum', "Vorratsraum Licht", 'other'),
    ]

    shutters = [
        DeviceInfo('Wohnzimmer Fenster Rollo', 'Wohnzimmer Fenster Rollo', 'living_area'),
        DeviceInfo('Terassentür Rollo', 'Wohnzimmer Terrassentür Rollo', 'living_area'),
        DeviceInfo('Küchenfenster Rollo', 'Küche Fenster Rollo', 'living_area'),
        DeviceInfo('Esszimmerfenster Rollo', 'Esszimmer Fenster Rollo', 'living_area'),
    ]

    switches = [
        # Bells
        DeviceInfo("KlingelOben", "Klingel Oben", 'first_floor'),
        DeviceInfo("Klingel Innen", "Klingel Innentür", 'other'),
        DeviceInfo("Klingel Aussen", "Klingen Außentür", 'other'),
        # Bewegungsmelder LEDs
        DeviceInfo("BewegungsmelderMitte LED", "Bewegungsmelder Mitte LED", 'hallway'),
        DeviceInfo("BewegungsmelderWest LED", "Bewegungsmelder West LED", 'hallway'),
        DeviceInfo("BewegungsmelderOst LED", "Bewegungsmelder Ost LED", 'hallway'),
    ]
    scene_button_names = ['ObenLinks', 'ObenRechts', 'MitteLinks', 'MitteRechts', 'UntenLinks', 'UntenRechts']
    scene_button_names = [(i, e) for i, e in enumerate(scene_button_names)]
    switches += [DeviceInfo(f"SzeneEsszimmer{n}", f"Esszimmer Szene {i}") for i, n in scene_button_names]
    switches += [DeviceInfo(f"SzeneWohnzimmer{n}", f"Wohnzimmer Szene {i}") for i, n in scene_button_names]
    switches += [DeviceInfo(f"SzeneEingang{n}", f"Eingang Szene {i}") for i, n in scene_button_names[2:]]
    switches += [DeviceInfo(f"SzeneTerrassentuer{n}", f"Wohnzimmer Terrassentür Szene {i}")
                 for i, n in scene_button_names[2: 4]]

    power_plugs = [
        # Vorratsraum
        DeviceInfo("VorratsraumSteckdose1", "Vorratsraum Steckdose 1", 'other'),
        DeviceInfo("VorratsraumSteckdose2", "Vorratsraum Steckdose 2", 'other'),
        DeviceInfo("VorratsraumSteckdose3", "Gefrierschrank", 'other'),
        # Waschraum
        DeviceInfo("Trockner", "Trockner", 'other'),
        DeviceInfo("Waschmaschine", "Waschmaschine", 'other'),
        # Küche
        DeviceInfo("KücheSteckdose1", "Küche Steckdosen rechts"),
        DeviceInfo("KücheSteckdose2", "Dunstabzug"),
        DeviceInfo("Spülmaschine", "Spülmaschine", 'living_area'),
        DeviceInfo("Backofen", "Backofen", 'living_area'),
        DeviceInfo("HerdP1", "Herd Phase 1", 'living_area'),
        DeviceInfo("HerdP2", "Herd Phase 2", 'living_area'),
        DeviceInfo("HerdP3", "Herd Phase 3", 'living_area'),
        # Rest
        DeviceInfo("ArbeitszimmerSteckdose", "Arbeitszimmer Steckdose", 'office_martin'),
        DeviceInfo("WohnzimmerSteckdose1", "Wohnzimmer Steckdose 1", 'living_area'),
        DeviceInfo("WohnzimmerSteckdose2", "Wohnzimmer Steckdose 2", 'living_area'),
    ]

    motion_sensors = {
        'sensor': [
            # Brightness
            {'name': 'Gang Bewegungsmelder West Helligkeit',
             'state_address': imported_csv['BewegungsmelderWest Helligkeit'],
             'type': 'illuminance'},
            {'name': 'Gang Bewegungsmelder Mitte Helligkeit',
             'state_address': imported_csv['BewegungsmelderMitte Helligkeit'],
             'type': 'illuminance'},
            {'name': 'Gang Bewegungsmelder Ost Helligkeit',
             'state_address': imported_csv['BewegungsmelderOst Helligkeit'],
             'type': 'illuminance'},
            # Temperature
            {'name': 'Esszimmer Temperatur',
             'state_address': imported_csv['TemperaturEsszimmer'],
             'type': 'temperature'},
            {'name': 'Treppenhaus Oben Temperatur',
             'state_address': imported_csv['TreppenhausObenTemperatur'],
             'type': 'temperature'},
            {'name': 'Eingangsbereich Temperatur',
             'state_address': imported_csv['TemperaturGang'],
             'type': 'temperature'},
            {'name': 'Wohnzimmer Gang Temperatur',
             'state_address': imported_csv['TemperaturWohnzimmerGangTuer'],
             'type': 'temperature'},
            {'name': 'Wohnzimmer Terassentür Temperatur',
             'state_address': imported_csv['TemperaturWohnzimmerAussenTuer'],
             'type': 'temperature'},
        ],
        'binary_sensor': [
            {'name': 'Gang Bewegungsmelder Schlafzimmer',
             'state_address': imported_csv['BewegungsmelderWest Motion Links'],
             'device_class': 'motion'},
            {'name': 'Gang Bewegungsmelder Wohnzimmer',
             'state_address': imported_csv['BewegungsmelderWest Motion Rechts'],
             'device_class': 'motion'},
            {'name': 'Gang Bewegungsmelder Mitte',
             'state_address': imported_csv['BewegungsmelderMitte Motion'],
             'device_class': 'motion'},
            {'name': 'Gang Bewegungsmelder Ost',
             'state_address': imported_csv['BewegungsmelderOst Motion'],
             'device_class': 'motion'},
        ]
    }
    knx.extent(devices, motion_sensors)
    knx.extent(devices, knx.create_lights(lights, imported_csv))
    knx.extent(devices, knx.create_shutters(shutters, imported_csv))
    knx.extent(devices, knx.create_switches(switches, imported_csv))
    knx.extent(devices, knx.create_power_plug(power_plugs, imported_csv))

    for device_type, devices in [('light', lights), ('cover', shutters), ('switch', switches)]:
        for device in devices:
            add_to_group(groups, device.groups, device.display_name, device_type)


def add_fhem_devices(devices, groups):
    fhem_yaml_path = os.path.join(script_path, 'fhem.yaml')
    fhem_yaml = yaml.load(open(fhem_yaml_path))
    for device_type, device_list in fhem_yaml.items():
        if device_type not in devices:
            devices[device_type] = []
        for device in device_list:
            device['platform'] = 'fhem'
            if 'groups' in device:
                add_to_group(groups, device['groups'], device['name'], device_type)
                del device['groups']
            devices[device_type].append(device)


def add_light_groups(groups):
    light_groups = {
        f"light_{group_id}": {
            'name': content['name'] + " Lichter",
            'entities': [e for e in content['entities'] if e.startswith('light.')],
        }
        for group_id, content in groups.items() if 'name' in content
    }
    groups.update(light_groups)


def make_sensor_exclude_list(all_devices, name_fragments):
    exclude_list = []

    def handle_sensor(s):
        name = s['name'] if 'name' in s else s['friendly_name']
        if any(t in name for t in name_fragments):
            exclude_list.append(name_to_id(name, 'sensor'))

    for sensor_device in all_devices['sensor']:
        if 'platform' in sensor_device and sensor_device['platform'] == 'time_date':
            continue
        if 'platform' in sensor_device and sensor_device['platform'] == 'template':
            for nested_sensor_dev in sensor_device['sensors'].values():
                handle_sensor(nested_sensor_dev)
        elif 'name' in sensor_device:
            handle_sensor(sensor_device)
        else:
            raise NotImplementedError(f"Sensor {sensor_device}")
    return exclude_list


def logbook_config(all_devices):
    sensor_excludes = ['Helligkeit', 'Betriebsstunden', 'Verbrauch', 'Bewegungsmelder']

    exclude_list = []
    exclude_list.extend(sensor_exclude_list)
    exclude_list.extend(make_sensor_exclude_list(all_devices, sensor_excludes))
    return {
        'logbook': {
            'exclude': {
                'entities': exclude_list,
                'domains': ['group']
            }
        }
    }


def recorder_config(all_devices):
    recorder_excludes = ['Betriebsstunden', 'Verbrauch mA']
    exclude_list = ['sun.sun']
    exclude_list.extend(sensor_exclude_list)
    exclude_list.extend(make_sensor_exclude_list(all_devices, recorder_excludes))
    return {
        'recorder': {
            'exclude': {
                'entities': exclude_list,
            }
        }
    }


def create_config(target_directory, development=False):
    groups_yaml_path = os.path.join(script_path, 'groups.yaml')
    manual_config_path = os.path.join(script_path, 'manual_config.yaml')
    group_dict = yaml.load(open(groups_yaml_path))
    manual_config_dict = yaml.load(open(manual_config_path))

    all_devices = {}
    add_knx_devices(all_devices, group_dict)
    add_fhem_devices(all_devices, group_dict)

    with open(os.path.join(target_directory, 'configuration.yaml'), 'w') as output:
        output.write("# Dont' edit manually! this is generated!\n\n")
        for key in ['sensor', 'switch', 'light', 'cover']:
            if key in manual_config_dict:
                all_devices[key].extend(manual_config_dict[key])
                del manual_config_dict[key]
        yaml.dump(manual_config_dict, output)
        yaml.dump(all_devices, output)
        yaml.dump(logbook_config(all_devices), output)
        yaml.dump(recorder_config(all_devices), output)

    add_light_groups(group_dict)

    with open(os.path.join(target_directory, 'groups.yaml'), 'w') as output:
        output.write("# Dont' edit manually! this is generated!\n\n")
        yaml.dump(group_dict, output)

    with open(os.path.join(target_directory, 'secrets.yaml'), 'w') as output:
        output.write("# Dont' edit manually! this is generated!\n\n")
        output.write(open(os.path.join(script_path, 'secrets.yaml'), 'r').read())
        output.write("\n\n")
        additional_file = 'secrets_development.yaml' if development else 'secrets_deploy.yaml'
        output.write(open(os.path.join(script_path, additional_file), 'r').read())


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("-d", "--dev", help="create config file for development", action="store_true")
    args = parser.parse_args()
    create_config(target_directory=os.getcwd(), development=args.dev)