File: __init__.py

package info (click to toggle)
cloud-init 22.4.2-1%2Bdeb12u3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 9,088 kB
  • sloc: python: 108,898; sh: 4,091; makefile: 147; xml: 22
file content (52 lines) | stat: -rw-r--r-- 1,576 bytes parent folder | download | duplicates (4)
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)