Mini Shell
3
{)c_� � @ sd d dl Z d dlZd dlZd dlZd dlZejdd� d dlmZ ddl m
Z
ddl m Z ddl m
Z
mZmZmZmZ dd lmZmZ dd
lmZmZmZ dd lmZmZ dd lm Z dd
l m Z ddl m Z ddl!m"Z" ddl m#Z# ddl$m%Z% ddl&m'Z'm(Z( ddl)m*Z* ddl+m,Z, d dl-Z-e-j.d�Z/eefZ0ej1e*�G dd� de2��Z3G dd� de3e%e"�Z4dS )� N�BlockDevz2.0)r � )�
ActionList)� callbacks)� DeviceError�DeviceTreeError� StorageError�DuplicateUUIDError�InvalidMultideviceSelection)�ActionDestroyDevice�ActionDestroyFormat)� BTRFSDevice�NoDevice�PartitionDevice)�LVMLogicalVolumeDevice�LVMVolumeGroupDevice)�Tags)�formats)�lvm)�EventHandlerMixin)�util)�PopulatorMixin)�log_method_call�log_method_return)�SynchronizedMeta)�lvs_infoZblivetc @ sl e Zd ZdZdMdd�ZdNdd�Zdd� Zed d
� �Zed d � �Z dOdd�Z
dPdd�Z dQdd�Z edd� �Z
dd� Zdd� Zdd� Zd d � ZdRd d �Zd!d"� Zd#d$� Zd%d&� ZdSd'd(�ZdTd)d*�ZdUd+d,�ZdVd-d.�ZdWd/d0�ZdXd1d2�ZdYd3d4�Z dZd5d6�Z ed7d8� �Z ed9d:� �Z ed;d<� �Z ed=d>� �Z!ed?d@� �Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dS )[�DeviceTreeBasea� A quasi-tree that represents the devices in the system.
The tree contains a list of :class:`~.devices.StorageDevice` instances,
which does not necessarily reflect the actual state of the system's
devices. :class:`~.deviceaction.DeviceAction` is used to perform
modifications to the tree, except when initially populating the tree.
:class:`~.deviceaction.DeviceAction` instances are registered, possibly
causing the addition or removal of :class:`~.devices.StorageDevice`
instances to/from the tree. A :class:`~.deviceaction.DeviceAction`
is reversible up to the time its 'execute' method is called.
Only one action of any given type/object pair should exist for
any given device at any given time.
:class:`~.deviceaction.DeviceAction` instances can only be registered
for leaf devices, except for resize actions.
Nc C s | j ||� dS )z�
:keyword ignored_disks: ignored disks
:type ignored_disks: list
:keyword exclusive_disks: exclusive didks
:type exclusive_disks: list
N)�reset)�self�
ignored_disks�exclusive_disks� r! � /usr/lib/python3.6/devicetree.py�__init__K s zDeviceTreeBase.__init__c C sD g | _ t| j| jd�| _g | _tj� |p,g | _|p6g | _ i | _
dS )z� Reset the instance to its initial state.
:keyword ignored_disks: ignored disks
:type ignored_disks: list
:keyword exclusive_disks: exclusive didks
:type exclusive_disks: list
)ZaddfuncZ
removefuncN) �_devicesr �_register_action�_cancel_action�_actions�_hiddenr Zlvm_devices_resetr r �edd_dict)r r r r! r! r" r T s
zDeviceTreeBase.resetc sF g � � �fdd��dd� | j D �}d}x|D ]}|�|d�7 }q,W |S )Nc sd | � k}dd| | f }� j | � |r>|dd|d f 7 }n"x | jD ]}|�||d �7 }qFW |S )Nz%s%s
z z%s...
r )�append�children)�root�depthZabbreviate_subtree�sZchild)�done� show_subtreer! r" r0 m s
z,DeviceTreeBase.__str__.<locals>.show_subtreec S s g | ]}|j s|�qS r! )�parents)�.0�dr! r! r" �
<listcomp>x s z*DeviceTreeBase.__str__.<locals>.<listcomp>� r )r$ )r �rootsZtreer, r! )r/ r0 r" �__str__j s
zDeviceTreeBase.__str__c C s^ g }xT| j D ]J}t|dd�s q |jrL|jdd� |D �krLt|t� rLtd��|j|� q W |S )z' List of devices currently in the tree �completeTc S s g | ]
}|j �qS r! )�uuid)r2 r3 r! r! r" r4 � s z*DeviceTreeBase.devices.<locals>.<listcomp>z duplicate uuids in device tree)r$ �getattrr9 �
isinstancer r r* )r �devices�devicer! r! r" r< � s zDeviceTreeBase.devicesc s� t tjj� �}g � xL| j| j D ]<}|jdks<|jjd� r |jdkr |j� kr � j |j� q W dd� | j
j ddd�D ��� j � �fd d
�|D �� � S ) z List of devices names � partitionZreqz btrfs volumec S s g | ] }|j j�qS r! )r= �name)r2 Zacr! r! r" r4 � s z(DeviceTreeBase.names.<locals>.<listcomp>Zdestroyr= )Z action_typeZ object_typec 3 s" | ]}|� kr|�kr|V qd S )Nr! )r2 �n)�names�
removed_namesr! r" � <genexpr>� s z'DeviceTreeBase.names.<locals>.<genexpr>)
�listr �cache�keysr$ r( �typer? �
startswithr* �actions�find�extend)r Zlv_info�devr! )rA rB r" rA � s
zDeviceTreeBase.namesTc C s� |j rh|j dd� | jD �krht|t� rh| j|j ddd�}|j|jkrPtd��ntd|j |j|jf ��x |jD ]}|| jkrptd��qpW |j |d� | jj
|� t j |d � t
jd
|j|j|j� d S ) z� Add a device to the tree.
:param newdev: the device to add
:type newdev: a subclass of :class:`~.devices.StorageDevice`
Raise DeviceTreeError if the device's identifier is already
in the list.
c S s g | ]
}|j �qS r! )r9 )r2 r3 r! r! r" r4 � s z.DeviceTreeBase._add_device.<locals>.<listcomp>T)�
incomplete�hiddenz&Trying to add already existing device.z5Duplicate UUID '%s' found for devices: '%s' and '%s'.zparent device not in tree)�new)r= z"added %s %s (id %d) to device treeN)r9 r$ r; r �get_device_by_uuidr? r r r1 �add_hookr* r Z device_added�log�inforG �id)r ZnewdevrO rL �parentr! r! r" � _add_device� s
zDeviceTreeBase._add_devicec C s� || j krtd|j ��|j rV| rVtjd|jtjdd� |jD ��� td|j ��|j |d� |r�t
|t �r�|j dk r�x,| j D ]"}t
|t �r�|j |j kr�|j
� q�W | j j|� tj|d� tjd |j|j|j� dS )
a� Remove a device from the tree.
:param dev: the device to remove
:type dev: a subclass of :class:`~.devices.StorageDevice`
:keyword force: whether to force removal of a non-leaf device
:type force: bool
:keyword modparent: update parent device to account for removal
:type modparent: bool
.. note::
Only leaves may be removed.
zDevice '%s' not in treez%s has children %sc s s | ] }|j V qd S )N)r? )r2 �cr! r! r" rC � s z0DeviceTreeBase._remove_device.<locals>.<genexpr>z"Cannot remove non-leaf device '%s')� modparentN)r= z&removed %s %s (id %d) from device tree)r$ �
ValueErrorr? �isleafrR �debug�pprintZpformatr+ Z remove_hookr; r �diskZ update_name�remover Zdevice_removedrS rG rT )r rL �forcerX r= r! r! r" �_remove_device� s"
"
z DeviceTreeBase._remove_devicec C s0 t jd|j� | j|�}|j� x�|r�t jddd� |D �� dd� |D �}t jddd� |D �� xn|D ]f}|r�|jjr�|j r�|j r�| j j
t |�� | j j
t |�� n|js�d|_| j
||d � |j|� qfW q"W |js�|r�| j j
t |�� nd|_|�r,|j �r,|�r | j j
t |�� n| j
||d � dS )
a� Remove a device after removing its dependent devices.
:param :class:`~.devices.StorageDevice` device: the device to remove
:keyword bool actions: whether to schedule actions for the removal
:keyword bool modparent: whether to update parent device upon removal
:keyword bool remove_device: whether to remove the root device
If the device is not a leaf, all of its dependents are removed
recursively until it is a leaf device. At that point the device is
removed, unless it is a disk. If the device is a disk, its
formatting is removed but no attempt is made to actually remove the
disk device.
z removing %szdevices to remove: %sc S s g | ]
}|j �qS r! )r? )r2 r3 r! r! r" r4 s z3DeviceTreeBase.recursive_remove.<locals>.<listcomp>c S s g | ]}|j r|�qS r! )rZ )r2 r3 r! r! r" r4 s zleaves to remove: %sc S s g | ]
}|j �qS r! )r? )r2 r3 r! r! r" r4 s N)rX )rR r[ r? �get_dependent_devices�reverse�format�exists� protectedZformat_immutablerI �addr r r` r^ �is_disk)r r= rI Z
remove_devicerX r< �leavesZleafr! r! r" �recursive_remove� s2
z DeviceTreeBase.recursive_removec C s | j S )N)r' )r r! r! r" rI $ s zDeviceTreeBase.actionsc C s& |j o
|j r$|j| jkr$td��n |j rD|jrD|j| jkrDtd��|j r�|jr�t|jt�r�|jjr�x$|jjD ]}|| jkrn| j|� qnW | j |j� n�|j
r�|jr�| j|j� t|jt�r�|jjr�x\|jjD ]}| j |dd� q�W n<|j o�|j �r"t|jj t
jj��r"|jj j| jk�r"td��dS )a Register an action to be performed at a later time.
:param action: the action
:type action: :class:`~.deviceaction.DeviceAction`
Modifications to the Device instance are handled before we
get here.
zdevice is not in the treez device is already in the treeF)rO zmountpoint already in useN)� is_create� is_devicer= r$ r r; r �from_lvsr` rV �
is_destroyZ is_formatrc r ZfsZFS�
mountpoint� filesystems)r �action�lvr! r! r" r% ( s*
z DeviceTreeBase._register_actionc C s� |j rN|jrN| j|j� t|jt�r�|jjr�xv|jjD ]}| j|dd� q6W nV|jr�|jr�t|jt�r�|jjr�x$|jjD ]}|| j krx| j|� qxW | j|jdd� dS )ai Cancel a registered action.
:param action: the action
:type action: :class:`~.deviceaction.DeviceAction`
This will unregister the action and do any required
modifications to the device list.
Actions all operate on a Device, so we can use the devices
to determine dependencies.
F)rO N)
rj rk r` r= r; r rl rV rm r$ )r rp rq r! r! r" r& L s
z DeviceTreeBase._cancel_actionc C sf x`| j D ]V}|jrqy|jdd� W q ttjfk
r\ } ztjd|j|� W Y dd}~X qX qW dS )z& Run teardown methods on all devices. T)� recursivezteardown of %s failed: %sN) rh re Zteardownr �blockdevZ
BlockDevErrorrR rS r? )r r= �er! r! r" � teardown_allm s zDeviceTreeBase.teardown_allc C sT xN| j D ]D}y |j� W q tk
rJ } ztjd|j|� W Y dd}~X qX qW dS )z# Run setup methods on all devices. zsetup of %s failed: %sN)rh Zsetupr rR �errorr? )r r= rt r! r! r" � setup_allx s
zDeviceTreeBase.setup_allFc C s� g }t | ||d� |jr,| r,tjd� |S | jdd� }|rJ|j| j� x2|D ]*}tjd|j|j� |j|�rP|j |� qPW |S )a= Return a list of devices that depend on dep.
The list includes both direct and indirect dependents.
:param dep: the device whose dependents we are looking for
:type dep: :class:`~.devices.StorageDevice`
:keyword bool hidden: include hidden devices in search
)�deprN z
dep is a leafNz checking if %s depends on %s)
r rZ rR r[ r$ rK r( r? �
depends_onr* )r rx rN Z
dependentsr< r= r! r! r" ra � s
z$DeviceTreeBase.get_dependent_devicesc C s t dd� | j|dd�D ��S )a$ Return disks related to disk by container membership.
:param :class:`~.devices.StorageDevice` disk: the disk
:returns: related disks
:rtype: set of :class:`~.devices.StorageDevice`
.. note::
The disk may be hidden.
c s s | ]}|j D ]
}|V q qd S )N)�disks)r2 rx r3 r! r! r" rC � s z3DeviceTreeBase.get_related_disks.<locals>.<genexpr>T)rN )�setra )r r] r! r! r" �get_related_disks� s z DeviceTreeBase.get_related_disksc sR t � �x2| jD ](� t� fdd�|D ��r�j� jj� qW �fdd�| jD �}|S )a� Return a list of actions related to the specified disk.
:param disks: list of disks
:type disk: list of :class:`~.devices.StorageDevices`
:returns: list of related actions
:rtype: list of :class:`~.deviceaction.DeviceAction`
This includes all actions on the specified disks, plus all actions
on disks that are in any way connected to the specified disk via
container devices.
c 3 s | ]}� j j|�V qd S )N)r= ry )r2 r3 )rp r! r" rC � s z2DeviceTreeBase.get_disk_actions.<locals>.<genexpr>c s" g | ]}t |jj�j� �r|�qS r! )r{ r= rz � intersection)r2 �a)�
related_disksr! r" r4 � s z3DeviceTreeBase.get_disk_actions.<locals>.<listcomp>)r{ rI �any�updater= rz )r rz Zrelated_actionsr! )rp r r" �get_disk_actions� s z DeviceTreeBase.get_disk_actionsc C s, | j |�}x t|�D ]}| jj|� qW dS )a Cancel all actions related to the specified disk.
:param disks: list of disks
:type disk: list of :class:`~.devices.StorageDevices`
This includes actions related directly and indirectly (via container
membership, for example).
N)r� �reversedrI r^ )r rz rI rp r! r! r" �cancel_disk_actions� s
z"DeviceTreeBase.cancel_disk_actionsc C sZ |r,dd� | j dd� | jdd� D �}ndd� | j dd� D �}|sVdd� |D �}|S )a6 Return list of devices modified according to parameters.
:param bool incomplete: include incomplete devices in result
:param bool hidden: include hidden devices in result
:returns: a generator of devices
:rtype: generator of :class:`~.devices.Device`
c s s | ]
}|V qd S )Nr! )r2 r3 r! r! r" rC � s z1DeviceTreeBase._filter_devices.<locals>.<genexpr>Nc s s | ]
}|V qd S )Nr! )r2 r3 r! r! r" rC � s c s s | ]}t |d d�r|V qdS )r8 TN)r: )r2 r3 r! r! r" rC � s )r$ r( )r rM rN r< r! r! r" �_filter_devices� s (z DeviceTreeBase._filter_devicesc sN t | � ||d� d}� r@| j||d�}tj� fdd�|D �d�}t| |� |S )ab Return a list of devices with a matching sysfs path.
:param str path: the sysfs path to match
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)�pathrM rN N)rM rN c 3 s | ]}|j � kr|V qd S )N)Z
sysfs_path)r2 r3 )r� r! r" rC � s z:DeviceTreeBase.get_device_by_sysfs_path.<locals>.<genexpr>)r r� �six�nextr )r r� rM rN �resultr< r! )r� r" �get_device_by_sysfs_path� s
z'DeviceTreeBase.get_device_by_sysfs_pathc sN t | � ||d� d}� r@| j||d�}tj� fdd�|D �d�}t| |� |S )aV Return a list of devices with a matching UUID.
:param str uuid: the UUID to match
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)r9 rM rN N)rM rN c 3 s( | ] }|j � ks |jj � kr|V qd S )N)r9 rc )r2 r3 )r9 r! r" rC s z4DeviceTreeBase.get_device_by_uuid.<locals>.<genexpr>)r r� r� r� r )r r9 rM rN r� r< r! )r9 r" rP � s
z!DeviceTreeBase.get_device_by_uuidc sN t | � ||d� d}� r@| j||d�}tj� fdd�|D �d�}t| |� |S )af Return a device with a matching filesystem label.
:param str label: the filesystem label to match
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)�labelrM rN N)rM rN c 3 s$ | ] }t |jd d�� kr|V qdS )r� N)r: rc )r2 r3 )r� r! r" rC s z5DeviceTreeBase.get_device_by_label.<locals>.<genexpr>)r r� r� r� r )r r� rM rN r� r< r! )r� r" �get_device_by_label
s
z"DeviceTreeBase.get_device_by_labelc sN t | � ||d� d}� r@| j||d�}tj� fdd�|D �d�}t| |� |S )aP Return a device with a matching name.
:param str name: the name to look for
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)r? rM rN N)rM rN c 3 s8 | ]0}|j � ks,t|t�r|j � jd d�kr|V qdS )z--�-N)r? r; �_LVM_DEVICE_CLASSES�replace)r2 r3 )r? r! r" rC + s z4DeviceTreeBase.get_device_by_name.<locals>.<genexpr>)r r� r� r� r )r r? rM rN r� r< r! )r? r" �get_device_by_name s
z!DeviceTreeBase.get_device_by_namec sV t | � ||d� d}� rH| j||d�}tj� fdd�tt|��D �d�}t| |� |S )a� Return a device with a matching path.
If there is more than one device with a matching path,
prefer a leaf device to a non-leaf device.
:param str path: the path to match
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)r� rM rN N)rM rN c 3 s8 | ]0}|j � ks,t|t�r|j � jd d�kr|V qdS )z--r� N)r� r; r� r� )r2 r3 )r� r! r" rC E s z4DeviceTreeBase.get_device_by_path.<locals>.<genexpr>)r r� r� r� r� rD r )r r� rM rN r� r< r! )r� r" �get_device_by_path1 s
z!DeviceTreeBase.get_device_by_pathc sF t | � ||d� | j||d�}tj� fdd�|D �d�}t| |� |S )aT Return a device with specified device id.
:param int id_num: the id to look for
:param bool incomplete: include incomplete devices in search
:param bool hidden: include hidden devices in search
:returns: the first matching device found
:rtype: :class:`~.devices.Device`
)�id_numrM rN )rM rN c 3 s | ]}|j � kr|V qd S )N)rT )r2 r3 )r� r! r" rC W s z2DeviceTreeBase.get_device_by_id.<locals>.<genexpr>N)r r� r� r� r )r r� rM rN r< r� r! )r� r" �get_device_by_idL s
z DeviceTreeBase.get_device_by_idc / C s� d}|j d�r`|jd�d }|j d�r0|jd�sD|j d�rP|jd�rP|dd*� }| jj|�}�n~|j d�r�|jd�d }|j d�r�|jd�s�|j d�r�|jd�r�|dd+� }| jj|�}�n"|r�d |jd
�kr�| j|�}|s�| j|�}�n�t j
d |��rF|jd �\}} }
t |d
�} x2| j j
� D ]$\} }
|
| k�r | j| |
�}P �q W | �rt|j d� �rt| j|�}|�std| }|�sv|j d��r�tjj|�}|j d��r�ytjj|dd� �}W n8 tjk
�r� } ztjd||� d}W Y dd}~X nX |�r�d| }t j
d|��rlytjj|dd� �}W n8 tjk
�r\ } ztjd||� d}W Y dd}~X nX |�rld| }| j|�}|dk�r�|�r |j|�}|�r tjd|� |jd�}|�r�| j|�}|�r�|j}nd}tjd|� |�r |j �r |j j dk�r |j j }tjd|� | j|�}|�r |}|dk�r�|�r�|j d��r�|j|jd �d, �}|�r�|d }y|j d }W n$ t k
�r� } zW Y dd}~X nX nL|dk�r�|dd� }|jd �\}}}|�r�d |k�r�d ||f }| j|�}|�r�|j j d ��r�|�r�t!|d!|�}d} d"|k�r$d#} t"j#d$|�} n*d%|k�r@d&} t"j#d'|�} n|j$�rN|j$}| �r�| �r�x(|j%D ] } t!| | d�| k�rb| }P �qbW |�r�tjd(||j|j � n tjd)|� |S )-a� Return the device matching the provided device specification.
The spec can be anything from a device name (eg: 'sda3') to a device
node path (eg: '/dev/mapper/fedora-root' or '/dev/dm-2') to
something like 'UUID=xyz-tuv-qrs' or 'LABEL=rootfs'.
:param devspec: a string describing a block device
:type devspec: str
:keyword blkid_tab: blkid info
:type blkid_tab: :class:`~.BlkidTab`
:keyword crypt_tab: crypto info
:type crypt_tab: :class:`~.CryptTab`
:keyword options: mount options
:type options: str
:returns: the device
:rtype: :class:`~.devices.StorageDevice` or None
NzUUID=�=� �"�'r zLABEL=Znodev�,z(0x)?[A-Fa-f0-9]{2}(p\d+)?$�p� z/dev/z
/dev/disk/z/dev/dm-� zfailed to resolve %s: %sz /dev/mapper/z/dev/md\d+(p\d+)?$z/dev/md/z found blkid.tab entry for '%s'ZUUID�Nonezfound device '%s' in treeZluksz luks device; map name is '%s'�/r= r z%s-%s�btrfsZvolumezsubvol=r? �subvolz subvolid=Zvol_idZsubvolidzresolved '%s' to '%s' (%s)zfailed to resolve '%s'���r� r� )&rH r> �endswith�uuids�get�labels�splitr� r� �re�match�intr) �items�osr� �realpathrs ZdmZname_from_nodeZDMErrorrR rS ZmdZ MDRaidErrorr[ rP r? rc rG �map_namer+ �
IndexErrorr: r Zget_option_valueZdefault_subvolumeZ
subvolumes) r ZdevspecZ blkid_tabZ crypt_tabZoptionsr= r9 r� ZdriveZ_pZpartnum�specZedd_nameZ
edd_numberZdm_namert Zmd_nameZ
blkid_tab_entZdevstrr� Z
mapped_devZ
crypt_tab_entZluks_devr? Zvg_nameZ_slashZlv_namerq �attr�valr� r! r! r" �resolve_device[ s�
z DeviceTreeBase.resolve_devicec C s dd� | j D �}|S )z8 List of all devices upon which no other devices exist. c S s g | ]}|j r|�qS r! )rZ )r2 r3 r! r! r" r4 � s z)DeviceTreeBase.leaves.<locals>.<listcomp>)r$ )r rh r! r! r" rh � s zDeviceTreeBase.leavesc C s8 g }x.| j D ]$}|jr t|jdd�r |j|j� q W |S )z List of filesystems. rn N)rh rc r: r* )r ro rL r! r! r" ro � s
zDeviceTreeBase.filesystemsc C sz i }xp| j D ]f}y
|j}W n tk
r2 d}Y nX |r@|||< y |jj}W n tk
rd d}Y nX |r |||< q W |S )z; Dict with uuid keys and :class:`~.devices.Device` values. N)r$ r9 �AttributeErrorrc )r r� rL r9 r! r! r" r� s
zDeviceTreeBase.uuidsc C sH i }x>| j D ]4}t|jdd�r |jjdks4t|t�r |||jj< q W |S )zh Dict with label keys and Device values.
FIXME: duplicate labels are a possibility
r� Nr� )r$ r: rc rG r; r
r� )r r� rL r! r! r" r� s zDeviceTreeBase.labelsc C s4 i }x*| j D ] }|jjr |jjr |||jj< q W |S )z. Dict with mountpoint keys and Device values. )r< rc Z mountablern )r ro r= r! r! r" � mountpoints. s
zDeviceTreeBase.mountpointsc C s� || j krdS |jr | j|g� x|jD ]}| j|� q(W tjd|� |jsPdS | j|ddd� | j j |� |j
j dkr�t j
|j� dS )a� Hide the specified device.
:param device: the device to hide
:type device: :class:`~.devices.StorageDevice`
Hiding a device will cancel all actions that involve the device and
will remove the device from the device list.
If the device is not a leaf device, all devices that depend on it
will be hidden leaves-first until the device is a leaf device.
If a device exists, performs some special actions and places
it on a list of hidden devices.
Mixes recursion and side effects, most significantly in the code
that removes all the actions. However, this code is a null op
in every case except the first base case that is reached,
where all actions are removed. This means that when a device
is removed explicitly in this function by means of a direct call to
_remove_devices it is guaranteed that all actions have already
been canceled.
If a device does not exist then it must have been removed by the
cancelation of all the actions, so it does not need to be removed
explicitly.
Most devices are considered leaf devices if they have no children,
however, some devices must satisfy more stringent requirements.
_remove_device() will raise an exception if the device it is
removing is not a leaf device. hide() guarantees that any
device that it removes will have no children, but it does not
guarantee that the more stringent requirements will be enforced.
Therefore, _remove_device() is invoked with the force parameter
set to True, to skip the isleaf check.
Nzhiding device %sTF)r_ rX �lvmpv)r( rg r� r+ �hiderR rS rd r` r* rc rG r Zlvm_devices_remover� )r r= r3 r! r! r" r� : s $
zDeviceTreeBase.hidec s� x�t � j�D ]�}||ks<|j|�r t� fdd�|jD �� r tjd|j|j|j � � jj
|� � j j |� |j
dd� |jjdkr tj|j� q W dS )a! Restore a device's visibility.
:param device: the device to restore/unhide
:type device: :class:`~.devices.StorageDevice`
.. note::
Actions canceled while hiding the device are not rescheduled
automatically.
c 3 s | ]}|� j kV qd S )N)r( )r2 rU )r r! r" rC � s z(DeviceTreeBase.unhide.<locals>.<genexpr>z unhiding device %s %s (id %d)F)rO r� N)r� r( ry r� r1 rR rS rG r? rT r^ r$ r* rQ rc r Zlvm_devices_addr� )r r= rN r! )r r" �unhideu s
zDeviceTreeBase.unhidec C s� t � }| jdd� }xh|D ]`}|jd�rp|dd� }|tjkrJtd| ��x.|D ]}||jkrP|j|j� qPW q|j|� qW |S )a9 Expands tags in input list into devices.
:param taglist: list of strings
:returns: set of devices
:rtype: set of strings
Raise ValueError if unknown tag encountered in taglist
.. note::
Returns empty set if taglist is empty
N�@r z unknown tag '@%s' encountered) r{ r$ rH r � __members__rY �tagsrf r? )r �taglistr� r< �item�tagr= r! r! r" �expand_taglist� s
z DeviceTreeBase.expand_taglistc C sV |j |krdS dd� |D �}x4|D ],}|tjkr<td| ��t|�|jkr"dS q"W dS )NTc S s" g | ]}|j d �r|dd� �qS )r� r N)rH )r2 �tr! r! r" r4 � s z3DeviceTreeBase._disk_in_taglist.<locals>.<listcomp>z(unknown ignoredisk tag '@%s' encounteredF)r? r r� rY r� )r r] r� r� r� r! r! r" �_disk_in_taglist� s
z DeviceTreeBase._disk_in_taglistc C s* | j r| j|| j �p(| jo(| j|| j� S )z{ Checks config for lists of exclusive and ignored disks
and returns if the given one should be ignored
)r r� r )r r] r! r! r" �_is_ignored_disk� s z DeviceTreeBase._is_ignored_diskc s2 �x*dd� � j D �D �]}� jo,� j|� j�}� jo@� j|� j�}|r�t|j�dkr�t� fdd�|jd jD ��sztd��� j |jd � x|jD ]}� j |� q�W � j |� � jo�| rd}|jr�td d� |jD ��r�t
� fd
d�|jD ��}n*|j j �r t|j�dk�r � j
|jd �}|r� j |� qW d S ) Nc S s g | ]}|j r|�qS r! )rg )r2 r3 r! r! r" r4 � s z6DeviceTreeBase._hide_ignored_disks.<locals>.<listcomp>r c 3 s | ]}� j |�V qd S )N)r� )r2 r3 )r r! r" rC � s z5DeviceTreeBase._hide_ignored_disks.<locals>.<genexpr>r zFIncluding only a subset of raid/multipath member disks is not allowed.Tc s s | ]}|j jV qd S )N)rc rN )r2 r� r! r! r" rC � s c 3 s | ]}� j |�V qd S )N)r� )r2 r3 )r r! r" rC � s )r$ r r� r �lenr+ �allr1 r
r� r� rc rN r� )r r] Z
is_ignoredZ is_exclusiver� Zignoredr! )r r" �_hide_ignored_disks� s&
z"DeviceTreeBase._hide_ignored_disks)NN)NN)T)NT)TTT)F)FF)FF)FF)FF)FF)FF)FF)NNN))�__name__�
__module__� __qualname__�__doc__r# r r7 �propertyr< rA rV r` ri rI r% r& ru rw ra r| r� r� r� r� rP r� r� r� r� r� rh ro r� r� r� r� r� r� r� r� r� r! r! r! r" r 7 sL
'
8 $!
;
r c @ s e Zd Zddd�Zddd�ZdS )�
DeviceTreeNc C s, t j| ||d� tj| |d� tj| � d S )N)r r )� disk_images)r r# r r )r r r r� r! r! r" r# � s zDeviceTree.__init__c C s" t j| ||d� tj| |d� d S )N)r r )r� )r r r )r r r r� r! r! r" r � s zDeviceTree.reset)NNN)NNN)r� r� r� r# r r! r! r! r" r� � s
r� )5r� r\ r� r� ZgiZrequire_versionZ
gi.repositoryr rs Z
actionlistr r �errorsr r r r r
Z deviceactionr r r< r
r r r r Z devices.libr r5 r Z
devicelibsr Zevents.handlerr r Z populatorr Z storage_logr r Zthreadsr Z static_datar ZloggingZ getLoggerrR r� Z
add_metaclass�objectr r� r! r! r! r" �<module> sD
4