#!/usr/bin/env python

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

import os

import click

import aws_lambda

import logging

CURRENT_DIR = os.getcwd()

logging.getLogger('pip').setLevel(logging.CRITICAL)

@click.group()

def cli():

pass

@click.command(help="Create a new function for Lambda.")

@click.argument('folder', nargs=-1, type=click.Path(file_okay=False, writable=True))

def init(folder):

path = CURRENT_DIR

if len(folder) > 0:

path = "{}/{}".format(CURRENT_DIR, folder[0])

if not os.path.exists(path):

os.makedirs(path)

aws_lambda.init(path)

@click.command(help="Bundles package for deployment.")

@click.option('--local-package', default=None, help='Install local package as well.', type=click.Path())

def build(local_package):

aws_lambda.build(CURRENT_DIR, local_package, copy_file=None)

@click.command(help="Run a local test of your function.")

@click.option('--event-file', default=None, help='Alternate event file.')

@click.option('--verbose', '-v', is_flag=True)

def invoke(event_file, verbose):

aws_lambda.invoke(CURRENT_DIR, event_file, verbose)

@click.command(help="Register and deploy your code to lambda.")

@click.option('--local-package', default=None, help='Install local package as well.', type=click.Path())

@click.option('--copy-file', default=None, help='Copies file/directory', type=click.Path())

def deploy(local_package, copy_file):

aws_lambda.deploy(CURRENT_DIR, local_package, copy_file)

@click.command(help="Delete old versions of your functions")

@click.option("--keep-last", type=int, prompt="Please enter the number of recent versions to keep")

def cleanup(keep_last):

aws_lambda.cleanup_old_versions(CURRENT_DIR, keep_last)

if __name__ == '__main__':

cli.add_command(init)

cli.add_command(invoke)

cli.add_command(deploy)

cli.add_command(build)

cli.add_command(cleanup)

cli()