EVOLUTION-MANAGER
Edit File: Configuration.cpython-36.pyc
3 x?ag.7 � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd l m Z m Z dd l m Z mZ dd lZdd lZdd lZdd lmZ dd lZdd lZG dd� dee�ZG dd� de�Zd S )z� Configuration - VDO manager configuration file handling $Id: //eng/vdo-releases/aluminum/src/python/vdo/vdomgmnt/Configuration.py#6 $ � )�absolute_import)�division)�print_function)�unicode_literals� )� ArgumentError)�StateExitStatus)� VDOService)�Command� runCommand)�FileLock� YAMLObjectN)�ST_MTIMEc s( e Zd ZdZ� fdd�Zdd� Z� ZS )�BadConfigurationFileErrorzwException raised to indicate an error in processing the configuration file, such as a parse error or missing data. c s t t| �j||� || _d S )N)�superr �__init__�_msg)�self�msg�args�kwargs)� __class__� �#/usr/lib/python3.6/Configuration.pyr 3 s z"BadConfigurationFileError.__init__c C s | j S )N)r )r r r r �__str__8 s z!BadConfigurationFileError.__str__)�__name__� __module__� __qualname__�__doc__r r � __classcell__r r )r r r + s r c s0 e Zd ZdZejd�ZdgZi ZdZ dZ dZ e dd� �Z d:d d �Zd d � Zed d� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Ze d d � �Zed d � �Ze� fd!d"��Z� fd#d$�Z e� fd%d&��Z � fd'd(�Z d;� fd*d+� Z d,d-� Z d.d/� Z!d0d1� Z"d2d3� Z#d4d5� Z$d6d7� Z%e d8d9� �Z&� Z'S )<� Configurationz Configuration of VDO volumes.zvdo.vdomgmnt.Configurationi z /var/lock/vdo-config-singletonsz!ConfigurationFc C sJ d}t | jd��0}| jj|�}|dkr<t|dd�}|| j|< W dQ R X |S )aR Allocates, as necessary, and returns a modifiable, singleton Configuration instance for the specified filepath. Separate entities can thus share one in memory copy of the configuration file allowing for encapsulation of per-entity manipulation of the configuration. Args: filepath (str): path to config file Nzr+F)�readonly)r � singletonLock�modifiableSingltons�getr )�cls�filepath�config�fr r r �modifiableSingletonI s z!Configuration.modifiableSingletonc C sB | j � | jjdj|�� | r.| j|�r.dS || j|< d| _dS )at Adds or replaces a VDOService object in the configuration. Generates an assertion error if this object is read-only. Arguments: name -- name of the VDOService vdo -- the VDOService to add or replace replace -- if True, any existing VDOService will be replaced Returns: False if the VDOService exists and replace is False, True otherwise z!Adding vdo "{0}" to configurationFT)�_assertCanModify�log�debug�format�haveVdo�_vdos�_dirty)r �name�vdo�replacer r r �addVdo\ s zConfiguration.addVdoc C s" t j| j| d�dd�}|jdd�S )zIReturns the configuration's YAML representation to present to users. )�filenamer' F)�default_flow_stylez!!python/unicode � )�yaml�dumpr& r3 )r �sr r r � asYAMLForUserp s zConfiguration.asYAMLForUserc C s | j S )z0Returns the file path of the configuration file.)� _filename)r r r r r& z s zConfiguration.filepathc C s | j S )z#Retrieves a list of all known VDOs.)r/ )r r r r � getAllVdos� s zConfiguration.getAllVdosc C s> d}y| j | }W n& tk r8 ttd�j|���Y nX |S )zRetrieves a VDO by name.NzVDO volume {0} not found)r/ �KeyErrorr �_r- )r r1 r2 r r r �getVdo� s zConfiguration.getVdoc C s || j kS )z0Returns True if we have a VDO with a given name.)r/ )r r1 r r r r. � s zConfiguration.haveVdoc C s< t jj|�}x*| jD ] }|t jj| j| j�krdS qW dS )z�Returns a boolean indicating if the configuration contains a VDO using the specified device. Both the specified device and the device from the vdos present in the configuration are fully resolved for the check. TF)�os�path�realpathr/ �device)r rD r2 r r r �isDeviceConfigured� s z Configuration.isDeviceConfiguredc C s> | j r dS | js | jjd� dS | jjdj| j�� | j� rH| j� dS tj d| idd�}t j � r�t t d�� t |� d| _dS | jd }tjj|�r�tj|� t|d ��n}|jd � |jtj� |jd jt d ��� |jtj� |jd � |jtj� |j|� |j� tj|� W dQ R X tj|| j� | j� d| _dS ) a\ Writes out the Configuration if necessary. If the Configuration is read-only or has not been modified, this method will silently return. If Command.noRunMode is True, any new Configuration will be printed to stdout instead of the file. This method will generate an assertion failure if the configuration file is not open. Nz&Configuration is clean, not persistingz Writing configuration to {0}r' F)r6 z New configuration (not written):z.new�wzD####################################################################z# {0}z>THIS FILE IS MACHINE GENERATED. DO NOT EDIT THIS FILE BY HAND.)� _readonlyr0 r+ r, r- r& �_empty� _removeFiler8 r9 r � noRunMode�printr? rA rB �exists�remove�open�write�linesep�flush�fsync�rename�_fsyncDirectory)r r: ZnewFile�fhr r r �persist� sD zConfiguration.persistc C s | j � | j|= d| _dS )zRemoves a VDO by name.TN)r* r/ r0 )r r1 r r r � removeVdo� s zConfiguration.removeVdoc C s� i }d}y t j| j�}| j|td�< W n@ tk rf } z$|jtjkrF� td�|td�< W Y dd}~X nX |dkr�td�|td�< n tjdtj |t ��|td�< |S )zAReturns a dictionary representing the status of this object. NZFilezdoes not existz not availablez Last modifiedz%Y-%m-%d %H:%M:%S) rA �statr& r? �OSError�errno�ENOENT�timeZstrftimeZ localtimer )r �status�st�exr r r r] � s "zConfiguration.statusc C s | d�S )Nz/dev/YAMLInstancer )r% r r r �_yamlMakeInstance� s z Configuration._yamlMakeInstancec C s ddgS )N�version�vdosr )r r r r �_yamlAttributeKeys� s z Configuration._yamlAttributeKeysc s$ t t| �j}| j|d< | j|d< |S )Nra rb )r r � _yamlData�_schemaVersionr/ )r �data)r r r rd s zConfiguration._yamlDatac s( t t| �j|� |d | _|d | _d S )Nra rb )r r �_yamlSetAttributesra rb )r Z attributes)r r r rg s z Configuration._yamlSetAttributesc s t t| �j}|jddg� |S )Nra rb )r r � _yamlSpeciallyHandledAttributes�extend)r Zspecials)r r r rh s z-Configuration._yamlSpeciallyHandledAttributesc sD t t| �j|� |j| _|j| _x | jD ]}| j| j| � q(W d S )N)r r �_yamlUpdateFromInstancera re rb r/ Z setConfig)r �instancer2 )r r r rj s z%Configuration._yamlUpdateFromInstanceTc s� t t| �j� i | _tjj|�| _|| _d| _ || _ d| _ | j sN| j � d| _ | j rxtjj| j� rxttd�j| j���|r�dnd}y@tjj|�r�tjj|�dkr�t||��}| j|� W dQ R X W n. tk r� } ztt|���W Y dd}~X nX dS ) a Construct a Configuration. Args: filename (str): The path to the XML configuration file Kwargs: readonly (bool): If True, the configuration is read-only. mustExist (bool): If True, the configuration file must exist. Raises: ArgumentError Fi Tz&Configuration file {0} does not exist.�rza+r N)r r r r/ rA rB �abspathr< rG r0 Z _mustExistre �yaml_has_unicode_ctorsZ_fix_constructorsrL r& r r? r- �getsizerN �_read�IOError�str)r r5 r! Z mustExist�moderU r )r r r r s* zConfiguration.__init__c C s dj t| �j| j�S )Nz{0}({1}))r- �typer r& )r r r r r D s zConfiguration.__str__c C s | j std��dS )z<Asserts that mutative operations are allowed on this object.zConfiguration is read-onlyN)rG �AssertionError)r r r r r* J s z Configuration._assertCanModifyc C s t | j�dkS )z,Returns True if this configuration is empty.r )�lenr/ )r r r r rH O s zConfiguration._emptyc C sT t jj| j�}tj� r&td|g� dS t j|t j�}zt j |� W dt j |� X dS )zIOpen and issue an fsync on the directory containing the config file. rR N) rA rB �dirnamer& r rJ r rN �O_RDONLYrR �close)r rw �fdr r r rT T s z Configuration._fsyncDirectoryc & C s | j jdj| j�� y|jdd� tj|�}W n$ tjjk rR t t d���Y nX y |d }W n^ t k r� t t d���Y n@ t k r� } z$| j jdj|�� t t d���W Y dd}~X nX y| j |� W nB t k �r } z$| j jdj|�� t t d���W Y dd}~X nX d| _dS ) z%Reads in a Configuration from a file.z Reading configuration from {0}r z Not a valid configuration filer' z9Not a valid configuration file (missing 'config' section)z#Not a valid configuration file: {0}NF)r+ r, r- r& �seekr8 Z safe_load�scannerZ ScannerErrorr r? r> � Exceptionrj r0 )r rU Zconfr' r_ r r r rp b s( zConfiguration._readc C s| t j� rtd| jg� dS tjj| j�r<tj| j� | j� y&t | j d��}t j | j= W dQ R X W n t k rv Y nX dS )zfDeletes the current configuration file. In noRun mode, pretend that we're doing an rm of the file.ZrmNzr+)r rJ r r& rA rB rL rM rT r r" r r# r> )r r( r r r rI � s zConfiguration._removeFilec C s" || j kr ttd�j|d���dS )z�Checks a configuration file schema version string against the list of supported schemas. Args: ver (str): the schema version string to check Raises: BadConfigurationFileError: version not supported. z,Configuration file version {v} not supported)�vN)�supportedSchemaVersionsr r? r- )r% Zverr r r �_validateVersion� s z Configuration._validateVersion)F)TF)(r r r r �loggingZ getLoggerr+ r r# r" Zyaml_tagrn � classmethodr) r4 r; �propertyr& r= r@ r. rE rV rW r] r` rc rd rg rh rj r r r* rH rT rp rI r� r r r )r r r < s@ 6 $ r )r Z __future__r r r r r7 r r r Z vdo.utilsr r r r rZ r� rA rX r r\ r8 r} r r r r r r �<module> s"