py-lambda 1.1.5

Lambda framework for AWS

AWS-ALB-lambda-python

Framework for load-balancer and lambda function.

To learn more about writing AWS Lambda functions in python, go to the official documentation

Blogs about AWS lambda and python is here

Getting Started

This is the AWS lambda framework written in python. When the lambda function is triggered by ALB or API-Gateway the event variable that contain all the required info will be sent to the init function. By default this init function will be in lambda_function.py and the name will be lambda_handler. This framework contains the request resource mappin for the URL and response that you have to send back to Load-Balancer.

The package information is in Pypi info

pip install py-lambda

This source code contain the following files:

• lambda_function : Contains the code that will be triggered by ALB or API-Gateway etc.
• request : Contains the request object which is created from the event that is send as an input.
• response : Contains the object that should be sent back from lambda function to ALB.
• lambda_mapping : Is the file that contain the mapping for the routes. Add your routes in here.
• all_functions : Contain the function contains the respective code that can be used for the given path.
• test_LambdaHandler : Contains the test code that can be usedfortesting all the functions.

Load-Balancer input

## test_LambdaHandler.py
import py_lambda.test_LambdaHandler as lTest
print(lTest.EVENT_FROM_ALB)

The above code will give the sample of the request that will be send by ALB.

Mapping

## lambda_mapping.py

FUNCT_MAPPING = {
	"": {"": allFun.index},
	"test": {
		"": allFun.test,
		ld.URL_REST_ID_STR:{
			"": allFun.testUserId,
		},
	},
	"abc":{
		"": allFun.abc,
	},
}

This will have the mapping for each resources. The mapping is dictionary in which key is the URL resource and the value will be the function that you have to execute when the URL is called. The index URL in the mapping is the first one "": {"": allFun.index}. If all the function for the mapping are in all_Functions.py then corrponding function can be called in here.

If the resource contan some information that will change of each request then the mapping should contain URL_REST_ID_STR which is there in lambda_defines.py form wilch you will be able to map the URL.

Eg: mydomain.com/profile/USER_ID/image In this USER_ID may change for each user. In that case use URL_REST_ID_STR in the mapping

Input validaation

## inp_validatn.py
## VALID_DOMAINS contains all the valid domains.
## VALID_SUB_DOMAINS contails all the valid subdomains.

import py_lambda.funct_defines as funDef

ret_validation = validateDomain("http://some_domain.com")
##If the domain is valid then the function will return the success response
if ret_validation[funDef.FUNCTION_STAT] == funDef.SUCCESS:
	print("Valid domain")
else:
	print("invalid domain")

Input validations can be adde in this function. validateDomain will validate the given domain and return success if it is valid.

Request

## request.py
import py_lambda.request as Req
import py_lambda.test_LambdaHandler as lTest
reqObj = Req.Request(lTest.EVENT_FROM_ALB)

print("Path Str:",reqObj.getPathStr())
print("Req method:",reqObj.httpMeth())
print("Req from ALB:",reqObj.isAlb())
print("Cookies:",reqObj.getCookies())
print("Query params by input list:",reqObj.getQueryParams(['a']))
print("Query params all:",reqObj.getAllQueryParam())
print("Base64 encooded:",reqObj.isBase64())

When the request is sent to lambda function this request object object is created and we can use this to get all the information that we need like cookies, query parameters, headers etc from this.

Response

## response.py
import py_lambda.response as Res
respObj = Res.Response()
respObj = Res.Response()
print("respObj:", respObj())
respObj.setError()
print("respObj:", respObj())
respObj.setResp()
print("respObj:", respObj())
respObj.setResp(respBody={'a':1, 'b':2})
print("respObj:", respObj())
# respObj.setContentType("application/json")
# print("respObj:", respObj())
respObj.setHeader("Access-Control-Allow-Origin", "*")
print("respObj:", respObj())

Loab-Balancer requer the response of the lambda function in a specific format. This response object is created when the function is invoked and this will be changed accordingly to send the reponse intended.

Lambda Function

## lambda_function.py

import py_lambda.test_LambdaHandler as lTest
import py_lambda.lambda_function as lf

RESOURCE_PREFIX = ""
lTest.EVENT_FROM_ALB["path"]= RESOURCE_PREFIX+'/abc'
lTest.EVENT_FROM_ALB["httpMethod"] = "GET"

resp = lf.lambda_handler(event=lTest.EVENT_FROM_ALB,context= "some_context")
print(resp)

This will give you the reponse if you have a mapping in lambda_mapping.py. body in the response will contain whatever you set in the response. Even you can set the header and the response code accordingly using py_lambda.response.setResp and py_lambda.response.setHeader

Testing

## test_LambdaHandler.py

import py_lambda.test_LambdaHandler as lTest
print(lTest.testOptions())

This contain all the testing function that are used to build this.