HEX

File: //usr/lib/python3.6/site-packages/sos/__pycache__/utilities.cpython-36.pyc
3

3��d~H�@sbddlZddlZddlZddlmZmZmZddlZddlZddl	Z	ddl
Z
ddlZddlZddl
Z
ddlZddlZddlmZddlmZyddlmZWn ek
r�ddlmZYnXdZyddlZeje�dZWnPeefk
�r,ej d	�Z!dd
l"m#Z#dZ$e!j%de#e$d
dd�d�YnXdZ&dd�Z'dLdd�Z(dQdd�Z)dd�Z*dRdd�Z+d d!�Z,dSd"d#�Z-e&dddddddddf
d$d%�Z.dTd&d'�Z/dUd(d)�Z0dVd+d,�Z1ej2fd-d.�Z3d/d0�Z4d1d2�Z5d3d4�Z6d5d6�Z7d7d8�Z8ej9d9�d:d;�Z:d<d=�Z;gfd>d?�Z<d@dA�ZGdBdC�dC�Z=GdDdE�dEe
j>�Z?GdFdG�dGe@�ZAGdHdI�dI�ZBGdJdK�dKeC�ZDdS)W�N)�Popen�PIPE�STDOUT)�closing)�deque)�
parse_version)�parseFT�sos)�fillz�WARNING: Failed to load 'magic' module version >= 0.4.20 which sos aims to use for detecting binary files. A less effective method will be used. It is recommended to install proper python3-magic package with the module.
�
�H)Zreplace_whitespacei,c
Cs<t|d��(}tj|�j|kr*|j|d�|j�SQRXdS)z,Returns the last number_of_bytes of filename�rb�N)�open�os�stat�st_size�seek�read)�filenameZnumber_of_bytes�f�r�/usr/lib/python3.6/utilities.py�tail2sr�rcCsXt|t�rLy
t||�Stk
rHtjd�}|jd|�ttj	��SXnt|�SdS)z@Returns a file-like object that can be used as a context managerr	zfileobj: %s could not be openedN)
�
isinstance�strr�IOError�logging�	getLogger�debugr�io�StringIO)Zpath_or_file�mode�logrrr�fileobj:s


r%��
���(cCsdt|�}||krd||S||kr0d||S||krDd||S||krXd||Sd|SdS)zCConverts a number of bytes to a shorter, more human friendly formatz%.1fTz%.1fGz%.1fMz%.1fKz%dN)�float)Zbytes_�K�M�G�T�fnrrr�
convert_bytesGsr1cs�trNy4tj|��ddg}�jdko8t�fdd�|D��Stk
rLYnXt|d��*}y|jd�dStk
r|d	SXWd
QRXd
S)azHelper to determine if a given file contains binary content or not.

    This is especially helpful for `sos clean`, which cannot obfuscate binary
    data and instead, by default, will remove binary files.

    :param fname:   The full path of the file to check binaryness of
    :type fname:    ``str``

    :returns:   True if binary, else False
    :rtype:     ``bool``
    ztext/zinode/�binaryc3s|]}�jj|�VqdS)N)Z	mime_type�
startswith)�.0Z_mt)�_ftuprr�	<genexpr>hsz!file_is_binary.<locals>.<genexpr>Ztrr&FTN)	�	magic_mod�magic�detect_from_filename�encoding�any�	Exceptionrr�UnicodeDecodeError)�fnameZ_mimesZtfiler)r5r�file_is_binaryVs


r?c	cs�|r"tjj|�jtjj�}||7}xrtj|�D]d\}}}|rX|jtjj�|krX|dd�=|rltj||�rlq.x$tj||�D]}tjj||�VqzWq.WdS)z�Generator function to find files recursively.
    Usage::

        for filename in find("*.properties", "/var/log/foobar"):
            print filename
    N)	r�path�dirname�count�sep�walk�fnmatch�filter�join)	Zfile_patternZtop_dirZ	max_depthZpath_patternZ
base_depthr@ZdirlistZfilelist�namerrr�findxs
rIcsBg}x8|D]0}t|��}|j�fdd�|D��WdQRXq
W|S)z�Returns lines matched in fnames, where fnames can either be pathnames to
    files to grep through or open file objects to grep through line by linec3s|]}tj�|�r|VqdS)N)�re�match)r4�line)�patternrrr6�szgrep.<locals>.<genexpr>N)r%�extend)rMZfiles_or_pathsZmatchesZfopZfor)rMr�grep�s


&rOcs\tjjdd�jtjj�}�g�fdd�|D�}�rJ|�fdd�|D�7}tdd�|D��S)z6Returns if a command matches an executable on the PATH�PATH�csg|]}tjj|���qSr)rr@rG)r4�p)�commandrr�
<listcomp>�sz!is_executable.<locals>.<listcomp>cs g|]}tjj�|jd���qS)�/)rr@rG�lstrip)r4�c)�sysrootrrrT�scss|]}tj|tj�VqdS)N)r�access�X_OK)r4r@rrrr6�sz is_executable.<locals>.<genexpr>)r�environ�get�splitr@�pathsepr;)rSrX�pathsZ
candidatesr)rSrXr�
is_executable�sr`csb��fdd�}�fdd�}tjj�}
d|
d<|r`x.|j�D]"\}}|rP||
|<q:|
j|d�q:W|r�td�r�d	|rvd
nd||f}tj|�}g}xN|D]F}|jd�r�d
|kr�t	j	|�}|r�|j
|�q�|j|�q�|j|�q�W|
r�t|
d�}nt
}y�t|d||�rtnt
d|
d|d�}|
�s.t|j||�}n
t||�}��rXxr|j�rT||��q@WnZy|j|�rh|nd�Wn@tk
�r�|j�|
�r�|j�d|_d|j�|jd�SX|
�r�|j�x|j�dk�rԐq�W|j�}|j}WnPtk
�r6}z2|
�r|j�|jtjk�r"dddd�S|�WYdd}~XnX|jdk�sP|jdk�rTd}|j||d�S)z�Execute a command and return a dictionary of status and output,
    optionally changing root or current working directory before
    executing command.
    cs(�r�dkrtj���r$tj��dS)NrU)r�chroot�chdirr)rbrarr�_child_prep_fn�s
z.sos_get_command_output.<locals>._child_prep_fncs,��s|j�dkr|j�t�tjd�dS)N�|g{�G�z�?)�poll�	terminate�SoSTimeoutError�time�sleep)�proc)�pollerrr�
_check_poller�sz-sos_get_command_output.<locals>._check_pollerzC.UTF-8�LC_ALLN�timeoutztimeout %s %ds %sz--foregroundrQrU�*�wFr&T)�shell�stdout�stderr�bufsize�envZ	close_fdsZ
preexec_fnrd)Zstatus�output�	truncated��~����)rr[�copy�items�popr`�shlexr]r3�globrN�appendrrrr�AsyncReaderrr�
FakeReader�running�waitr<rf�close�get_contents�is_fullre�OSError�errno�ENOENT�
returncode)rSrnrsrarbruZ
foregroundr2�	sizelimitrkZto_filercrlZcmd_env�key�value�argsZ
expanded_args�argZexpanded_argZ_outputrR�readerrrrw�er)rbrarkr�sos_get_command_output�s~








r�csX�jd�d}t�t�t�|g�}�fdd�tj|tj�D�}�rT�fdd�|D�}|S)aImports the module module_fqname and returns a list of defined classes
    from that module. If superclasses is defined then the classes returned will
    be subclasses of the specified superclass or superclasses. If superclasses
    is plural it must be a tuple of classes.�.r&csg|]\}}|j�kr|�qSr)�
__module__)r4ZcnameZclass_)�
module_fqnamerrrTsz!import_module.<locals>.<listcomp>csg|]}t|��r|�qSr)�
issubclass)r4�m)�superclassesrrrT"sr{)�
rpartition�
__import__�globals�locals�inspectZ
getmembersZisclass)r�r�Zmodule_name�module�modulesr)r�r�r�
import_modules
r�cCst||||d�dS)zeShell out to an external command and return the output or the empty
    string in case of error.
    )rnrarbrv)r�)�cmdrnraZrunatrrr�	shell_out'sr�rcCsJdddddg}d}x$|dkr6|dkr6|d	7}|d
}qWd||||fS)N�BZKiBZMiBZGiBZTiBri�r&g�@z%.*f%sr)�sizeZ	precision�suffixesZsuffixindexrrr�get_human_readable/sr�cCs>|r,|tjkr,|j|�s,tjj||jd��}t||�}||�S)NrU)rrCr3r@rGrV�getattr)r@rX�methodr�Z_methrrr�_os_wrapper9s


r�cCs$d|krt||dtd�St||d�S)Nror�)r��exists)r�r�)r@rXrrr�path_existsAsr�cCst||d�S)N�isdir)r�)r@rXrrr�
path_isdirGsr�cCst||d�S)N�isfile)r�)r@rXrrr�path_isfileKsr�cCst||d�S)N�islink)r�)r@rXrrr�path_islinkOsr�cCst||dt�S)N�listdir)r�r)r@rXrrrr�Ssr�)rXcGs8|r&|j|�r&tjj||jtj��}tjj|f|��S)N)r3rr@rGrVrC)r@rXrRrrr�	path_joinWsr�cCsd|dS)a"Helper to make text bold in console output, without pulling in
    dependencies to the project unneccessarily.

    :param text:    The text to make bold
    :type text:     ``str``

    :returns:       The text wrapped in the ASCII codes to display as bold
    :rtype:         ``str``
    z[1mz[0mr)�textrrr�bold]s
r�cs�g}g�|j|�t|t�r�x�|j�D]�\}��j|�|sF||krV|jt���q&t�t�r&y$|jt�fdd�|D��d�Wntk
r�YnX�j�j��q&Wn
|j|��fdd�|D�S)a	Recursively compile all elements of a potentially nested dict by a set
    of keys. If a given key is a dict within ``dobj``, then _all_ elements
    within that dict, regardless of child keys, will be returned.

    For example, if a Plugin searches the devices dict for the 'storage' key,
    then all storage devices under the that dict (e.g. block, fibre, etc...)
    will be returned. However, if the Plugin specifies 'block' via ``keys``,
    then only the block devices within the devices['storage'] dict will be
    returned.

    Any elements passed here that are _not_ keys within the dict or any nested
    dicts will also be returned.

    :param dobj:    The 'top-level' dict to intially search by
    :type dobj:     ``dict``

    :param keys:    Which keys to compile elements from within ``dobj``. If no
                    keys are given, all nested elements are returned
    :param keys:    ``list`` of ``str``

    :returns:       All elements within the dict and any nested dicts
    :rtype:         ``list``
    c3s|]}|�kr�|VqdS)Nr)r4r�)�vrrr6�sz/recursive_dict_values_by_key.<locals>.<genexpr>rcsg|]}|�kr|�qSrr)r4�d)�_filtrrrT�sz0recursive_dict_values_by_key.<locals>.<listcomp>)rNr�dictr}r��recursive_dict_values_by_key�
IndexError�keys)Zdobjr�Z_items�kr)r�r�rr�js&




r�cCst|�S)zParse the version string
    )�
version_parse)�versionrrrr�src@s8eZdZdZdd�Zedd��Zdd�Zedd	��Zd
S)r�z�Used as a replacement AsyncReader for when we are writing directly to
    disk, and allows us to keep more simplified flows for executing,
    monitoring, and collecting command output.
    cCs||_||_dS)N)�processr2)�selfr�r2rrr�__init__�szFakeReader.__init__cCsdS)NFr)r�rrrr��szFakeReader.is_fullcCs|js
dSdS)NrQrz)r2)r�rrrr��szFakeReader.get_contentscCs|jj�dkS)N)r�re)r�rrrr��szFakeReader.runningN)	�__name__r��__qualname__�__doc__r��propertyr�r�r�rrrrr��s
r�cs<eZdZdZ�fdd�Zdd�Zdd�Zedd	��Z�Z	S)
r�z�Used to limit command output to a given size without deadlocking
    sos.

    Takes a sizelimit value in MB, and will compile stdout from Popen into a
    string that is limited to the given sizelimit.
    csbtt|�j�||_||_d|_d|_|rB|d}t||j�|_t|jd�|_d|_	|j
�dS)Nii)�maxlenT)�superr�r��chanr2�	chunksize�slots�intrr��start)r�Zchannelr�r2)�	__class__rrr��szAsyncReader.__init__cCsNy*x$|jj|j�}|sP|jj|�qWWnttfk
rBYnXd|_dS)alReads from the channel (pipe) that is the output pipe for a
        called Popen. As we are reading from the pipe, the output is added
        to a deque. After the size of the deque exceeds the sizelimit
        earlier (older) entries are removed.

        This means the returned output is chunksize-sensitive, but is not
        really byte-sensitive.
        FN)r�rr�rr��
ValueErrorrr�)r�rLrrr�run�s	zAsyncReader.runcCsLx|jrtjd�qW|js2djdd�|jD��Sdjdd�|jD��SdS)z-Returns the contents of the deque as a stringg{�G�z�?rQcss|]}|jdd�VqdS)zutf-8�ignoreN)�decode)r4�lnrrrr6�sz+AsyncReader.get_contents.<locals>.<genexpr>rzcss|]
}|VqdS)Nr)r4r�rrrr6�sN)r�rhrir2rGr)r�rrrr��s
zAsyncReader.get_contentscCs|js
dSt|j�|jkS)z?Checks if the deque is full, implying that output was truncatedF)r��lenr)r�rrrr��szAsyncReader.is_full)
r�r�r�r�r�r�r�r�r��
__classcell__rr)r�rr��s

r�c@s8eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�ZdS)
�ImporterHelperz�Provides a list of modules that can be imported in a package.
    Importable modules are located along the module __path__ list and modules
    are files that end in .py.
    cCs
||_dS)zopackage is a package module
        import my.package.module
        helper = ImporterHelper(my.package.module)N)�package)r�r�rrrr��szImporterHelper.__init__cCs tjj|�}tjj|�\}}|S)z-Returns the plugin module name given the path)rr@�basename�splitext)r�r@�baserHZextrrr�_plugin_nameszImporterHelper._plugin_namecs�fdd�|D�}|j�|S)Ncs(g|] }d|kr|jd�r�j|��qS)r�z.py)�endswithr�)r4Zplugin)r�rrrTsz9ImporterHelper._get_plugins_from_list.<locals>.<listcomp>)�sort)r�Zlist_�pluginsr)r�r�_get_plugins_from_lists
z%ImporterHelper._get_plugins_from_listcCs4tjj|�r0ttd|��}|j|�}|r,|SgSdS)Nz*.py)rr@r��listrIr�)r�r@Zpy_filesZpnamesrrr�_find_plugins_in_dirs
z#ImporterHelper._find_plugins_in_dircCs6g}x,|jjD] }tjj|�r|j|j|��qW|S)zQReturns the list of importable modules in the configured python
        package. )r��__path__rr@r�rNr�)r�r�r@rrr�get_moduless
zImporterHelper.get_modulesN)	r�r�r�r�r�r�r�r�r�rrrrr��s	r�c@s$eZdZdd�Zdd�Zdd�ZdS)�TempFileUtilcCs||_g|_dS)N)�tmp_dir�files)r�r�rrrr�(szTempFileUtil.__init__cCs2tj|jd�\}}tj|d�}|jj||f�|S)N)�dirzw+)�tempfileZmkstempr�r�fdopenr�r�)r��fdr>Zfobjrrr�new,szTempFileUtil.newcCslx`|jD]V\}}y|j�|j�Wntk
r8YnXytj|�Wqtk
r\YqXqWg|_dS)N)r��flushr�r<r�unlink)r�r>rrrr�clean3s
zTempFileUtil.cleanN)r�r�r�r�r�r�rrrrr�&sr�c@seZdZdS)rgN)r�r�r�rrrrrgAsrg)r���@�)r�r�r�r�)NN)N)N)r)NN)r)ErrJr��
subprocessrrrrrEr�rr�r�Z	threadingrhr!�
contextlibr�collectionsrZ
pkg_resourcesrr��SyntaxErrorZpackaging.versionrr7r8r9�__file__�ImportError�AttributeErrorrr$�textwrapr
�msgZwarningZTIMEOUT_DEFAULTrr%r1r?rIrOr`r�r�r�r�r@r�r�r�r�r�r�rCr�r�r�r�ZThreadr��objectr�r�r�rgrrrr�<module>	st

"

"

n




6=-