U ]@sfdZddlZddlZddlmZmZddlmZddlm Z ddlm Z e e Z GdddeZdS) zPlugin storage class.N)AnyDict)errors)os) filesystemc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) PluginStoragez4Class implementing storage functionality for pluginscCs"||_||_d|_d|_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_data _storagepath)selfZconfigZclasskeyr9/usr/lib/python3/dist-packages/certbot/plugins/storage.py__init__s zPluginStorage.__init__cCs&tj|jjd|_|d|_dS)zhInitializes PluginStorage data and reads current state from the disk if the storage json exists.z.pluginstorage.jsonTN)rpathjoinrZ config_dirr _loadr )r rrr_initialize_storage sz!PluginStorage._initialize_storagec Cst}d}z$t|jd}|}W5QRXWnVtk r}z8d|jt|}tj |jrtt |t |W5d}~XYnXzt|}WnHtk r|st d|jn d|j}t |t |YnX||_dS)zReads 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.)dictopenr readIOErrorformatstrrrisfileloggererrorrPluginStorageErrorjsonloads ValueErrordebugr )r dataZfiledatafheerrmsgrrrr(s4  zPluginStorage._loadc Cs|jsd}t|t|zt|j}WnDtk rr}z&d t |}t|t|W5d}~XYnXz@t t |jt jt jBt jBdd}||W5QRXWnHtk r}z*d |jt |}t|t|W5d}~XYnXdS)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}Niwz4Could not write PluginStorage data to file {0} : {1})r rrrr r!dumpsr TypeErrorrrrfdopenrrr O_WRONLYO_CREATO_TRUNCwriter)r r(Z serializedr'r&rrrsaveFs:    zPluginStorage.savecCs@|js||j|jkr,t|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 rr r keysr)r keyvaluerrrputds zPluginStorage.putcCs|js||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 rr r )r r3rrrfetchqszPluginStorage.fetchN) __name__ __module__ __qualname____doc__rrrr1r5r6rrrrrs r)r:r!ZloggingZacme.magic_typingrrZcertbotrZcertbot.compatrrZ getLoggerr7robjectrrrrrs