"""Amazon Step Functions Connector.""" import boto3 from video import config from video.constants.activities import ACTIVITIES_TO_HANDLERS def get_stepfunctions_client(): """Get stepfunctions client. Returns: boto3.client.SFN: stepfunctions client. """ return boto3.client( 'stepfunctions', region_name=config.AWS_REGION, config=boto3.session.Config( read_timeout=70 ), ) def list_activities(): """List all daemon-video activities. Returns: list: List of daemon-video activities. """ return [ { 'name': activity_name, 'activityArn': '{}{}'.format( config.ACTIVITY_ARN_PREFIX, activity_name) } for activity_name in list(ACTIVITIES_TO_HANDLERS.keys()) ]