← Back
Editing: storage.cpython-38.pyc
U �]� � @ sf d Z ddlZddlZddlmZmZ ddlmZ ddlm Z ddlm Z e� e �Z G dd� de�ZdS ) zPlugin storage class.� N)�Any�Dict)�errors)�os)� filesystemc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )� PluginStoragez4Class implementing storage functionality for pluginsc C s" || _ || _d| _d| _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�_data� _storagepath)�selfZconfigZclasskey� 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 Z config_dirr �_loadr )r r r r �_initialize_storage s z!PluginStorage._initialize_storagec C s� t � }d}z$t| jd��}|�� }W 5 Q R X W nV tk r� } z8d�| jt|��}tj� | j�rtt � |� t � |��W 5 d}~X Y nX zt�|�}W nH tk r� |s�t �d| j� n d�| j�}t � |� t � |��Y nX || _dS )z�Reads PluginStorage content from the disk to a dict structure :raises .errors.PluginStorageError: when unable to open or read the file � �rz1Could not read PluginStorage data file: {0} : {1}Nz2Plugin storage file %s was empty, no values loadedz$PluginStorage file {0} is corrupted.)�dict�openr �read�IOError�format�strr r �isfile�logger�errorr �PluginStorageError�json�loads� ValueError�debugr )r �dataZfiledata�fh�e�errmsgr r r r ( s4 � �� zPluginStorage._loadc C s | j s d}t�|� t�|��zt�| j�}W nD tk rr } z&d� t |��}t�|� t�|��W 5 d}~X Y nX z@t � t �| jt jt jB t jB d�d��}|�|� W 5 Q R X W nH tk r� } z*d� | jt |��}t�|� t�|��W 5 d}~X Y nX 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 �saveF s: � �� � zPluginStorage.savec C s@ | j s| �� | j| j�� kr,t� | 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 �keysr )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 r3 r r r �fetchq s zPluginStorage.fetchN) �__name__� __module__� __qualname__�__doc__r r r r1 r5 r6 r r r r r s r )r: r! ZloggingZacme.magic_typingr r Zcertbotr Zcertbot.compatr r Z getLoggerr7 r �objectr r r r r �<module> s
Save File
Cancel