from datetime import datetime
from Feed import Feed
from FeedEntry import FeedEntry
import feedparser
import logging

STATUS_OK = 200
logger = logging.getLogger("FeedUpdater")

def updateAllFeeds(session):
    allFeeds = session.query(Feed)
    for feed in allFeeds:
        try:
            FeedUpdater(session, feed).update()
        except FeedUpdateException:
            logger.warn("problems while updating feed " + feed.rss_url)
        session.commit()

def createNewFeed(url, session):
    result = feedparser.parse(url)
    title = result["feed"].title
    newFeed = Feed(title, url)
    session.add(newFeed)
    session.commit()
    FeedUpdater(newFeed).update()
    session.commit()

class FeedUpdater(object):
    def __init__(self, session, feed):
        self.session = session
        self.feed = feed
        
    def update(self):
        result = self.getFeed()
        for entry in result.entries:
            self.processEntry(entry)

    def getFeed(self):
        result = feedparser.parse(self.feed.rss_url)
        if result["status"] is not STATUS_OK:
            raise FeedUpdateException()
        return result

    def processEntry(self, entry):
        feedEntry = FeedEntry.findById(entry.id, self.session)
        if feedEntry is None:
            self.createFeedEntry(entry)
            
    def createFeedEntry(self, entry):
        new = FeedEntry()
        new.id = entry.id
        new.link = entry.link
        new.title = entry.title
        new.updated = datetime(*entry.updated_parsed[:6])
        new.summary = entry.summary
        new.feed = self.feed
        self.session.add(new)

class FeedUpdateException(Exception):
    pass