def test_print_config(self):
config_file = 'tests/fixtures/config_flat.yaml'
cfg = config.config_load('myenv', config_file)
self.assertIsInstance(cfg, dict)
self.assertEqual(cfg['env'], 'myenv')
self.assertEqual(cfg['false_boolean'], False)
self.assertEqual(cfg['zero'], 0)
def test_config_dir_override(self):
config_file = 'tests/fixtures/config_flat.yaml'
config_dir = 'tests/fixtures/config.d'
cfg = config.config_load('myenv', config_file, config_dir)
self.assertIsInstance(cfg, dict)
self.assertEqual(cfg['env'], 'myenv')
self.assertEqual(cfg['foo'], 'baz')
self.assertEqual(cfg['comes_from'], '20-config')
def test_config_load_flat(self):
config_file = 'tests/fixtures/config_flat.yaml'
cfg = config.config_load('myenv', config_file)
self.assertIsInstance(cfg, dict)
self.assertEqual(cfg['env'], 'myenv')
self.assertEqual(cfg['foo'], 'bar')
def test_config_load_with_envs(self):
config_file = 'tests/fixtures/config_with_envs.yaml'
cfg = config.config_load('myenv', config_file)
self.assertIsInstance(cfg, dict)
self.assertEqual(cfg['env'], 'myenv')
self.assertEqual(cfg['foo'], 'foo-value-in-myenv')
def test_config_load_no_file(self):
cfg = config.config_load('myenv')
self.assertIsInstance(cfg, dict)
self.assertEqual(cfg['env'], 'myenv')
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
now = datetime.now(tz=pytz.UTC)
if args.subcommand == 'config':
print_config(config,
args.property_name,
output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
# Figure out profile value in the following order
# - cli arg
# - env variable
# - default profile if exists
if args.profile:
profile = args.profile
elif os.environ.get('AWS_DEFAULT_PROFILE'):
profile = os.environ.get('AWS_DEFAULT_PROFILE')
elif profile_exists('default'):
profile = 'default'
elif args.profile and not profile_exists(args.profile):
print('Profile "{}" does not exist.'.format(args.profile))
sys.exit(1)
else:
profile = None
# Figure out region value in the following order
# - cli arg
# - env variable
# - region from config
if args.region:
region = args.region
elif os.environ.get('AWS_DEFAULT_REGION'):
region = os.environ.get('AWS_DEFAULT_REGION')
elif get_region_name(profile):
region = get_region_name(profile)
else:
region = get_default_region_name()
if not region:
print('Region is not specified.')
sys.exit(1)
config['region'] = region
# Not great, but try to catch everything. Above should be refactored in a
# function which handles setting up connections to different aws services
try:
ec2_conn = boto.ec2.connect_to_region(region, profile_name=profile)
vpc_conn = boto.vpc.connect_to_region(region, profile_name=profile)
cf_conn = boto.cloudformation.connect_to_region(region,
profile_name=profile)
r53_conn = boto.route53.connect_to_region(region, profile_name=profile)
s3_conn = boto.s3.connect_to_region(region, profile_name=profile)
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
# TODO(alekna): Fix too broad exception
except:
print(sys.exc_info()[1])
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
stack_name = cf.create_stack(cf_conn,
args.name,
args.template,
config,
dry=args.dry_run)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name,
args.events_follow)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
else:
stack_name = cf.create_stack(
cf_conn,
args.name,
args.template,
config,
update=True,
dry=args.dry_run,
create_on_update=args.create_on_update)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn,
stack_name,
args.events_follow,
from_dt=now)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, region, profile, args.yes)
if args.events_follow:
stack_status = cf.print_events(cf_conn,
args.name,
args.events_follow,
from_dt=now)
if stack_status in FAILED_STACK_STATES:
sys.exit(1)
if args.subcommand == 'events':
cf.print_events(cf_conn, args.name, args.events_follow, args.lines)
if args.subcommand == 'diff':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
cf.print_stack_diff(cf_conn, args.name, args.template, config)
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
now = datetime.now(tz=pytz.UTC)
if args.subcommand == 'config':
print_config(config, args.property_name, output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
# Figure out profile value in the following order
# - cli arg
# - env variable
# - default profile if exists
if args.profile:
profile = args.profile
elif os.environ.get('AWS_DEFAULT_PROFILE'):
profile = os.environ.get('AWS_DEFAULT_PROFILE')
elif profile_exists('default'):
profile = 'default'
elif args.profile and not profile_exists(args.profile):
print('Profile "{}" does not exist.'.format(args.profile))
sys.exit(1)
else:
profile = None
# Figure out region value in the following order
# - cli arg
# - env variable
# - region from config
if args.region:
region = args.region
elif os.environ.get('AWS_DEFAULT_REGION'):
region = os.environ.get('AWS_DEFAULT_REGION')
elif get_region_name(profile):
region = get_region_name(profile)
else:
region = get_default_region_name()
if not region:
print('Region is not specified.')
sys.exit(1)
config['region'] = region
# Not great, but try to catch everything. Above should be refactored in a
# function which handles setting up connections to different aws services
try:
ec2_conn = boto.ec2.connect_to_region(region, profile_name=profile)
vpc_conn = boto.vpc.connect_to_region(region, profile_name=profile)
cf_conn = boto.cloudformation.connect_to_region(region, profile_name=profile)
r53_conn = boto.route53.connect_to_region(region, profile_name=profile)
s3_conn = boto.s3.connect_to_region(region, profile_name=profile)
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
# TODO(alekna): Fix too broad exception
except:
print(sys.exc_info()[1])
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
stack_name = cf.create_stack(cf_conn, args.name, args.template, config, dry=args.dry_run)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name, args.events_follow)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
else:
stack_name = cf.create_stack(cf_conn, args.name, args.template, config, update=True, dry=args.dry_run,
create_on_update=args.create_on_update)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name, args.events_follow, from_dt=now)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, region, profile, args.yes)
if args.events_follow:
stack_status = cf.print_events(cf_conn, args.name, args.events_follow, from_dt=now)
if stack_status in FAILED_STACK_STATES:
sys.exit(1)
if args.subcommand == 'events':
cf.print_events(cf_conn, args.name, args.events_follow, args.lines)
if args.subcommand == 'diff':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
cf.print_stack_diff(cf_conn, args.name, args.template, config)
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
if args.subcommand == 'config':
print_config(config,
args.property_name,
output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
session_kwargs = {}
if args.profile:
session_kwargs['profile_name'] = args.profile
if args.region:
session_kwargs['region_name'] = args.region
try:
botosession = boto3.Session(**session_kwargs)
config['region'] = botosession.region_name
s3_conn = botosession.client('s3')
ec2_conn = botosession.resource('ec2')
vpc_conn = ec2_conn
r53_conn = botosession.client('route53')
cf_conn = botosession.client('cloudformation')
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
except botocore.exceptions.ClientError as e:
print(e)
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
cf.create_stack(cf_conn,
args.name,
args.template,
config,
dry=args.dry_run,
follow=args.events_follow)
else:
cf.create_stack(cf_conn,
args.name,
args.template,
config,
update=True,
dry=args.dry_run,
follow=args.events_follow,
create_on_update=args.create_on_update)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, botosession.region_name,
botosession.profile_name, args.yes)
if args.events_follow:
cf.get_events(cf_conn, args.name, args.events_follow, 10)
if args.subcommand == 'events':
cf.get_events(cf_conn, args.name, args.events_follow, args.lines)