def partition_defaultboot(partition):
try:
settings_path = partition_mount(
config.DATA['system']['settings_partition'], 'rw')
conf_path = os.path.join(settings_path, 'noobs.conf')
noobs_conf = ConfigParser()
noobs_conf.read(conf_path)
section = 'General'
if not noobs_conf.has_section(section):
noobs_conf.add_section(section)
if partition == config.DATA['system']['recovery_partition']:
noobs_conf.remove_option(section, 'default_partition_to_boot')
noobs_conf.remove_option(section, 'sticky_boot')
else:
noobs_conf.set(section, 'default_partition_to_boot',
str(partition_number(partition)))
noobs_conf.set(section, 'sticky_boot',
str(partition_number(partition)))
output = StringIO()
noobs_conf.write(output)
write_file(conf_path, output.getvalue())
except:
raise
finally:
partition_umount(config.DATA['system']['settings_partition'])
def run(self, args, **kwargs):
if not args.password:
args.password = getpass.getpass()
instance = self.resource(ttl=args.ttl) if args.ttl else self.resource()
cli = BaseCLIApp()
# Determine path to config file
try:
config_file = cli._get_config_file_path(args)
except ValueError:
# config file not found in args or in env, defaulting
config_file = config_parser.ST2_CONFIG_PATH
# Retrieve token
manager = self.manager.create(instance,
auth=(args.username, args.password),
**kwargs)
cli._cache_auth_token(token_obj=manager)
# Update existing configuration with new credentials
config = ConfigParser()
config.read(config_file)
# Modify config (and optionally populate with password)
if not config.has_section('credentials'):
config.add_section('credentials')
config.set('credentials', 'username', args.username)
if args.write_password:
config.set('credentials', 'password', args.password)
else:
# Remove any existing password from config
config.remove_option('credentials', 'password')
config_existed = os.path.exists(config_file)
with open(config_file, 'w') as cfg_file_out:
config.write(cfg_file_out)
# If we created the config file, correct the permissions
if not config_existed:
os.chmod(config_file, 0o660)
return manager
def run(self, args, **kwargs):
if not args.password:
args.password = getpass.getpass()
instance = self.resource(ttl=args.ttl) if args.ttl else self.resource()
cli = BaseCLIApp()
# Determine path to config file
try:
config_file = cli._get_config_file_path(args)
except ValueError:
# config file not found in args or in env, defaulting
config_file = config_parser.ST2_CONFIG_PATH
# Retrieve token
manager = self.manager.create(instance, auth=(args.username, args.password), **kwargs)
cli._cache_auth_token(token_obj=manager)
# Update existing configuration with new credentials
config = ConfigParser()
config.read(config_file)
# Modify config (and optionally populate with password)
if not config.has_section('credentials'):
config.add_section('credentials')
config.set('credentials', 'username', args.username)
if args.write_password:
config.set('credentials', 'password', args.password)
else:
# Remove any existing password from config
config.remove_option('credentials', 'password')
with open(config_file, 'w') as cfg_file_out:
config.write(cfg_file_out)
return manager
inifile=args[0]
# Set up the configuration for the sub-dags
prior_cp=ConfigParser()
prior_cp.optionxform = str
prior_cp.readfp(open(inifile))
main_cp=ConfigParser()
main_cp.optionxform = str
main_cp.readfp(open(inifile))
# Remove gps start and end time options
for option in ['gps-start-time','gps-end-time']:
if main_cp.has_option('input',option):
main_cp.remove_option('input',option)
rundir=os.path.abspath(opts.run_path)
if opts.daglog_path is not None:
prior_cp.set('paths','daglogdir',os.path.join(os.path.abspath(opts.daglog_path),'prior'))
main_cp.set('paths','daglogdir',os.path.join(os.path.abspath(opts.daglog_path),'main'))
daglogdir=os.path.abspath(opts.daglog_path)
else:
prior_cp.set('paths','daglogdir',os.path.join(os.path.abspath(opts.run_path),'prior'))
main_cp.set('paths','daglogdir',os.path.join(os.path.abspath(opts.run_path),'main'))
daglogdir=os.path.abspath(opts.run_path)
webdir=main_cp.get('ppanalysis','webdir')
priordir=os.path.join(rundir,'prior')
maindir=os.path.join(rundir,'main')