def __init__(self, config_file=None, section='stackdio'):
super(StackdioConfig, self).__init__()
self.section = section
self._cfg_file = os.path.abspath(config_file or CFG_FILE)
self._config = ConfigParser()
self.usable_file = os.path.isfile(self._cfg_file)
if self.usable_file:
self._config.read(self._cfg_file)
self.usable_section = self._config.has_section(self.section)
self.usable_config = self.usable_file and self.usable_section
if not self.usable_section:
self._config.add_section(section)
def __init__(self, config_file=None, section='stackdio'):
super(StackdioConfig, self).__init__()
self.section = section
self._cfg_file = os.path.abspath(config_file or CFG_FILE)
self._config = ConfigParser()
self.usable_file = os.path.isfile(self._cfg_file)
if self.usable_file:
self._config.read(self._cfg_file)
self.usable_section = self._config.has_section(self.section)
self.usable_config = self.usable_file and self.usable_section
if not self.usable_section:
self._config.add_section(section)
class StackdioConfig(object):
"""
A wrapper around python's ConfigParser class
"""
KEYRING_SERVICE = 'stackdio_cli'
BOOL_MAP = {
str(True): True,
str(False): False,
}
def __init__(self, config_file=None, section='stackdio'):
super(StackdioConfig, self).__init__()
self.section = section
self._cfg_file = os.path.abspath(config_file or CFG_FILE)
self._config = ConfigParser()
self.usable_file = os.path.isfile(self._cfg_file)
if self.usable_file:
self._config.read(self._cfg_file)
self.usable_section = self._config.has_section(self.section)
self.usable_config = self.usable_file and self.usable_section
if not self.usable_section:
self._config.add_section(section)
def save(self):
full_path = os.path.dirname(self._cfg_file)
if not os.path.isdir(full_path):
os.makedirs(full_path)
with open(self._cfg_file, 'w') as f:
self._config.write(f)
def get_password(self, username=None):
username = username or self.get('username')
if username is not None:
return keyring.get_password(self.KEYRING_SERVICE, username)
else:
return None
def set_password(self, new_password):
username = self.get('username')
if username is None:
raise KeyError('Not username provided')
keyring.set_password(self.KEYRING_SERVICE, username, new_password)
def __contains__(self, item):
try:
self._config.get(self.section, item)
return True
except NoOptionError:
return False
def __getitem__(self, item):
try:
ret = self._config.get(self.section, item)
if ret in self.BOOL_MAP:
return self.BOOL_MAP[ret]
else:
return str(ret)
except NoOptionError:
raise KeyError(item)
def __setitem__(self, key, value):
if isinstance(value, bool):
value = str(value)
self._config.set(self.section, key, value)
def get(self, item, default=None):
try:
return self[item]
except KeyError:
return default
def items(self):
return self._config.items(self.section)
def prompt_for_config(self):
self.get_url()
self.get_username()
self.get_blueprint_dir()
# Save when we're done
self.save()
def _test_url(self, url):
try:
r = requests.get(url, verify=self.get('verify', True))
return (200 <= r.status_code < 300) or r.status_code == 403
except ConnectionError:
return False
except MissingSchema:
click.echo('You might have forgotten http:// or https://')
return False
def _test_credentials(self, username, password):
try:
r = requests.get(self['url'],
verify=self.get('verify', True),
auth=(username, password))
return 200 <= r.status_code < 300
except (ConnectionError, MissingSchema):
click.echo('There is something wrong with your URL.')
return False
def get_url(self):
if self.get('url') is not None:
if click.confirm('Keep existing url ({0})?'.format(self['url']),
default=True):
return
self['verify'] = not click.confirm(
'Does your stackd.io server have a self-signed '
'SSL certificate?')
new_url = None
while new_url is None:
url = click.prompt('What is the URL of your stackd.io server',
prompt_suffix='? ')
if url.endswith('api'):
url += '/'
elif url.endswith('api/'):
pass
elif url.endswith('/'):
url += 'api/'
else:
url += '/api/'
if self._test_url(url):
new_url = url
else:
click.echo(
'There was an error while attempting to contact that server. '
'Try again.')
self['url'] = new_url
def get_username(self):
valid_creds = False
while not valid_creds:
keep_username = False
username = self.get('username')
if username is not None:
if click.confirm(
'Keep existing username ({0})?'.format(username),
default=True):
keep_username = True
if not keep_username:
username = click.prompt('What is your stackd.io username',
prompt_suffix='? ')
password = self.get_password(username)
keep_password = False
if password is not None:
if click.confirm('Keep existing password for user {0}?'.format(
username),
default=True):
keep_password = True
if not keep_password:
password = click.prompt(
'What is the password for {0}'.format(username),
prompt_suffix='? ',
hide_input=True)
if self._test_credentials(username, password):
self['username'] = username
self.set_password(password)
valid_creds = True
else:
click.echo('Invalid credentials. Please try again.')
valid_creds = False
def get_blueprint_dir(self):
blueprints = self.get('blueprint_dir')
if blueprints is not None:
if click.confirm(
'Keep existing blueprints directory ({0})?'.format(
blueprints),
default=True):
return
self['blueprint_dir'] = click.prompt(
'Where are your blueprints stored',
prompt_suffix='? ',
type=UserPath(exists=True, file_okay=False, resolve_path=True))
class StackdioConfig(object):
"""
A wrapper around python's ConfigParser class
"""
KEYRING_SERVICE = 'stackdio_cli'
BOOL_MAP = {
str(True): True,
str(False): False,
}
def __init__(self, config_file=None, section='stackdio'):
super(StackdioConfig, self).__init__()
self.section = section
self._cfg_file = os.path.abspath(config_file or CFG_FILE)
self._config = ConfigParser()
self.usable_file = os.path.isfile(self._cfg_file)
if self.usable_file:
self._config.read(self._cfg_file)
self.usable_section = self._config.has_section(self.section)
self.usable_config = self.usable_file and self.usable_section
if not self.usable_section:
self._config.add_section(section)
def save(self):
full_path = os.path.dirname(self._cfg_file)
if not os.path.isdir(full_path):
os.makedirs(full_path)
with open(self._cfg_file, 'w') as f:
self._config.write(f)
def get_password(self, username=None):
username = username or self.get('username')
if username is not None:
return keyring.get_password(self.KEYRING_SERVICE, username)
else:
return None
def set_password(self, new_password):
username = self.get('username')
if username is None:
raise KeyError('Not username provided')
keyring.set_password(self.KEYRING_SERVICE, username, new_password)
def __contains__(self, item):
try:
self._config.get(self.section, item)
return True
except NoOptionError:
return False
def __getitem__(self, item):
try:
ret = self._config.get(self.section, item)
if ret in self.BOOL_MAP:
return self.BOOL_MAP[ret]
else:
return str(ret)
except NoOptionError:
raise KeyError(item)
def __setitem__(self, key, value):
if isinstance(value, bool):
value = str(value)
self._config.set(self.section, key, value)
def get(self, item, default=None):
try:
return self[item]
except KeyError:
return default
def items(self):
return self._config.items(self.section)
def prompt_for_config(self):
self.get_url()
self.get_username()
self.get_blueprint_dir()
# Save when we're done
self.save()
def _test_url(self, url):
try:
r = requests.get(url, verify=self.get('verify', True))
return (200 <= r.status_code < 300) or r.status_code == 403
except ConnectionError:
return False
except MissingSchema:
click.echo('You might have forgotten http:// or https://')
return False
def _test_credentials(self, username, password):
try:
r = requests.get(self['url'],
verify=self.get('verify', True),
auth=(username, password))
return 200 <= r.status_code < 300
except (ConnectionError, MissingSchema):
click.echo('There is something wrong with your URL.')
return False
def get_url(self):
if self.get('url') is not None:
if click.confirm('Keep existing url ({0})?'.format(self['url']), default=True):
return
self['verify'] = not click.confirm('Does your stackd.io server have a self-signed '
'SSL certificate?')
new_url = None
while new_url is None:
url = click.prompt('What is the URL of your stackd.io server', prompt_suffix='? ')
if url.endswith('api'):
url += '/'
elif url.endswith('api/'):
pass
elif url.endswith('/'):
url += 'api/'
else:
url += '/api/'
if self._test_url(url):
new_url = url
else:
click.echo('There was an error while attempting to contact that server. '
'Try again.')
self['url'] = new_url
def get_username(self):
valid_creds = False
while not valid_creds:
keep_username = False
username = self.get('username')
if username is not None:
if click.confirm('Keep existing username ({0})?'.format(username), default=True):
keep_username = True
if not keep_username:
username = click.prompt('What is your stackd.io username', prompt_suffix='? ')
password = self.get_password(username)
keep_password = False
if password is not None:
if click.confirm('Keep existing password for user {0}?'.format(username),
default=True):
keep_password = True
if not keep_password:
password = click.prompt('What is the password for {0}'.format(username),
prompt_suffix='? ', hide_input=True)
if self._test_credentials(username, password):
self['username'] = username
self.set_password(password)
valid_creds = True
else:
click.echo('Invalid credentials. Please try again.')
valid_creds = False
def get_blueprint_dir(self):
blueprints = self.get('blueprint_dir')
if blueprints is not None:
if click.confirm('Keep existing blueprints directory ({0})?'.format(blueprints),
default=True):
return
self['blueprint_dir'] = click.prompt('Where are your blueprints stored',
prompt_suffix='? ',
type=UserPath(exists=True, file_okay=False,
resolve_path=True))
