from Feed import Feed
from FeedEntry import FeedEntry
from sqlalchemy.orm import joinedload

def getFeeds(session):
#    return AllFeeds(session).getFeeds()
    return UnreadFeeds(session).getFeeds()

class AllFeeds(object):
    '''A simple feed list that returns all feeds stored in the database'''

    def __init__(self, session):
        self.session = session
    
    def getFeeds(self):
        return Feed.all(self.session)
    
class UnreadFeeds(object):
    '''A feed list that returns only Feed objects that have unread entries'''
    
    def __init__(self, session):
        self.session = session
        
    def getFeeds(self):
        query = self.session.query(FeedEntry).filter(FeedEntry.read == 0)
        queryWithOptions = query.options(joinedload("feed"))
        result = queryWithOptions.all()
        return self._collectFeeds(result)
    
    def _collectFeeds(self, feedEntries):
        feeds = [entry.feed for entry in feedEntries]
        uniqueFeeds = set(feeds)
        return list(uniqueFeeds)