import common
import yaml
import io

N = common.N
E = common.E
R = common.R
RO = common.RO
exec_cmd = common.exec_cmd
parse = common.parse
err = common.err
check_file_exists = common.check_file_exists
LOG = common.LOG

class ConfigureSettings(object):

    def __init__(self, yaml_config_dir, env_id, dea):
        self.yaml_config_dir = yaml_config_dir
        self.env_id = env_id
        self.dea = dea

    def download_settings(self):
        LOG.debug('Download settings for environment %s\n' % self.env_id)
        r, c = exec_cmd('fuel settings --env %s --download --dir %s'
                        % (self.env_id, self.yaml_config_dir))

    def upload_settings(self):
        LOG.debug('Upload settings for environment %s\n' % self.env_id)
        r, c = exec_cmd('fuel settings --env %s --upload --dir %s'
                        % (self.env_id, self.yaml_config_dir))

    def config_settings(self):
        LOG.debug('Configure settings\n')
        self.download_settings()
        self.modify_settings()
        self.upload_settings()

    def modify_settings(self):
        LOG.debug('Modify settings for environment %s\n' % self.env_id)
        settings_yaml = (self.yaml_config_dir + '/settings_%s.yaml'
                         % self.env_id)
        check_file_exists(settings_yaml)

        settings = self.dea.get_settings()

        with io.open(settings_yaml, 'w') as stream:
            yaml.dump(settings, stream, default_flow_style=False)