EVOLUTION-MANAGER
Edit File: VDOOperation.cpython-36.pyc
3 x?ag�q � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd lm Z dd lm Z dd lm Z ddlmZ ddlmZmZmZ ddlmZmZmZmZ ddlmZmZmZ ddlmZmZ ddl m Z ddl Z ddl Z ddl Z!ddl"Z"ddl#Z#ddl$Z$ddl%Z%e&� Z'dd� Z(dd� Z)dd� Z*G dd � d ee+�Z,G d d � d e-�Z.G d d � d e.�Z/G d!d"� d"e.�Z0G d#d$� d$e.�Z1G d%d&� d&e.�Z2G d'd(� d(e.�Z3G d)d*� d*e.�Z4G d+d,� d,e.�Z5G d-d.� d.e.�Z6G d/d0� d0e.�Z7G d1d2� d2e.�Z8G d3d4� d4e.�Z9G d5d6� d6e.�Z:G d7d8� d8e.�Z;G d9d:� d:e.�Z<G d;d<� d<e.�Z=G d=d>� d>e.�Z>G d?d@� d@e.�Z?G dAdB� dBe?�Z@G dCdD� dDe?�ZAG dEdF� dFe?�ZBG dGdH� dHe?�ZCe#jDdI�ZEdJdK� ZFdLdM� ZGeGeH� dS )Nz� VDOOperation - an object representing a vdo script command $Id: //eng/vdo-releases/aluminum/src/python/vdo/vdomgmnt/VDOOperation.py#12 $ � )�absolute_import)�division)�print_function)�unicode_literals� )� ArgumentError)� CommandLock)� Configuration)� Constants)�Defaults)� MgmntUtils)�Service)�VDOKernelModuleService)� VDOService�VDOServiceError� VDOServicePreviousOperationError)� ExitStatus�StateExitStatus�SystemExitStatus�UserExitStatus)�Command� CommandError� runCommand)� Transaction� transactional)�partialNc O sb |d }t jj|j�}t jj|�}|jdd�d }t jjtj|�}t || �� |||�S Q R X d S )Nr �/�_z.lock) �os�path�abspath�confFile�realpath�replace�joinr ZLOCK_DIRr )Z isExclusive�func�args�kwargsZ commandArgsr! Z lockFileBaseZlockFile� r( �"/usr/lib/python3.6/VDOOperation.py�lock4 s r* c s � fdd�}|S )zDDecorator that locks the configuration for exclusive (write) access.c s t d� f| �|�S )NF)r* )r& r' )r% r( r) �wrapC s zexclusivelock.<locals>.wrapr( )r% r+ r( )r% r) � exclusivelockA s r, c s � fdd�}|S )z@Decorator that locks the configuration for shared (read) access.c s t d� f| �|�S )NF)r* )r& r' )r% r( r) r+ I s zsharedlock.<locals>.wrapr( )r% r+ r( )r% r) � sharedlockG s r- c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�OperationErrorz=Exception raised to indicate an error executing an operation.c s t t| �j||� || _d S )N)�superr. �__init__�_msg)�self�msgr& r' )� __class__r( r) r0 \ s zOperationError.__init__c C s | j S )N)r1 )r2 r( r( r) �__str__` s zOperationError.__str__)�__name__� __module__� __qualname__�__doc__r0 r5 � __classcell__r( r( )r4 r) r. N s r. c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )� VDOOperationz�Every instance of this class runs one of the subcommands requested when 'vdo [<options>] <subcommand>' is called via the execute() method.c C s> g }|j r |j|j� j� � n | j|� |j|j|j�� |S )z�Return a list of VDOService objects to be operated on depending on the settings of the --name and --all options. Arguments: args: The arguments passed into vdo. conf: The config file Raises: ArgumentError )�all�extend� getAllVdos�values� _checkForName�append�getVdo�name)r2 r& �confZservicesr( r( r) �getVdoServicesl s zVDOOperation.getVdoServicesc K s� d}d}|j dd�r$t| jf|�}n tj| j�}xJ| j||�D ]:}y|||� W q> tk rv } z |}W Y dd}~X q>X q>W |j dd�s�|j� |dk r�|�dS )a� Apply a method to all specified VDOs. An exception applying the method to some VDO will not prevent it from being applied to any other VDO, however any exception will result in applyToVDOs raising an exception. If the 'readonly' keyword argument is False, the configuration will be persisted after the method has been applied to all the VDOs (whether or not it succeeded for any of them). Arguments: args (dict): The command line arguments method (callable): The method to call on each VDO; will be called as method(args, vdo) kwargs: Keyword args controlling what gets returned and to use when making the Configuration N�readonlyT)�getr r! �modifiableSingletonrE � Exception�persist)r2 r& �methodr' Z exceptionrD �vdo�exr( r( r) � applyToVDOs~ s zVDOOperation.applyToVDOsc C s | j jtd�j| j�� t�dS )a! Execute this operation. This method should be overridden by operation classes, and is intended only to be called from within the run() method below. Arguments: unused_args (dict): The command line arguments (used by subclasses) Raises: NotImplementedError z{0} unimplementedN)�log�errorr �formatrC �NotImplementedError)r2 � unused_argsr( r( r) �execute� s zVDOOperation.executec C s� | j r.tj� dkr.td�j| j�}t|td��| jrfx0d D ](}t j |�s:td�j|�}t|t d��q:W | j r�t j� r�td�j| j�}t|td��d S ) z�Perform checks prior to actually executing the command. Arguments: args (dict): The command line arguments (used by subclasses) Raises: OperationError r z)You must be root to use the "{0}" command)� exitStatus� vdodumpconfig�vdoforcerebuild� vdoformatz#executable '{0}' not found in $PATHz&{0} command not available with --noRunN)rV rW rX )� requiresRootr �getuidr rQ rC r. r � checkBinariesr Zwhichr �requiresRunModer Z noRunMode)r2 r& r3 � executabler( r( r) � preflight� s zVDOOperation.preflightc C sB t j| _y |j| _W n tk r( Y nX | j|� | j|� dS )z�Run this operation. This is the external entry point for users of VDOOperation. Arguments: args (dict): The command line arguments N)r r! �KeyErrorr^ rT )r2 r& r( r( r) �run� s zVDOOperation.runc K sr t jddt| �j�}|d j� |dd � | _tjd�| _|j dd�| _ |j dd �| _ |j d d �| _ | t | j< d S ) Nz Operation$� r r zvdo.vdomgmnt.VDOOperationrY Tr[ Fr\ )�re�sub�typer6 �lowerrC �loggingZ getLoggerrO rG rY r[ r\ � vdoOperations)r2 r' rC r( r( r) r0 � s zVDOOperation.__init__c C s( |j dks|j j� dkr$ttd���dS )z|Check that the args contain a non-None name. Arguments: args: the args passed Raises: ArgumentError Nra z"Missing required argument '--name')rC �stripr r )r2 r& r( r( r) r@ � s zVDOOperation._checkForNameN) r6 r7 r8 r9 rE rN rT r^ r` r0 r@ r( r( r( r) r; d s $ r; c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )�ActivateOperationz Implements the activate command.c s t t| �j� d S )N)r/ ri r0 )r2 )r4 r( r) r0 � s zActivateOperation.__init__c C s | j || jdd� d S )NF)rF )rN � _activate)r2 r& r( r( r) rT � s zActivateOperation.executec C s |j � d S )N)Zactivate)r2 r& rL r( r( r) rj s zActivateOperation._activate) r6 r7 r8 r9 r0 r, rT rj r: r( r( )r4 r) ri � s ri c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )�ChangeWritePolicyOperationz)Implements the changeWritePolicy command.c s t t| �j� d S )N)r/ rk r0 )r2 )r4 r( r) r0 s z#ChangeWritePolicyOperation.__init__c C s0 t |d�| _| jd krd S | j|| jdd� d S )NZ writePolicyF)rF )�getattr�_newWritePolicyrN �_changeWritePolicy)r2 r& r( r( r) rT s z"ChangeWritePolicyOperation.executec C s |j | j� d S )N)ZsetWritePolicyrm )r2 r& rL r( r( r) rn s z-ChangeWritePolicyOperation._changeWritePolicy) r6 r7 r8 r9 r0 r, rT rn r: r( r( )r4 r) rk s rk c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )�ConvertOperationz Implements the convert command.c s t t| �j� d S )N)r/ ro r0 )r2 )r4 r( r) r0 ( s zConvertOperation.__init__c C s6 t j| j�}| j|� |j|j�}|j� |j� d S )N)r rH r! r@ rB rC ZconvertrJ )r2 r& rD rL r( r( r) rT , s zConvertOperation.execute)r6 r7 r8 r9 r0 r, rT r: r( r( )r4 r) ro " s ro c s< e Zd ZdZ� fdd�Z� fdd�Zeedd� ��Z� Z S )�CreateOperationz Implements the create command.c s t t| �jdd� d S )NT)r[ )r/ rp r0 )r2 )r4 r( r) r0 < s zCreateOperation.__init__c s8 t t| �j|� |js"ttd���|js4ttd���d S )Nz"Missing required argument '--name'z$Missing required argument '--device')r/ rp r^ rC r r �device)r2 r& )r4 r( r) r^ @ s zCreateOperation.preflightc C sj t j| j�}t|�j� }|d }|d= t|j|f|�}tj� }|j |j � |j |j � |j � |j� d S )NrC )r rH r! �vars�copyr rC r � transactionZcreate�forceZ addUndoStage�removerJ � announceReady)r2 r& rD �argsDictrC rL rt r( r( r) rT J s zCreateOperation.execute) r6 r7 r8 r9 r0 r^ r, r rT r: r( r( )r4 r) rp 6 s rp c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )�DeactivateOperationz"Implements the deactivate command.c s t t| �j� d S )N)r/ ry r0 )r2 )r4 r( r) r0 d s z DeactivateOperation.__init__c C s | j || jdd� d S )NF)rF )rN � _deactivate)r2 r& r( r( r) rT h s zDeactivateOperation.executec C s |j � d S )N)Z deactivate)r2 r& rL r( r( r) rz o s z DeactivateOperation._deactivate) r6 r7 r8 r9 r0 r, rT rz r: r( r( )r4 r) ry ^ s ry c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )�GrowLogicalOperationz#Implements the growLogical command.c s t t| �j� d S )N)r/ r{ r0 )r2 )r4 r( r) r0 y s z GrowLogicalOperation.__init__c C s: t j| j�}| j|� |j|j�}|j|j� |j� d S )N) r rH r! r@ rB rC Z growLogicalZvdoLogicalSizerJ )r2 r& rD rL r( r( r) rT } s z GrowLogicalOperation.execute)r6 r7 r8 r9 r0 r, rT r: r( r( )r4 r) r{ s s r{ c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )�GrowPhysicalOperationz$Implements the growPhysical command.c s t t| �j� d S )N)r/ r| r0 )r2 )r4 r( r) r0 � s z GrowPhysicalOperation.__init__c C s6 t j| j�}| j|� |j|j�}|j� |j� d S )N)r rH r! r@ rB rC Z growPhysicalrJ )r2 r& rD rL r( r( r) rT � s z GrowPhysicalOperation.execute)r6 r7 r8 r9 r0 r, rT r: r( r( )r4 r) r| � s r| c s< e Zd ZdZ� fdd�Z� fdd�Zeedd� ��Z� Z S )�ImportOperationz Implements the import command.c s t t| �jdd� d S )NT)r[ )r/ r} r0 )r2 )r4 r( r) r0 � s zImportOperation.__init__c s8 t t| �j|� |js"ttd���|js4ttd���d S )Nz"Missing required argument '--name'z$Missing required argument '--device')r/ r} r^ rC r r rq )r2 r& )r4 r( r) r^ � s zImportOperation.preflightc C sJ t j| j�}t|�j� }|d }|d= t|j|f|�}|j� |j� d S )NrC ) r rH r! rr rs r rC Z importDevicerw )r2 r& rD rx rC rL r( r( r) rT � s zImportOperation.execute) r6 r7 r8 r9 r0 r^ r, r rT r: r( r( )r4 r) r} � s r} c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� ListOperationz Implements the list command.c s t t| �j� d S )N)r/ r~ r0 )r2 )r4 r( r) r0 � s zListOperation.__init__c C s� t � }dddddg}x^t|dd�j� D ]J}|jdd �jd �\}}}|jd �d d kr&|jtjj |jd �d �� q&W |j r�t | j �}|t |j � j� �O }t|�}|j� ttjj|�� d S )NZlsblkz--pairsz--pathsz--outputz type,nameT)�noThrow�"ra � �=r rL )�setr � splitlinesr# � partition�split�addr r �basenamer< r r! r> �keys�list�sort�print�linesepr$ ) r2 r& �vdos�cmd�lineZblktypeZfoorC rD r( r( r) rT � s zListOperation.execute)r6 r7 r8 r9 r0 rT r: r( r( )r4 r) r~ � s r~ c s@ e Zd ZdZ� fdd�Zedd� �Z� fdd�Zdd � Z� Z S ) �ModifyOperationz Implements the modify command.c s t t| �j� d S )N)r/ r� r0 )r2 )r4 r( r) r0 � s zModifyOperation.__init__c C s | j || jdd� d S )NF)rF )rN � _modifyVDO)r2 r& r( r( r) rT � s zModifyOperation.executec s t t| �j|� tj|� d S )N)r/ r� r^ r ZvalidateModifiableOptions)r2 r& )r4 r( r) r^ � s zModifyOperation.preflightc C s |j |� d S )N)ZsetModifiableOptions)r2 r& rL r( r( r) r� � s zModifyOperation._modifyVDO) r6 r7 r8 r9 r0 r, rT r^ r� r: r( r( )r4 r) r� � s r� c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )�PrintConfigFileOperationz'Implements the printConfigFile command.c s t t| �jddd� d S )NFT)rY r\ )r/ r� r0 )r2 )r4 r( r) r0 � s z!PrintConfigFileOperation.__init__c C s t | jdd�}t|j� � d S )NT)Z mustExist)r r! r� Z asYAMLForUser)r2 r& rD r( r( r) rT s z PrintConfigFileOperation.execute)r6 r7 r8 r9 r0 r- rT r: r( r( )r4 r) r� � s r� c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )�RemoveOperationz Implements the remove command.c s t t| �j� d S )N)r/ r� r0 )r2 )r4 r( r) r0 s zRemoveOperation.__init__c C s | j || jdd� dS )z Implements the remove command.F)rF N)rN � _removeVDO)r2 r& r( r( r) rT s zRemoveOperation.executec C sh g }y|j |j|d� W nJ tk rb ttd�� ttd�� ttd�� ttjj|�� � Y nX d S )N)� removeStepsz A previous operation failed.z;Recovery from the failure either failed or was interrupted.z;Add '--force' to 'remove' to perform the following cleanup.)rv ru r r� r r r� r$ )r2 r& rL r� r( r( r) r� s zRemoveOperation._removeVDO) r6 r7 r8 r9 r0 r, rT r� r: r( r( )r4 r) r� s r� c s8 e Zd ZdZ� fdd�Zedd� �Zedd� �Z� Z S )�StartOperationz Implements the start command.c s t t| �jdd� d S )NT)r[ )r/ r� r0 )r2 )r4 r( r) r0 / s zStartOperation.__init__c C s | j || jdd� d S )NF)rF )rN � _startVDO)r2 r& r( r( r) rT 3 s zStartOperation.executec C s@ |j s&td�j|j� d�}t|td��|j|j� |jd� d S )Nz VDO volume {name} not activated)rC )rU F) Z activatedr rQ �getNamer. r �startZ forceRebuildrw )r2 r& rL r3 r( r( r) r� : s zStartOperation._startVDO) r6 r7 r8 r9 r0 r, rT r r� r: r( r( )r4 r) r� ) s r� c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )�StatusOperationz Implements the status command.c s t t| �jdddd� d S )NFT)rY r[ r\ )r/ r� r0 )r2 )r4 r( r) r0 I s zStatusOperation.__init__c C s� t | jdd�}|jsd|_�y|i }td�|i}tddgddd�|td�< td d gddd�|td �< tj� d kr~td�|td�< td�t� j � i}td�|j � i}i }td�|i}xb| j ||�D ]R} y| j |j �|| j � < W q� t k �r } zt| �|| j � < W Y d d } ~ X q�X q�W ttj|dd�d d� � ttj|dd�d d� � ttj|dd�d d� � ttj|dtd�d�d d � � tjj� tjj� W n6 tk �r� } z| jjdj| �� W Y d d } ~ X nX d S ) NF)rF Tz VDO status�unamez-n)r rh ZNodeZdatez--rfc-3339=secondsZDater zNot running as root,z some status may be unavailableZNotez Kernel moduler ZVDOs)�default_flow_styler �inf)r� �widthzexception ignored: {0}z3Not running as root, some status may be unavailable���r� r� r� )r r! rC r< r r r rZ r ZstatusrE �pendingr� r �strr� �yamlZ safe_dump�float�sys�stdout�flush�stderr�IOErrorrO �debugrQ ) r2 r& rD r? Z vdoStatusZ kernelStatusZ confStatusr� Z perVdoStatusrL rM r( r( r) rT O s< & zStatusOperation.execute)r6 r7 r8 r9 r0 r, rT r: r( r( )r4 r) r� C s r� c s4 e Zd ZdZ� fdd�Zedd� �Zdd� Z� ZS )� StopOperationz Implements the stop command.c s t t| �j� d S )N)r/ r� r0 )r2 )r4 r( r) r0 � s zStopOperation.__init__c C s | j || jdd� d S )NF)rF )rN �_stopVDO)r2 r& r( r( r) rT � s zStopOperation.executec C s |j |j� d S )N)�stopru )r2 r& rL r( r( r) r� � s zStopOperation._stopVDO) r6 r7 r8 r9 r0 r, rT r� r: r( r( )r4 r) r� ~ s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�VersionOperationz Implements the version command.c s t t| �jdddd� d S )NFT)rY r[ r\ )r/ r� r0 )r2 )r4 r( r) r0 � s zVersionOperation.__init__c C s2 t ddgdd�}tjd|�}tddj|� � d S )NrX z --versionT)r z\d+z VDO version: �.)r rb �findallr� r$ )r2 rS �output�versionr( r( r) rT � s zVersionOperation.execute)r6 r7 r8 r9 r0 rT r: r( r( )r4 r) r� � s r� c s6 e Zd ZdZd � fdd� Zedd� �Zdd� Z� ZS ) � OptionTogglezCBase class for operations which either enable or disable an option.Fc s" t t| �j|d� || _|| _d S )N)r[ )r/ r� r0 Z_enableZ _optionName)r2 �enableZ optionNamer[ )r4 r( r) r0 � s zOptionToggle.__init__c C s | j || jdd� d S )NF)rF )rN � _configure)r2 r& r( r( r) rT � s zOptionToggle.executec C s | j jtd�j| j�� t�dS )zlActually update the configuration for this operation. This method must be overridden by derived classes.z{0} unimplementedN)rO rP r rQ rC rR )r2 r& rL r( r( r) r� � s zOptionToggle._configure)F) r6 r7 r8 r9 r0 r, rT r� r: r( r( )r4 r) r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�DisableCompressionOperationz*Implements the disableCompression command.c s t t| �jdd� d S )NF� compression)r/ r� r0 )r2 )r4 r( r) r0 � s z$DisableCompressionOperation.__init__c C s |j d� d S )NF)�setCompression)r2 r& rL r( r( r) r� � s z&DisableCompressionOperation._configure)r6 r7 r8 r9 r0 r� r: r( r( )r4 r) r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� DisableDeduplicationOperationz-Implements the disable deduplication command.c s t t| �jdd� d S )NF� deduplication)r/ r� r0 )r2 )r4 r( r) r0 � s z&DisableDeduplicationOperation.__init__c C s |j d� d S )NF)�setDeduplication)r2 r& rL r( r( r) r� � s z(DisableDeduplicationOperation._configure)r6 r7 r8 r9 r0 r� r: r( r( )r4 r) r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�EnableCompressionOperationz)Implements the enableCompression command.c s t t| �jdd� d S )NTr� )r/ r� r0 )r2 )r4 r( r) r0 � s z#EnableCompressionOperation.__init__c C s |j d� d S )NT)r� )r2 r& rL r( r( r) r� � s z%EnableCompressionOperation._configure)r6 r7 r8 r9 r0 r� r: r( r( )r4 r) r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� EnableDeduplicationOperationz+Implements the enableDeduplication command.c s t t| �jdddd� d S )NTr� )r[ )r/ r� r0 )r2 )r4 r( r) r0 � s z%EnableDeduplicationOperation.__init__c C s |j d� d S )NT)r� )r2 r& rL r( r( r) r� � s z'EnableDeduplicationOperation._configure)r6 r7 r8 r9 r0 r� r: r( r( )r4 r) r� � s r� z^[A-Z].*Operation$c C s t j| �o | jdko tj| j�S )Nr; )�inspectZisclassr6 � operationRE�match)�memberr( r( r) � _isOperation� s r� c C s* x$t jtj| t�D ]}|d � qW d S )Nr )r� Z getmembersr� �modulesr� )Z moduleNameZ operationr( r( r) �makeOperations s r� )Ir9 Z __future__r r r r ra r r r r r r r r r r r r r r r Z vdo.utilsr r r r r � functoolsr r� rf �__main__�mainr rb r� r� �dictrg r* r, r- rI r. �objectr; ri rk ro rp ry r{ r| r} r~ r� r� r� r� r� r� r� r� r� r� r� r� �compiler� r� r� r6 r( r( r( r) �<module> sp ($ ;