summaryrefslogtreecommitdiff
path: root/cloudinit/reporting/__init__.py
blob: b839eaae19e5050394ea47532690b034cbfd9082 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# Copyright (C) 2015 Canonical Ltd.
#
# This file is part of cloud-init. See LICENSE file for license information.

"""
cloud-init reporting framework

The reporting framework is intended to allow all parts of cloud-init to
report events in a structured manner.
"""

from typing import Type

from cloudinit.registry import DictRegistry
from cloudinit.reporting.handlers import HandlerType, available_handlers

DEFAULT_CONFIG = {
    "logging": {"type": "log"},
}


def update_configuration(config):
    """Update the instantiated_handler_registry.

    :param config:
        The dictionary containing changes to apply.  If a key is given
        with a False-ish value, the registered handler matching that name
        will be unregistered.
    """
    for handler_name, handler_config in config.items():
        if not handler_config:
            instantiated_handler_registry.unregister_item(
                handler_name, force=True
            )
            continue
        handler_config = handler_config.copy()
        cls: Type[HandlerType] = available_handlers.registered_items[
            handler_config.pop("type")
        ]
        instantiated_handler_registry.unregister_item(handler_name)
        instance = cls(**handler_config)  # pyright: ignore
        instantiated_handler_registry.register_item(handler_name, instance)


def flush_events():
    handler: HandlerType
    for handler in instantiated_handler_registry.registered_items.values():
        handler.flush()


instantiated_handler_registry = DictRegistry()
update_configuration(DEFAULT_CONFIG)