28 lines
1,012 B
Python
28 lines
1,012 B
Python
|
from DataSourceInterface import DataSourceInterface
|
||
|
|
from datetime import datetime, timezone, timedelta
|
||
|
|
|
||
|
|
class RssInterface(DataSourceInterface):
|
||
|
|
"""Interface for fetching and processing rss post information."""
|
||
|
|
def __init__(self):
|
||
|
|
self.loaded_posts = self.__get_posts__()
|
||
|
|
self.__sort_posts__()
|
||
|
|
|
||
|
|
def __get_posts__(self):
|
||
|
|
raise NotImplementedError("Functions needs to be implemented")
|
||
|
|
|
||
|
|
def get_latest_posts(self, count=10):
|
||
|
|
return self.loaded_posts[0:count]
|
||
|
|
|
||
|
|
def get_today_posts(self):
|
||
|
|
return self.get_day_posts(datetime.now())
|
||
|
|
|
||
|
|
def get_day_posts(self, day):
|
||
|
|
return self.__get_posts_to_filter__(lambda x : x.datetime.strftime('%d-%m-%y') == date.strftime('%d-%m-%y'))
|
||
|
|
|
||
|
|
def __get_posts_to_filter__(self, post_filter):
|
||
|
|
if self.loaded_posts is None:
|
||
|
|
return []
|
||
|
|
return [post for post in self.loaded_posts if post_filter(post)]
|
||
|
|
|
||
|
|
def __sort_posts__(self):
|
||
|
|
self.loaded_posts.sort(key=lambda x : x.datetime)
|