Gecko [ 167.86.127.34 ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	30 B	-rw-r--r--
common.py	16.9 KB	-rw-r--r--
common_test.py	16.53 KB	-rw-r--r--
disco.py	9.92 KB	-rw-r--r--
disco_test.py	11.34 KB	-rw-r--r--
dns_common.py	11.7 KB	-rw-r--r--
dns_common_lexicon.py	5.39 KB	-rw-r--r--
dns_common_lexicon_test.py	651 B	-rw-r--r--
dns_common_test.py	8.25 KB	-rw-r--r--
dns_test_common.py	1.61 KB	-rw-r--r--
dns_test_common_lexicon.py	5.48 KB	-rw-r--r--
enhancements.py	5.58 KB	-rw-r--r--
enhancements_test.py	2.36 KB	-rw-r--r--
manual.py	10.6 KB	-rw-r--r--
manual_test.py	7.37 KB	-rw-r--r--
null.py	1.34 KB	-rw-r--r--
null_test.py	624 B	-rw-r--r--
selection.py	13.55 KB	-rw-r--r--
selection_test.py	7.76 KB	-rw-r--r--
standalone.py	11.36 KB	-rw-r--r--
standalone_test.py	9.26 KB	-rw-r--r--
storage.py	4.08 KB	-rw-r--r--
storage_test.py	5.37 KB	-rw-r--r--
util.py	1.7 KB	-rw-r--r--
util_test.py	1.61 KB	-rw-r--r--
webroot.py	11.9 KB	-rw-r--r--
webroot_test.py	11.95 KB	-rw-r--r--

Code Editor : storage.py

"""Plugin storage class."""
import json
import logging
import os

from acme.magic_typing import Any, Dict  # pylint: disable=unused-import, no-name-in-module
from certbot import errors

logger = logging.getLogger(__name__)

class PluginStorage(object):
    """Class implementing storage functionality for plugins"""

def __init__(self, config, classkey):
        """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

"""

self._config = config
        self._classkey = classkey
        self._initialized = False
        self._data = None
        self._storagepath = None

def _initialize_storage(self):
        """Initializes PluginStorage data and reads current state from the disk
        if the storage json exists."""

self._storagepath = os.path.join(self._config.config_dir, ".pluginstorage.json")
        self._load()
        self._initialized = True

def _load(self):
        """Reads PluginStorage content from the disk to a dict structure

:raises .errors.PluginStorageError: when unable to open or read the file
        """
        data = dict()  # type: Dict[str, Any]
        filedata = ""
        try:
            with open(self._storagepath, 'r') as fh:
                filedata = fh.read()
        except IOError as e:
            errmsg = "Could not read PluginStorage data file: {0} : {1}".format(
                self._storagepath, str(e))
            if os.path.isfile(self._storagepath):
                # Only error out if file exists, but cannot be read
                logger.error(errmsg)
                raise errors.PluginStorageError(errmsg)
        try:
            data = json.loads(filedata)
        except ValueError:
            if not filedata:
                logger.debug("Plugin storage file %s was empty, no values loaded",
                             self._storagepath)
            else:
                errmsg = "PluginStorage file {0} is corrupted.".format(
                    self._storagepath)
                logger.error(errmsg)
                raise errors.PluginStorageError(errmsg)
        self._data = data

def save(self):
        """Saves PluginStorage content to disk

:raises .errors.PluginStorageError: when unable to serialize the data
            or write it to the filesystem
        """
        if not self._initialized:
            errmsg = "Unable to save, no values have been added to PluginStorage."
            logger.error(errmsg)
            raise errors.PluginStorageError(errmsg)

try:
            serialized = json.dumps(self._data)
        except TypeError as e:
            errmsg = "Could not serialize PluginStorage data: {0}".format(
                str(e))
            logger.error(errmsg)
            raise errors.PluginStorageError(errmsg)
        try:
            with os.fdopen(os.open(self._storagepath,
                                   os.O_WRONLY | os.O_CREAT | os.O_TRUNC,
                                   0o600), 'w') as fh:
                fh.write(serialized)
        except IOError as e:
            errmsg = "Could not write PluginStorage data to file {0} : {1}".format(
                self._storagepath, str(e))
            logger.error(errmsg)
            raise errors.PluginStorageError(errmsg)

def put(self, key, value):
        """Put configuration value to PluginStorage

:param str key: Key to store the value to
        :param value: Data to store
        """
        if not self._initialized:
            self._initialize_storage()

if not self._classkey in self._data.keys():
            self._data[self._classkey] = dict()
        self._data[self._classkey][key] = value

def fetch(self, key):
        """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
        """
        if not self._initialized:
            self._initialize_storage()

return self._data[self._classkey][key]

${this.title}

Code Editor : storage.py