<h2>Editing File: /usr/lib/python3.6/site-packages/sos/collector/clusters/saltstack.py</h2><form method='post'><textarea name='content' class='file-edit'># Copyright Red Hat 2022, Trevor Benson &lt;trevor.benson@gmail.com&gt;

# This file is part of the sos project: https://github.com/sosreport/sos
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# version 2 of the GNU General Public License.
#
# See the LICENSE file in the source distribution for further information.

import json
from shlex import quote
from sos.collector.clusters import Cluster


class saltstack(Cluster):
    &quot;&quot;&quot;
    The saltstack cluster profile is intended to be used on saltstack
    clusters (Salt Project).
    &quot;&quot;&quot;

    cluster_name = &quot;Saltstack&quot;
    packages = (&quot;salt-master&quot;,)
    sos_plugins = [&quot;saltmaster&quot;]
    strict_node_list = True
    option_list = [
        (&quot;compound&quot;, &quot;&quot;, &quot;Filter node list to those matching compound&quot;),
        (&quot;glob&quot;, &quot;&quot;, &quot;Filter node list to those matching glob pattern&quot;),
        (&quot;grain&quot;, &quot;&quot;, &quot;Filter node list to those with matching grain&quot;),
        (&quot;minion_id_unresolvable&quot;, False, &quot;Returns the FQDN grain of each&quot;
         &quot; minion in the node list when the minion ID is not a hostname.&quot;),
        (&quot;nodegroup&quot;, &quot;&quot;, &quot;Filter node list to those matching nodegroup&quot;),
        (&quot;pillar&quot;, &quot;&quot;, &quot;Filter node list to those with matching pillar&quot;),
        (&quot;subnet&quot;, &quot;&quot;, &quot;Filter node list to those in subnet&quot;),
    ]
    targeted = False

    node_cmd = &quot;salt-run --out=pprint manage.status&quot;

    def _parse_manage_status(self, output: str) -&gt; list:
        nodes = []
        salt_json_output = json.loads(output.replace(&quot;'&quot;, '&quot;'))
        for _, value in salt_json_output.items():
            nodes.extend(value)
        return nodes

    def _get_hostnames_from_grain(self, manage_status: dict) -&gt; list:
        hostnames = []
        for status, minions in manage_status.items():
            if status == &quot;down&quot;:
                self.log_warn(f&quot;Node(s) {minions} are status down.&quot;)
                hostnames.extend(minions)
            else:
                for minion in minions:
                    node_cmd = (
                        f&quot;salt --out=newline_values_only {minion} &quot;
                        f&quot;grains.get fqdn&quot;
                    )
                    hostnames.append(
                        self.exec_primary_cmd(node_cmd)[&quot;output&quot;].strip()
                    )
        return hostnames

    def _get_nodes(self) -&gt; list:
        res = self.exec_primary_cmd(self.node_cmd)
        if res[&quot;status&quot;] != 0:
            raise Exception(&quot;Node enumeration did not return usable output&quot;)
        if self.get_option(&quot;minion_id_unresolvable&quot;):
            status = json.loads(res[&quot;output&quot;].replace(&quot;'&quot;, '&quot;'))
            return self._get_hostnames_from_grain(status)
        return self._parse_manage_status(res[&quot;output&quot;])

    def get_nodes(self):
        # Default to all online nodes
        for option in self.option_list:
            if option[0] != &quot;minion_id_unresolvable&quot;:
                opt = self.get_option(option[0])
                if opt:
                    self.node_cmd += f&quot; tgt={quote(opt)} tgt_type={option[0]}&quot;
                    break
        return self._get_nodes()


# vim: set et ts=4 sw=4 :
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fusr%2Flib%2Fpython3.6%2Fsite-packages%2Fsos%2Fcollector%2Fclusters'>Back to Directory</a>