def step_impl(context): # Pass as the import to the lambda didn't fail context.sns_client = get_sns_provider(test=True) context.sfn_client = get_sfn_provider(test=True) context.checkin_model = get_checkin_provider(test=True) context.consultant_model = get_consultants_provider(test=True) context.request_model = get_requests_provider(test=True) context.customer_model = get_customers_provider(test=True) context.contract_model = get_contracts_provider(test=True) pass
def get(event, context): ''' AWS Event Handler ''' print("Context: ", context) print("Event: ", event) requests_client = get_requests_provider() customer_model = get_customers_provider() consultant_model = get_consultants_provider() contract_model = get_contracts_provider() import_customer(requests_client, customer_model, consultant_model, contract_model)
def sub(event, context): ''' AWS Serverless Handler - :param event: AWS event :param context: AWS Lambda context ''' sqs_client = get_sqs_provider() customers_model = get_customers_provider() checkin_model = get_checkin_provider() contract_model = get_contracts_provider() weights_model = get_weights_provider() make_prediction(event, context, sqs_client, customers_model,\ checkin_model, contract_model, weights_model)
def __init__(self, payload: Dict, test: bool = False): ''' Init for class - :param payload: Payload of the event :param test: true for test ''' self.payload = payload self.metadata = read_metadata(payload) self.sqs_client = get_sqs_provider(test) self.requests_client = get_requests_provider(test) self.checkin_model = get_checkin_provider(test) self.customers_model = get_customers_provider(test) self.consultant_model = get_consultants_provider(test) self.contract_model = get_contracts_provider(test) self.stepfunctions_client = get_sfn_provider(test)
def con(event, context): ''' SQS Consume Handler :param event: AWS event :param context: AWS Lambda context ''' checkin_model = get_checkin_provider() consultant_model = get_consultants_provider() sns_client = get_sns_provider() stepfunctions_client = get_sfn_provider() request_client = get_requests_provider() customer_model = get_customers_provider() contract_model = get_contracts_provider() checkin_consumer(event, context, sns_client, stepfunctions_client, checkin_model, consultant_model, request_client, customer_model, contract_model)