forked from nficano/python-lambda
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlambda
More file actions
executable file
·101 lines (83 loc) · 3.11 KB
/
lambda
File metadata and controls
executable file
·101 lines (83 loc) · 3.11 KB
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#!/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.option(
'--minimal', default=False, is_flag=True,
help='Exclude any unnecessary template files',
)
@click.argument(
'folder', nargs=-1,
type=click.Path(file_okay=False, writable=True),
)
def init(folder, minimal):
path = CURRENT_DIR
if len(folder) > 0:
path = os.path.join(CURRENT_DIR, *folder)
if not os.path.exists(path):
os.makedirs(path)
aws_lambda.init(path, minimal=minimal)
@click.command(help='Bundles package for deployment.')
@click.option(
'--use-requirements', default=False, is_flag=True,
help='Install all packages defined in requirements.txt',
)
@click.option(
'--local-package', default=None, type=click.Path(),
help='Install local package as well.', multiple=True,
)
def build(use_requirements, local_package):
aws_lambda.build(CURRENT_DIR, use_requirements, local_package)
@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(
'--use-requirements', default=False, is_flag=True,
help='Install all packages defined in requirements.txt',
)
@click.option(
'--local-package', default=None, type=click.Path(),
help='Install local package as well.', multiple=True,
)
def deploy(use_requirements, local_package):
aws_lambda.deploy(CURRENT_DIR, use_requirements, local_package)
@click.command(help='Upload your lambda to S3.')
@click.option(
'--use-requirements', default=False, is_flag=True,
help='Install all packages defined in requirements.txt',
)
@click.option(
'--local-package', default=None, type=click.Path(),
help='Install local package as well.', multiple=True,
)
def upload(use_requirements, local_package):
aws_lambda.upload(CURRENT_DIR, use_requirements, local_package)
@click.command(help="Deploy your lambda via S3.")
@click.option('--use-requirements', default=False, is_flag=True, help='Install all packages defined in requirements.txt')
@click.option('--local-package', default=None, help='Install local package as well.', type=click.Path(), multiple=True)
def deploy_s3(use_requirements, local_package):
aws_lambda.deploy_s3(CURRENT_DIR, use_requirements, local_package)
@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(upload)
cli.add_command(deploy_s3)
cli.add_command(build)
cli.add_command(cleanup)
cli()