Github to Codebuild Lambda Function

main.py

import boto3, json

def lambda_handler(event, context):
  client = boto3.client('codebuild')

  for e in event['Records']:
    source = e['EventSource']
    if source == 'aws:sns':
      message = json.loads(e['Sns']['Message'])
      project = message['repository']['name']
      treeish = message['ref'].replace('refs/heads/', '')
      sha = message['after']
      print(treeish, sha)

      response = client.start_build(
          projectName=project,
          sourceVersion=sha
      )
      print(response)
    else:
      print("Unrecognized event source: {}".format(source))