vcs.maemo.org Git - feedingit/blob - src/rss.py

   1 #!/usr/bin/env python2.5
   2
   3 #
   4 # Copyright (c) 2007-2008 INdT.
   5 # This program is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU Lesser General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 #  This program is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #  GNU Lesser General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU Lesser General Public License
  16 #  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18
  19 # ============================================================================
  20 # Name        : FeedingIt.py
  21 # Author      : Yves Marcoz
  22 # Version     : 0.1
  23 # Description : PyGtk Example
  24 # ============================================================================
  25
  26 from os.path import isfile
  27 from os.path import isdir
  28 from os import remove
  29 import pickle
  30 import md5
  31 import feedparser
  32 import time
  33
  34 CONFIGDIR="/home/user/.feedingit/"
  35
  36 def getId(string):
  37     return md5.new(string).hexdigest()
  38
  39 class Feed:
  40     # Contains all the info about a single feed (articles, ...), and expose the data
  41     def __init__(self, name, url):
  42         self.entries = []
  43         self.readItems = {}
  44         self.countUnread = 0
  45         self.name = name
  46         self.url = url
  47         self.updateTime = "Never"
  48
  49     def saveFeed(self):
  50         file = open(CONFIGDIR+getId(self.name), "w")
  51         pickle.dump(self, file )
  52         file.close()
  53
  54     def updateFeed(self):
  55         tmp=feedparser.parse(self.url)
  56         # Check if the parse was succesful (number of entries > 0, else do nothing)
  57         if len(tmp["entries"])>0:
  58            self.tmpReadItems = self.readItems
  59            self.readItems = {}
  60            self.updateTime = time.asctime()
  61            self.entries = tmp["entries"]
  62            self.countUnread = 0
  63            # Initialize the new articles to unread
  64            for index in range(self.getNumberOfEntries()):
  65                if not self.tmpReadItems.has_key(self.getTitle(index)):
  66                    self.readItems[self.getTitle(index)] = False
  67                else:
  68                    self.readItems[self.getTitle(index)] = self.tmpReadItems[self.getTitle(index)]
  69                if self.readItems[self.getTitle(index)]==False:
  70                   self.countUnread = self.countUnread + 1
  71            del tmp
  72            self.saveFeed()
  73
  74     def setEntryRead(self, index):
  75         if self.readItems[self.getTitle(index)]==False:
  76             self.countUnread = self.countUnread - 1
  77             self.readItems[self.getTitle(index)] = True
  78
  79     def isEntryRead(self, index):
  80         return self.readItems[self.getTitle(index)]
  81
  82     def getTitle(self, index):
  83         return self.entries[index]["title"]
  84
  85     def getUpdateTime(self):
  86         return self.updateTime
  87
  88     def getEntries(self):
  89         try:
  90             return self.entries
  91         except:
  92             return []
  93
  94     def getNumberOfUnreadItems(self):
  95         return self.countUnread
  96
  97     def getNumberOfEntries(self):
  98         return len(self.entries)
  99
 100     def getItem(self, index):
 101         try:
 102             return self.entries[index]
 103         except:
 104             return []
 105
 106     def getArticle(self, index):
 107         self.setEntryRead(index)
 108         entry = self.entries[index]
 109         title = entry.get('title', 'No title')
 110         #content = entry.get('content', entry.get('summary_detail', {}))
 111         if entry.has_key('content'):
 112             content = entry.content[0].value
 113         else:
 114             content = entry.get('summary', '')
 115         #print content.keys()
 116         #.get('value', "No Data")
 117         link = entry.get('link', 'NoLink')
 118         date = time.strftime("%a, %d %b %Y %H:%M:%S",entry["updated_parsed"])
 119         #text = '''<div style="color: black; background-color: white;">'''
 120         text = '<meta http-equiv="Content-Type" content="text/html; charset=UTF-8"/>'
 121         text = text + '<div><a href=\"' + link + '\">' + title + "</a>"
 122         text = text + "<BR /><small><i>Date: " + date + "</i></small></div>"
 123         text = text + "<BR /><BR />"
 124         text = text + content
 125         return text
 126
 127
 128 class Listing:
 129     # Lists all the feeds in a dictionary, and expose the data
 130     def __init__(self):
 131         self.feeds = {}
 132         if isfile(CONFIGDIR+"feeds.pickle"):
 133             file = open(CONFIGDIR+"feeds.pickle")
 134             self.listOfFeeds = pickle.load(file)
 135             file.close()
 136         else:
 137             self.listOfFeeds = {getId("Slashdot"):{"title":"Slashdot", "url":"http://rss.slashdot.org/Slashdot/slashdot"}, }
 138         for key in self.listOfFeeds.keys():
 139             if isfile(CONFIGDIR+key):
 140                 file = open(CONFIGDIR+key)
 141                 self.feeds[key] = pickle.load(file)
 142                 file.close()
 143             else:
 144                 self.feeds[key] = Feed(self.listOfFeeds[key]["title"], self.listOfFeeds[key]["url"])
 145         self.saveConfig()
 146
 147     def updateFeeds(self):
 148         for key in self.listOfFeeds.keys():
 149             self.feeds[key].updateFeed()
 150
 151     def getFeed(self, key):
 152         return self.feeds[key]
 153
 154     def getFeedUpdateTime(self, key):
 155         return self.feeds[key].getUpdateTime()
 156
 157     def getFeedNumberOfUnreadItems(self, key):
 158         return self.feeds[key].getNumberOfUnreadItems()
 159
 160     def getFeedTitle(self, key):
 161         return self.listOfFeeds[key]["title"]
 162
 163     def getFeedUrl(self, key):
 164         return self.listOfFeeds[key]["url"]
 165
 166     def getListOfFeeds(self):
 167         return self.listOfFeeds.keys()
 168
 169     def addFeed(self, title, url):
 170         self.listOfFeeds[getId(title)] = {"title":title, "url":url}
 171         self.saveConfig()
 172         self.feeds[getId(title)] = Feed(title, url)
 173
 174     def removeFeed(self, key):
 175         del self.listOfFeeds[key]
 176         del self.feeds[key]
 177         if isfile(CONFIGDIR+key):
 178            remove(CONFIGDIR+key)
 179
 180     def saveConfig(self):
 181         file = open(CONFIGDIR+"feeds.pickle", "w")
 182         pickle.dump(self.listOfFeeds, file)
 183         file.close()