electrum-bitcoinprivate/lib/simple_config.py

import json
import os
from util import user_dir

class SimpleConfig:

    default_options = {"gui": "lite"}
    
    def __init__(self):
        # Find electrum data folder
        self.config_folder = user_dir()
        # Read the file
        if os.path.exists(self.config_file_path()):
            self.load_config()
        else:
            self.config = self.default_options
            # Make config directory if it does not yet exist.
            if not os.path.exists(self.config_folder):
                os.mkdir(self.config_folder)
            self.save_config()
        
    def set_key(self, key, value, save = True):
        self.config[key] = value
        if save == True:
            self.save_config()
    
    def save_config(self):
        f = open(self.config_file_path(), "w+")
        f.write(json.dumps(self.config))
    
    def load_config(self):
        f = open(self.config_file_path(), "r")
        file_contents = f.read()
        if file_contents:
            self.config = json.loads(file_contents)
        else:
            self.config = self.default_options
            self.save_config()
    
    def config_file_path(self):
        return "%s" % (self.config_folder + "/config.json")