deluge/plugins/TorrentSearch/plugin.py

class plugin_Search:
	def __init__(self, path, deluge_core, deluge_interface):
		import common, gtk, gtk.glade, dgtk, pref
		self.core = deluge_core
		self.interface = deluge_interface
		self.conf_file = common.CONFIG_DIR + "/search.conf"
		if not os.path.isfile(self.conf_file):
			f = open(self.conf_file, mode='w')
			f.flush()
			f.close()
		glade = gtk.glade.XML(path + "/searchdlg.glade")
		self.dlg = glade.get_widget("search_dialog")
		self.dlg.set_icon_from_file(common.get_pixmap("deluge32.png"))
		self.view = glade.get_widget("search_view")
		model = gtk.ListStore(str, str)
		self.view.set_model(model)
		dgtk.add_text_column(self.view, "Name", 0)
		dgtk.add_text_column(self.view, "Search String", 1)
		self.field_name = glade.get_widget("field_name")		
		self.field_search = glade.get_widget("field_search")
		self.button_add = glade.get_widget("button_addsearch")
		self.button_del = glade.get_widget("button_delsearch")
		dic = {	"add_clicked"	: self.add_clicked,
				"del_clicked"	: self.del_clicked,
				"row_clicked"	: self.row_clicked,
				"text_changed"	: self.text_changed }
		glade.signal_autoconnect(dic)
		self.view.get_selection().set_select_function(self.row_clicked)
		### Note: All other plugins should use self.interface.toolbar
		### when adding items to the toolbar
		self.se = ''
		self.toolbar = self.interface.wtree.get_widget("tb_right") 
		self.engines = pref.Preferences(self.conf_file, False)
		self.search_entry = gtk.Entry()
		self.search_item = gtk.ToolItem()
		self.search_item.add(self.search_entry)
		self.search_icon = gtk.Image()
		self.search_icon.set_from_stock(gtk.STOCK_FIND, gtk.ICON_SIZE_MENU)
		self.menu_button = gtk.MenuToolButton(self.search_icon, "Choose an Engine")
		self.menu_button.set_is_important(True)
		self.menu_button.connect("clicked", self.torrent_search)
		self.menu = gtk.Menu()
		self.manage_item = gtk.ImageMenuItem("Manage Engines")
		self.image = gtk.Image()
		self.image.set_from_stock(gtk.STOCK_PREFERENCES, gtk.ICON_SIZE_MENU)
		self.manage_item.set_image(self.image)
		self.manage_item.connect("activate", self.configure)
		self.menu.add(self.manage_item)
		self.menu_button.set_menu(self.menu)
		self.toolbar.insert(self.search_item, -1)
		self.toolbar.insert(self.menu_button, -1)
		self.populate_search_menu()
		self.toolbar.show_all()
		self.search_item.show_all()
		self.menu_button.show_all()
		self.menu.show_all()
	
	def unload(self):
		self.engines.save(self.conf_file)
		self.toolbar.remove(self.search_item)
		self.toolbar.remove(self.menu_button)
	
	def text_changed(self, args):
		a = (self.field_name.get_text() != "")
		b = (self.field_search.get_text() != "")
		if(a and b):
			self.button_add.set_sensitive(1)
		else:
			self.button_add.set_sensitive(0)

	def add_clicked(self, args):
		self.view.get_model().append([self.field_name.get_text(),
					self.field_search.get_text()])
		self.field_name.set_text("")
		self.field_search.set_text("")
			
	def del_clicked(self, args):
		(model, selection) = self.view.get_selection().get_selected()
		model.remove(selection)
		self.button_del.set_sensitive(0)
			
	def row_clicked(self, args):
		self.button_del.set_sensitive(1)
		return True
	
	def configure(self, widget=None):
		import common, gtk, gtk.glade
		self.dlg.show_all()
		model = self.view.get_model()
		model.clear()
		for name in self.engines.keys():
			self.view.get_model().append( (name, self.engines.get(name)) )
		self.button_add.set_sensitive(0)
		self.button_del.set_sensitive(0)
		result = self.dlg.run()
		self.dlg.hide_all()
		if result == 1:
			self.engines.clear()
			the_iter = model.get_iter_first()
			while the_iter is not None:
				self.engines.set(model.get_value(the_iter, 0), model.get_value(the_iter, 1))
				the_iter = model.iter_next(the_iter)
			self.engines.save(self.conf_file)
		self.populate_search_menu()
		

	
	def update(self):
		pass
		
	def torrent_search(self, widget=None):
		import common
		print "Searching with engine", self.se
		url = self.engines.get(self.se)
		entry = self.search_entry.get_text()
		print 'URL =', url
		print 'Entry =', entry
		entry = entry.replace(' ', '+')
		print 'URL =', url
		print 'Entry =', entry
		url = url.replace('${query}', entry)
		print 'URL =', url
		print 'Entry =', entry
		common.open_url_in_browser(None, url)
		
	def populate_search_menu(self):
		import gtk
		self.menu_button.set_label("Choose an Engine")
		for child in self.menu.get_children():
			self.menu.remove(child)
		group = None
		i = 0
		for engine in self.engines.keys():
			rmi = gtk.RadioMenuItem(None, engine)
			rmi.eng_name = engine
			rmi.connect("activate", self.select_search, rmi.eng_name)
			if (group != None):
				rmi.set_group(group)
			else:
				group = rmi
				rmi.set_active(1)
			self.menu.insert(rmi, i)
			i = i + 1
			rmi.show()
		self.menu.insert(self.manage_item, i)
		self.menu.show()

	def select_search(self, menuitem, engine_string):
		self.menu_button.set_label("Search " + engine_string)
		self.se = engine_string
	

register_plugin("Torrent Search",
				plugin_Search,
				"Zach Tibbitts",
				"0.5",
				"A searchbar for torrent search engines",
				config=True
				)