Current File : //usr/lib/python2.7/site-packages/salt/modules/disk.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l m Z d d l
m Z d d l Z d d l Z d d l Z d d l Z d d l m Z i d d 6Z e j e � Z e j j j d � d k Z e j j j d
� d k Z d � Z d � Z! d
� Z" d d � Z# d d � Z$ d d � Z% e j j& j j d � d d � � Z' d � Z( d � Z) d d � Z* d � Z+ e j j& j j d � e j j& j j d � d d d e, d � � � Z- e j j& j j. d d g � d � � Z/ e j j& j0 e � e1 d � � Z2 e j j& j0 e � d d � � Z3 e j j& j0 e � d d � � Z4 d d d! � Z5 e j j& j0 e � d" d# d d$ � � Z6 d% � Z7 d& � Z8 d' � Z9 d( � Z: d S() u,
Module for managing disks and blockdevices
i ( t absolute_importt print_functiont unicode_literalsN( t six( t zip( t CommandExecutionErroru formatu format_u hdparmu iostatc C@ s t j j j � r t d f St S( u)
Only work on POSIX-like systems
u$ This module doesn't work on Windows.( t saltt utilst platformt
is_windowst Falset True( ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt __virtual__$ s
c C@ s� | j � r t j | � Sy� i d d 6d d 6d d 6d d 6d d
6d d 6d
d 6d d 6} | d | j � k r� t j | d � } | t j | | d � } | St j | � SWn t k
r� | SXd S( u�
Convert a string to a number, allowing for a K|M|G|T postfix, 32.8K.
Returns a decimal number if the string is a real number,
or the string unchanged otherwise.
u 10E3u Ku 10E6u Mu 10E9u Gu 10E12u Tu 10E15u Pu 10E18u Eu 10E21u Zu 10E24u Yi����N( t isdigitt decimalt Decimalt keyst
ValueError( t textt postPrefixest v( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt _parse_numbers- s
>
c C@ s_ d } | d k r | Sd } x< | D]4 } | | k rB | | 7} q# t d j | � � � q# W| S( u'
Sanitize flags passed into df
u u au Bu hu Hu iu ku lu Pu tu Tu xu vu Invalid flag passed to {0}N( u au Bu hu Hu iu ku lu Pu tu Tu xu v( t NoneR t format( t argst callert flagst allowedt flag( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt _clean_flagsB s
c C@ s� t | d � } t j j d � rh t d d k rh t j d � t j d � d k rd t j d � n i St d d k r� d } nH t d d
k s� t d d k r� d } n t d d
k r� d } n d } | r� | d j | � 7} n i } t d | d t
�j � } d* } x�| D]�} | s&qn | j
d � r;qn | rR| d | } n | j � } t | � d k r|| } qn d* } xR t | � d k r�| d j � r�d j | d | d � | d <| j d � q�Wt | � d k r�qn y� t d d k rki | d d 6| d d 6| d d 6| d d 6| d d 6| d! d" 6| d# d$ 6| d% d& 6| | d' <nE i | d d 6| d d( 6| d d 6| d d 6| d d 6| | d! <Wqt k
r�t j d) � i } qXqW| S(+ u�
Return usage information for volumes mounted on this minion
.. versionchanged:: 2019.2.0
Default for SunOS changed to 1 kilobyte blocks
CLI Example:
.. code-block:: bash
salt '*' disk.usage
u
disk.usageu /etc/mtabu kernelu Linuxu df cannot run without /etc/mtabu virtual_subtypeu LXCu� df command failed and LXC detected. If you are running a Docker container, consider linking /proc/mounts to /etc/mtab or consider running Docker with -privilegedu df -Pu OpenBSDu AIXu df -kPu SunOSu df -ku dfu -{0}u cmd.runt python_shellu
Filesystemu i i u {0} {1}i u Darwinu
filesystemu
512-blocksu usedi u availablei u capacityi u iusedi u ifreei u %iusedi u 1K-blocksu& Problem parsing disk usage informationN( R t ost patht isfilet
__grains__t logt errort getR t __salt__R
t
splitlinesR t
startswitht splitt lenR
t popt
IndexError( R R t cmdt rett outt oldlinet linet comps( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt usageT sp #
&
c C@ s� t | d � } t d d k r( d } n d } | rJ | d j | � 7} n i } t d | d t �j � } xu| D]m} | j d � r� qs n | j � } | s� qs n y
t d d
k ri t | d � t | d � d
6| d d 6| d d 6| d d 6| d d 6| | d <n� t d d k rmi | d d
6| d d 6| d d 6| d d 6| d d 6| | d <nE i | d d
6| d d 6| d d 6| d d 6| d d 6| | d <Wqs t t
f k
r�t j d � i } qs Xqs W| S( u�
Return inode usage information for volumes mounted on this minion
CLI Example:
.. code-block:: bash
salt '*' disk.inodeusage
u disk.inodeusageu kernelu AIXu df -iu df -iPu -{0}u cmd.runR u
Filesystemu OpenBSDi i u inodesu usedu freei u usei u
filesystemi i i i i u' Problem parsing inode usage information(
R R" R R& R
R' R( R) t intR, R R# R$ ( R R R- R. R/ R1 R2 ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt
inodeusage� sN
c C@ s� t d d k r d } n/ t d d k s9 t d d k rB d } n d } i } t d | d t �j � } x
| D]} | s� qq n | j d
� r� qq n | j � } xR t | � d k r� | d j � r� d
j | d | d � | d <| j d � q� Wt | � d k rqq n y; t d d k r9| d | | d <n | d | | d <Wqq t
k
rrt j d � i } qq Xqq W| r�| | k r�t j d | � i } n | r�| | S| S( u�
Return partition information for volumes mounted on this minion
CLI Example:
.. code-block:: bash
salt '*' disk.percent /var
u kernelu Linuxu df -Pu OpenBSDu AIXu df -kPu dfu cmd.runR u
Filesystemi i u {0} {1}i u Darwini i i u& Problem parsing disk usage informationuF Problem parsing disk usage information: Partition '%s' does not exist!(
R" R& R
R' R( R) R* R
R R+ R, R# R$ ( R R- R. R/ R1 R2 ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt percent� sB
&
u blkidc
C@ s d } | r d | } n i } t d d | d t �} | d d k rM | Sx� | d j � D]� } | sp q^ n | j � } | d d } i } t j d
| j d � d � } xR t t | � g d � D]7 \ } } | j d � j d � } | j d
� | | <q� W| | | <q^ W| S( u�
Return block device attributes: UUID, LABEL, etc. This function only works
on systems where blkid is available.
CLI Example:
.. code-block:: bash
salt '*' disk.blkid
salt '*' disk.blkid /dev/sda
u u u cmd.run_allu blkidR u retcodei u stdouti����u "*"i u =u "( R& R
R' R) t ret partitionR t itert strip(
t deviceR R. t blkid_resultR1 R2 t infot device_attributest keyt value( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt blkid s&
&c K@ s i d d 6d d 6d d 6d d 6} d } g } x� | D]� } | | k r5 | | } | d k ry | j | j d
d � � n
| j d � | | d
k s� | | t k r� | d j | � 7} q� | d j | | | � 7} q5 q5 Wd j | | � } t d | d t �j � } t | | � S( ud
Set attributes for the specified device
CLI Example:
.. code-block:: bash
salt '*' disk.tune /dev/sda1 read-ahead=1024 read-write=True
Valid options are: ``read-ahead``, ``filesystem-read-ahead``,
``read-only``, ``read-write``.
See the ``blockdev(8)`` manpage for a more complete description of these
options.
u setrau
read-aheadu setfrau filesystem-read-aheadu setrou read-onlyu setrwu
read-writeu u setu getu getrou Trueu --{0} u
--{0} {1} u blockdev {0}{1}u cmd.runR ( t appendt replaceR R R& R
R' t dump( R; t kwargst kwarg_mapt optsR R? t switchR- R/ ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt tune4 s$
! c C@ sv d j | � } y t d | d t �} Wn t j k
rB } t SX| d d k rW t St j d | | d � t Sd S( u}
Remove the filesystem information
CLI Example:
.. code-block:: bash
salt '*' disk.wipe /dev/sda1
u
wipefs -a {0}u cmd.run_allR u retcodei u Error wiping device %s: %su stderrN( R R& R
t
subprocesst CalledProcessErrorR R# R$ ( R; R- R/ t err( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt wipe[ s c C@ s d j | � } i } g | j � D] } | j d � r" | d ^ q" } t d | d t �} | d d k r� g | d j � D] } | r~ | ^ q~ } d } x& | D] } | | | | <| d } q� W| r� i }
x | D] } | | |
| <q� W|
S| Sn t Sd
S( u�
Return all contents of dumpe2fs for a specified device
CLI Example:
.. code-block:: bash
salt '*' disk.dump /dev/sda1
u� blockdev --getro --getsz --getss --getpbsz --getiomin --getioopt --getalignoff --getmaxsect --getsize --getsize64 --getra --getfra {0}u --i u cmd.run_allR u retcodei u stdouti N( R R) R( R& R
R' ( R; R R- R. t cRG R/ R1 t linest countt temp_rett arg( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyRD r s$ 2)
c C@ s[ d j | � } y t d | d t �} Wn t j k
rB } t SX| d d k rW t Sd S( uw
Resizes the filesystem.
CLI Example:
.. code-block:: bash
salt '*' disk.resize2fs /dev/sda1
u
resize2fs {0}u cmd.run_allR u retcodei N( R R& R
RJ RK R ( R; R- R/ RL ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt resize2fs� s u syncu mkfsu ext4c C@ sb d d t j | � g } | d k r� | d d k rS | j d t j | � g � q� | d k r� | j d d j | � g � q� n | d k r� | d d k r� | j d d j | � g � q� n | r| d d k r� | j d
� q| d k r| j d � qn | j t j | � � t d | d
t �d k } t d d d
t �d k } t | | g � S( u}
Format a filesystem onto a device
.. versionadded:: 2016.11.0
device
The device in which to create the new filesystem
fs_type
The type of filesystem to create
inode_size
Size of the inodes
This option is only enabled for ext and xfs filesystems
lazy_itable_init
If enabled and the uninit_bg feature is enabled, the inode table will
not be fully initialized by mke2fs. This speeds up filesystem
initialization noticeably, but it requires the kernel to finish
initializing the filesystem in the background when the filesystem
is first mounted. If the option value is omitted, it defaults to 1 to
enable lazy inode table zeroing.
This option is only enabled for ext filesystems
force
Force mke2fs to create a filesystem, even if the specified device is
not a partition on a block special device. This option is only enabled
for ext and xfs filesystems
This option is dangerous, use it with caution.
CLI Example:
.. code-block:: bash
salt '*' disk.format /dev/sdX1
u mkfsu -ti u extu -iu xfsu size={0}u -Eu lazy_itable_init={0}u -Fu -fu cmd.retcodet ignore_retcodei u syncN( R t text_typeR t extendR RB R& R t all( R; t fs_typet
inode_sizet lazy_itable_initt forceR- t mkfs_successt sync_success( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt format_� s$ . " " u lsblku dfc C@ s7 t j j j d � rf t d d j | � � j � } t | � d k rf | d j � } | rc | Sqf n t j j j d � r3t d d k r� t
j j d � r� t d d j | � � j � } t | � d
k r0| d
} | r� | Sq0q3t d d j | � � j � } t | � d k r3| d } | r0| Sq3n d S(
u�
Return the filesystem name of the specified device
.. versionadded:: 2016.11.0
device
The name of the device
CLI Example:
.. code-block:: bash
salt '*' disk.fstype /dev/sdX1
u lsblku cmd.runu lsblk -o fstype {0}i u dfu kernelu AIXu /usr/sysv/bin/dfu /usr/sysv/bin/df -n {0}i u df -T {0}u (
R R R t whichR& R R' R* R: R" R R! R) ( R; t lsblk_outRX t df_out( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt fstype� s&
"
c C@ sr d j | � } t d | � } | d d k rj d j | | d � } | rZ t | � � qj t j | � n | d S( uT
Execute hdparm
Fail hard when required
return output when possible
u
hdparm {0}u cmd.run_allu retcodei u {0}: {1}u stderru stdout( R R& R R# t warning( R t failhardR- t resultt msg( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt _hdparm s c
C@ s1 d } | d k r | } n' t | t t f � rB d j | � } n t | t t f � sc | g } n i } x�| D]�} | j d � s� d j | � } n i } xt d j | | � t � j � D]\} | j
� } t | � d k s� | | d k r� q� n d | k r5| j d d � \ } } t
j d d | � } n* d
| k r� | j d
d � \ } } n q� | j
� j � j d d � } | j
� } g } t
j d
| � r�| j d � } | j t | d � � | j | d j
d � � n-i }
x� t
j d | j
� � D]� } | j
� } y t | � } | j | � Wqt k
r�d
| k r�| j d
d � \ } } | j
� } | j
� } t | � r�| |
| <q�q�t | � r�| j | � q�qXqWt |
� r�| j |
� n t | � d k r�q� n t | � d k r| d } n | | | <q� W| | | <qp W| S( u�
Retrieve all info's for all disks
parse 'em into a nice dict
(which, considering hdparms output, is quite a hassle)
.. versionadded:: 2016.3.0
CLI Example:
.. code-block:: bash
salt '*' disk.hdparms /dev/sda
u aAbBcCdgHiJkMmNnQrRuWu u /devu /dev/{0}u -{0} {1}i u :i u is$u =u u _u
[0-9]+ \(.*\)u ()u [/,]N( R t
isinstancet listt tuplet joinR( R Rg R
R' R: R* R) R7 t subt lowerRC t matchRB R4 t Exception(
t disksR t all_parmsR/ t diskt disk_dataR1 R? t valst rvalst valdictt valt deep_key( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt hdparms$ sj
( "
c C@ s` i } x� t | d � j � D]t \ } } | j � d \ } } } | | k sY d | k rm i | d 6| | <q i | d 6| d 6| | d 6| | <q W| d
k r� | Sx� | j � D]� \ } } y | d t | � } WnQ t k
r(d | k r)t | j d � � } d | | d } | d :} q)n X| d k rB| d } n t d j | | � � q� Wd
S( u�
Get/set Host Protected Area settings
T13 INCITS 346-2001 (1367D) defines the BEER (Boot Engineering Extension Record)
and PARTIES (Protected Area Run Time Interface Extension Services), allowing
for a Host Protected Area on a disk.
It's often used by OEMS to hide parts of a disk, and for overprovisioning SSD's
.. warning::
Setting the HPA might clobber your data, be very careful with this on active disks!
.. versionadded:: 2016.3.0
CLI Example:
.. code-block:: bash
salt '*' disk.hpa /dev/sda
salt '*' disk.hpa /dev/sda 5%
salt '*' disk.hpa /dev/sda 10543256
u Ni u disabledu totalu visibleu hiddenu %id u' --yes-i-know-what-i-am-doing -Np{0} {1}N( Ry t itemst valuesR R4 Ro R: Rg R ( Rp t sizet hpa_dataRr t datat visiblet totalt status( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt hpaq s. "
c C@ sg | j d � s d | } n d j | � } t d | d d �} | d d k rd t | d � � n t | d j � � } g } x\ | D]T } | j d
� r� t j d | j � � } g | d D] } | j � ^ q� } Pq� q� W| d k rg | D] } | | k r
| n d
^ q� } n i } xA| D]9} t j d | � sBPn t j d | j � d t | � �} t
| d � }
| d k r�|
| k r�q&n t t | | d � � } y | d
=Wn t k
r�n Xx� | D]~ } | | } y t
| � } WnQ t k
rFy, g | j d � D] }
t
|
� ^ q} WqGt k
rBqGXn X| | | <q�W| | |
<q&W| S( u�
Fetch SMART attributes
Providing attributes will deliver only requested attributes
Providing values will deliver only requested values for attributes
Default is the Backblaze recommended
set (https://www.backblaze.com/blog/hard-drive-smart-stats/):
(5,187,188,197,198)
.. versionadded:: 2016.3.0
CLI Example:
.. code-block:: bash
salt '*' disk.smart_attributes /dev/sda
salt '*' disk.smart_attributes /dev/sda attributes=(5,187,188,197,198)
u /dev/u smartctl --attributes {0}u cmd.run_allt output_loglevelu quietu retcodei u stderru stdoutu ID#u \s+i u _u [\s]*\dt maxsplitu N( R( R R& R R9 R' R7 R) R: Rm R Rn R* R4 t dictR Ro ( t devt
attributesR{ R- t smart_resultt fieldsR1 R? t fieldt
smart_attrt attrR~ Rw R@ ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt smart_attributes� sP
# .
$
,
i i c C@ sj t j j j � r" t | | | � St j j j � rD t | | | � St j j j � rf t | | | � Sd S( u�
Gather and return (averaged) IO stats.
.. versionadded:: 2016.3.0
.. versionchanged:: 2016.11.4
Added support for AIX
CLI Example:
.. code-block:: bash
salt '*' disk.iostat 1 5 disks=sda
N( R R R t is_linuxt
_iostat_linuxt
is_freebsdt _iostat_fbsdt is_aixt _iostat_aix( t intervalRP Rp ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt iostat� s c C@ s` g t | � D]4 } t t | � t | � j t j d � � � ^ q
} t t | | � � } | S( u�
Transpose collected data, average it, stomp it in dict using header
Use Decimals so we can properly calc & round, convert to float 'caus' we can't transmit Decimals over 0mq
u .01( R t floatt sumR* t quantizeR R R� ( t headert statst stat( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt
_iostats_dict
s Gc C@ s� | d k r! d j | | � } nH t | t j � rK d j | | | � } n d j | | d j | � � } g } t j t � } g } g } d } t t
d | d d �j � � } x� | D]� }
|
j d � s� q� n t
| � s� |
j � d } n x� |
t k r�t | t � }
|
s)|
d
j � r-Pn |
j � }
|
d
} g |
d D] } t j | � ^ qN}
t
|
� | k r�t
|
� } n | | j |
� q� Wq� Wi } | t
| � k r_| | } | d
| !} xt | j � D]f \ } }
t
|
d
� | k r�g |
D] } | | ^ q} g |
D] } | d
| !^ q%| | <q�q�Wt | | � | d <n x- | j � D] \ } }
t | |
� | | <qlW| S(
u[
Tested on FreeBSD, quite likely other BSD's only need small changes in cmd syntax
u iostat -xC -w {0} -c {1} u iostat -x -w {0} -c {1} {2}u i� u cmd.run_stdoutR� u quietu devicei i u sysN( R R Rh R t string_typesRk t collectionst defaultdictRi R9 R& R' R( R* R) R
t nextt isalnumR R RB Rz R� ( R� RP Rp t
iostat_cmdt sys_statst dev_statst
sys_headert
dev_headert h_lenR. R1 Rr t xR� t iostatsR� ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyR� sL "
&
+c C@ sM | d k r! d j | | � } nH t | t j � rK d j | | | � } n d j | | d j | � � } g } t j t � } g } g } t t
d | d d �j � � } xB| D]:} | j d � r+t
| � s� t | j � d � } n g t | � j � D] }
t j |
� ^ q } | j | � q� | j d � r� t
| � s_t | j � d � } n x� | t k r�t | t � } | s�| d
j � r�Pn | j � } | d
} g | d D] }
t j |
� ^ q�} | | j | � qbWq� q� Wi }
t
| � rt | | � |
d <n x- | j � D] \ } } t | | � |
| <q&W|
S( Nu iostat -x {0} {1} u iostat -xd {0} {1} {2}u u cmd.run_stdoutR� u quietu avg-cpu:i u Device:i u sys( R R Rh R R� Rk R� R� Ri R9 R& R' R( R* Rj R) R� R R RB R
R� R� Rz ( R� RP Rp R� R� R� R� R� R. R1 R� Rr R� R� ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyR� M sB "
.
& c C@ s6 t j d � | d k r. d j | | � } nH t | t j � rX d j | | | � } n d j d j | � | | � } i } d } g } d } d } t j t
� } xt d | � j � D]�}
|
s� |
j
d � s� |
j
d � r� q� n t j d |
� s�|
j d
� } | d j � } | d } | d } | d j � } | | j � k r� g | | <t | | � } | | j i � i | | | | <| | | | | d
<g | | | | d <q� q� n d
|
k r]|
j d
� }
|
d j � } |
d j � } | | | d j � k r�i | | d | <| | | d | d
<g | | d | d <q�q� |
j � }
g |
D] } t | � ^ qq} | | d | d j | � q� Wi } x | j � D]q \ } } i | | <xX | D]P } xG | j � D]9 } | | d
} | | d } t | | � | | | <q�Wq�Wq�W| S( u?
AIX support to gather and return (averaged) IO stats.
u DGM disk iostat entryu iostat -dD {0} {1} u iostat -dD {0} {1} {2}u u u cmd.runu Systemu -----------u \su :i i u fieldsu statsN( R# t debugR R Rh R R� Rk R� R� Ri R& R' R( R7 Rn R) R R* RB t lstripR Rz R� ( R� RP Rp R� R. t procnR� t disk_namet disk_modeR� R1 t dsk_compst
dsk_firstsR2 R� R� R� Rr t
list_modest modes( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyR� x sb
$%
!
#(; t __doc__t
__future__R R R t loggingR RJ R7 R� R t salt.extR t salt.ext.six.movesR t salt.utils.decoratorsR t salt.utils.decorators.patht salt.utils.patht salt.utils.platformt salt.exceptionsR t __func_alias__t getLoggert __name__R# R R R_ R t
HAS_HDPARMt
HAS_IOSTATR R R R3 R5 R6 t
decoratorsRA RI RM RD RS R
R^ t which_binRb t dependsR Rg Ry R� R� R� R� R� R� R� ( ( ( s5 /usr/lib/python2.7/site-packages/salt/modules/disk.pyt <module> sf
O : 1% ' >'*L7I 8 +
Mini Shell