Mini Shell
3
{)c}% � @ s� d dl Z d dlZejd�ZddlmZ ddlmZ ddlm Z ddl
m Z m Z dd l
mZ dd
lmZ dd lmZ G d d
� d
e�ZG dd� de�ZdS )� NZblivet� )�
StorageDevice� )� stratis_info)�log_method_call)� DeviceError� StratisError)�Size)� availability)�
devicelibsc s( e Zd ZdZdZdZddgZdZdZe j
e j gZ � fdd �Z
ed
d � �Zed d
� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Zejdd� �Zedd � �Zd d � Zeed �Zed d!� �Z� fd"d#�Z d$d%� Z � fd&d'�Z d(d)� Z d0� fd*d+� Z d1� fd,d-� Z!d.d/� Z"� Z#S )2�StratisPoolDevicez A stratis pool device z stratis poolF�stratisdz stratis-cliz /dev/stratisTc s@ |j dd�| _|j dd�| _|j dd�| _tt| �j||� dS )a
:encrypted: whether this pool is encrypted or not
:type encrypted: bool
:keyword passphrase: device passphrase
:type passphrase: str
:keyword key_file: path to a file containing a key
:type key_file: str
� encryptedF�
passphraseN�key_file)�pop�
_encrypted� _StratisPoolDevice__passphrase� _key_file�superr �__init__)�self�args�kwargs)� __class__� � /usr/lib/python3.6/stratis.pyr . s zStratisPoolDevice.__init__c C s | j dd� S )z# A list of this pool block devices N)�parents)r r r r � blockdevs= s zStratisPoolDevice.blockdevsc C s | j dd� S )z' A list of this pool block filesystems N)Zchildren)r r r r � filesystemsB s z StratisPoolDevice.filesystemsc C s t dd� | jD ��S )z The size of this pool c s s | ] }|j V qd S )N)�size)�.0�parentr r r � <genexpr>K s z)StratisPoolDevice.size.<locals>.<genexpr>)�sumr )r r r r r G s zStratisPoolDevice.sizec C s4 | j r*tj| j�}|s$td| j ��|jS | jS d S )Nz'Failed to get information about pool %s)�existsr �
get_pool_info�namer Z
physical_sizer )r � pool_infor r r �_physical_sizeM s z StratisPoolDevice._physical_sizec C s t jjdd� | jD �| j�S )Nc S s g | ]
}|j �qS r )r )r! �bdr r r �
<listcomp>Y s z9StratisPoolDevice._pool_metadata_size.<locals>.<listcomp>)r �stratisZ pool_usedr r )r r r r �_pool_metadata_sizeW s z%StratisPoolDevice._pool_metadata_sizec C s0 t d�}x| jD ]}||j7 }qW || j7 }|S )Nr )r r � used_sizer- )r Z
physical_usedZ
filesystemr r r �_physical_used\ s
z StratisPoolDevice._physical_usedc C s | j | j S )z3 Free space in the pool usable for new filesystems )r) r/ )r r r r �
free_spacei s z StratisPoolDevice.free_spacec C s | j S )z# True if this device is encrypted. )r )r r r r r n s zStratisPoolDevice.encryptedc C s
|| _ d S )N)r )r r r r r r s s c C s | j S )z. Path to key file to be used in /etc/crypttab )r )r r r r r w s zStratisPoolDevice.key_filec C s
|| _ dS )z0 Set the passphrase used to access this device. N)r )r r r r r �_set_passphrase| s z!StratisPoolDevice._set_passphrase)�fsetc C s | j dkp | jo tj| jtj�S )N� )r3 N)r r �os�access�R_OK)r r r r �has_key� s
zStratisPoolDevice.has_keyc s( t t| �j� | jr$| j r$td��d S )Nz0cannot create encrypted stratis pool without key)r r � _pre_creater r7 r )r )r r r r8 � s z StratisPoolDevice._pre_createc C sD t | | j| jd� dd� | jD �}tjj| j|| j| j| j d� dS )z Create the device. )�statusc S s g | ]
}|j �qS r )�path)r! r* r r r r+ � s z-StratisPoolDevice._create.<locals>.<listcomp>)r' Zdevicesr r r N)
r r' r9 r r r, Z create_poolr r r )r Zbd_listr r r �_create� s
zStratisPoolDevice._createc sd t t| �j� d| j_tj| j�}|s4td| j ��|j | _ x"| j
D ]}| j|j_ | j |j_ qDW d S )NTz5Failed to get information about newly created pool %s)
r r � _post_create�formatr% r r& r' r �uuidr � pool_name� pool_uuid)r r( r" )r r r r< � s
z StratisPoolDevice._post_createc C s$ t | | j| jd� tjj| j� dS )z Destroy the device. )r9 N)r r' r9 r r, Z remove_poolr> )r r r r �_destroy� s zStratisPoolDevice._destroyc sB t t| �j|d� |rd S x"| jD ]}| j|j_| j|j_q"W d S )N)�new) r r �add_hookr r' r= r? r> r@ )r rB r" )r r r rC � s
zStratisPoolDevice.add_hookc s: |r$x | j D ]}d |j_d |j_q W tt| �j|d� d S )N)� modparent)r r= r? r@ r r � remove_hook)r rD r" )r r r rE � s
z StratisPoolDevice.remove_hookc C s t d| j g�S )Nzstratis.rootfs.pool_uuid=%s)�setr> )r r r r �dracut_setup_args� s z#StratisPoolDevice.dracut_setup_args)T)T)$�__name__�
__module__� __qualname__�__doc__�_type�
_resizable� _packages�_dev_dirZ_format_immutabler
�STRATISPREDICTUSAGE_APP� STRATIS_DBUS�_external_dependenciesr �propertyr r r r) r- r/ r0 r �setterr r1 r r7 r8 r; r< rA rC rE rG �
__classcell__r r )r r r $ s8
r c s� e Zd ZdZdZdZddgZdZej ej
gZ e d�Z
d � fd d
� Z� fd d �Zed
d� �Zedd� �Zedd� �Z� fdd�Zdd� Z� fdd�Zdd� Zdd � Z� ZS ) �StratisFilesystemDevicez A stratis pool device zstratis filesystemFr
z stratis-cliz /dev/stratisz512 MiBNc sB | r$|d j tjj|�kr$td��tt| �j|||||d� d S )Nr zGcannot create new stratis filesystem, not enough free space in the pool)r' r r> r r% )r0 r r, �filesystem_md_sizer r rV r )r r' r r r> r% )r r r r � s z StratisFilesystemDevice.__init__c s. | j dk r d| j j| jf S tt| �j� S dS )z This device's name. Nz%s/%s)�poolr' �_namer rV � _get_name)r )r r r rZ � s
z!StratisFilesystemDevice._get_namec C s | j S )z8 The Stratis filesystem name (not including pool name). )rY )r r r r �fsname� s z StratisFilesystemDevice.fsnamec C s | j s
d S | j d S )Nr )r )r r r r rX � s z StratisFilesystemDevice.poolc C sB | j stjj| j�S tj| jj| j �}|s8t
d| j ��|j S dS )z* Size used by this filesystem in the pool z-Failed to get information about filesystem %sN) r% r r, rW r r �get_filesystem_inforX r' r[ r r. )r �fs_infor r r r. � s z!StratisFilesystemDevice.used_sizec sd t | | j| j| j|d� t|t�s*td��| jsPtj j
|�}|| j j krPt
d��tt| �j|� d S )N)r9 r �newsizeznew size must of type Sizez not enough free space in pool)r r' r9 Z_size�
isinstancer �
ValueErrorr% r r, rW rX r0 r r rV � _set_size)r r^ Zmd_size)r r r ra � s
z!StratisFilesystemDevice._set_sizec C s0 t | | j| jd� tjj| j| jj| j d� dS )z Create the device. )r9 )r' r@ Zfs_sizeN)
r r' r9 r r, Zcreate_filesystemr[ rX r> r )r r r r r; � s z StratisFilesystemDevice._createc sH t t| �j� tj| jj| j�}|s2td| j ��|j | _ |j
| j _
d S )Nz;Failed to get information about newly created filesystem %s) r rV r< r r\ rX r' r[ r r> r@ r= )r r] )r r r r< s z$StratisFilesystemDevice._post_createc C s* t | | j| jd� tjj| jj| j� dS )z Destroy the device. )r9 N)r r' r9 r r, Zremove_filesystemrX r> )r r r r rA s z StratisFilesystemDevice._destroyc C s t d| j g�S )Nzroot=%s)rF r: )r r r r rG s z)StratisFilesystemDevice.dracut_setup_args)NNNF)rH rI rJ rK rL rM rN rO r
rP rQ rR r Z _min_sizer rZ rS r[ rX r. ra r; r< rA rG rU r r )r r rV � s"
rV )r4 ZloggingZ getLogger�logZstorager Z static_datar Z storage_logr �errorsr r r r Ztasksr
r3 r r rV r r r r �<module> s