ec2-k8.py

#!/root/kubernates-ansible-aws/ansible/bin/python
import boto3
import json
import pprint
def get_hosts(ec2,fv):
   f={'Name':'tag:node-type','Values':[fv]}
   hosts=[]
   for each_in in ec2.instances.filter(Filters=[f]):
#      print(each_in.private_ip_address)
       hosts.append(each_in.private_ip_address)
   return hosts

def main():
   ec2=boto3.resource("ec2")
   master_group=get_hosts(ec2,"master")
   worker_group=get_hosts(ec2,"worker")
   
   all_groups={
	            'master': {
			'hosts': master_group,
			'vars': {
			     'group_name': 'k8-master',
                             'ansible_user': 'ubuntu'
                               }
			},
		    'worker': {
			'hosts': worker_group,
			'vars': {
			     'group_name': 'k8-node',
                             'ansible_user': 'ubuntu'
                               }
			}
     	      }
   print(json.dumps(all_groups))
   

if __name__=="__main__":
 main()