controller/contrib/ec2/gen-json.py at 374ff64ea09e825df81c5fa2d8b6b2f8032ec6bc · drycc/controller · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/usr/bin/env python
import json
import os

template = json.load(open("deis.template.json",'r'))

with open('../coreos/user-data','r') as f:
  lines = f.readlines()

template['Resources']['CoreOSServerLaunchConfig']['Properties']['UserData']['Fn::Base64']['Fn::Join'] = [ '', lines ]
template['Parameters']['ClusterSize']['Default'] = str(os.getenv('DEIS_NUM_INSTANCES', 3))

VPC_ID = os.getenv('VPC_ID', None)
VPC_SUBNETS = os.getenv('VPC_SUBNETS', None)
VPC_ZONES = os.getenv('VPC_ZONES', None)

if VPC_ID and VPC_SUBNETS and VPC_ZONES:
  for resource in template['Resources'].keys():
    resource_type = template['Resources'][resource]['Type']
    if resource_type == 'AWS::EC2::SecurityGroup':
      template['Resources'][resource]['Properties']['VpcId'] = VPC_ID
    elif resource_type == 'AWS::EC2::SecurityGroupIngress':
      template['Resources'][resource]['Properties']['GroupId'] = template['Resources'][resource]['Properties']['GroupName']
      del template['Resources'][resource]['Properties']['GroupName']
      template['Resources'][resource]['Properties']['SourceSecurityGroupId'] = {
        'Ref': template['Resources'][resource]['Properties']['SourceSecurityGroupId']['Fn::GetAtt'][0]
      }
    elif resource_type == 'AWS::AutoScaling::LaunchConfiguration':
      template['Resources'][resource]['Properties']['AssociatePublicIpAddress'] = False
    elif resource_type == 'AWS::ElasticLoadBalancing::LoadBalancer':
      del template['Resources'][resource]['Properties']['AvailabilityZones']
      template['Resources'][resource]['Properties']['Subnets'] = VPC_SUBNETS.split(',')
    elif resource_type == 'AWS::AutoScaling::AutoScalingGroup':
      template['Resources'][resource]['Properties']['VPCZoneIdentifier'] = VPC_SUBNETS.split(',')
      template['Resources'][resource]['Properties']['AvailabilityZones'] = VPC_ZONES.split(',')

print json.dumps(template)