def redshift_handler(event, context):
"""
Function for handling `run_grizli_fit` events
"""
import matplotlib.pyplot as plt
import traceback
import time
t0 = time.time()
print(event) #['s3_object_path'], event['verbose'])
#print(context)
try:
run_grizli_fit(event)
except:
print(traceback.format_exc(limit=2))
# Clean up
beams_file = os.path.basename(event['s3_object_path'])
root = beams_file.split('_')[0]
if 'clean' in event:
run_clean = event['clean'] in TRUE_OPTIONS
else:
run_clean = True
if run_clean:
clean(root=root, verbose=True)
t1 = time.time()
return (event['s3_object_path'], t0, t1)
def run_test(s3_object_path=TESTER):
import os
from importlib import reload
import grizli.aws.lambda_handler
reload(grizli.aws.lambda_handler)
from grizli.aws.lambda_handler import run_grizli_fit, clean
#event = {'s3_object_path': s3_object_path, 'verbose':'True'}
obj = 'j224916m4432_02983'
bucket = 'grizli'
root, id = obj.split('_')
s3_object_path = 'Pipeline/{0}/Extractions/{0}_{1:05d}.beams.fits'.format(root, int(id))
event = {'s3_object_path':s3_object_path,
'bucket':bucket,
'verbose': 'True',
'check_wcs': 'True',
'zr':'0.1,0.3',
}
run_grizli_fit(event)
beams_file = os.path.basename(event['s3_object_path'])
root = beams_file.split('_')[0]
clean(root=root, verbose=True)
