CasperSecurity
o
6��aB � @ sl d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
e� e �Z
G dd � d �ZdS )
zPlugin storage class.� N)�Any)�Dict)�errors)�
filesystem)�osc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
PluginStoragez4Class implementing storage functionality for pluginsc C s || _ || _d| _| | dS )z�Initializes PluginStorage object storing required configuration
options.
:param .configuration.NamespaceConfig config: Configuration object
:param str classkey: class name to use as root key in storage file
FN)�_config� _classkey� _initialized)�self�config�classkey� r �9/usr/lib/python3/dist-packages/certbot/plugins/storage.py�__init__ s
zPluginStorage.__init__c C s& t j�| jjd�| _| �� d| _dS )zhInitializes PluginStorage data and reads current state from the disk
if the storage json exists.z.pluginstorage.jsonTN)r �path�joinr �
config_dir� _storagepath�_loadr
)r r r r �_initialize_storage s
z!PluginStorage._initialize_storagec
C s� i }d}z t | jd�� }|�� }W d � n1 sw Y W n, tyL } z d�| jt|��}tj�| j�rBt �
|� t � |��W Y d}~nd}~ww zt
�|�}W n# tyw |set �d| j� nd�| j�}t �
|� t � |��Y nw || _dS )z�Reads PluginStorage content from the disk to a dict structure
:raises .errors.PluginStorageError: when unable to open or read the file
� �rNz1Could not read PluginStorage data file: {0} : {1}z2Plugin storage file %s was empty, no values loadedz$PluginStorage file {0} is corrupted.)�openr �read�IOError�format�strr r �isfile�logger�errorr �PluginStorageError�json�loads�
ValueError�debug�_data)r �data�filedata�fh�e�errmsgr r r r ( s@
��
�
��� ��
��
zPluginStorage._loadc
C s | j sd}t�|� t�|��zt�| j�}W n ty4 } zd� t
|��}t�|� t�|��d}~ww z-t � t
�| jt jt jB t jB d�d��}|�|� W d � W dS 1 s[w Y W dS ty� } zd� | jt
|��}t�|� t�|��d}~ww )z�Saves PluginStorage content to disk
:raises .errors.PluginStorageError: when unable to serialize the data
or write it to the filesystem
z;Unable to save, no values have been added to PluginStorage.z+Could not serialize PluginStorage data: {0}Ni� �wz4Could not write PluginStorage data to file {0} : {1})r
r r r r! r" �dumpsr& � TypeErrorr r r �fdopenr r r �O_WRONLY�O_CREAT�O_TRUNC�writer )r r+ �
serializedr* r) r r r �saveF sB
�
���� &�
�
��zPluginStorage.savec C s: | j s| �� | j| jvri | j| j< || j| j |< dS )z�Put configuration value to PluginStorage
:param str key: Key to store the value to
:param value: Data to store
N)r
r r r& )r �key�valuer r r �putd s
zPluginStorage.putc C s | j s| �� | j| j | S )z�Get configuration value from PluginStorage
:param str key: Key to get value from the storage
:raises KeyError: If the key doesn't exist in the storage
)r
r r& r )r r6 r r r �fetchq s zPluginStorage.fetchN)
�__name__�
__module__� __qualname__�__doc__r r r r5 r8 r9 r r r r r s
r )r= r" �logging�typingr r �certbotr �certbot.compatr r � getLoggerr: r r r r r r �<module> s