Source code for octodns.secret.environ

#
#
#

from os import environ

from .base import BaseSecrets
from .exception import SecretsException




class EnvironSecretsException(SecretsException):
    pass





class EnvironSecrets(BaseSecrets):


    def fetch(self, name, source):
        # check for a default value, format is VARIABLE_NAME/DEFAULT_VALUE
        default = None
        if '/' in name:
            name, default = name.split('/', 1)

        # expand env variables
        try:
            v = environ[name]
        except KeyError:
            if default is not None:
                v = default
            else:
                self.log.exception('Invalid provider config')
                raise EnvironSecretsException(
                    f'Incorrect provider config, missing env var {name}, {source.context}'
                )

        try:
            if '.' in v:
                # has a dot, try converting it to a float
                v = float(v)
            else:
                # no dot, try converting it to an int
                v = int(v)
        except ValueError:
            # just leave it as a string
            pass

        return v