Mini Shell
3
�XZ�@ �
@ s� d dl Z d dlZd dlmZ d dlmZ G dd� de�Zi Ze j � Z
e
ee
j < x<ye j e
�Z
e
ee
j < W qP e e
e jfk
r� P Y qPX qPW i Ze jd �Ze je�ee< dZx*er�e je�Zes�dZq�e je�ee< q�W dS )� N)�
CachedList)�localeCc @ sF e Zd ZdZedbdd��Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
e edd� ��Z e edd� ��Z
e edd� ��Ze edd� ��Ze edd� ��Ze edd� ��Ze edd � ��Zed d � �Zed d � �Zed!d"� �Zed#d$� �Ze d%d&� �Ze d'd(� �Ze d)d*� d+d*� �Zed,d-� �Zed.d/� �Zed0d1� �Z ed2d3� �Z ed4d5� �Z ed6d7� �Z ed8d9� �Z edcd:d;��Z!eddd<d=��Z"ed>d?� �Z#ed@dA� �Z$ededBdC��Z%edfdDdE��Z&edgdFdG��Z'edHdI� �Z(edJdK� �Z)edLdM� �Z*dNdO� Z+dPdQ� Z,dRdS� Z-dTdU� Z.dVdW� Z/edXdY� �Z0edZd[� �Z1ed\d]� �Z2ed^d_� �Z3d`da� Z4dS )h�Diskz�Disk()
A Disk object describes a type of device in the system. Disks
can hold partitions. A Disk is a basic operating system-specific
object.Nc sn |r.|� _ |dkr&tj� j jd�� _qX|� _n*|dkrBtjd��ntj|j� �� _ |� _t � fdd��� _
dS )z�Create a new Disk object from the device and type specified. The
device is a Device object and type is a string matching a key in
the diskType hash.N)Z PedDevicezno device specifiedc s � j � S )N)�_Disk__getPartitions� )�selfr � /usr/lib64/python3.6/disk.py�<lambda>8 s z Disk.__init__.<locals>.<lambda>) � _Disk__disk�partedZDeviceZdev�_device�
DiskException�_pedr Z getPedDevicer � _partitions)r �device�PedDiskr )r r �__init__% s z
Disk.__init__c C sn dd l }t| j�t|j�kr dS |jj| j|j�}x6y t|�\}}||krNdS W q4 tk
rd dS X q4W d S )Nr FT)�six�len�
partitionsZmoves�zip�next�
StopIteration)r �otherr ZpartIter�left�rightr r r �
_hasSameParts: s zDisk._hasSamePartsc C s | j |� S )N)�__ne__)r r r r r �__eq__I s z Disk.__eq__c C s( t | |j�sdS | j|jkp&| j|� S )NT)�
isinstance� __class__r r )r r r r r r L s z Disk.__ne__c C s* d| j | j| j| j| j| j| jd� }|S )Nz�parted.Disk instance --
type: %(type)s primaryPartitionCount: %(primaryCount)s
lastPartitionNumber: %(last)s maxPrimaryPartitionCount: %(max)s
partitions: %(partitions)s
device: %(device)r
PedDisk: %(ped)r)�typeZ primaryCountZlast�maxr r Zped)r! �primaryPartitionCount�lastPartitionNumber�maxPrimaryPartitionCountr r r
)r �sr r r �__str__R s z Disk.__str__c C sZ g }| j � }xH|rT|jtj@ s6|jtj@ s6|jtj@ r@|j� }q|j|� |j� }qW |S )z�Construct a list of partitions on the disk. This is called only as
needed from the self.partitions property, which just happens to be
a CachedList.)�getFirstPartitionr! r �PARTITION_FREESPACEZPARTITION_METADATAZPARTITION_PROTECTEDZ
nextPartition�append)r r � partitionr r r Z__getPartitions_ s
zDisk.__getPartitionsc C s
| j j� S )z.The number of primary partitions on this disk.)r
Zget_primary_partition_count)r r r r r# r s zDisk.primaryPartitionCountc C s
| j j� S )z:The last assigned partition number currently on this disk.)r
Zget_last_partition_num)r r r r r$ x s zDisk.lastPartitionNumberc C s
| j j� S )z>The maximum number of primary partitions allowed on this disk.)r
Z get_max_primary_partition_count)r r r r r% ~ s z Disk.maxPrimaryPartitionCountc C s
| j j� S )z6The maximum number of partitions allowed on this disk.)r
Z!get_max_supported_partition_count)r r r r �maxSupportedPartitionCount� s z Disk.maxSupportedPartitionCountc C s | j j� }tj|d�S )z"Partition start address Alignment.)Z PedAlignment)r
Zget_partition_alignmentr Z Alignment)r Z alignmentr r r �partitionAlignment� s
zDisk.partitionAlignmentc C s
| j j� S )z4Max Partition Length the disk's label can represent.)r
Zmax_partition_length)r r r r �maxPartitionLength� s zDisk.maxPartitionLengthc C s
| j j� S )z:Max Partition Start Sector the disk's label can represent.)r
Zmax_partition_start_sector)r r r r �maxPartitionStartSector� s z Disk.maxPartitionStartSectorc C s | j j|�S )z�Get the value of a particular flag on the disk. Valid flags
are the _ped.DISK_* constants. See _ped.disk_flag_get_name() and
_ped.disk_flag_get_by_name() for more help working with disk flags.
)r
Zget_flag)r �flagr r r �getFlag� s z Disk.getFlagc C s | j j|d�S )z�Set the flag on this disk. On error, an Exception will be raised.
See getFlag() for more help on working with disk flags.� )r
�set_flag)r r0 r r r �setFlag� s z Disk.setFlagc C s | j j|d�S )z�Unset the flag on this disk. On error, an Exception will be raised.
See getFlag() for more help on working with disk flags.r )r
r3 )r r0 r r r � unsetFlag� s zDisk.unsetFlagc C s | j j|�S )zJReturn True if flag is available on this Disk, False
otherwise.)r
Zis_flag_available)r r0 r r r �isFlagAvailable� s zDisk.isFlagAvailablec C s | j S )z.The list of partitions currently on this disk.)r )r r r r r � s zDisk.partitionsc C s | j S )z7The underlying Device holding this disk and partitions.)r )r r r r r � s z Disk.devicec C s
| j jjS )N)r
r! �name)r& r r r r � s z
Disk.<lambda>c C s t | jdtj| �S )Nr! )�setattrr
r �diskType)r& �vr r r r � s c C s t | jj� d�S )z Make a deep copy of this Disk.)r )r r
� duplicate)r r r r r; � s zDisk.duplicatec C s
| j j� S )z<Closes the Disk ensuring all outstanding writes are flushed.)r
�destroy)r r r r r< � s z Disk.destroyc C s | j j� | jj� S )z�Writes in-memory changes to a partition table to disk and
informs the operating system of the changes. Equivalent to
calling self.commitToDevice() then self.commitToOS().)r �
invalidater
�commit)r r r r r> � s
z Disk.commitc C s | j j� | jj� S )zbWrite the changes made to the in-memory description of a
partition table to the device.)r r= r
Z
commit_to_dev)r r r r �commitToDevice� s
zDisk.commitToDevicec C s | j j� | jj� S )zZTell the operating system kernel about the partition table
layout of this Disk.)r r= r
Z commit_to_os)r r r r �
commitToOS� s
zDisk.commitToOSc C s
| j j� S )z;Perform a sanity check on the partition table of this Disk.)r
�check)r r r r rA � s z
Disk.checkc C s | j jj|�S )z7Check that the disk type supports the provided feature.)r
r! Z
check_feature)r Zfeaturer r r �supportsFeature� s zDisk.supportsFeaturec C sj |r | j j|j� |j� �}n |s,tjd��n| j j|j� �}|rbtj|j� jd�|_| j j
� dS dS dS )z;Add a new Partition to this Disk with the given Constraint.z$no partition or constraint specified)� PedGeometryTFN) r
Z
add_partition�getPedPartition�getPedConstraintr r
�Geometry�geomZgeometryr r= )r r+ �
constraint�resultr r r � addPartition� s
zDisk.addPartitionc C s4 |st jd��| jj|j� �r,| jj� dS dS dS )a Removes specified Partition from this Disk. NOTE: If the
Partition is an extended partition, it must not contain any
logical partitions. Also note that the partition is not
actually destroyed unless you use the deletePartition()
method.zno partition specifiedTFN)r r
r
Zremove_partitionrD r r= )r r+ r r r �removePartition s
zDisk.removePartitionc C s& | j j|j� �r | jj� dS dS dS )z�Removes specified Partition from this Disk under the same
conditions as removePartition(), but also destroy the
removed Partition.TFN)r
Zdelete_partitionrD r r= )r r+ r r r �deletePartition s
zDisk.deletePartitionc C s | j j� r| jj� dS dS dS )z1Removes and destroys all Partitions in this Disk.TFN)r
Z
delete_allr r= )r r r r �deleteAllPartitions s
zDisk.deleteAllPartitionsc C sF | s | rt jd��| s"| r,t jd��| jj|j� |j� ||�S )z�Sets the Geometry of the specified Partition using the given
Constraint and start and end sectors. Note that this method
does not modify the partition contents, just the partition
table.z$no partition or constraint specifiedz"no start or end geometry specified)r r
r
Zset_partition_geomrD rE )r r+ rH �start�endr r r �setPartitionGeometry& s
zDisk.setPartitionGeometryc C s< |st jd��|r(| jj|j� |j� �S | jj|j� �S dS )zWGrow the Partition's Geometry to the maximum possible subject
to Constraint.zno partition specifiedN)r r
r
Zmaximize_partitionrD rE )r r+ rH r r r �maximizePartition6 s
zDisk.maximizePartitionc C sL |st jd��|r0t j| jj|j� |j� �d�S t j| jj|j� �d�S dS )zcGet the maximum Geometry the Partition can be grown to,
subject to the given Constraint.zno partition specified)rC N)r r
rF r
Zget_max_partition_geometryrD rE )r r+ rH r r r � calculateMaxPartitionGeometryC s
z"Disk.calculateMaxPartitionGeometryc C s | j j� }|r| jj� |S )z�Reduce the size of the extended partition to a minimum while
still wrapping its logical partitions. If there are no logical
partitions, remove the extended partition.)r
Zminimize_extended_partitionr r= )r �retr r r �minimizeExtendedPartitionO s
z Disk.minimizeExtendedPartitionc C s t j| | jj|�d�S )z�Returns the Partition that contains the sector. If the sector
lies within a logical partition, then the logical partition is
returned (not the extended partition).)�disk� PedPartition)r � Partitionr
Zget_partition_by_sector)r Zsectorr r r �getPartitionBySector[ s zDisk.getPartitionBySectorc
C s$ yt j| | jj� d�S dS dS )z4Return the extended Partition, if any, on this Disk.)rU rV N)r rW r
Zextended_partition)r r r r �getExtendedPartitionb s zDisk.getExtendedPartitionc s � fdd�| j D �S )Nc s g | ]}� |�r|�qS r r )�.0�part)�fnr r �
<listcomp>l s z+Disk.__filterPartitions.<locals>.<listcomp>)r )r r\ r )r\ r Z__filterPartitionsk s zDisk.__filterPartitionsc C s | j dd� �S )z1Return a list of logical Partitions on this Disk.c S s | j o| jtj@ S )N)�activer! r ZPARTITION_LOGICAL)�pr r r r p s z+Disk.getLogicalPartitions.<locals>.<lambda>)�_Disk__filterPartitions)r r r r �getLogicalPartitionsn s zDisk.getLogicalPartitionsc C s | j dd� �S )z=Return a list of primary (or normal) Partitions on this Disk.c S s | j tjkS )N)r! r ZPARTITION_NORMAL)r_ r r r r t s z+Disk.getPrimaryPartitions.<locals>.<lambda>)r` )r r r r �getPrimaryPartitionsr s zDisk.getPrimaryPartitionsc C s | j dd� �S )z:Return a list of RAID (or normal) Partitions on this Disk.c S s | j o| jtj�S )N)r^ r1 r ZPARTITION_RAID)r_ r r r r x s z(Disk.getRaidPartitions.<locals>.<lambda>)r` )r r r r �getRaidPartitionsv s zDisk.getRaidPartitionsc C s | j dd� �S )z>Return a list of physical volume-type Partitions on this Disk.c S s | j o| jtj�S )N)r^ r1 r Z
PARTITION_LVM)r_ r r r r | s z'Disk.getLVMPartitions.<locals>.<lambda>)r` )r r r r �getLVMPartitionsz s zDisk.getLVMPartitionsc C sH g }| j j� }x4|rB|jtj@ r4|jtj|jd�� | j j|�}qW |S )zhReturn a list of Geometry objects representing the available
free space regions on this Disk.)rC )r
�next_partitionr! r r) r* rF rG )r � freespacer[ r r r �getFreeSpaceRegions~ s
zDisk.getFreeSpaceRegionsc C sH g }| j j� }x4|rB|jtj@ r4|jtj| |d�� | j j|�}qW |S )ziReturn a list of Partition objects representing the available
free space regions on this Disk.)rU rV )r
re r! r r) r* rW )r rf r[ r r r �getFreeSpacePartitions� s
zDisk.getFreeSpacePartitionsc C s t j| | jj� d�S )zUReturn the first Partition object on the disk or None if
there is not one.)rU rV )r rW r
re )r r r r r( � s zDisk.getFirstPartitionc C s" x | j D ]}|j|kr|S qW dS )z�Return a Partition object associated with the partition device
path, such as /dev/sda1. Returns None if no partition is found.N)r �path)r ri r+ r r r �getPartitionByPath� s
zDisk.getPartitionByPathc C s | j S )z]Return the _ped.Disk object contained in this Disk. For internal
module use only.)r
)r r r r �
getPedDisk� s zDisk.getPedDisk)NN)NN)N)NNNN)NN)NN)5�__name__�
__module__� __qualname__�__doc__r r r r r r' r �propertyr# r$ r% r, r- r. r/ r1 r4 r5 r6 r r r! r; r< r> r? r@ rA rB rJ rK rL rM rP rQ rR rT rX rY r` ra rb rc rd rg rh r( rj rk r r r r r sz
r TF)r r Zparted.cachedlistr Zparted.decoratorsr �objectr r9 Zdisk_type_get_nextZ__typer7 �
IndexError� TypeErrorZUnknownTypeExceptionZdiskFlagZdisk_flag_nextZ__flagZdisk_flag_get_nameZ __readFlagsr r r r �<module> s2