FilesExpand file tree

 master

/

build_aws_lambda_layer.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

92 lines (80 loc) · 2.89 KB

 master

/

build_aws_lambda_layer.py

Copy path

Top

File metadata and controls

• Code
• Blame

92 lines (80 loc) · 2.89 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import os

import shutil

import subprocess

import tempfile

from sentry_sdk.consts import VERSION as SDK_VERSION

DIST_PATH = "dist" # created by "make dist" that is called by "make aws-lambda-layer"

PYTHON_SITE_PACKAGES = "python" # see https://docs.aws.amazon.com/lambda/latest/dg/configuration-layers.html#configuration-layers-path

class LayerBuilder:

def __init__(

self,

base_dir, # type: str

):

# type: (...) -> None

self.base_dir = base_dir

self.python_site_packages = os.path.join(self.base_dir, PYTHON_SITE_PACKAGES)

self.out_zip_filename = f"sentry-python-serverless-{SDK_VERSION}.zip"

def make_directories(self):

# type: (...) -> None

os.makedirs(self.python_site_packages)

def install_python_packages(self):

# type: (...) -> None

sentry_python_sdk = os.path.join(

DIST_PATH,

f"sentry_sdk-{SDK_VERSION}-py2.py3-none-any.whl", # this is generated by "make dist" that is called by "make aws-lamber-layer"

)

subprocess.run(

[

"pip",

"install",

"--no-cache-dir", # always access PyPI

"--quiet",

sentry_python_sdk,

"--target",

self.python_site_packages,

],

check=True,

)

def create_init_serverless_sdk_package(self):

# type: (...) -> None

"""

Method that creates the init_serverless_sdk pkg in the

sentry-python-serverless zip

"""

serverless_sdk_path = (

f"{self.python_site_packages}/sentry_sdk/"

f"integrations/init_serverless_sdk"

)

if not os.path.exists(serverless_sdk_path):

os.makedirs(serverless_sdk_path)

shutil.copy(

"scripts/init_serverless_sdk.py", f"{serverless_sdk_path}/__init__.py"

)

def zip(self):

# type: (...) -> None

subprocess.run(

[

"zip",

"-q", # Quiet

"-x", # Exclude files

"**/__pycache__/*", # Files to be excluded

"-r", # Recurse paths

self.out_zip_filename, # Output filename

PYTHON_SITE_PACKAGES, # Files to be zipped

],

cwd=self.base_dir,

check=True, # Raises CalledProcessError if exit status is non-zero

)

shutil.copy(

os.path.join(self.base_dir, self.out_zip_filename),

os.path.abspath(DIST_PATH)

)

def build_packaged_zip():

with tempfile.TemporaryDirectory() as base_dir:

layer_builder = LayerBuilder(base_dir)

layer_builder.make_directories()

layer_builder.install_python_packages()

layer_builder.create_init_serverless_sdk_package()

layer_builder.zip()

if __name__ == "__main__":

build_packaged_zip()