mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 18:00:35 +00:00
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from base import Task
|
|
from common import phases
|
|
import host
|
|
|
|
|
|
class GetCredentials(Task):
|
|
description = 'Getting AWS credentials'
|
|
phase = phases.preparation
|
|
|
|
def run(self, info):
|
|
info.credentials = self.get_credentials(info.manifest)
|
|
|
|
def get_credentials(self, manifest):
|
|
from os import getenv
|
|
# manifest overrides environment
|
|
if(manifest.credentials['access-key'] and manifest.credentials['secret-key']):
|
|
return {'access_key': manifest.credentials['access-key'],
|
|
'secret_key': manifest.credentials['secret-key']}
|
|
if(getenv('EC2_ACCESS_KEY') and getenv('EC2_SECRET_KEY')):
|
|
return {'access_key': getenv('EC2_ACCESS_KEY'),
|
|
'secret_key': getenv('EC2_SECRET_KEY')}
|
|
|
|
if(bool(manifest.credentials['access-key']) != bool(manifest.credentials['secret-key'])):
|
|
raise RuntimeError('Both the access key and secret key must be specified in the manifest.')
|
|
if(bool(getenv('EC2_ACCESS_KEY')) != bool(getenv('EC2_SECRET_KEY'))):
|
|
raise RuntimeError('Both the access key and secret key must be specified as environment variables.')
|
|
|
|
raise RuntimeError('No ec2 credentials found.')
|
|
|
|
|
|
class Connect(Task):
|
|
description = 'Connecting to EC2'
|
|
phase = phases.preparation
|
|
after = [GetCredentials, host.GetInfo]
|
|
|
|
def run(self, info):
|
|
from boto.ec2 import connect_to_region
|
|
info.connection = connect_to_region(info.host['region'],
|
|
aws_access_key_id=info.credentials['access_key'],
|
|
aws_secret_access_key=info.credentials['secret_key'])
|