FilesExpand file tree

 master

/

lambda_ec2_ssm.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

48 lines (37 loc) · 1.38 KB

 master

/

lambda_ec2_ssm.py

Copy path

Top

File metadata and controls

• Code
• Blame

48 lines (37 loc) · 1.38 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

38

39

40

41

42

43

44

45

46

47

48

# -*- coding: utf-8 -*-

__author__ = "Chirag Rathod (Srce Cde)"

__license__ = "MIT"

__email__ = "chiragr83@gmail.com"

__maintainer__ = "Chirag Rathod (Srce Cde)"

import time

import json

import boto3

def lambda_handler(event, context):

# boto3 client

client = boto3.client("ec2")

ssm = boto3.client("ssm")

# getting instance information

describeInstance = client.describe_instances()

InstanceId = []

# fetchin instance id of the running instances

for i in describeInstance["Reservations"]:

for instance in i["Instances"]:

if instance["State"]["Name"] == "running":

InstanceId.append(instance["InstanceId"])

# looping through instance ids

for instanceid in InstanceId:

# command to be executed on instance

response = ssm.send_command(

InstanceIds=[instanceid],

DocumentName="AWS-RunShellScript",

Parameters={

"commands": ["command_to_be_executed"]

}, # replace command_to_be_executed with command

)

# fetching command id for the output

command_id = response["Command"]["CommandId"]

time.sleep(3)

# fetching command output

output = ssm.get_command_invocation(CommandId=command_id, InstanceId=instanceid)

print(output)

return {"statusCode": 200, "body": json.dumps("Thanks from Srce Cde!")}