"""Methods for retrieving Activity Detector 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_ACTIVITY_DETECTOR_TABLE = dynamodb.Table( config.FEED_ACTIVITY_DETECTOR_TABLE, ) def get_activity_detector_feed_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_ACTIVITY_DETECTOR_TABLE.query( KeyConditionExpression=( Key('activity_type_name').eq(feed_name) & Key('date').between(from_date, to_date) ) ) return response.get('Items')