def get_most_read(key):
"""
Gets the most read news from a given page (national or international)
:param key: the key of the source page (e.g: g1, localDF, WP, etc.)
:return: a list with the most read news from the page
"""
# Check if the News Source is national or internacional
if key in news_source_national.strategies:
return news_source_national.get_most_read(key)
elif key in news_source_international.strategies:
return news_source_international.get_most_read(key)
elif key in news_source_sports.strategies:
return news_source_sports.get_most_read(key)
elif key in news_source_entertainment.strategies:
return news_source_entertainment.get_most_read(key)
elif key in news_source_technology.strategies:
return news_source_technology.get_most_read(key)
elif key in news_source_music.strategies:
return news_source_music.get_most_read(key)
elif key in news_source_reddit.strategies:
return news_source_reddit.get_most_read(key)
else:
raise ValueError('No news source found for SOURCE={0}'.format(key))
def test_local_ma(self):
news, title = news_source_national.get_most_read('localMA')
self.assertEqual(len(news), 6)
def test_veja(self):
news, title = news_source_national.get_most_read('veja')
self.assertEqual(len(news), 10)
def test_carta(self):
news, title = news_source_national.get_most_read('carta')
self.assertEqual(len(news), 5)
def test_r7(self):
news, title = news_source_national.get_most_read('r7')
self.assertEqual(len(news), 7)
def test_g1(self):
news, title = news_source_national.get_most_read('g1')
self.assertEqual(len(news), 5)
def test_local_to(self):
news, title = news_source_national.get_most_read('localTO')
self.assertEqual(len(news), 5)