Source code for geowombat.config

import ast
import configparser
from pathlib import Path

config = {}

config_file = str(Path(__file__).parent.resolve() / 'config.ini')
config_parser = configparser.ConfigParser()
config_parser.read(config_file)

ASSOCIATIONS = {}

for section in config_parser.sections():
    for k, v in config_parser[section].items():
        ASSOCIATIONS[k] = section


def _update_config(config_parser, config_dict):
    for section in config_parser.sections():
        for k, v in config_parser[section].items():

            if v in ['True', 'False']:
                config_dict[k] = True if v == 'True' else False

            elif v == 'None':
                config_dict[k] = None
            else:

                try:
                    config_dict[k] = ast.literal_eval(v)
                except (SyntaxError, ValueError):
                    config_dict[k] = v

    return config_dict


config = _update_config(config_parser, config)


def _set_defaults(d):

    config_parser.read(config_file)
    d = _update_config(config_parser, d)


def _set_bool(d):
    d['with_config'] = True


class update(object):
    """Updates the global configuration parameters. See 'config.ini' for
    parameter options and defaults.

    Note that ``nodata`` is only used/applied as a value setter during
    array warping. I.e., the ``nodata`` argument is not the input 'no data'
    value itself. Rather, it is used to replace 'no data' values in the opened,
    warped array.

    Example:
        >>> with gw.config.update(sensor='l8'):
        >>>     with gw.open('image.tif') as ds:
        >>>         print(ds.gw.config)
    """

    def __init__(self, config=config, **kwargs):
        self.config = config
        self.__set_defaults(config)

        if kwargs:
            self._assign(config, **kwargs)

    def __enter__(self):
        _set_bool(self.config)
        return self.config

    def __exit__(self, type, value, traceback):
        d = self.config
        self.__set_defaults(d)

    @staticmethod
    def __set_defaults(d):
        _set_defaults(d)

    @staticmethod
    def _assign(d, **kwargs):

        if kwargs:
            for k, v in kwargs.items():
                config_parser[ASSOCIATIONS[k]][k] = str(v)

        d = _update_config(config_parser, d)