deluge/plugins/RSS/plugin.py

class plugin_RSS:
	def __init__(self, path, deluge_core, deluge_interface):
		#set up system thingies
		import gtk, gtk.glade, os, ConfigParser, feedparser
		import deluge.common, deluge.dgtk, deluge.pref
		from time import asctime
		self.path = path
		self.core = deluge_core
		self.interface = deluge_interface
		#set up feeds file
		self.feeds_config = ConfigParser.ConfigParser()
		self.feeds_file = deluge.common.CONFIG_DIR + "/feeds.conf"
		if not os.path.isfile(self.feeds_file):
			f = open(self.feeds_file, mode='w')
			f.flush()
			f.close()
		self.feeds_config.read(self.feeds_file)
		#set up filters file
		self.filters_config = ConfigParser.ConfigParser()
		self.filters_file = deluge.common.CONFIG_DIR + "/filters.conf"
		if not os.path.isfile(self.filters_file):
			f = open(self.filters_file, mode='w')
			f.flush()
			f.close()
		self.filters_config.read(self.filters_file)
		#set up the preferences dialog
		glade = gtk.glade.XML(path + "/rss.glade")
		self.dlg = glade.get_widget("dialog")
		self.dlg.set_icon_from_file(self.path + "/rss.png")
		#set up the feeds list viewer
		self.feeds_view = glade.get_widget("feeds_view")
		model = gtk.ListStore(str, str, str)
		self.feeds_view.set_model(model)
		deluge.dgtk.add_text_column(self.feeds_view, _("Name"), 0)
		deluge.dgtk.add_text_column(self.feeds_view, _("URL"), 1)
		deluge.dgtk.add_text_column(self.feeds_view, _("Last Entry Date"), 2)
		#set up the torrents list viewer
		self.torrents_view = glade.get_widget("torrents_view")
		model = gtk.ListStore(str, str, str, str)
		self.torrents_view.set_model(model)
		deluge.dgtk.add_text_column(self.torrents_view, _("Feed"), 0)
		deluge.dgtk.add_text_column(self.torrents_view, _("Name"), 1)
		deluge.dgtk.add_text_column(self.torrents_view, _("URL"), 2)
		deluge.dgtk.add_text_column(self.torrents_view, _("Date"), 3)
		# Setup the filters list viewer
		self.filters_view = glade.get_widget("filters_view")
		model = gtk.ListStore(str)
		self.filters_view.set_model(model)
		deluge.dgtk.add_text_column(self.filters_view, _("Name"), 0)
		# set up the feed choice combobox
		self.feedchoice_combo = glade.get_widget("feedchoice_combo")
		model = gtk.ListStore(str)
		self.feedchoice_combo.set_model(model)
		cell = gtk.CellRendererText()
		self.feedchoice_combo.pack_start(cell, True)
		self.feedchoice_combo.add_attribute(cell, 'text', 0) 
		# set up the torrents feed choice combobox
		self.torrents_fcc = glade.get_widget("torrents_fcc")
		model = gtk.ListStore(str)
		self.torrents_fcc.set_model(model)
		cell = gtk.CellRendererText()
		self.torrents_fcc.pack_start(cell, True)
		self.torrents_fcc.add_attribute(cell, 'text', 0) 
		#set up the rest of the GUI elements
		self.name_entry = glade.get_widget("name_entry")		
		self.url_entry = glade.get_widget("url_entry")
		self.button_addfeed = glade.get_widget("button_addfeed")
		self.button_delfeed = glade.get_widget("button_delfeed")
		self.chkfeeds = glade.get_widget("chkfeeds")
		self.filtername_entry = glade.get_widget("filtername_entry")
		self.filterexp_entry = glade.get_widget("filterexp_entry")
		self.button_delfilter = glade.get_widget("button_delfilter")
		self.checkonstart_button = glade.get_widget("checkonstart_button")
		self.update_entry = glade.get_widget("update_entry")
		#Connect event signals
		self.filters_view.connect("cursor-changed", self.filter_row_clicked)
		dic = {	"addfeed_clicked" : self.addfeed_clicked,
				"delfeed_clicked" : self.delfeed_clicked,
				"addfilter_clicked" : self.addfilter_clicked,
				"delfilter_clicked" : self.delfilter_clicked,
				"row_clicked" : self.row_clicked,
				"feedtext_changed" : self.feedtext_changed,
				"filtername_lostfocus" : self.filtername_lostfocus,
				"filterexp_lostfocus" : self.filterexp_lostfocus,
				"feedchoice_combo_changed" : self.feedchoice_combo_changed,
				"torrents_fcc_changed" : self.torrents_fcc_changed,
				"torrents_view_row_activated" : self.torrents_view_row_activated,
				"chkfeeds_clicked" : self.chkfeeds_clicked }
		glade.signal_autoconnect(dic)
		self.feeds_view.get_selection().set_select_function(self.row_clicked)
		self.timer = 0
		self.torrents_list = []
		#self.checkfeeds()
		# Access the interface's toolbar
		self.toolbar = self.interface.toolbar
		# Make a toolbar button
		icon = gtk.Image()
		icon.set_from_file(self.path + "/rss.png") # Toolbar items should be 22x22 pixel images
		self.button = gtk.ToolButton(icon_widget=icon, label=_("RSS"))
		self.button.connect("clicked", self.rss_clicked) # Connect the signal handler for the button
		self.toolbar.add(self.button) # Add button to toolbar
		self.button.show_all() # Show the button
		self.checkonstart = False
		#load options
		if self.feeds_config.has_option("DEFAULT", "interval"):
			self.interval = self.feeds_config.getint("DEFAULT", "interval")
			self.update_entry.set_text(self.feeds_config.get("DEFAULT", "interval"))
		else:
			self.interval = 900
			self.feeds_config.set("DEFAULT", "interval", 900)
		if self.feeds_config.has_option("DEFAULT", "checkonstart"):
			self.checkonstart = self.feeds_config.getboolean("DEFAULT", "checkonstart")
			self.checkonstart_button.set_active(self.checkonstart)
		else:
			self.checkonstart = False
			self.feeds_config.set("DEFAULT", "checkonstart", False)
		if self.checkonstart_button.get_active():
			self.timer = self.interval - 5

	def rss_clicked(self, button):
		self.configure(self)
	
	def unload(self):
		self.toolbar.remove(self.button) # Remove the button from the toolbar
		f = open(self.feeds_file, "w")
		self.feeds_config.write(f)
		f.close()
	
	def feedtext_changed(self, args):
		a = (self.name_entry.get_text() != "")
		b = (self.url_entry.get_text() != "")
		if(a and b):
			self.button_addfeed.set_sensitive(1)
		else:
			self.button_addfeed.set_sensitive(0)

	def addfeed_clicked(self, args):
		self.feeds_view.get_model().append([self.name_entry.get_text(),
					self.url_entry.get_text(), ""])
		self.feedchoice_combo.append_text(self.name_entry.get_text())
		self.torrents_fcc.append_text(self.name_entry.get_text())
		self.feeds_config.add_section(self.name_entry.get_text())
		self.feeds_config.set(self.name_entry.get_text(), "url", self.url_entry.get_text())
		self.feeds_config.set(self.name_entry.get_text(), "lastchecked", "")
		self.name_entry.set_text("")
		self.url_entry.set_text("")

	def delfeed_clicked(self, args):
		(model, selection) = self.feeds_view.get_selection().get_selected()
		text = self.feeds_view.get_model().get_value(selection, 0)
		model2 = self.feedchoice_combo.get_model()
		model3 = self.torrents_fcc.get_model()
		the_iter = model2.get_iter_first()
		print text
		while the_iter is not None:
			print model2.get_value(the_iter, 0)
			if (model2.get_value(the_iter, 0) == text):
				remove_iter = the_iter
			the_iter = model2.iter_next(the_iter)
		model2.remove(remove_iter)
		the_iter = model3.get_iter_first()
		while the_iter is not None:
			print model3.get_value(the_iter, 0)
			if (model3.get_value(the_iter, 0) == text):
				remove_iter = the_iter
			the_iter = model3.iter_next(the_iter)
		model3.remove(remove_iter)
		model.remove(selection)
		for filt in self.filters_config.sections():
			if self.filters_config.get(filt, "feed") == text:
				self.filters_config.set(filt, "feed", "All")
		self.feeds_config.remove_section(text)
		self.button_delfeed.set_sensitive(0)
		
	def row_clicked(self, args):
		self.button_delfeed.set_sensitive(1)
		return True

	def addfilter_clicked(self, args):
		unique = True
		for filt in self.filters_config.sections():
			if filt == _("New Filter"):
				unique = False

		if unique:
			self.current_filter = _("New Filter")
			self.filters_config.add_section(_("New Filter"))
			new_iter = self.filters_view.get_model().append([_("New Filter")])
			self.filters_view.get_selection().select_iter(new_iter)
			self.filters_config.set(_("New Filter"), "filterexp", "")
			self.filters_config.set(_("New Filter"), "feed", "All")
			self.filtername_entry.set_text(_("New Filter"))
			self.feedchoice_combo.set_active(0)

			self.filtername_entry.set_sensitive(1)
			self.filterexp_entry.set_sensitive(1)
			self.feedchoice_combo.set_sensitive(1)
			self.filterexp_entry.set_text("")
			self.filtername_entry.grab_focus()
			self.button_delfilter.set_sensitive(1)

	def delfilter_clicked(self, args):
		model, selection = self.filters_view.get_selection().get_selected()
		self.filters_config.remove_section(self.current_filter)
		model.remove(selection)
		self.current_filter = None

		self.filtername_entry.set_text("")
		self.filterexp_entry.set_text("")
		self.feedchoice_combo.set_active(-1)
		self.filtername_entry.set_sensitive(0)
		self.filterexp_entry.set_sensitive(0)
		self.feedchoice_combo.set_sensitive(0)
		self.button_delfilter.set_sensitive(0)

	def filter_row_clicked(self, widget):
		selection = self.filters_view.get_selection()
		model, selection_iter = selection.get_selected()
		print model
		print selection_iter
		self.current_filter = self.filters_view.get_model().get_value(selection_iter, 0)
		self.filtername_entry.set_text(self.current_filter)
		self.filterexp_entry.set_text(self.filters_config.get(self.current_filter, "filterexp"))
		feed = self.filters_config.get(self.current_filter, "feed")
		model2 = self.feedchoice_combo.get_model()
		the_iter = model2.get_iter_first()
		while the_iter is not None:
			#print model2.get_value(the_iter, 0)
			if (model2.get_value(the_iter, 0) == feed):
				set_iter = the_iter
			the_iter = model2.iter_next(the_iter)
		self.feedchoice_combo.set_active_iter(set_iter)

		self.filtername_entry.set_sensitive(1)
		self.filterexp_entry.set_sensitive(1)
		self.feedchoice_combo.set_sensitive(1)
		self.button_delfilter.set_sensitive(1)

	def filtername_lostfocus(self, args, spare):
		(model, selection) = self.filters_view.get_selection().get_selected()
		old_filtername = self.filters_view.get_model().get_value(selection, 0)
		self.filters_config.remove_section(old_filtername)
		model.remove(selection)

		self.current_filter = self.filtername_entry.get_text()
		new_iter = self.filters_view.get_model().append([self.current_filter])
		self.filters_view.get_selection().select_iter(new_iter)
		self.filters_config.add_section(self.current_filter)
		self.filters_config.set(self.current_filter, "filterexp", self.filterexp_entry.get_text())
		self.filters_config.set(self.current_filter, "feed", self.feedchoice_combo.get_active_text())

	def filterexp_lostfocus(self, args, spare):
		self.filters_config.set(self.current_filter, "filterexp", self.filterexp_entry.get_text())

	def feedchoice_combo_changed(self, args):
		self.filters_config.set(self.current_filter, "feed", self.feedchoice_combo.get_active_text())

	def torrents_fcc_changed(self, args):
		model = self.torrents_view.get_model()
		model.clear()
		if self.torrents_fcc.get_active_text() == _("All"):
			for (date, feed, title, link) in self.torrents_list:
				self.torrents_view.get_model().append((feed, title, link, date))
		else:
			for (date, feed, title, link) in self.torrents_list:
				if feed == self.torrents_fcc.get_active_text():
					self.torrents_view.get_model().append((feed, title, link, date))

	def torrents_view_row_activated(self, widget, spare1, spare2):
		selection = widget.get_selection()
		model, selection_iter = selection.get_selected()
		self.interface.interactive_add_torrent_url(widget.get_model().get_value(selection_iter, 2))
		

	def chkfeeds_clicked(self, args):
		self.checkfeeds()

	def checkfeeds(self):
		import feedparser, datetime
		from time import asctime, strptime

		avail = {}
		sorted = {}
		self.torrents_list = []
		for name in self.feeds_config.sections():
			print "Attempting to parse " + name
			tmp = feedparser.parse(self.feeds_config.get(name, "url"))
			try:
				print "Parsed " + tmp['feed']['title']
				avail[name] = True
			except:
				print "Failed to download/parse " + name
				avail[name] = False
			if avail[name]:
				entries = []
				entries.extend( tmp[ "items" ] )
				decorated = [(entry["date_parsed"], entry) for entry in entries]
				tmplist = [(entry["date_parsed"], name, entry["title"], entry.enclosures[0].href) for entry in entries]
				decorated.sort()
				self.torrents_list.extend(tmplist)
				#decorated.reverse() # for most recent entries first
				sorted[name] = [entry for (date,entry) in decorated]

		model = self.torrents_view.get_model()
		model.clear()
		self.torrents_list.sort()
		self.torrents_list.reverse()
		#self.torrents_view.get_model().append([entry for entry in self.torrents_list])
		for (date,feed,title,link) in self.torrents_list:
			self.torrents_view.get_model().append((feed, title, link, date))
		#for key in sorted.keys():
		#	print "listing entries in " + key
		#	
		#	for entry in sorted[key]:
		#		print entry.title
		#		for enclosure in entry.enclosures:
		#			self.torrents_view.get_model().append( (key, entry.title, enclosure.href, entry.date_parsed) )

		checked = {}
		for name in self.filters_config.sections():
			checkfiltername = name
			checkfilterexp = self.filters_config.get(name, "filterexp")
			checkfilterfeed = self.filters_config.get(name, "feed")
			print "filter: " + checkfiltername
			print "search: " + checkfilterexp
			print "feed: " + checkfilterfeed
			if checkfilterfeed == "All":
				#print "made it to 'All'"
				for feedname in sorted.keys():
					if avail[feedname]:
						print feedname + " last checked: " + self.feeds_config.get(feedname, "lastchecked")
						if self.feeds_config.get(feedname, "lastchecked") != "":
							lastchecked = strptime(self.feeds_config.get(feedname, "lastchecked"))
						else:
							lastchecked = strptime(asctime(sorted[feedname][0].date_parsed))
						#print "searching feed: " + feedname
						for entry in sorted[feedname]:
							#print entry.title + ": " + asctime(entry.date_parsed)
							if (strptime(asctime(entry.date_parsed)) > lastchecked):
								#print entry.date_parsed
								#print " > "
								#print lastchecked
								if entry.title.find(checkfilterexp) != -1:
									#print "contains" + checkfilterexp
									for enclosure in entry.enclosures:
										print enclosure.href
										self.interface.interactive_add_torrent_url(enclosure.href)
										#self.feeds_config.set(feedname, "lastchecked", asctime(entry.date_parsed))
			else:
				if avail[checkfilterfeed]:
					print "searching feed: " + checkfilterfeed
					if self.feeds_config.get(checkfilterfeed, "lastchecked") != "":
						lastchecked = strptime(self.feeds_config.get(checkfilterfeed, "lastchecked"))
					else:
						#print sorted[checkfilterfeed][1].date_parsed
						lastchecked = strptime(asctime(sorted[checkfilterfeed][0].date_parsed))
					print "lastchecked: " + asctime(lastchecked)
					for entry in sorted[checkfilterfeed]:
						#print entry.title + ": " + asctime(entry.date_parsed)
						if (strptime(asctime(entry.date_parsed)) > lastchecked):
							#print entry.date_parsed
							#print " > "
							#print lastchecked
							if (entry.title.find(checkfilterexp) != -1):
								#print "contains" + checkfilterexp
								for enclosure in entry.enclosures:
									print enclosure.href
									self.interface.interactive_add_torrent_url(enclosure.href)
									#self.feeds_config.set(checkfilterfeed, "lastchecked", asctime(entry.date_parsed))

		for name in avail.keys():
			if avail[name]:
				self.feeds_config.set(name, "lastchecked", asctime(sorted[name][len(sorted[name])-1].date_parsed))

		self.timer = 0




	def configure(self, widget=None):
		import gtk, gtk.glade
		from deluge import common
		self.dlg.show_all()
		model = self.feeds_view.get_model()
		model.clear()
		model2 = self.feedchoice_combo.get_model()
		model2.clear()
		model3 = self.filters_view.get_model()
		model3.clear()
		model4 = self.torrents_fcc.get_model()
		model4.clear()
		self.filtername_entry.set_text("")
		self.filterexp_entry.set_text("")
		self.name_entry.set_text("")
		self.url_entry.set_text("")
		self.feedchoice_combo.append_text(_("All"))
		self.torrents_fcc.append_text(_("All"))
		self.torrents_fcc.set_active(0)
		for name in self.feeds_config.sections():
			self.feeds_view.get_model().append( (name, self.feeds_config.get(name, "url"), self.feeds_config.get(name, "lastchecked") ) )
			self.feedchoice_combo.append_text(name)
			self.torrents_fcc.append_text(name)
		for filters in self.filters_config.sections():
			self.filters_view.get_model().append( ([filters]) )
		#self.checkfeeds()
		self.button_addfeed.set_sensitive(0)
		self.button_delfeed.set_sensitive(0)
		self.filtername_entry.set_sensitive(0)
		self.filterexp_entry.set_sensitive(0)
		self.feedchoice_combo.set_sensitive(0)
		self.button_delfilter.set_sensitive(0)
		self.update_entry.set_text(str(self.interval))
		self.checkonstart_button.set_active(self.checkonstart)
		result = self.dlg.run()
		self.dlg.hide_all()
		if result == 1:
			self.interval = int(self.update_entry.get_text())
			self.feeds_config.set("DEFAULT", "interval", self.update_entry.get_text())
			self.feeds_config.set("DEFAULT", "checkonstart", self.checkonstart_button.get_active())
			f = open(self.filters_file, "w")
			self.filters_config.write(f)
			f.close()
			f = open(self.feeds_file, "w")
			self.feeds_config.write(f)
			f.close()
			

	def update(self):
		print "tick: " + str(self.timer)
		print "interval: " + str(self.interval)
		self.timer += 1
		if self.timer >= self.interval:
			print "BONG!"
			self.checkfeeds()
			self.timer = 0