AnonymousFox

//lib/python2.7/site-packages/sos/plugins/python.py

# 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.

from sos.plugins import Plugin, RedHatPlugin, DebianPlugin, UbuntuPlugin
import sos.policies
import os
import json
import hashlib

class Python(Plugin, DebianPlugin, UbuntuPlugin):
    """Python runtime
    """

plugin_name = 'python'
    profiles = ('system',)

packages = ('python',)

python_version = "python -V"

def setup(self):
        self.add_cmd_output(
            self.python_version, suggest_filename="python-version"
        )
        self.add_cmd_output("pip list")

class RedHatPython(Python, RedHatPlugin):

packages = ('python', 'python36', 'python2', 'platform-python')
    option_list = [
        ('hashes', "gather hashes for all python files", 'slow',
         False)]

def setup(self):
        self.add_cmd_output(['python2 -V', 'python3 -V'])
        if isinstance(self.policy, sos.policies.redhat.RHELPolicy) and \
                self.policy.dist_version() > 7:
            self.python_version = "/usr/libexec/platform-python -V"
        super(RedHatPython, self).setup()

if self.get_option('hashes'):
            digests = {
                'digests': []
            }

py_paths = [
                '/usr/lib',
                '/usr/lib64',
                '/usr/local/lib',
                '/usr/local/lib64'
            ]

for py_path in py_paths:
                for root, _, files in os.walk(py_path):
                    for file_ in files:
                        filepath = os.path.join(root, file_)
                        if filepath.endswith('.py'):
                            try:
                                with open(filepath, 'rb') as f:
                                    digest = hashlib.sha256()
                                    chunk = 1024
                                    while True:
                                        data = f.read(chunk)
                                        if data:
                                            digest.update(data)
                                        else:
                                            break

digest = digest.hexdigest()

digests['digests'].append({
                                        'filepath': filepath,
                                        'sha256': digest
                                    })
                            except IOError:
                                self._log_error(
                                    "Unable to read python file at %s" %
                                    filepath
                                )

self.add_string_as_file(json.dumps(digests), 'digests.json')

# vim: set et ts=4 sw=4 :