Mini Shell
3
{)c�( � @ s� d dl Z d dlZddlmZ ddlmZ ddlmZmZmZ ddlm Z m
Z
ddl m Z ddl
mZ dd lmZ d
d lmZ d
d lmZmZmZ d
d
lmZ d
dlmZ d
dlm Z dd� Z G dd� de �Z dS )� N� )�Blivet)� callbacks)�DEVICE_TYPE_PARTITION�DEVICE_TYPE_LVM�DEVICE_TYPE_LVM_THINP)�DEVICE_TYPE_MD�DEVICE_TYPE_BTRFS)� StorageError)�Size)�ObjectID� )�
DBusAction)�BLIVET_INTERFACE�BLIVET_OBJECT_PATH�BUS_NAME)�
DBusDevice)�
DBusFormat)�
DBusObjectc C s t | dd� d�}dd� |D �S )Nc S s | j S )N)�id)�o� r � /usr/lib/python3.6/blivet.py�<lambda>' s z/sorted_object_paths_from_list.<locals>.<lambda>)�keyc S s g | ]
}|j �qS r )� object_path)�.0r r r r �
<listcomp>( s z1sorted_object_paths_from_list.<locals>.<listcomp>)�sorted)Zobj_list�objectsr r r � sorted_object_paths_from_list& s r c sf e Zd ZdZ� fdd�Zdd� Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
d:dd�Z dd� Z d;dd�Z
dd� Zdd� Zdd� Zdd � Zd<d d �Zd=d d!�Zejjed"�d#d$� �Zejjed"�d%d&� �Zejjed'd(�d)d*� �Zejjed+d,d-�d.d/� �Zejjed,d0�d1d2� �Z ejjed,d0�d3d4� �Z ejjed"�d5d6� �Z ejjed7d,d-�d8d9� �Z � Z S )>�
DBusBlivetz� This class provides the main entry point to the Blivet1 service.
It provides methods for controlling the blivet service and querying its
state.
c s: t t| �j|� t� | _t� j| _| jj | � | j
� d S )N) �superr! �__init__r �_blivetr r �_id�_manager�
add_object�_set_up_callbacks)�selfZmanager)� __class__r r r# 1 s
zDBusBlivet.__init__c C sf t jj| j� t jj| j� t jj| j� t jj| j � t j
j| j � t j j| j
� t jj| j� d S )N)r Z device_added�add�
_device_addedZdevice_removed�_device_removedZ format_added�
_format_addedZformat_removed�_format_removedZ action_added�
_action_addedZaction_removed�_action_removedZaction_executed�_action_executed)r) r r r r( 8 s z DBusBlivet._set_up_callbacksc C s | j S )N)r% )r) r r r r A s z
DBusBlivet.idc C s t S )N)r )r) r r r r E s zDBusBlivet.object_pathc C s t S )N)r )r) r r r � interfaceI s zDBusBlivet.interfacec C s | j � tttttd�}|S )N)ZDevicesr r r r r )� ListDevicesr r r r r )r) Zpropsr r r �
propertiesM s zDBusBlivet.propertiesTc C s~ | j j|jj�}|r.|jr.| j||j|d� n |rJ| rJ| j||jdd� | j j|j�}| j j|� |rzd|_| j j|� dS )z; Update ObjectManager interface after a device is removed. )�keepFN)r& �get_object_by_id�formatr �presentr/ �
remove_objectr' )r) �devicer6 Z removed_fmt�removedr r r r- W s
zDBusBlivet._device_removedc C s6 | j j|j�}|rd|_n t|| j �}| j j|� dS )z9 Update ObjectManager interface after a device is added. TN)r& r7 r r9 r r' )r) r; �addedr r r r, g s
zDBusBlivet._device_addedc C s@ | j j|j�}|d krd S | j j|� |r<d|_| j j|� d S )NF)r& r7 r r: r9 r' )r) r; �fmtr6 r< r r r r/ r s zDBusBlivet._format_removedc C s6 | j j|j�}|rd|_n t|| j �}| j j|� d S )NT)r& r7 r r9 r r' )r) r; r> r= r r r r. } s
zDBusBlivet._format_addedc C s | j j|j�}| j j|� d S )N)r& r7 r r: )r) �actionr< r r r r1 � s zDBusBlivet._action_removedc C s t || j�}| jj|� d S )N)r r& r' )r) r? r= r r r r0 � s zDBusBlivet._action_addedc C sF |j r8|jr | j|jdd� n|jr8| j|j|jdd� | j|� d S )NF)r6 )Z
is_destroyZ is_devicer- r; Z is_formatr/ r8 r1 )r) r? r r r r2 � s zDBusBlivet._action_executedFc s$ dd� | j jD �}� fdd�|D �S )Nc s s | ]}t |t�r|V qd S )N)�
isinstancer )r �dr r r � <genexpr>� s z0DBusBlivet._list_dbus_devices.<locals>.<genexpr>c s g | ]}� s|j r|�qS r )r9 )r rA )r< r r r � s z1DBusBlivet._list_dbus_devices.<locals>.<listcomp>)r& r )r) r< Z dbus_devicesr )r< r �_list_dbus_devices� s z DBusBlivet._list_dbus_devicesc C s` | j j|�}|dks t|t� r6tjjdt d| ��|j rZ| rZtjjdt d| ��|j S )z; Return the StorageDevice corresponding to an object path. Nz%s.DeviceNotFoundz&No device found with object path "%s".z6Device with object path "%s" has already been removed.)
r& Zget_object_by_pathr@ r �dbus�
exceptions�
DBusExceptionr r9 Z_device)r) r r< � dbus_devicer r r �_get_device_by_object_path� s z%DBusBlivet._get_device_by_object_path)�dbus_interfacec C sX | j jdd� }x|D ]}| j|dd� qW x | j jjD ]}| j|� q8W | j j� dS )z9 Reset the Blivet instance and populate the device tree. NF)r; r6 )r$ Zdevicesr- �
devicetreeZactionsr1 �reset)r) Z old_devicesr< r? r r r �Reset� s
zDBusBlivet.Resetc C s t jd� dS )z Stop the blivet service. r N)�sys�exit)r) r r r �Exit� s zDBusBlivet.ExitZao)rI �
out_signaturec C s t | j� �}tj|dd�S )zA Return a list of strings describing the devices in this system. r )Z signature)r rC rD ZArray)r) Z object_pathsr r r r4 � s zDBusBlivet.ListDevices�sr )rI � in_signaturerP c C s< | j jj|�}|dkr,tjjdt d| ��| jj|j �j
S )zH Return a string describing the device the given specifier resolves to. Nz%s.DeviceLookupFailedz<No device was found that matches the device descriptor "%s".) r$ rJ Zresolve_devicerD rE rF r r& r7 r r )r) �specr; r r r �
ResolveDevice� s zDBusBlivet.ResolveDevice)rI rR c C s | j |�}| jjj|� dS )z. Remove a device and all devices built on it. N)rH r$ rJ Zrecursive_remove)r) r r; r r r � RemoveDevice� s
zDBusBlivet.RemoveDevicec C s$ | j |� | j|�}| jj|� dS )z, Clear a disk and create a disklabel on it. N)rU rH r$ Zinitialize_disk)r) r r; r r r �InitializeDisk� s
zDBusBlivet.InitializeDiskc C sX y| j j� W nD tk
rR } z(tjjdt|jjf dt |� ��W Y dd}~X nX dS )z! Commit pending changes to disk. z%s.%sz9An error occured while committing the changes to disk: %sN)
r$ Zdo_itr
rD rE rF r r* �__name__�str)r) �er r r �Commit� s zDBusBlivet.Commitza{sv}c s� � fdd�|j dg �D �}||d< |j dd �}|rD� j|�}||d< |j dd �}|d k rdt|�|d< y� jjf |�}W nD tk
r� } z(tjjdt |j
j f dt |� ��W Y d d }~X nX |d kr�d}n� j
j|j�j}|S ) Nc s g | ]}� j |��qS r )rH )r �p)r) r r r � s z&DBusBlivet.Factory.<locals>.<listcomp>�disksr; �sizez%s.%sz1An error occured while configuring the device: %s�/)�poprH r r$ Zfactory_devicer
rD rE rF r r* rW rX r& r7 r r )r) �kwargsr\ rG r; r] rY r r )r) r �Factory� s&
zDBusBlivet.Factory)T)T)F)F)!rW �
__module__� __qualname__�__doc__r# r( �propertyr r r3 r5 r- r, r/ r. r1 r0 r2 rC rH rD Zservice�methodr rL rO r4 rT rU rV rZ ra �
__classcell__r r )r* r r! + s0
r! ) rM rD � r r Z
devicefactoryr r r r r �errorsr
r] r �utilr r? r Z constantsr r r r; r r8 r �objectr r r! r r r r �<module> s