File: //usr/lib/python3.6/site-packages/ipalib/__pycache__/sysrestore.cpython-36.opt-1.pyc
3
�d[e]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�eZeje�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 || _ tjj| j |�| _tj� | _i | _| j� 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 � /usr/lib/python3.6/sysrestore.py�__init__= s
zFileStore.__init__c C sp t jd| j� i | _tdd�}t|_|j| j� x:|j� D ].}|dkr:x |j |�D ]\}}|| j|< qRW q:W dS )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)�
interpolationr )
�logger�debugr
r r �str� optionxform�read�sections�items)r �pZsection�key�valuer r r r L s
zFileStore._loadc C s� t jd| j� t| j�dkrDt jd� tjj| j�r@tj| j� dS t dd�}t
|_ |j d� x(| jj
� D ]\}}|jd|t
|�� qjW t| jd��}|j|� W dQ R X 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 fileN)r 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 jd|� tjj|�s td��tjj|�s<t jd|� dS tjj|�\}}t|d��}t |j
� �j � }W dQ R X dj ||d�}tjj
| j|�}tjj|�r�t jd|� dS tj||� tj|�}d } | j ||d
�| j|< | 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'zAbsolute 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" �shutilZcopy2r+ r r) )
r r Z_reldirZ
backupfiler( Z cont_hash�filename� backup_pathr+ �templater r r � backup_filet s*
zFileStore.backup_filec C sB d}x8| j j� D ]*\}}|jdd�\}}}}||krd}P qW |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 r/ ) r r �resultZ_keyr �_mode�_uid�_gid�filepathr r r �has_file� s zFileStore.has_fileNc C s> |dkrt jd|� nt jd||� tjj|�s8td��|dk rVtjj|� rVtd��d}d}d}d}x8| jj� D ]*\}}|jdd�\}}}} | |krr|}P qrW |s�td��tjj | j
|�}
tjj |
�s�t jd |
� d
S |dk r�|}t j
|
|� tj|
� tj|t|�t|�� tj|t|�� tj|� | j|= | 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'zAbsolute path requiredzAbsolute new path requiredr6 r7 z No such file name in the indexz' -> Not restoring - '%s' doesn't existFT)r r r
r r, r- r r r/ r r r" r1 �copyr# �chown�int�chmodr �restore_contextr) ) r r �new_path�mode�uid�gidr2 r r r< r3 r r r � restore_file� sF
zFileStore.restore_filec C s� t | j�dkrdS x�| jj� D ]�\}}|jdd�\}}}}tjj| j|�}tjj|�sdt j
d|� q t j ||� tj
|� tj|t|�t|�� tj|t|�� tj|� q W i | _| j� 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 Fr6 r7 z' -> Not restoring - '%s' doesn't existT)r! r r r/ r
r r r r" r r r1 r> r# r? r@ rA r rB r) )r r2 r rD rE rF r r3 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 jd|� tjj|�s td��d}x8| jj� D ]*\}}|jdd�\}}}}||kr0|}P q0W |sjtd��tjj | j
|�} tjj | �s�t jd| � dS ytj | � W n4 t
k
r� }
zt jd | t|
�� W Y dd}
~
X nX | j|= | 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'zAbsolute path requiredNr6 r7 z No such file name in the indexz' -> Not restoring - '%s' doesn't existFzError removing %s: %sT)r r r
r r, r- r r r/ r r r" �unlink� Exception�errorr r) ) r r r2 r r r9 r: r; r< r3 �er r r � untrack_file s. $zFileStore.untrack_file)N)�__name__�
__module__� __qualname__�__doc__�SYSRESTORE_PATH�SYSRESTORE_INDEXFILEr r r) r5 r= rG rH rI rN 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 jj||�| _i | _| j� 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 jd| j� i | _tdd�}t|_|j| j� xb|j� D ]V}i | j|< xF|j |�D ]8\}}|td�krnd}n|td�kr~d}|| j| |< qTW q:W dS )zwLoad the modules from the file @_path. @modules will
be an empty dictionary if the file doesn't exist.
zLoading StateFile from '%s'N)r TF)
r r r rV r r r r r r )r r �moduler r r r r r S s
zStateFile._loadc C s� t jd| j� x,t| j�D ] }t| j| �dkr| j|= qW t| j�dkrrt jd� tjj| j�rntj | j� dS t
dd�}t |_ xH| jj
� D ]:\}}|j|� x&|j
� D ]\}}|j||t |�� q�W q�W t| jd��}|j|� W dQ R X 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 fileN)r r )r r r �listrV r! r
r r" r# r r r r r$ r% r& r' )r rW r �valsr r r( r r r r) h s"
zStateFile.savec C sX t |tttf�std��| j� || jkr4i | j|< || jkrL|| j| |< | 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 rV r) )r rW r r r r r � backup_state� s
zStateFile.backup_statec C s( | j � || jkrdS | j| 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 rV �get)r rW r r r r � get_state� s
zStateFile.get_statec C s: | j � y| j| |= W n tk
r, Y n
X | j� dS )z�Delete system state from @module, identified by the string
@key.
If the item doesn't exist, no change is done.
N)r rV �KeyErrorr) )r rW r r r r � delete_state� s zStateFile.delete_statec C s$ | j ||�}|dk r | j||� |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)r_ ra )r rW r r r r r �
restore_state� s zStateFile.restore_statec C s
|| j kS )z�Return True or False if there is any state stored for @module.
Can be used to determine if a service is configured.
)rV )r rW r r r � has_state� s zStateFile.has_stateN)rO rP rQ rR rS �SYSRESTORE_STATEFILEr r r) r] r_ ra rb rc r r r r rU 5 s rU ) Z
__future__r Zloggingr
Zos.pathr1 r Zhashlibr ZsixZPY3Z configparserr r Zipaplatform.tasksr Zipaplatform.pathsr r r\ Z getLoggerrO r ZTMPrS rT rd r rU r r r r �<module> s*
|