File: //lib/python3.9/site-packages/certbot/plugins/__pycache__/storage.cpython-39.pyc
a
}|�g� � @ sx d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z e� e
�ZG d d
� d
�ZdS ) zPlugin storage class.� N)�Any)�Dict)�
configuration)�errors)�
filesystem)�osc @ sp e Zd ZdZejedd�dd�Zdd�dd�Zdd�d d
�Z dd�d d �Z
ee dd
�dd�Z ee d�dd�Z
dS )�
PluginStoragez4Class implementing storage functionality for pluginsN)�config�classkey�returnc 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)�selfr r
� r �;/usr/lib/python3.9/site-packages/certbot/plugins/storage.py�__init__ s
zPluginStorage.__init__)r 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 Z
config_dir� _storagepath�_loadr )r r r r �_initialize_storage! s z!PluginStorage._initialize_storagec
C s� i }d}z8t | jd��}|�� }W d � n1 s40 Y W nX ty� } z@d�| jt|��}tj�| j�r�t �
|� t � |��W Y d}~n
d}~0 0 zt
�|�}W nF ty� |s�t �d| j� n d�| j�}t �
|� t � |��Y n0 || _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�OSError�format�strr r �isfile�logger�errorr �PluginStorageError�json�loads�
ValueError�debug�_data)r �dataZfiledata�fh�e�errmsgr r r r ) s2 *
�
��
zPluginStorage._loadc
C s | j s d}t�|� t�|��zt�| j�}W nF tyt } z.d� t
|��}t�|� t�|��W Y d}~n
d}~0 0 zTt � t
�| jt jt jB t jB d�d��}|�|� W d � n1 s�0 Y W nL t�y } z2d� | jt
|��}t�|� t�|��W Y d}~n
d}~0 0 dS )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+ Z
serializedr* r) r r r �saveG s8
�
��,
�
zPluginStorage.save)�key�valuer c C s: | j s| �� | j| jvr&i | 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 r5 r6 r r r �pute s
zPluginStorage.put)r5 r c 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 r5 r r r �fetchr s zPluginStorage.fetch)�__name__�
__module__� __qualname__�__doc__r ZNamespaceConfigr r r r r4 r r7 r8 r r r r r s
r )r< r# Zlogging�typingr r Zcertbotr r Zcertbot.compatr r Z getLoggerr9 r r r r r r �<module> s