From a82b6316d8a780a7a8cbfbabeb52fa50b3fb1032 Mon Sep 17 00:00:00 2001 From: Alexander Graul Date: Tue, 18 Jan 2022 19:07:34 +0100 Subject: [PATCH] Revert "Fixing a use case when multiple inotify beacons are defined but when notifications are fired the configuration fron the first beacon are used." Revert "Adding a util function to remove hidden (options starting with underscore) from the beacon configuration. This is used when the beacons loop through the configuration, eg. status beacon, and expect certain options." This reverts commit 68a891ab2fe53ebf329b9c83b875f3575e87e266. This reverts commit 66c58dedf8c364eaeb35c5adce8bcc8fe5c1219a. --- salt/beacons/__init__.py | 1 - salt/beacons/diskusage.py | 3 --- salt/beacons/inotify.py | 24 ++++++++-------------- salt/beacons/napalm_beacon.py | 6 ++---- salt/beacons/status.py | 4 ---- tests/pytests/unit/beacons/test_inotify.py | 5 +---- tests/pytests/unit/test_beacons.py | 16 --------------- 7 files changed, 11 insertions(+), 48 deletions(-) diff --git a/salt/beacons/__init__.py b/salt/beacons/__init__.py index 414142c262..13e2ae78db 100644 --- a/salt/beacons/__init__.py +++ b/salt/beacons/__init__.py @@ -71,7 +71,6 @@ class Beacon: beacon_name = current_beacon_config["beacon_module"] else: beacon_name = mod - b_config[mod].append({"_beacon_name": mod}) fun_str = "{}.beacon".format(beacon_name) validate_str = "{}.validate".format(beacon_name) if fun_str in self.beacons: diff --git a/salt/beacons/diskusage.py b/salt/beacons/diskusage.py index 1216abf79b..897bc90a6a 100644 --- a/salt/beacons/diskusage.py +++ b/salt/beacons/diskusage.py @@ -8,7 +8,6 @@ Beacon to monitor disk usage. import logging import re -import salt.utils.beacons import salt.utils.platform try: @@ -81,8 +80,6 @@ def beacon(config): it will override the previously defined threshold. """ - whitelist = [] - config = salt.utils.beacons.remove_hidden_options(config, whitelist) parts = psutil.disk_partitions(all=True) ret = [] for mounts in config: diff --git a/salt/beacons/inotify.py b/salt/beacons/inotify.py index b6e7334eee..c44bd49fb0 100644 --- a/salt/beacons/inotify.py +++ b/salt/beacons/inotify.py @@ -65,19 +65,17 @@ def _get_notifier(config): """ Check the context for the notifier and construct it if not present """ - beacon_name = config.get("_beacon_name", "inotify") - notifier = "{}.notifier".format(beacon_name) - if notifier not in __context__: + if "inotify.notifier" not in __context__: __context__["inotify.queue"] = collections.deque() wm = pyinotify.WatchManager() - __context__[notifier] = pyinotify.Notifier(wm, _enqueue) + __context__["inotify.notifier"] = pyinotify.Notifier(wm, _enqueue) if ( "coalesce" in config and isinstance(config["coalesce"], bool) and config["coalesce"] ): - __context__[notifier].coalesce_events() - return __context__[notifier] + __context__["inotify.notifier"].coalesce_events() + return __context__["inotify.notifier"] def validate(config): @@ -237,9 +235,6 @@ def beacon(config): being at the Notifier level in pyinotify. """ - whitelist = ["_beacon_name"] - config = salt.utils.beacons.remove_hidden_options(config, whitelist) - config = salt.utils.beacons.list_to_dict(config) ret = [] @@ -262,7 +257,7 @@ def beacon(config): break path = os.path.dirname(path) - excludes = config["files"].get(path, {}).get("exclude", "") + excludes = config["files"][path].get("exclude", "") if excludes and isinstance(excludes, list): for exclude in excludes: @@ -349,9 +344,6 @@ def beacon(config): def close(config): - config = salt.utils.beacons.list_to_dict(config) - beacon_name = config.get("_beacon_name", "inotify") - notifier = "{}.notifier".format(beacon_name) - if notifier in __context__: - __context__[notifier].stop() - del __context__[notifier] + if "inotify.notifier" in __context__: + __context__["inotify.notifier"].stop() + del __context__["inotify.notifier"] diff --git a/salt/beacons/napalm_beacon.py b/salt/beacons/napalm_beacon.py index ec8cf63fca..164b29cdf8 100644 --- a/salt/beacons/napalm_beacon.py +++ b/salt/beacons/napalm_beacon.py @@ -168,9 +168,10 @@ with a NTP server at a stratum level greater than 5. """ import logging + +# Import Python std lib import re -import salt.utils.beacons import salt.utils.napalm log = logging.getLogger(__name__) @@ -301,9 +302,6 @@ def beacon(config): """ Watch napalm function and fire events. """ - whitelist = [] - config = salt.utils.beacons.remove_hidden_options(config, whitelist) - log.debug("Executing napalm beacon with config:") log.debug(config) ret = [] diff --git a/salt/beacons/status.py b/salt/beacons/status.py index aa5aa13b47..e2c3177ea8 100644 --- a/salt/beacons/status.py +++ b/salt/beacons/status.py @@ -91,7 +91,6 @@ import datetime import logging import salt.exceptions -import salt.utils.beacons import salt.utils.platform log = logging.getLogger(__name__) @@ -119,9 +118,6 @@ def beacon(config): log.debug(config) ctime = datetime.datetime.utcnow().isoformat() - whitelist = [] - config = salt.utils.beacons.remove_hidden_options(config, whitelist) - if not config: config = [ { diff --git a/tests/pytests/unit/beacons/test_inotify.py b/tests/pytests/unit/beacons/test_inotify.py index f5befb2756..dfaf1d499a 100644 --- a/tests/pytests/unit/beacons/test_inotify.py +++ b/tests/pytests/unit/beacons/test_inotify.py @@ -273,7 +273,6 @@ def test_multi_files_exclude(tmp_path): # Check __get_notifier and ensure that the right bits are in __context__ -# including a beacon_name specific notifier is found. def test__get_notifier(): config = { "files": { @@ -303,10 +302,8 @@ def test__get_notifier(): }, }, "coalesce": True, - "beacon_module": "inotify", - "_beacon_name": "httpd.inotify", } ret = inotify._get_notifier(config) assert "inotify.queue" in inotify.__context__ - assert "httpd.inotify.notifier" in inotify.__context__ + assert "inotify.notifier" in inotify.__context__ diff --git a/tests/pytests/unit/test_beacons.py b/tests/pytests/unit/test_beacons.py index 27940c6f65..a347f3f27f 100644 --- a/tests/pytests/unit/test_beacons.py +++ b/tests/pytests/unit/test_beacons.py @@ -70,19 +70,3 @@ def test_beacon_module(): ] assert ret == _expected - # Ensure that "beacon_name" is available in the call to the beacon function - name = "ps.beacon" - mocked = {name: MagicMock(return_value=_expected)} - mocked[name].__globals__ = {} - calls = [ - call( - [ - {"processes": {"apache2": "stopped"}}, - {"beacon_module": "ps"}, - {"_beacon_name": "watch_apache"}, - ] - ) - ] - with patch.object(beacon, "beacons", mocked) as patched: - beacon.process(mock_opts["beacons"], mock_opts["grains"]) - patched[name].assert_has_calls(calls) -- 2.34.1