git.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.2.0
  23 # Description : Simple RSS Reader
  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.getUniqueId(index)):
  66                    self.readItems[self.getUniqueId(index)] = False
  67                else:
  68                    self.readItems[self.getUniqueId(index)] = self.tmpReadItems[self.getUniqueId(index)]
  69                if self.readItems[self.getUniqueId(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.getUniqueId(index)]==False:
  76             self.countUnread = self.countUnread - 1
  77             self.readItems[self.getUniqueId(index)] = True
  78
  79     def isEntryRead(self, index):
  80         return self.readItems[self.getUniqueId(index)]
  81
  82     def getTitle(self, index):
  83         return self.entries[index]["title"]
  84
  85     def getUniqueId(self,index):
  86         entry = self.entries[index]
  87         return getId(time.strftime("%a, %d %b %Y %H:%M:%S",entry["updated_parsed"]) + entry["title"])
  88
  89     def getUpdateTime(self):
  90         return self.updateTime
  91
  92     def getEntries(self):
  93         try:
  94             return self.entries
  95         except:
  96             return []
  97
  98     def getNumberOfUnreadItems(self):
  99         return self.countUnread
 100
 101     def getNumberOfEntries(self):
 102         return len(self.entries)
 103
 104     def getItem(self, index):
 105         try:
 106             return self.entries[index]
 107         except:
 108             return []
 109
 110     def getContent(self, index):
 111         entry = self.entries[index]
 112         if entry.has_key('content'):
 113             content = entry.content[0].value
 114         else:
 115             content = entry.get('summary', '')
 116         return content
 117
 118     def getArticle(self, index):
 119         self.setEntryRead(index)
 120         entry = self.entries[index]
 121         title = entry.get('title', 'No title')
 122         #content = entry.get('content', entry.get('summary_detail', {}))
 123         content = self.getContent(index)
 124
 125         link = entry.get('link', 'NoLink')
 126         date = time.strftime("%a, %d %b %Y %H:%M:%S",entry["updated_parsed"])
 127         #text = '''<div style="color: black; background-color: white;">'''
 128         text = '<meta http-equiv="Content-Type" content="text/html; charset=UTF-8"/>'
 129         text = text + '<div><a href=\"' + link + '\">' + title + "</a>"
 130         text = text + "<BR /><small><i>Date: " + date + "</i></small></div>"
 131         text = text + "<BR /><BR />"
 132         text = text + content
 133         return text
 134
 135
 136 class Listing:
 137     # Lists all the feeds in a dictionary, and expose the data
 138     def __init__(self):
 139         self.feeds = {}
 140         if isfile(CONFIGDIR+"feeds.pickle"):
 141             file = open(CONFIGDIR+"feeds.pickle")
 142             self.listOfFeeds = pickle.load(file)
 143             file.close()
 144         else:
 145             self.listOfFeeds = {getId("Slashdot"):{"title":"Slashdot", "url":"http://rss.slashdot.org/Slashdot/slashdot"}, }
 146         print self.listOfFeeds.has_key("feedingit-order")
 147         if self.listOfFeeds.has_key("feedingit-order"):
 148             self.sortedKeys = self.listOfFeeds["feedingit-order"]
 149         else:
 150             self.sortedKeys = self.listOfFeeds.keys()
 151             self.sortedKeys.sort(key=lambda obj: self.getFeedTitle(obj))
 152         for key in self.sortedKeys:
 153             if isfile(CONFIGDIR+key):
 154                 file = open(CONFIGDIR+key)
 155                 self.feeds[key] = pickle.load(file)
 156                 file.close()
 157             else:
 158                 self.feeds[key] = Feed(self.listOfFeeds[key]["title"], self.listOfFeeds[key]["url"])
 159         self.saveConfig()
 160
 161     def updateFeeds(self):
 162         print self.sortedKeys
 163         for key in self.getListOfFeeds():
 164             print key, self.sortedKeys
 165             self.feeds[key].updateFeed()
 166         print self.sortedKeys
 167
 168     def updateFeed(self, key):
 169         self.feeds[key].updateFeed()
 170
 171     def getFeed(self, key):
 172         return self.feeds[key]
 173
 174     def getFeedUpdateTime(self, key):
 175         return self.feeds[key].getUpdateTime()
 176
 177     def getFeedNumberOfUnreadItems(self, key):
 178         return self.feeds[key].getNumberOfUnreadItems()
 179
 180     def getFeedTitle(self, key):
 181         return self.listOfFeeds[key]["title"]
 182
 183     def getFeedUrl(self, key):
 184         return self.listOfFeeds[key]["url"]
 185
 186     def getListOfFeeds(self):
 187         return self.sortedKeys
 188
 189     def addFeed(self, title, url):
 190         self.listOfFeeds[getId(title)] = {"title":title, "url":url}
 191         self.sortedKeys.append(getId(title))
 192         self.saveConfig()
 193         self.feeds[getId(title)] = Feed(title, url)
 194
 195     def removeFeed(self, key):
 196         del self.listOfFeeds[key]
 197         self.sortedKeys.remove(key)
 198         del self.feeds[key]
 199         if isfile(CONFIGDIR+key):
 200            remove(CONFIGDIR+key)
 201
 202     def saveConfig(self):
 203         self.listOfFeeds["feedingit-order"] = self.sortedKeys
 204         file = open(CONFIGDIR+"feeds.pickle", "w")
 205         pickle.dump(self.listOfFeeds, file)
 206         file.close()
 207
 208     def moveUp(self, key):
 209         index = self.sortedKeys.index(key)
 210         self.sortedKeys[index] = self.sortedKeys[index-1]
 211         self.sortedKeys[index-1] = key
 212
 213     def moveDown(self, key):
 214         index = self.sortedKeys.index(key)
 215         index2 = (index+1)%len(self.sortedKeys)
 216         self.sortedKeys[index] = self.sortedKeys[index2]
 217         self.sortedKeys[index2] = key