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 ZdZejeddddZddddZddd d
Z ddd d Z
ee dd
ddZ ee dddZ
dS )
PluginStoragez4Class implementing storage functionality for pluginsN)configclasskeyreturnc C s || _ || _d| _| | dS )zInitializes 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 pathjoinr 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| jrt
| t |W Y d}~n
d}~0 0 zt
|}W nF ty |st d| j n d| j}t
| t |Y n0 || _dS )zReads 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 readOSErrorformatstrr r isfileloggererrorr PluginStorageErrorjsonloads
ValueErrordebug_data)r dataZfiledatafheerrmsgr 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 dd}|| W d n1 s0 Y W nL ty } z2d | jt
|}t| t|W Y d}~n
d}~0 0 dS )zSaves 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_WRONLYO_CREATO_TRUNCwriter )r r+ Z
serializedr* r) r r r saveG s8
,
zPluginStorage.save)keyvaluer c C s: | j s| | j| jvr&i | j| j< || j| j |< dS )zPut 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 )zGet 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# Zloggingtypingr r Zcertbotr r Zcertbot.compatr r Z getLoggerr9 r r r r r r s