lambda.py

import json
import os
import zipfile

import boto3

bucket = os.environ.get('bucket')


def lambda_handler(event, context):
    print("started")
    s3 = boto3.client('s3')
    if not os.path.isdir('/mnt/cern_root/chroot'):
        print("ROOT not present")
        print("Downloading ROOT")
        s3.download_file(bucket, 'aws_root.zip', '/tmp/aws_root.zip')
        print("Streamed ROOT")

        with zipfile.ZipFile('/tmp/aws_root.zip', 'r') as zip_ref:
            print("Starting extracting to EFS")
            zip_ref.extractall('/mnt/cern_root')

        print("Finished extracting ROOT")
    else:
        print("ROOT already extracted")

    result = os.system('''
        export PATH=/mnt/cern_root/chroot/usr/local/sbin:/mnt/cern_root/chroot/usr/local/bin:/mnt/cern_root/chroot/usr/sbin:/mnt/cern_root/chroot/usr/bin:/mnt/cern_root/chroot/sbin:/mnt/cern_root/chroot/bin:$PATH && \
        export LD_LIBRARY_PATH=/mnt/cern_root/chroot/usr/lib64:/mnt/cern_root/chroot/usr/lib:/usr/lib64:/usr/lib:$LD_LIBRARY_PATH && \
        export CPATH=/mnt/cern_root/chroot/usr/include:$CPATH && \
        export roothome=/mnt/cern_root/root_install && \
        chmod 777 /mnt/cern_root/chroot/usr/bin/python3.7 && \
        chmod 777 /mnt/cern_root/root_install/bin/root-config && \
        cd /tmp && . ${roothome}/bin/thisroot.sh  && \
        python3.7 -m pip install numpy cloudpickle --target=${roothome}
    ''')
    # move inside os.system if wanting to test ASAP for internet connection and working ROOT
    # /mnt/cern_root/chroot/usr/bin/python3.7 ${roothome}/PyRDF/xrootd.py

    if os.WEXITSTATUS(result) != 0:
        return {
            'statusCode': 500,
            'body': json.dumps('Failed to extract ROOT to EFS!'),
            'result': json.dumps(os.WEXITSTATUS(result))
        }

    return {
        'statusCode': 200,
        'body': json.dumps('Extracted ROOT to EFS!'),
    }