"""Logic for retrieving highwatermark dates.""" from oto import response as oto_response from analytics.utils import data_availability def get_highwatermark(highwatermark_type): """Return highwatermark for to the store, videos and downloads. Returns: oto.Response dict: {'highwatermark':yyyy-mm-dd}. """ available_feeds = data_availability.get_available_feeds() highwatermark = None if highwatermark_type == "streams": highwatermark = data_availability.get_max_available_date(available_feeds) elif highwatermark_type == "videos": highwatermark = data_availability.get_videos_max_available_date(available_feeds) elif highwatermark_type == "downloads": highwatermark = data_availability.get_downloads_max_available_date( available_feeds ) return oto_response.Response({"highwatermark": highwatermark})