162 lines
5.0 KiB
Python
162 lines
5.0 KiB
Python
|
|
|
|
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)
|
|
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_to_file(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_to_file(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
|
|
)
|