Mini Shell
3
{)c( � @ s d dl Z e jdd� e jdd� d dlmZmZ d dlZd dlZddlmZ ddl m
Z
dd l m Z dd
l
mZ dd lmZ dd lmZ d
ZdZed Zed Zed Zed Ze
d�ZdZd&dd�Zdd� Zdd� Z d d � Z d d � Z d d!� Z d"d#� Z d'd$d%�Z!dS )(� N�GLibz2.0�Gio)r r � )� StratisError)�Size)� stratis_info)� availability)� safe_dbus)�utilzorg.storage.stratis3z/org/storage/stratis3z.pool.r0z.filesystem.r0z .blockdev.r0z .Manager.r0z1 TiBz 0-9a-zA-Z._-Fc C s� t jjstdt jj ��t jjdg}x"| D ]}|jdt|j� �g� q*W |rV|jd� t j
|�\}}|rptd��yt j |�}W n. t j
k
r� } ztd�|�W Y d d }~X nX t|d �S )Nz<Utility for predicting stratis pool usage '%s' not availableZpoolz
--device-sizez --encryptedz(Failed to predict usage for stratis poolz Failed to get stratis pool usage�used)r �STRATISPREDICTUSAGE_APP� availabler �name�extend�str� get_bytes�appendr
� run_program_and_capture_output�json�loads�JSONDecodeErrorr )Z dev_sizes� encrypted�cmd�size�rc�out�pred�e� r � /usr/lib/python3.6/stratis.py� pool_used5 s
r c C s� t jjstdt jj ��tjt jjddt| j� �g�\}}|rJtd| ��yt j
|�}W n. t j k
r� } ztd�|�W Y d d }~X nX t |d �S )Nz<Utility for predicting stratis pool usage '%s' not availableZ
filesystemz--filesystem-sizez2Failed to predict usage for stratis filesystem: %sz&Failed to get stratis filesystem usager )
r r r
r r r
r r r r r r r )�fs_sizer r r r r r r �filesystem_md_sizeK s r" c C s� t jjstd��tj� | tjj� kr2td| ��tj| }y,tj t
t t dt
jd|jf��\\}}}}W n4 tjk
r� } ztdt|� ��W Y d d }~X nX |s�td||f ��d S )Nz"Stratis DBus service not availablez#Stratis pool with UUID %s not foundZ DestroyPoolz(o)z!Failed to remove stratis pool: %sz&Failed to remove stratis pool: %s (%d))r � STRATIS_DBUSr
r r �
drop_cache�pools�keysr � call_sync�STRATIS_SERVICE� STRATIS_PATH�STRATIS_MANAGER_INTFr �Variant� object_path�
DBusCallErrorr )� pool_uuid� pool_info�succ�_uuidr �errr r r r � remove_pool] s
"r3 c C s� t jjstd��tj� | tjj� kr2td| ��|tjj� krLtd| ��tj| }tj| }y0t j
t |j t
dtjd|j gf��\\}}}}W n4 t jk
r� } ztdt|� ��W Y d d }~X nX |s�td||f ��d S )Nz"Stratis DBus service not availablez#Stratis pool with UUID %s not foundz)Stratis filesystem with UUID %s not foundZDestroyFilesystemsz(ao)z'Failed to remove stratis filesystem: %sz,Failed to remove stratis filesystem: %s (%d))r r# r
r r r$ r% r& Z filesystemsr r' r( r, �STRATIS_POOL_INTFr r+ r- r ) r. Zfs_uuidr/ Zfs_infor0 r1 r r2 r r r r �remove_filesystemv s&
""r5 c C s� |r(t j� \}}t j||jd�� |}n|r:t j|t j�}tj� }|j|� z�y0t j
t t t
dtjd| df�|d�\\}}} }
W n4 t jk
r� } ztdt| � ��W Y d d } ~ X nX | dkr�td|
| f ��W d |r�t j|� |r�t j|� X d S )Nzutf-8ZSetKeyz(sh)r )Zfdsz"Failed to set key for new pool: %sz'Failed to set key for new pool: %s (%d))�os�pipe�write�encode�open�O_RDONLYr Z
UnixFDListr r r' r( r) r* r r+ r- r r �close) �key_desc�
passphrase�key_file�readr8 �fdZfd_listZ_changedZ_setr r2 r r r r �set_key� s,
$"
rB c C s� t jjstd��y(tjtttdt j
d| df��\}}}W n4 tj k
rl } ztdt |� ��W Y d d }~X nX |s~td| ��d S )Nz"Stratis DBus service not availableZ
UnlockPoolz(ss)ZkeyringzFailed to unlock pool: %s)
r r# r
r r r' r( r) r* r r+ r- r )r. r0 r2 Z
_blockdevsr r r r � unlock_pool� s "rC c C s� t jjstd��|r&|p| r&td��tjdd�}|r\d| }t|||� tjdd|f�}n tjdd�}tjd
d�}y2tjt t
t d tjd | ||||f��\\} }
} } W n4 tj k
r� }
ztd
t
|
� ��W Y d d }
~
X nX | � s�td| | f ��tj� d S )Nz"Stratis DBus service not availablez;Passphrase or key file must be specified for encrypted poolz(bq)Fr z blivet-%sz(bs)T� z(b(ss))Z
CreatePoolz(s(bq)as(bs)(b(ss)))z!Failed to create stratis pool: %sz&Failed to create stratis pool: %s (%d))Fr )FrD �rD rD )FrE )r r# r
r r r+ rB r r' r( r) r* r- r r r$ )r Zdevicesr r> r? Zraid_optr= Zkey_optZ
clevis_optr0 �_pathsr r2 r r r r � create_pool� s0
"rG c
C s t jjstd��tj� |tjj� kr2td| ��tj| }|rZtj ddt
|j � �f�}n tj dd �}y:t j
t|jtdtj dtj d | |f�gf��\\}}}}W n: t jk
r� } z td
|jt
| �f ��W Y d d } ~ X nX |s�td |j||f ��tj� d S )
Nz"Stratis DBus service not availablez#Stratis pool with UUID %s not foundz(bs)TFrD ZCreateFilesystemsz
(a(s(bs)))z(s(bs))z/Failed to create stratis filesystem on '%s': %sz4Failed to create stratis filesystem on '%s': %s (%d))FrD )r r# r
r r r$ r% r& r r+ r r r r' r( r, r4 r- r )
r r. r! r/ Zsize_optr0 rF r r2 r r r r �create_filesystem� s(
,(rH )F)N)"ZgiZrequire_versionZ
gi.repositoryr r r r6 �errorsr r r Z static_datar Ztasksr rD r r
r( r) r4 ZSTRATIS_FILESYSTEM_INTFZSTRATIS_BLOCKDEV_INTFr* ZSTRATIS_FS_SIZEZsafe_name_charactersr r" r3 r5 rB rC rG rH r r r r �<module> s6
$