-
Notifications
You must be signed in to change notification settings - Fork 0
/
handler.py
54 lines (44 loc) · 1.49 KB
/
handler.py
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
49
50
51
52
53
54
"""
@author Rishabh Srivastava (rishabh080598@gmail.com)
"""
import json
import boto3
import os
def lambda_handler(event, context):
if "queryStringParameters" in event:
params = event['queryStringParameters']
if 'user_name' not in params or 'email' not in params or 'pincode' not in params:
return{
"statusCode": 400,
"body" : json.dumps("invalid input")
}
client = boto3.client("stepfunctions")
response = client.start_execution(
stateMachineArn=os.environ["STEP_FUNCTION"],
input=json.dumps(params)
)
return {
'statusCode': 200,
'body': json.dumps(f"{params['user_name']} is added to database.")
}
else:
dynamo_client = boto3.client("dynamodb")
op=[]
for item in scan_table(dynamo_client, TableName=os.environ["TABLE"]):
op.append(item)
res = []
for item in op:
i = {
'id' : item['id']['S'],
'user_name' : item['user_name']['S'],
'pincode' : item['pincode']['S'],
}
res.append(i)
return{
'statusCode': 200,
'body': json.dumps(res)
}
def scan_table(dynamo_client, *, TableName, **kwargs):
paginator = dynamo_client.get_paginator("scan")
for page in paginator.paginate(TableName=TableName, **kwargs):
yield from page["Items"]