Skip to content

Instantly share code, notes, and snippets.

@it-am

it-am/invoke_slave.py

Last active September 17, 2019 03:08
Show Gist options
  • Save it-am/6862a04cbd979f705cb477fa84d122e2 to your computer and use it in GitHub Desktop.
Save it-am/6862a04cbd979f705cb477fa84d122e2 to your computer and use it in GitHub Desktop.
Download ZIP
medium
Raw
invoke_slave.py
import boto3
import datetime
import time
stsclient = boto3.client('sts')
s3client = boto3.resource('s3')
def lambda_handler(event, context):
# -----------------------------------------------------------------------
# initiating a session using ARN of the IAM role
# -----------------------------------------------------------------------
rolearn = event['ARN']
awsaccount = stsclient.assume_role(
RoleArn=rolearn,
RoleSessionName='awsaccount_session'
)
ACCESS_KEY = awsaccount['Credentials']['AccessKeyId']
SECRET_KEY = awsaccount['Credentials']['SecretAccessKey']
SESSION_TOKEN = awsaccount['Credentials']['SessionToken']
# -----------------------------------------------------------------------
# create a list of all currently available aws regions
# -----------------------------------------------------------------------
ec2 = boto3.client('ec2', aws_access_key_id=ACCESS_KEY, aws_secret_access_key=SECRET_KEY, aws_session_token=SESSION_TOKEN)
final_awsregionslist = []
awsregions = ec2.describe_regions()
awsregions_list = awsregions['Regions']
for region in awsregions_list:
final_awsregionslist.append(region['RegionName'])
# -----------------------------------------------------------------------
start = '::'
end = ':'
awsaccountid = rolearn[rolearn.find(start)+len(start):rolearn.rfind(end)] # getting awsaccount ID from IAM Role ARN
# -----------------------------------------------------------------------
# Building HTML page/table using jquery datatables
# -----------------------------------------------------------------------
date_now = datetime.date.today()
time_now = time.strftime("%H:%M:%S")
creationdatetime = f'last update: {date_now} {time_now} UTC'
payload_start = """<html><head><script src="https://code.jquery.com/jquery-3.3.1.min.js"></script><link rel="stylesheet" type="text/css" href="https://cdn.datatables.net/1.10.16/css/jquery.dataTables.css"><script type="text/javascript" charset="utf8" src="https://cdn.datatables.net/1.10.16/js/jquery.dataTables.js"></script><script>$(document).ready( function () {$('#example').DataTable();} );</script></head><body><table id="example" class="display"><thead><tr><th><font face="arial">AWS Account Id</font></th><th><font face="arial">AWS Region</font></th><th><font face="arial">EIP</font></th></tr></thead><tbody>"""
# -----------------------------------------------------------------------
# loop through all exisiting aws regions
# -----------------------------------------------------------------------
for awsregion in final_awsregionslist:
# =================== THIS IS WHERE YOUR JOB STARTS ==================
# ----------------------------------------------------------------------
# Open ec2 session for current aws account (arn) and region
ec2client = boto3.client('ec2', aws_access_key_id=ACCESS_KEY, aws_secret_access_key=SECRET_KEY, aws_session_token=SESSION_TOKEN, region_name=awsregion)
# ----------------------------------------------------------------------
response = ec2client.describe_addresses()
elasticipslist = response['Addresses']
for eip in elasticipslist:
if 'AssociationId' not in eip:
elastic_ip = eip['PublicIp']
# --------------------------------------------------------------
# Building HTML page/table
loopstring = f'<tr><td><font face="arial">{awsaccountid}</font></td><td><font face="arial">{awsregion}</font></td><td><font face="arial">{elastic_ip}</font></td></tr>'
payload_start = payload_start + loopstring
payload_end = f'</tbody></table><p>{creationdatetime}</p></body></html>'
finalpayload = payload_start + payload_end
# --------------------------------------------------------------
domain = 'your_s3_bucket_name_here' # S3 bucket name where HTML page will be saved (must be changed)
htmlfilename = f'awsaccount-{awsaccountid}-EIPs.html' # making unique name for HTML file
s3client.Object(domain, htmlfilename).put(Body=finalpayload, ContentType='text/html')
# =================== THIS IS WHERE YOUR JOB ENDS ===================
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment