D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
lib
/
python3.9
/
site-packages
/
ipalib
/
__pycache__
/
Filename :
sysrestore.cpython-39.pyc
back
Copy
a }�f]6 � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl Z e j r\d dl m Z n d dl m Z d dlmZ d dlmZ e j r�eZe�e�ZejZdZd ZG d d � d �ZG d d � d �Z dS )� )�absolute_importN)�sha256)� ConfigParser)�SafeConfigParser)�tasks)�pathszsysrestore.indexzsysrestore.statec @ s` e Zd ZdZeefdd�Zdd� Zdd� Zdd � Z d d � Z dd d�Z dd� Z dd� Z dd� Zd S )� FileStorez.Class for handling backup and restore of filesc C s4 || _ tj�| j |�| _t�� | _i | _| �� dS )z�Create a _StoreFiles object, that uses @path as the base directory. The file @path/sysrestore.index is used to store information about the original location of the saved files. N) �_path�os�path�join�_index�randomZRandom�files�_load)�selfr Z index_file� r �5/usr/lib/python3.9/site-packages/ipalib/sysrestore.py�__init__= s zFileStore.__init__c C sh t �d| j� i | _tdd�}t|_|�| j� |�� D ]*}|dkr8|� |�D ]\}}|| j|< qNq8dS )zvLoad the file list from the index file. @files will be an empty dictionary if the file doesn't exist. z Loading Index file from '%s'N�Z interpolationr ) �logger�debugr r r �str� optionxform�read�sections�items)r �p�section�key�valuer r r r L s zFileStore._loadc C s� t �d| j� t| j�dkrDt �d� tj�| j�r@t�| j� dS t dd�}t |_ |� d� | j� � D ]\}}|�d|t |�� qht| jd��}|�|� W d � n1 s�0 Y dS )zkSave the file list to @_index. If @files is an empty dict, then @_index should be removed. zSaving Index File to '%s'r z -> no files, removing fileNr r �w)r r r �lenr r r �exists�remover r r � add_sectionr �set�open�write)r r r r �fr r r �save^ s zFileStore.savec C s� t �d|� tj�|�s td��tj�|�s<t �d|� dS tj�|�\}}t|d�� }t |� � �� � }W d � n1 s|0 Y dj ||d�}tj� | j|�}tj�|�r�t �d|� dS t�||� t�|�}d } | j ||d �| j|< | �� dS ) z�Create a copy of the file at @path - as long as an exact copy does not already exist - which will be restored to its original location by restore_files(). z)Backing up system configuration file '%s'�Absolute path requiredz( -> Not backing up - '%s' doesn't existN�rbz{hexhash}-{bcppath})ZhexhashZbcppathz1 -> Not backing up - already have a copy of '%s'z1{stat.st_mode},{stat.st_uid},{stat.st_gid},{path})�statr )r r r r �isabs� ValueError�isfile�splitr' r r Z hexdigest�formatr r r# �shutil�copy2r- r r* ) r r Z_reldirZ backupfiler) Z cont_hash�filename� backup_pathr- �templater r r � backup_filet s. .� � zFileStore.backup_filec C s@ d}| j �� D ],\}}|�dd�\}}}}||krd} q<q|S )z�Checks whether file at @path was added to the file store Returns #True if the file exists in the file store, #False otherwise F�,� T)r r r1 ) r r �resultZ_keyr �_mode�_uid�_gid�filepathr r r �has_file� s zFileStore.has_fileNc C s: |du rt �d|� nt �d||� tj�|�s8td��|durTtj�|�sTtd��d}d}d}d}| j�� D ],\}}|�dd�\}}}} | |krn|} q�qn|s�td��tj� | j |�} tj� | �s�t �d | � d S |dur�|}t � | |� t�| � t�|t|�t|�� t�|t|�� t�|� | j|= | �� d S ) a< Restore the copy of a file at @path to its original location and delete the copy. Takes optional parameter @new_path which specifies the location where the file is to be restored. Returns #True if the file was restored, #False if there was no backup file to restore Nz(Restoring system configuration file '%s'z0Restoring system configuration file '%s' to '%s'r+ zAbsolute new path requiredr9 r: � No such file name in the index�' -> Not restoring - '%s' doesn't existFT)r r r r r. r/ r r r1 r r r# r3 �copyr$ �chown�int�chmodr �restore_contextr* ) r r �new_path�mode�uid�gidr5 r r r? r6 r r r � restore_file� sL �� � zFileStore.restore_filec C s� t | j�dkrdS | j�� D ]�\}}|�dd�\}}}}tj�| j|�}tj�|�sbt � d|� q t � ||� t� |� t�|t|�t|�� t�|t|�� t�|� q i | _| �� dS )z�Restore the files in the inbdex to their original location and delete the copy. Returns #True if the file was restored, #False if there was no backup file to restore r Fr9 r: rB T)r" r r r1 r r r r r# r r r3 rC r$ rD rE rF r rG r* )r r5 r rI rJ rK r r6 r r r �restore_all_files� s$ � zFileStore.restore_all_filesc C s t | j�dkS )zReturn True or False if there are any files in the index Can be used to determine if a program is configured. r )r" r )r r r r � has_files s zFileStore.has_filesc C s� t �d|� tj�|�s td��d}| j�� D ],\}}|�dd�\}}}}||kr.|} q\q.|shtd��tj� | j |�} tj� | �s�t �d| � dS zt� | � W n6 t y� } z t �d | t| �� W Y d} ~ n d} ~ 0 0 | j|= | �� d S ) z�Remove file at path @path from list of backed up files. Does not remove any files from the filesystem. Returns #True if the file was untracked, #False if there was no backup file to restore z)Untracking system configuration file '%s'r+ Nr9 r: rA rB FzError removing %s: %sT)r r r r r. r/ r r r1 r r r# �unlink� Exception�errorr r* ) r r r5 r r r< r= r> r? r6 �er r r � untrack_file s0 �(zFileStore.untrack_file)N)�__name__� __module__� __qualname__�__doc__�SYSRESTORE_PATH�SYSRESTORE_INDEXFILEr r r* r8 r@ rL rM rN rS r r r r r : s $ ;#r c @ sV e Zd ZdZeefdd�Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dS )� StateFileaf A metadata file for recording system state which can be backed up and later restored. StateFile gets reloaded every time to prevent loss of information recorded by child processes. But we do not solve concurrency because there is no need for it right now. The format is something like: [httpd] running=True enabled=False c C s" t j�||�| _i | _| �� dS )a� Create a StateFile object, loading from @path. The dictionary @modules, a member of the returned object, is where the state can be modified. @modules is indexed using a module name to return another dictionary containing key/value pairs with the saved state of that module. The keys in these latter dictionaries are arbitrary strings and the values may either be strings or booleans. N)r r r r �modulesr )r r Z state_filer r r r B s zStateFile.__init__c C s� t �d| j� i | _tdd�}t|_|�| j� |�� D ]R}i | j|< |� |�D ]8\}}|td�krjd}n|td�krzd}|| j| |< qPq8dS )zwLoad the modules from the file @_path. @modules will be an empty dictionary if the file doesn't exist. zLoading StateFile from '%s'Nr TF) r r r r[ r r r r r r )r r �moduler r r r r r S s zStateFile._loadc C s� t �d| j� t| j�D ] }t| j| �dkr| j|= qt| j�dkrnt �d� tj�| j�rjt� | j� dS t dd�}t |_ | j� � D ]6\}}|�|� |� � D ]\}}|�||t |�� q�q�t| jd��}|�|� W d � n1 s�0 Y dS )ziSave the modules to @_path. If @modules is an empty dict, then @_path should be removed. zSaving StateFile to '%s'r z -> no modules, removing fileNr r! )r r r �listr[ r" r r r# r$ r r r r r% r& r' r( )r r\ r Zvalsr r r) r r r r* h s" zStateFile.savec C sX t |tttf�std��| �� || jvr4i | j|< || jvrL|| j| |< | �� dS )z�Backup an item of system state from @module, identified by the string @key and with the value @value. @value may be a string or boolean. z7Only strings, booleans or unicode strings are supportedN)� isinstancer �bool�unicoder/ r r[ r* �r r\ r r r r r � backup_state� s zStateFile.backup_statec C s( | � � || jvrdS | j| �|d�S )z�Return the value of an item of system state from @module, identified by the string @key. If the item doesn't exist, #None will be returned, otherwise the original string or boolean value is returned. N)r r[ �get�r r\ r r r r � get_state� s zStateFile.get_statec C s8 | � � z| j| |= W n ty* Y n 0 | �� dS )z�Delete system state from @module, identified by the string @key. If the item doesn't exist, no change is done. N)r r[ �KeyErrorr* rd r r r � delete_state� s zStateFile.delete_statec C s$ | � ||�}|dur | �||� |S )a Return the value of an item of system state from @module, identified by the string @key, and remove it from the backed up system state. If the item doesn't exist, #None will be returned, otherwise the original string or boolean value is returned. N)re rg ra r r r � restore_state� s zStateFile.restore_statec C s || j v S )z�Return True or False if there is any state stored for @module. Can be used to determine if a service is configured. )r[ )r r\ r r r � has_state� s zStateFile.has_stateN)rT rU rV rW rX �SYSRESTORE_STATEFILEr r r* rb re rg rh ri r r r r rZ 5 s rZ ) Z __future__r Zloggingr Zos.pathr3 r Zhashlibr ZsixZPY3Z configparserr r Zipaplatform.tasksr Zipaplatform.pathsr r r` Z getLoggerrT r ZTMPrX rY rj r rZ r r r r �<module> s* |