Source code for mountain_tapir.config

# -*- coding: utf-8 -*-
#
# Mountain Tapir Collage Maker is a tool for combining images into collages.
# Copyright (c) 2016, tttppp
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

try:
    import configparser
except ImportError:
    import ConfigParser as configparser

import os


class Config:
    def __init__(self):
        self.config = configparser.RawConfigParser()
        home = os.path.expanduser('~')
        dotDir = os.path.join(home, '.mountain_tapir')
        homeConfigFile = os.path.join(dotDir, 'mountain_tapir.properties')
        readFiles = self.config.read(['mountain_tapir.properties', homeConfigFile])
        # Try to create a properties file if one doesn't already exist.
        if len(readFiles) == 0:
            try:
                if not os.path.exists(dotDir):
                    os.mkdir(dotDir)
                open(homeConfigFile, 'a').close()
                self.persistFile = homeConfigFile
            except:
                # If we can't manage to create one then just run without persisting config.
                self.persistFile = None
        else:
            self.persistFile = readFiles[0]

    def get(self, section, key, default=None, valueType='String'):
        """Get a value from the configuration.

        :param section: The name of the config section.
        :param key: The name of the option within the section.
        :param default: Optional parameter specifying the default value."""
        try:
            if valueType == 'int':
                value = self.config.getint(section, key)
            elif valueType == 'boolean':
                value = self.config.getboolean(section, key)
            else:
                value = self.config.get(section, key)
        except (configparser.NoSectionError, configparser.NoOptionError):
            value = default
        if value is None:
            value = default
        return value

    def update(self, section, key, value):
        """Update a config entry and save the resulting options.

        :param section: The name of the config section.
        :param key: The name of the option within the section.
        :param value: The value to set."""
        if not self.config.has_section(section):
            self.config.add_section(section)
        self.config.set(section, key, value)
        if self.persistFile is not None:
            try:
                with open(self.persistFile, 'w') as outFile:
                    self.config.write(outFile)
            except:
                print('Failed to write to config file: {0}'.format(self.persistFile))