"""Methods for retrieving Outgoing Feed status from DynamoDB.""" import boto3 from boto3.dynamodb.conditions import Key from feed_status import config dynamodb = boto3.resource('dynamodb', region_name=config.AWS_REGION) FEED_SENDER_TABLE = dynamodb.Table(config.FEED_SENDER_TABLE) def get_ingestion_history(feed_name, from_date, to_date): """Look up feed status in DynamoDB. Args: feed_name (str): feed name of interest. from_date (date): start date of interest. to_date (date): end date of interest. Returns: list: list of statuses. """ response = FEED_SENDER_TABLE.query( KeyConditionExpression=( Key('feed_name').eq(feed_name) & Key('date').between(from_date, to_date) ) ) return response.get('Items')