class Search:
def __init__(self, searches, text, id, mode, remember, showtab):
self.searches = searches
self.frame = searches.frame
# Build the window
builder = Gtk.Builder()
builder.set_translation_domain('nicotine')
builder.add_from_file(os.path.join(os.path.dirname(os.path.realpath(__file__)), "ui", "search.ui"))
self.search_tab = builder.get_object("SearchTab")
for i in builder.get_objects():
try:
self.__dict__[Gtk.Buildable.get_name(i)] = i
except TypeError:
pass
self.search_tab.remove(self.Main)
self.search_tab.destroy()
builder.connect_signals(self)
self.text = text
self.searchterm_words_include = [p for p in text.lower().split() if not p.startswith('-')]
self.searchterm_words_ignore = [p[1:] for p in text.lower().split() if p.startswith('-') and len(p) > 1]
self.id = id
self.mode = mode
self.remember = remember
self.showtab = showtab
self.usersiters = {}
self.directoryiters = {}
self.users = set()
self.all_data = []
self.filters = None
self.resultslimit = 2000
self.numvisibleresults = 0
self.operators = {
'<': operator.lt,
'<=': operator.le,
'==': operator.eq,
'!=': operator.ne,
'>=': operator.ge,
'>': operator.gt
}
fill_file_grouping_combobox(self.ResultGrouping)
self.ResultGrouping.set_active(self.frame.np.config.sections["searches"]["group_searches"])
self.ResultGrouping.connect("changed", self.on_group)
self.ExpandButton.set_active(self.frame.np.config.sections["searches"]["expand_searches"])
self.ExpandButton.connect("toggled", self.on_toggle_expand_all)
if mode > 0:
self.RememberCheckButton.set_sensitive(False)
self.RememberCheckButton.set_active(remember)
""" Columns """
self.ResultsList.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.ResultsList.set_enable_tree_lines(True)
self.ResultsList.set_headers_clickable(True)
self.ResultsList.set_rubber_banding(True)
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
self.ResultsList.set_show_expanders(True)
else:
self.ResultsList.set_show_expanders(False)
self.resultsmodel = Gtk.TreeStore(
GObject.TYPE_UINT64, # (0) num
str, # (1) user
GObject.TYPE_OBJECT, # (2) flag
str, # (3) immediatedl
str, # (4) h_speed
str, # (5) h_queue
str, # (6) directory
str, # (7) filename
str, # (8) h_size
str, # (9) h_bitrate
str, # (10) length
GObject.TYPE_UINT64, # (11) bitrate
str, # (12) fullpath
str, # (13) country
GObject.TYPE_UINT64, # (14) size
GObject.TYPE_UINT64, # (15) speed
GObject.TYPE_UINT64 # (16) queue
)
widths = self.frame.np.config.sections["columns"]["filesearch_widths"]
cols = initialise_columns(
self.ResultsList,
[_("ID"), widths[0], "text", self.cell_data_func],
[_("User"), widths[1], "text", self.cell_data_func],
[_("Country"), widths[2], "pixbuf"],
[_("Immediate Download"), widths[3], "center", self.cell_data_func],
[_("Speed"), widths[4], "number", self.cell_data_func],
[_("In queue"), widths[5], "center", self.cell_data_func],
[_("Directory"), widths[6], "text", self.cell_data_func],
[_("Filename"), widths[7], "text", self.cell_data_func],
[_("Size"), widths[8], "number", self.cell_data_func],
[_("Bitrate"), widths[9], "number", self.cell_data_func],
[_("Length"), widths[10], "number", self.cell_data_func]
)
self.col_num, self.col_user, self.col_country, self.col_immediate, self.col_speed, self.col_queue, self.col_directory, self.col_file, self.col_size, self.col_bitrate, self.col_length = cols
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
self.ResultsList.get_columns()[0].set_visible(False)
self.ExpandButton.show()
hide_columns(cols, self.frame.np.config.sections["columns"]["filesearch_columns"])
self.col_num.set_sort_column_id(0)
self.col_user.set_sort_column_id(1)
self.col_country.set_sort_column_id(13)
self.col_immediate.set_sort_column_id(3)
self.col_speed.set_sort_column_id(15)
self.col_queue.set_sort_column_id(16)
self.col_directory.set_sort_column_id(6)
self.col_file.set_sort_column_id(7)
self.col_size.set_sort_column_id(14)
self.col_bitrate.set_sort_column_id(11)
self.col_length.set_sort_column_id(10)
self.col_country.get_widget().hide()
self.ResultsList.set_model(self.resultsmodel)
self.ResultsList.connect("button_press_event", self.on_list_clicked)
self.change_colours()
""" Filters """
self.filter_bitrate_model = Gtk.ListStore(GObject.TYPE_STRING)
self.FilterBitrate.set_model(self.filter_bitrate_model)
self.FilterBitrate.set_entry_text_column(0)
self.filter_size_model = Gtk.ListStore(GObject.TYPE_STRING)
self.FilterSize.set_model(self.filter_size_model)
self.FilterSize.set_entry_text_column(0)
self.filter_country_model = Gtk.ListStore(GObject.TYPE_STRING)
self.FilterCountry.set_model(self.filter_country_model)
self.FilterCountry.set_entry_text_column(0)
self.filter_in_model = Gtk.ListStore(GObject.TYPE_STRING)
self.FilterIn.set_model(self.filter_in_model)
self.FilterIn.set_entry_text_column(0)
self.filter_out_model = Gtk.ListStore(GObject.TYPE_STRING)
self.FilterOut.set_model(self.filter_out_model)
self.FilterOut.set_entry_text_column(0)
self.populate_filters()
self.FilterSize.clear()
sizecell = Gtk.CellRendererText()
sizecell.set_property("xalign", 1)
self.FilterSize.pack_start(sizecell, True)
self.FilterSize.add_attribute(sizecell, "text", 0)
self.FilterBitrate.clear()
bit_cell = Gtk.CellRendererText()
bit_cell.set_property("xalign", 1)
self.FilterBitrate.pack_start(bit_cell, True)
self.FilterBitrate.add_attribute(bit_cell, "text", 0)
self.FilterIn.connect("changed", self.on_filter_changed)
self.FilterOut.connect("changed", self.on_filter_changed)
self.FilterSize.connect("changed", self.on_filter_changed)
self.FilterBitrate.connect("changed", self.on_filter_changed)
self.FilterCountry.connect("changed", self.on_filter_changed)
self.FilterIn.get_child().connect("activate", self.on_refilter)
self.FilterOut.get_child().connect("activate", self.on_refilter)
self.FilterSize.get_child().connect("activate", self.on_refilter)
self.FilterBitrate.get_child().connect("activate", self.on_refilter)
self.FilterCountry.get_child().connect("activate", self.on_refilter)
""" Popup """
self.popup_menu_users = PopupMenu(self.frame, False)
self.popup_menu = popup = PopupMenu(self.frame)
popup.setup(
("#" + _("_Download file(s)"), self.on_download_files),
("#" + _("Download file(s) _to..."), self.on_download_files_to),
("#" + _("Download _folder(s)"), self.on_download_folders),
("#" + _("Download f_older(s) to..."), self.on_download_folders_to),
("#" + _("View Metadata of file(s)"), self.on_search_meta),
("", None),
("#" + _("Copy _URL"), self.on_copy_url),
("#" + _("Copy folder U_RL"), self.on_copy_dir_url),
("", None),
(1, _("User(s)"), self.popup_menu_users, self.on_popup_menu_users)
)
def on_tooltip(self, widget, x, y, keyboard_mode, tooltip):
return show_country_tooltip(widget, x, y, tooltip, 13, stripprefix='')
def on_filter_changed(self, widget):
iterator = widget.get_active_iter()
if iterator:
self.on_refilter(None)
def populate_filters(self):
if self.frame.np.config.sections["searches"]["enablefilters"]:
sfilter = self.frame.np.config.sections["searches"]["defilter"]
self.FilterIn.get_child().set_text(sfilter[0])
self.FilterOut.get_child().set_text(sfilter[1])
self.FilterSize.get_child().set_text(sfilter[2])
self.FilterBitrate.get_child().set_text(sfilter[3])
self.FilterFreeSlot.set_active(sfilter[4])
if(len(sfilter) > 5):
self.FilterCountry.get_child().set_text(sfilter[5])
self.filtersCheck.set_active(1)
for i in ['0', '128', '160', '192', '256', '320']:
self.FilterBitrate.get_model().append([i])
for i in [">10MiB", "<10MiB", "<5MiB", "<1MiB", ">0"]:
self.FilterSize.get_model().append([i])
s_config = self.frame.np.config.sections["searches"]
for i in s_config["filterin"]:
self.add_combo(self.FilterIn, i, True)
for i in s_config["filterout"]:
self.add_combo(self.FilterOut, i, True)
for i in s_config["filtersize"]:
self.add_combo(self.FilterSize, i, True)
for i in s_config["filterbr"]:
self.add_combo(self.FilterBitrate, i, True)
for i in s_config["filtercc"]:
self.add_combo(self.FilterCountry, i, True)
def add_combo(self, combobox, text, list=False):
text = text.strip()
if not text:
return False
model = combobox.get_model()
iterator = model.get_iter_first()
match = False
while iterator is not None:
value = model.get_value(iterator, 0)
if value.strip() == text:
match = True
iterator = model.iter_next(iterator)
if not match:
if list:
model.append([text])
else:
model.prepend([text])
def add_user_results(self, msg, user, country):
if user in self.users:
return
self.users.add(user)
counter = len(self.all_data) + 1
inqueue = msg.inqueue
ulspeed = msg.ulspeed
h_speed = human_speed(ulspeed)
if msg.freeulslots:
imdl = "Y"
inqueue = 0
else:
imdl = "N"
h_queue = humanize(inqueue)
append = False
maxstoredresults = self.searches.maxstoredresults
for result in msg.list:
if counter > maxstoredresults:
break
fullpath = result[1]
fullpath_lower = fullpath.lower()
if any(word in fullpath_lower for word in self.searchterm_words_ignore):
""" Filter out results with filtered words (e.g. nicotine -music) """
log.add_search(_("Filtered out excluded search result " + fullpath + " from user " + user))
continue
if not any(word in fullpath_lower for word in self.searchterm_words_include):
""" Some users may send us wrong results, filter out such ones """
log.add_search(_("Filtered out inexact or incorrect search result " + fullpath + " from user " + user))
continue
fullpath_split = reversed(fullpath.split('\\'))
name = next(fullpath_split)
directory = '\\'.join(fullpath_split)
size = result[2]
h_size = human_size(size)
h_bitrate, bitrate, h_length = get_result_bitrate_length(size, result[4])
self.append([counter, user, self.get_flag(user, country), imdl, h_speed, h_queue, directory, name, h_size, h_bitrate, h_length, bitrate, fullpath, country, size, ulspeed, inqueue])
append = True
counter += 1
if append:
# If this search wasn't initiated by us (e.g. wishlist), and the results aren't spoofed, show tab
if not self.showtab:
self.searches.show_tab(self, self.id, self.text, self.mode)
self.showtab = True
# Update counter
self.Counter.set_text("Results: %d/%d" % (self.numvisibleresults, len(self.all_data)))
# Update tab notification
self.frame.searches.request_changed(self.Main)
if self.frame.MainNotebook.get_current_page() != self.frame.MainNotebook.page_num(self.frame.searchvbox):
self.frame.SearchTabLabel.get_child().set_image(self.frame.images["online"])
def get_flag(self, user, flag=None):
if flag is not None:
flag = "flag_" + flag.lower()
self.frame.flag_users[user] = flag
else:
flag = self.frame.get_user_flag(user)
return self.frame.get_flag_image(flag)
def append(self, row):
self.all_data.append(row)
if self.numvisibleresults >= self.searches.maxdisplayedresults:
return
if not self.check_filter(row):
return
iterator = self.add_row_to_model(row)
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
if self.ExpandButton.get_active():
path = None
if iterator is not None:
path = self.resultsmodel.get_path(iterator)
if path is not None:
self.ResultsList.expand_to_path(path)
else:
collapse_treeview(self.ResultsList, self.ResultGrouping.get_active())
def add_row_to_model(self, row):
counter, user, flag, immediatedl, h_speed, h_queue, directory, filename, h_size, h_bitrate, length, bitrate, fullpath, country, size, speed, queue = row
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
if user not in self.usersiters:
self.usersiters[user] = self.resultsmodel.append(
None,
[0, user, self.get_flag(user, country), immediatedl, h_speed, h_queue, "", "", "", "", "", 0, "", country, 0, speed, queue]
)
parent = self.usersiters[user]
if self.ResultGrouping.get_active() == 1:
# Group by folder
if directory not in self.directoryiters:
self.directoryiters[directory] = self.resultsmodel.append(
self.usersiters[user],
[0, user, self.get_flag(user, country), immediatedl, h_speed, h_queue, directory, "", "", "", "", 0, fullpath.rsplit('\\', 1)[0] + '\\', country, 0, speed, queue]
)
row = row[:]
row[6] = "" # Directory not visible for file row if "group by folder" is enabled
parent = self.directoryiters[directory]
else:
parent = None
try:
iterator = self.resultsmodel.append(parent, row)
self.numvisibleresults += 1
except Exception as e:
types = []
for i in row:
types.append(type(i))
log.add_warning(_("Search row error: %(exception)s %(row)s"), {'exception': e, 'row': row})
iterator = None
return iterator
def check_digit(self, sfilter, value, factorize=True):
op = ">="
if sfilter[:1] in (">", "<", "="):
op, sfilter = sfilter[:1] + "=", sfilter[1:]
if not sfilter:
return True
factor = 1
if factorize:
base = 1024
if sfilter[-1:].lower() == 'b':
sfilter = sfilter[:-1] # stripping off the b, we always assume it means bytes
if sfilter[-1:].lower() == 'i':
base = 1000
sfilter = sfilter[:-1]
if sfilter.lower()[-1:] == "g":
factor = pow(base, 3)
sfilter = sfilter[:-1]
elif sfilter.lower()[-1:] == "m":
factor = pow(base, 2)
sfilter = sfilter[:-1]
elif sfilter.lower()[-1:] == "k":
factor = base
sfilter = sfilter[:-1]
if not sfilter:
return True
try:
sfilter = int(sfilter) * factor
except ValueError:
return True
operation = self.operators.get(op)
return operation(value, sfilter)
def check_filter(self, row):
filters = self.filters
if not self.filtersCheck.get_active():
return True
# "Included text"-filter, check full file path (located at index 12 in row)
if filters[0] and not filters[0].search(row[12].lower()):
return False
# "Excluded text"-filter, check full file path (located at index 12 in row)
if filters[1] and filters[1].search(row[12].lower()):
return False
if filters[2] and not self.check_digit(filters[2], row[14]):
return False
if filters[3] and not self.check_digit(filters[3], row[11], False):
return False
if filters[4] and row[3] != "Y":
return False
if filters[5]:
for cc in filters[5]:
if not cc:
continue
if row[13] is None:
return False
if cc[0] == "-":
if row[13].upper() == cc[1:].upper():
return False
elif cc.upper() != row[13].upper():
return False
return True
def set_filters(self, enable, f_in, f_out, size, bitrate, freeslot, country):
self.filters = [None, None, None, None, freeslot, None]
if f_in:
try:
f_in = re.compile(f_in.lower())
self.filters[0] = f_in
except sre_constants.error:
self.frame.set_text_bg(self.FilterIn.get_child(), "red", "white")
else:
self.frame.set_text_bg(self.FilterIn.get_child())
if f_out:
try:
f_out = re.compile(f_out.lower())
self.filters[1] = f_out
except sre_constants.error:
self.frame.set_text_bg(self.FilterOut.get_child(), "red", "white")
else:
self.frame.set_text_bg(self.FilterOut.get_child())
if size:
self.filters[2] = size
if bitrate:
self.filters[3] = bitrate
if country:
self.filters[5] = country.upper().split(" ")
self.usersiters.clear()
self.directoryiters.clear()
self.resultsmodel.clear()
self.numvisibleresults = 0
for row in self.all_data:
if self.numvisibleresults >= self.searches.maxdisplayedresults:
break
if self.check_filter(row):
self.add_row_to_model(row)
self.Counter.set_text("Results: %d/%d" % (self.numvisibleresults, len(self.all_data)))
def on_popup_menu_users(self, widget):
self.select_results()
self.popup_menu_users.clear()
if len(self.selected_users) > 0:
items = []
for user in self.selected_users:
popup = PopupMenu(self.frame, False)
popup.setup(
("#" + _("Send _message"), popup.on_send_message),
("#" + _("Show IP a_ddress"), popup.on_show_ip_address),
("#" + _("Get user i_nfo"), popup.on_get_user_info),
("#" + _("Brow_se files"), popup.on_browse_user),
("#" + _("Gi_ve privileges"), popup.on_give_privileges),
("", None),
("$" + _("_Add user to list"), popup.on_add_to_list),
("$" + _("_Ban this user"), popup.on_ban_user),
("$" + _("_Ignore this user"), popup.on_ignore_user),
("#" + _("Select User's Results"), self.on_select_user_results)
)
popup.set_user(user)
items.append((1, user, popup, self.on_popup_menu_user, popup))
self.popup_menu_users.setup(*items)
return True
def on_popup_menu_user(self, widget, popup=None):
if popup is None:
return
menu = popup
user = menu.user
items = menu.get_children()
act = False
if len(self.selected_users) >= 1:
act = True
items[0].set_sensitive(act)
items[1].set_sensitive(act)
items[2].set_sensitive(act)
items[3].set_sensitive(act)
items[6].set_active(user in [i[0] for i in self.frame.np.config.sections["server"]["userlist"]])
items[7].set_active(user in self.frame.np.config.sections["server"]["banlist"])
items[8].set_active(user in self.frame.np.config.sections["server"]["ignorelist"])
for i in range(4, 9):
items[i].set_sensitive(act)
return True
def on_select_user_results(self, widget):
if len(self.selected_users) == 0:
return
selected_user = widget.get_parent().user
sel = self.ResultsList.get_selection()
fmodel = self.ResultsList.get_model()
sel.unselect_all()
iterator = fmodel.get_iter_first()
select_user_row_iter(fmodel, sel, 1, selected_user, iterator)
self.select_results()
def select_results(self):
self.selected_results = set()
self.selected_users = set()
self.ResultsList.get_selection().selected_foreach(self.selected_results_callback)
def change_colours(self):
self.frame.set_text_bg(self.FilterIn.get_child())
self.frame.set_text_bg(self.FilterOut.get_child())
self.frame.set_text_bg(self.FilterSize.get_child())
self.frame.set_text_bg(self.FilterBitrate.get_child())
self.frame.set_text_bg(self.FilterCountry.get_child())
font = self.frame.np.config.sections["ui"]["searchfont"]
self.frame.change_list_font(self.ResultsList, font)
def save_columns(self):
columns = []
widths = []
for column in self.ResultsList.get_columns():
columns.append(column.get_visible())
widths.append(column.get_width())
self.frame.np.config.sections["columns"]["filesearch_columns"] = columns
self.frame.np.config.sections["columns"]["filesearch_widths"] = widths
def selected_results_callback(self, model, path, iterator):
user = model.get_value(iterator, 1)
if user is None:
return
self.selected_users.add(user)
filepath = model.get_value(iterator, 12)
if filepath == "":
# Result is not a file or directory, don't add it
return
bitrate = model.get_value(iterator, 9)
length = model.get_value(iterator, 10)
size = model.get_value(iterator, 14)
self.selected_results.add((user, filepath, size, bitrate, length))
def on_list_clicked(self, widget, event):
if event.button == 3:
return self.on_popup_menu(widget, event)
else:
pathinfo = widget.get_path_at_pos(event.x, event.y)
if pathinfo is None:
widget.get_selection().unselect_all()
elif event.button == 1 and event.type == Gdk.EventType._2BUTTON_PRESS:
self.select_results()
self.on_download_files(widget)
self.ResultsList.get_selection().unselect_all()
return True
return False
def on_popup_menu(self, widget, event):
if event.button != 3:
return False
set_treeview_selected_row(widget, event)
self.select_results()
items = self.popup_menu.get_children()
users = len(self.selected_users) > 0
files = len(self.selected_results) > 0
for i in range(0, 5):
items[i].set_sensitive(files)
items[0].set_sensitive(False)
items[1].set_sensitive(False)
items[4].set_sensitive(False)
items[6].set_sensitive(False)
items[7].set_sensitive(files)
items[8].set_sensitive(users)
for result in self.selected_results:
if not result[1].endswith('\\'):
# At least one selected result is a file, activate file-related items
items[0].set_sensitive(True)
items[1].set_sensitive(True)
items[4].set_sensitive(True)
items[6].set_sensitive(True)
break
self.popup_menu.popup(None, None, None, None, event.button, event.time)
widget.stop_emission_by_name("button_press_event")
return True
def cell_data_func(self, column, cellrenderer, model, iterator, dummy="dummy"):
imdl = model.get_value(iterator, 3)
color_id = imdl == "Y" and "search" or "searchq"
color = self.frame.np.config.sections["ui"][color_id]
if color:
cellrenderer.set_property("foreground", color)
else:
cellrenderer.set_property("foreground-set", False)
def meta_box(self, title="Meta Data", message="", data=None, modal=True):
win = MetaDialog(self.frame, message, data, modal)
win.set_title(title)
win.show()
Gtk.main()
return win.ret
def selected_results_all_data(self, model, path, iterator, data):
filename = model.get_value(iterator, 7)
# We only want to see the metadata of files, not directories
if filename != "":
num = model.get_value(iterator, 0)
user = model.get_value(iterator, 1)
immediate = model.get_value(iterator, 3)
speed = model.get_value(iterator, 4)
queue = model.get_value(iterator, 5)
directory = model.get_value(iterator, 6)
size = model.get_value(iterator, 8)
bitratestr = model.get_value(iterator, 9)
length = model.get_value(iterator, 10)
fn = model.get_value(iterator, 12)
country = model.get_value(iterator, 13)
data[len(data)] = {
"user": user,
"fn": fn,
"position": num,
"filename": filename,
"directory": directory,
"size": size,
"speed": speed,
"queue": queue,
"immediate": immediate,
"bitrate": bitratestr,
"length": length,
"country": country
}
def on_search_meta(self, widget):
if not self.frame.np.transfers:
return
data = {}
self.ResultsList.get_selection().selected_foreach(self.selected_results_all_data, data)
if data != {}:
self.meta_box(title=_("Search Results"), message=_("<b>Metadata</b> for Search Query: <i>%s</i>") % self.text, data=data, modal=True)
def on_download_files(self, widget, prefix=""):
if not self.frame.np.transfers:
return
for file in self.selected_results:
# Make sure the selected result is not a directory
if not file[1].endswith('\\'):
self.frame.np.transfers.get_file(file[0], file[1], prefix, size=file[2], bitrate=file[3], length=file[4], checkduplicate=True)
def on_download_files_to(self, widget):
folder = choose_dir(self.frame.MainWindow, self.frame.np.config.sections["transfers"]["downloaddir"], multichoice=False)
if folder is None:
return
for folders in folder:
self.on_download_files(widget, folders)
break
def on_download_folders(self, widget):
requested_folders = {}
for i in self.selected_results:
user = i[0]
folder = i[1].rsplit('\\', 1)[0]
if user not in requested_folders:
requested_folders[user] = []
if folder not in requested_folders[user]:
""" Ensure we don't send folder content requests for a folder more than once,
e.g. when several selected resuls belong to the same folder. """
self.frame.np.process_request_to_peer(user, slskmessages.FolderContentsRequest(None, folder))
requested_folders[user].append(folder)
def on_download_folders_to(self, widget):
directories = choose_dir(self.frame.MainWindow, self.frame.np.config.sections["transfers"]["downloaddir"], multichoice=False)
if directories is None or directories == []:
return
destination = directories[0]
for i in self.selected_results:
user = i[0]
folder = i[1].rsplit('\\', 1)[0]
if user not in self.frame.np.requested_folders:
self.frame.np.requested_folders[user] = {}
if folder not in self.frame.np.requested_folders[user]:
""" Ensure we don't send folder content requests for a folder more than once,
e.g. when several selected resuls belong to the same folder. """
self.frame.np.requested_folders[user][folder] = destination
self.frame.np.process_request_to_peer(user, slskmessages.FolderContentsRequest(None, folder))
def on_copy_url(self, widget):
user, path = next(iter(self.selected_results))[:2]
self.frame.set_clipboard_url(user, path)
def on_copy_dir_url(self, widget):
user, path = next(iter(self.selected_results))[:2]
path = "\\".join(path.split("\\")[:-1])
if path[:-1] != "/":
path += "/"
self.frame.set_clipboard_url(user, path)
def on_group(self, widget):
self.on_refilter(widget)
self.ResultsList.set_show_expanders(widget.get_active())
self.frame.np.config.sections["searches"]["group_searches"] = self.ResultGrouping.get_active()
if widget.get_active():
self.ResultsList.get_columns()[0].set_visible(False)
self.ExpandButton.show()
else:
self.ResultsList.get_columns()[0].set_visible(True)
self.ExpandButton.hide()
def on_toggle_expand_all(self, widget):
active = self.ExpandButton.get_active()
if active:
self.ResultsList.expand_all()
self.expand.set_from_icon_name("list-remove-symbolic", Gtk.IconSize.BUTTON)
else:
collapse_treeview(self.ResultsList, self.ResultGrouping.get_active())
self.expand.set_from_icon_name("list-add-symbolic", Gtk.IconSize.BUTTON)
self.frame.np.config.sections["searches"]["expand_searches"] = active
def on_toggle_filters(self, widget):
if widget.get_active():
self.FiltersContainer.show()
self.on_refilter(None)
else:
self.FiltersContainer.hide()
self.ResultsList.set_model(None)
self.set_filters(0, None, None, None, None, None, "")
self.ResultsList.set_model(self.resultsmodel)
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
if self.ExpandButton.get_active():
self.ResultsList.expand_all()
else:
collapse_treeview(self.ResultsList, self.ResultGrouping.get_active())
def on_ignore(self, widget):
self.searches.searches[self.id][5] = True # ignored
self.searches.wish_list.remove_wish(self.text)
widget.set_sensitive(False)
def on_clear(self, widget):
self.all_data = []
self.usersiters.clear()
self.directoryiters.clear()
self.resultsmodel.clear()
def on_close(self, widget):
if not self.frame.np.config.sections["searches"]["reopen_tabs"]:
if self.text not in self.frame.np.config.sections["server"]["autosearch"]:
self.on_ignore(widget)
self.searches.remove_tab(self)
def on_copy_search_term(self, widget):
self.frame.clip.set_text(self.text, -1)
def on_toggle_remember(self, widget):
self.remember = widget.get_active()
search = self.searches.searches[self.id]
if not self.remember:
self.searches.wish_list.remove_wish(search[1])
else:
self.searches.wish_list.add_wish(search[1])
def push_history(self, widget, title):
text = widget.get_child().get_text()
if not text.strip():
return None
text = text.strip()
history = self.frame.np.config.sections["searches"][title]
if text in history:
history.remove(text)
elif len(history) >= 5:
del history[-1]
history.insert(0, text)
self.frame.np.config.write_configuration()
self.add_combo(widget, text)
widget.get_child().set_text(text)
return text
def on_refilter(self, widget):
f_in = self.push_history(self.FilterIn, "filterin")
f_out = self.push_history(self.FilterOut, "filterout")
f_size = self.push_history(self.FilterSize, "filtersize")
f_br = self.push_history(self.FilterBitrate, "filterbr")
f_free = self.FilterFreeSlot.get_active()
f_country = self.push_history(self.FilterCountry, "filtercc")
self.ResultsList.set_model(None)
self.set_filters(1, f_in, f_out, f_size, f_br, f_free, f_country)
self.ResultsList.set_model(self.resultsmodel)
if self.ResultGrouping.get_active() > 0:
# Group by folder or user
if self.ExpandButton.get_active():
self.ResultsList.expand_all()
else:
collapse_treeview(self.ResultsList, self.ResultGrouping.get_active())
def on_about_filters(self, widget):
self.frame.on_about_filters(widget)
class Downloads(TransferList):
def __init__(self, frame):
TransferList.__init__(self,
frame,
frame.DownloadList,
type='downloads')
self.myvbox = self.frame.downloadsvbox
self.frame.DownloadList.set_property("rules-hint", True)
self.accel_group = gtk.AccelGroup()
self.popup_menu_users = PopupMenu(self.frame, False)
self.popup_menu_clear = popup2 = PopupMenu(self.frame, False)
popup2.setup(
("#" + _("Clear finished/aborted"), self.OnClearFinishedAborted),
("#" + _("Clear finished"), self.OnClearFinished),
("#" + _("Clear aborted"), self.OnClearAborted),
("#" + _("Clear paused"), self.OnClearPaused),
("#" + _("Clear filtered"), self.OnClearFiltered),
("#" + _("Clear queued"), self.OnClearQueued))
self.popup_menu = popup = PopupMenu(frame)
popup.setup(
("#" + _("Get place in _queue"), self.OnGetPlaceInQueue),
("", None), ("#" + _("Copy _URL"), self.OnCopyURL),
("#" + _("Copy folder URL"), self.OnCopyDirURL),
("#" + _("Send to _player"), self.OnPlayFiles),
("#" + _("View Metadata of file(s)"), self.OnDownloadMeta),
("#" + _("Open Directory"), self.OnOpenDirectory),
("#" + _("Search"), self.OnFileSearch),
(1, _("User(s)"), self.popup_menu_users, self.OnPopupMenuUsers),
("", None), ("#" + _("_Retry"), self.OnRetryTransfer), ("", None),
("#" + _("Abor_t"), self.OnAbortTransfer),
("#" + _("Abort & Delete"), self.OnAbortRemoveTransfer),
("#" + _("_Clear"), self.OnClearTransfer), ("", None),
(1, _("Clear Groups"), self.popup_menu_clear, None))
frame.DownloadList.connect("button_press_event", self.OnPopupMenu,
"mouse")
frame.DownloadList.connect("key-press-event", self.on_key_press_event)
cols = frame.DownloadList.get_columns()
try:
for i in range(len(cols)):
parent = cols[i].get_widget().get_ancestor(gtk.Button)
if parent:
parent.connect("button_press_event", PressHeader)
# Read Show / Hide column settings from last session
cols[i].set_visible(self.frame.np.config.sections["columns"]
["downloads_columns"][i])
except IndexError:
# Column count in config is probably incorrect (outdated?), don't crash
pass
frame.clearFinishedAbortedButton.connect("clicked",
self.OnClearFinishedAborted)
frame.clearQueuedButton.connect("clicked", self.OnTryClearQueued)
frame.retryTransferButton.connect("clicked", self.OnRetryTransfer)
frame.abortTransferButton.connect("clicked",
self.OnSelectAbortTransfer)
frame.deleteTransferButton.connect("clicked",
self.OnAbortRemoveTransfer)
frame.banDownloadButton.connect("clicked", self.OnBan)
frame.DownloadList.expand_all()
self.frame.ToggleAutoRetry.set_active(
self.frame.np.config.sections["transfers"]["autoretry_downloads"])
frame.ToggleAutoRetry.connect("toggled", self.OnToggleAutoRetry)
self.frame.ToggleTreeDownloads.set_active(
self.frame.np.config.sections["transfers"]["groupdownloads"])
frame.ToggleTreeDownloads.connect("toggled", self.OnToggleTree)
self.OnToggleTree(None)
self.frame.ExpandDownloads.set_active(
self.frame.np.config.sections["transfers"]["downloadsexpanded"])
frame.ExpandDownloads.connect("toggled", self.OnExpandDownloads)
self.OnExpandDownloads(None)
def saveColumns(self):
columns = []
widths = []
for column in self.frame.DownloadList.get_columns():
columns.append(column.get_visible())
widths.append(column.get_width())
self.frame.np.config.sections["columns"]["downloads_columns"] = columns
self.frame.np.config.sections["columns"]["downloads_widths"] = widths
def OnToggleAutoRetry(self, widget):
self.frame.np.config.sections["transfers"][
"autoretry_downloads"] = self.frame.ToggleAutoRetry.get_active()
def OnTryClearQueued(self, widget):
direction = "down"
win = OptionDialog(self.frame,
_("Clear All Queued Downloads?"),
modal=True,
status=None,
option=False,
third="")
win.connect("response", self.frame.on_clear_response, direction)
win.set_title(_("Nicotine+") + ": " + _("Clear Queued Transfers"))
win.set_icon(self.frame.images["n"])
win.show()
def expandcollapse(self, path):
if self.frame.ExpandDownloads.get_active():
self.frame.DownloadList.expand_row(path, True)
else:
self.frame.DownloadList.collapse_row(path)
def OnExpandDownloads(self, widget):
expanded = self.frame.ExpandDownloads.get_active()
if expanded:
self.frame.DownloadList.expand_all()
self.frame.ExpandDownloadsImage.set_from_stock(gtk.STOCK_REMOVE, 4)
else:
self.frame.DownloadList.collapse_all()
self.frame.ExpandDownloadsImage.set_from_stock(gtk.STOCK_ADD, 4)
self.frame.np.config.sections["transfers"][
"downloadsexpanded"] = expanded
self.frame.np.config.writeConfiguration()
def OnToggleTree(self, widget):
self.TreeUsers = self.frame.ToggleTreeDownloads.get_active()
self.frame.np.config.sections["transfers"][
"groupdownloads"] = self.TreeUsers
if not self.TreeUsers:
self.frame.ExpandDownloads.hide()
else:
self.frame.ExpandDownloads.show()
self.RebuildTransfers()
def MetaBox(self,
title="Meta Data",
message="",
data=None,
modal=True,
Search=False):
win = MetaDialog(self.frame, message, data, modal, Search=Search)
win.set_title(title)
win.set_icon(self.frame.images["n"])
win.show()
gtk.main()
return win.ret
def SelectedResultsAllData(self, model, path, iter, data):
if iter in self.selected_users:
return
user = model.get_value(iter, 0)
filename = model.get_value(iter, 1)
fullname = model.get_value(iter, 10)
size = speed = "0"
length = bitrate = None # noqa: F841
queue = immediate = num = country = bitratestr = ""
for transfer in self.frame.np.transfers.downloads:
if transfer.user == user and fullname == transfer.filename:
size = HumanSize(transfer.size)
try:
speed = str(int(transfer.speed))
speed += _(" KB/s")
except Exception:
pass
bitratestr = str(transfer.bitrate)
length = str(transfer.length)
directory = fullname.rsplit("\\", 1)[0]
data[len(data)] = {
"user": user,
"fn": fullname,
"position": num,
"filename": filename,
"directory": directory,
"size": size,
"speed": speed,
"queue": queue,
"immediate": immediate,
"bitrate": bitratestr,
"length": length,
"country": country
}
def OnDownloadMeta(self, widget):
if not self.frame.np.transfers:
return
data = {}
self.widget.get_selection().selected_foreach(
self.SelectedResultsAllData, data)
if data != {}:
self.MetaBox(title=_("Nicotine+:") + " " + _("Downloads Metadata"),
message=_("<b>Metadata</b> for Downloads"),
data=data,
modal=True,
Search=False)
def OnOpenDirectory(self, widget):
downloaddir = self.frame.np.config.sections["transfers"]["downloaddir"]
incompletedir = self.frame.np.config.sections["transfers"][
"incompletedir"]
if incompletedir == "":
incompletedir = downloaddir
filemanager = self.frame.np.config.sections["ui"]["filemanager"]
transfer = self.selected_transfers[0]
complete_path = os.path.join(downloaddir, transfer.path)
if transfer.path == "":
if transfer.status == "Finished":
executeCommand(filemanager, downloaddir)
else:
executeCommand(filemanager, incompletedir)
elif os.path.exists(complete_path): # and tranfer.status is "Finished"
executeCommand(filemanager, complete_path)
else:
executeCommand(filemanager, incompletedir)
def RebuildTransfers(self):
if self.frame.np.transfers is None:
return
self.Clear()
self.update()
def select_transfers(self):
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
def OnBan(self, widgets):
self.select_transfers()
for user in self.selected_users:
self.frame.BanUser(user)
def OnSelectAbortTransfer(self, widget):
self.select_transfers()
self.OnAbortTransfer(widget, False)
def OnSelectUserTransfer(self, widget):
if len(self.selected_users) == 0:
return
selected_user = widget.get_parent().user
sel = self.frame.DownloadList.get_selection()
fmodel = self.frame.DownloadList.get_model()
sel.unselect_all()
for item in self.transfers:
user_file, iter, transfer = item
user, filepath = user_file
if selected_user == user:
ix = fmodel.get_path(iter)
sel.select_path(ix, )
self.select_transfers()
def on_key_press_event(self, widget, event):
key = Gdk.keyval_name(event.keyval)
if key in ("P", "p"):
self.OnPopupMenu(widget, event, "keyboard")
else:
self.select_transfers()
if key in ("T", "t"):
self.OnAbortTransfer(widget)
elif key in ("R", "r"):
self.OnRetryTransfer(widget)
elif key == "Delete":
self.OnAbortTransfer(widget, True, True)
def OnPlayFiles(self, widget, prefix=""):
start_new_thread(self._OnPlayFiles, (widget, prefix))
def _OnPlayFiles(self, widget, prefix=""):
executable = self.frame.np.config.sections["players"]["default"]
downloaddir = self.frame.np.config.sections["transfers"]["downloaddir"]
if "$" not in executable:
return
for fn in self.selected_transfers:
if fn.file is None:
continue
playfile = None
if os.path.exists(fn.file.name):
playfile = fn.file.name
else:
# If this file doesn't exist anymore, it may have finished downloading and have been renamed
# try looking in the download directory and match the original filename.
basename = str.split(fn.filename, '\\')[-1]
path = os.sep.join([downloaddir, basename])
if os.path.exists(path):
playfile = path
if playfile:
executeCommand(executable, playfile, background=False)
def OnPopupMenuUsers(self, widget):
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
self.popup_menu_users.clear()
if len(self.selected_users) > 0:
items = []
self.selected_users.sort(key=str.lower)
for user in self.selected_users:
popup = PopupMenu(self.frame, False)
popup.setup(
("#" + _("Send _message"), popup.OnSendMessage),
("#" + _("Show IP a_ddress"), popup.OnShowIPaddress),
("#" + _("Get user i_nfo"), popup.OnGetUserInfo),
("#" + _("Brow_se files"), popup.OnBrowseUser),
("#" + _("Gi_ve privileges"), popup.OnGivePrivileges),
("", None),
("$" + _("_Add user to list"), popup.OnAddToList),
("$" + _("_Ban this user"), popup.OnBanUser),
("$" + _("_Ignore this user"), popup.OnIgnoreUser),
("#" + _("Select User's Transfers"),
self.OnSelectUserTransfer))
popup.set_user(user)
items.append((1, user, popup, self.OnPopupMenuUser, popup))
self.popup_menu_users.setup(*items)
return True
def OnPopupMenuUser(self, widget, popup=None):
if popup is None:
return
menu = popup
user = menu.user
items = menu.get_children()
act = False
if len(self.selected_users) >= 1:
act = True
items[0].set_sensitive(act)
items[1].set_sensitive(act)
items[2].set_sensitive(act)
items[3].set_sensitive(act)
items[6].set_active(user in [
i[0] for i in self.frame.np.config.sections["server"]["userlist"]
])
items[7].set_active(
user in self.frame.np.config.sections["server"]["banlist"])
items[8].set_active(
user in self.frame.np.config.sections["server"]["ignorelist"])
for i in range(4, 9):
items[i].set_sensitive(act)
return True
def DoubleClick(self, event):
self.select_transfers()
dc = self.frame.np.config.sections["transfers"]["download_doubleclick"]
if dc == 1: # Send to player
self.OnPlayFiles(None)
elif dc == 2: # File manager
self.OnOpenDirectory(None)
elif dc == 3: # Search
self.OnFileSearch(None)
elif dc == 4: # Abort
self.OnAbortTransfer(None, False)
elif dc == 5: # Clear
self.OnClearTransfer(None)
elif dc == 6: # Retry
self.OnRetryTransfer(None)
def OnPopupMenu(self, widget, event, kind):
if kind == "mouse":
if event.button != 3:
if event.button == 1 and event.type == Gdk.EventType._2BUTTON_PRESS:
self.DoubleClick(event)
return False
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
users = len(self.selected_users) > 0
multi_users = len(self.selected_users) > 1 # noqa: F841
files = len(self.selected_transfers) > 0
multi_files = len(self.selected_transfers) > 1
self.SelectCurrentRow(event, kind)
items = self.popup_menu.get_children()
if users:
items[7].set_sensitive(True) # Users Menu
else:
items[7].set_sensitive(False) # Users Menu
if files:
act = True
else:
act = False
items[0].set_sensitive(act) # Place
items[4].set_sensitive(act) # Send to player
items[5].set_sensitive(act) # View Meta
items[6].set_sensitive(act) # File manager
items[8].set_sensitive(act) # Search filename
act = False
if not multi_files and files:
act = True
items[2].set_sensitive(act) # Copy URL
items[3].set_sensitive(act) # Copy Folder URL
if not users or not files:
# Disable options
# Abort, Abort and Remove, retry, clear
act = False
else:
act = True
for i in range(10, 15):
items[i].set_sensitive(act)
self.popup_menu.popup(None, None, None, None, 3, event.time)
if kind == "keyboard":
widget.emit_stop_by_name("key_press_event")
elif kind == "mouse":
widget.emit_stop_by_name("button_press_event")
return True
def update(self, transfer=None, forced=False):
TransferList.update(self, transfer, forced)
if transfer is None and self.frame.np.transfers is not None:
self.frame.np.transfers.SaveDownloads()
def OnGetPlaceInQueue(self, widget):
self.select_transfers()
for i in self.selected_transfers:
if i.status != "Queued":
continue
self.frame.np.ProcessRequestToPeer(
i.user, slskmessages.PlaceInQueueRequest(None, i.filename))
def OnFileSearch(self, widget):
self.select_transfers()
for transfer in self.selected_transfers:
self.frame.SearchEntry.set_text(
transfer.filename.rsplit("\\", 1)[1])
self.frame.ChangeMainPage(None, "search")
break
def OnRetryTransfer(self, widget):
self.select_transfers()
for transfer in self.selected_transfers:
if transfer.status in ["Finished", "Old"]:
continue
self.frame.np.transfers.AbortTransfer(transfer)
transfer.req = None
self.frame.np.transfers.getFile(transfer.user, transfer.filename,
transfer.path, transfer)
self.frame.np.transfers.SaveDownloads()
def OnAbortRemoveTransfer(self, widget):
self.select_transfers()
self.OnClearTransfer(widget)
class Search:
def __init__(self, searches, text, id, mode, remember, showtab):
self.searches = searches
self.frame = searches.frame
# Build the window
load_ui_elements(self,
os.path.join(self.frame.gui_dir, "ui", "search.ui"))
self.text = text
self.searchterm_words_include = [
p for p in text.lower().split() if not p.startswith('-')
]
self.searchterm_words_ignore = [
p[1:] for p in text.lower().split()
if p.startswith('-') and len(p) > 1
]
self.id = id
self.mode = mode
self.remember = remember
self.showtab = showtab
self.usersiters = {}
self.directoryiters = {}
self.users = set()
self.all_data = []
self.filters = None
self.clearing_filters = False
self.resultslimit = 2000
self.numvisibleresults = 0
self.active_filter_count = 0
self.operators = {
'<': operator.lt,
'<=': operator.le,
'==': operator.eq,
'!=': operator.ne,
'>=': operator.ge,
'>': operator.gt
}
if mode not in ("global", "wishlist"):
self.RememberCheckButton.set_sensitive(False)
self.RememberCheckButton.set_active(remember)
""" Columns """
self.ResultsList.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.ResultsList.set_enable_tree_lines(True)
self.ResultsList.set_headers_clickable(True)
self.ResultsList.set_rubber_banding(True)
self.resultsmodel = Gtk.TreeStore(
GObject.TYPE_UINT64, # (0) num
str, # (1) user
GObject.TYPE_OBJECT, # (2) flag
str, # (3) immediatedl
str, # (4) h_speed
str, # (5) h_queue
str, # (6) directory
str, # (7) filename
str, # (8) h_size
str, # (9) h_bitrate
str, # (10) length
GObject.TYPE_UINT64, # (11) bitrate
str, # (12) fullpath
str, # (13) country
GObject.TYPE_UINT64, # (14) size
GObject.TYPE_UINT64, # (15) speed
GObject.TYPE_UINT64 # (16) queue
)
self.cols = cols = initialise_columns(
"file_search", self.ResultsList,
["id", _("ID"), 50, "text", self.cell_data_func, None],
["user", _("User"), 200, "text", self.cell_data_func, None],
["country", _("Country"), 25, "pixbuf", None, None], [
"immediate_download",
_("Immediate Download"), 50, "center", self.cell_data_func,
None
], ["speed",
_("Speed"), 90, "number", self.cell_data_func, None], [
"in_queue",
_("In Queue"), 90, "center", self.cell_data_func, None
],
["folder",
_("Folder"), 400, "text", self.cell_data_func, None], [
"filename",
_("Filename"), 400, "text", self.cell_data_func, None
], ["size",
_("Size"), 100, "number", self.cell_data_func, None], [
"bitrate",
_("Bitrate"), 100, "number", self.cell_data_func, None
],
["length",
_("Length"), 0, "number", self.cell_data_func, None])
cols["id"].set_sort_column_id(0)
cols["user"].set_sort_column_id(1)
cols["country"].set_sort_column_id(13)
cols["immediate_download"].set_sort_column_id(3)
cols["speed"].set_sort_column_id(15)
cols["in_queue"].set_sort_column_id(16)
cols["folder"].set_sort_column_id(6)
cols["filename"].set_sort_column_id(7)
cols["size"].set_sort_column_id(14)
cols["bitrate"].set_sort_column_id(11)
cols["length"].set_sort_column_id(10)
cols["country"].get_widget().hide()
self.ResultsList.set_model(self.resultsmodel)
self.update_visuals()
""" Filters """
self.ShowFilters.set_active(
self.frame.np.config.sections["searches"]["filters_visible"])
self.populate_filters()
""" Popup """
self.popup_menu_users = PopupMenu(self.frame, False)
self.popup_menu = popup = PopupMenu(self.frame)
popup.setup(
("#" + _("_Download File(s)"), self.on_download_files),
("#" + _("Download File(s) _To..."), self.on_download_files_to),
("#" + _("Download _Folder(s)"), self.on_download_folders),
("#" + _("Download F_older(s) To..."),
self.on_download_folders_to),
("#" + _("_Browse Folder"), self.on_browse_folder),
("#" + _("File _Properties"), self.on_file_properties), ("", None),
("#" + _("Copy _File Path"), self.on_copy_file_path),
("#" + _("Copy _URL"), self.on_copy_url),
("#" + _("Copy Folder U_RL"), self.on_copy_dir_url), ("", None),
(1, _("User(s)"), self.popup_menu_users, self.on_popup_menu_users))
""" Grouping """
self.ResultGrouping.set_active(
self.frame.np.config.sections["searches"]["group_searches"])
self.ExpandButton.set_active(
self.frame.np.config.sections["searches"]["expand_searches"])
def on_tooltip(self, widget, x, y, keyboard_mode, tooltip):
return show_country_tooltip(widget, x, y, tooltip, 13, stripprefix='')
def populate_filters(self, set_default_filters=True):
for combobox in (self.FilterIn, self.FilterOut, self.FilterSize,
self.FilterBitrate, self.FilterCountry):
combobox.remove_all()
if set_default_filters and self.frame.np.config.sections["searches"][
"enablefilters"]:
sfilter = self.frame.np.config.sections["searches"]["defilter"]
self.FilterInEntry.set_text(sfilter[0])
self.FilterOutEntry.set_text(sfilter[1])
self.FilterSizeEntry.set_text(sfilter[2])
self.FilterBitrateEntry.set_text(sfilter[3])
self.FilterFreeSlot.set_active(sfilter[4])
if (len(sfilter) > 5):
self.FilterCountryEntry.set_text(sfilter[5])
self.on_refilter(None)
for i in ['0', '128', '160', '192', '256', '320']:
self.FilterBitrate.append_text(i)
for i in [">10MiB", "<10MiB", "<5MiB", "<1MiB", ">0"]:
self.FilterSize.append_text(i)
s_config = self.frame.np.config.sections["searches"]
for i in s_config["filterin"]:
self.add_combo(self.FilterIn, i, True)
for i in s_config["filterout"]:
self.add_combo(self.FilterOut, i, True)
for i in s_config["filtersize"]:
self.add_combo(self.FilterSize, i, True)
for i in s_config["filterbr"]:
self.add_combo(self.FilterBitrate, i, True)
for i in s_config["filtercc"]:
self.add_combo(self.FilterCountry, i, True)
def focus_combobox(self, button):
self.frame.focus_combobox(button)
def add_combo(self, combobox, text, list=False):
text = text.strip()
if not text:
return False
model = combobox.get_model()
iterator = model.get_iter_first()
match = False
while iterator is not None:
value = model.get_value(iterator, 0)
if value.strip() == text:
match = True
iterator = model.iter_next(iterator)
if not match:
if list:
combobox.append_text(text)
else:
combobox.prepend_text(text)
def add_user_results(self, msg, user, country):
if user in self.users:
return
self.users.add(user)
counter = len(self.all_data) + 1
inqueue = msg.inqueue
ulspeed = msg.ulspeed
h_speed = human_speed(ulspeed)
if msg.freeulslots:
imdl = "Y"
inqueue = 0
else:
imdl = "N"
h_queue = humanize(inqueue)
append = False
maxstoredresults = self.searches.maxstoredresults
for result in msg.list:
if counter > maxstoredresults:
break
fullpath = result[1]
fullpath_lower = fullpath.lower()
if any(word in fullpath_lower
for word in self.searchterm_words_ignore):
""" Filter out results with filtered words (e.g. nicotine -music) """
log.add_search(
_("Filtered out excluded search result " + fullpath +
" from user " + user))
continue
if not any(word in fullpath_lower
for word in self.searchterm_words_include):
""" Some users may send us wrong results, filter out such ones """
log.add_search(
_("Filtered out inexact or incorrect search result " +
fullpath + " from user " + user))
continue
fullpath_split = reversed(fullpath.split('\\'))
name = next(fullpath_split)
directory = '\\'.join(fullpath_split)
size = result[2]
h_size = human_size(size)
h_bitrate, bitrate, h_length = get_result_bitrate_length(
size, result[4])
self.append([
counter, user,
self.get_flag(user, country), imdl, h_speed, h_queue,
directory, name, h_size, h_bitrate, h_length, bitrate,
fullpath, country, size, ulspeed, inqueue
])
append = True
counter += 1
if append:
# If this search wasn't initiated by us (e.g. wishlist), and the results aren't spoofed, show tab
if not self.showtab:
self.searches.show_tab(self, self.id, self.text, self.mode)
self.showtab = True
# Update number of results
self.update_result_counter()
# Update tab notification
self.frame.searches.request_changed(self.Main)
self.frame.request_tab_icon(self.frame.SearchTabLabel)
def get_flag(self, user, flag=None):
if flag is not None:
flag = "flag_" + flag.lower()
self.frame.flag_users[user] = flag
else:
flag = self.frame.get_user_flag(user)
return self.frame.get_flag_image(flag)
def append(self, row):
self.all_data.append(row)
if self.numvisibleresults >= self.searches.maxdisplayedresults:
return
if not self.check_filter(row):
return
iterator = self.add_row_to_model(row)
if self.ResultGrouping.get_active_id() != "ungrouped":
# Group by folder or user
if self.ExpandButton.get_active():
path = None
if iterator is not None:
path = self.resultsmodel.get_path(iterator)
if path is not None:
self.ResultsList.expand_to_path(path)
else:
collapse_treeview(self.ResultsList,
self.ResultGrouping.get_active_id())
def add_row_to_model(self, row):
counter, user, flag, immediatedl, h_speed, h_queue, directory, filename, h_size, h_bitrate, length, bitrate, fullpath, country, size, speed, queue = row
if self.ResultGrouping.get_active_id() != "ungrouped":
# Group by folder or user
if user not in self.usersiters:
self.usersiters[user] = self.resultsmodel.append(
None, [
0, user,
self.get_flag(
user, country), immediatedl, h_speed, h_queue, "",
"", "", "", "", 0, "", country, 0, speed, queue
])
parent = self.usersiters[user]
if self.ResultGrouping.get_active_id() == "folder_grouping":
# Group by folder
if directory not in self.directoryiters:
self.directoryiters[directory] = self.resultsmodel.append(
self.usersiters[user], [
0, user,
self.get_flag(user, country), immediatedl, h_speed,
h_queue, directory, "", "", "", "", 0,
fullpath.rsplit('\\', 1)[0] + '\\', country, 0,
speed, queue
])
row = row[:]
row[6] = "" # Directory not visible for file row if "group by folder" is enabled
parent = self.directoryiters[directory]
else:
parent = None
try:
iterator = self.resultsmodel.append(parent, row)
self.numvisibleresults += 1
except Exception as e:
types = []
for i in row:
types.append(type(i))
log.add_warning(_("Search row error: %(exception)s %(row)s"), {
'exception': e,
'row': row
})
iterator = None
return iterator
def check_digit(self, sfilter, value, factorize=True):
op = ">="
if sfilter[:1] in (">", "<", "="):
op, sfilter = sfilter[:1] + "=", sfilter[1:]
if not sfilter:
return True
factor = 1
if factorize:
base = 1024 # Default to binary for "k", "m", "g" suffixes
if sfilter[-1:].lower() == 'b':
base = 1000 # Byte suffix detected, prepare to use decimal if necessary
sfilter = sfilter[:-1]
if sfilter[-1:].lower() == 'i':
base = 1024 # Binary requested, stop using decimal
sfilter = sfilter[:-1]
if sfilter.lower()[-1:] == "g":
factor = pow(base, 3)
sfilter = sfilter[:-1]
elif sfilter.lower()[-1:] == "m":
factor = pow(base, 2)
sfilter = sfilter[:-1]
elif sfilter.lower()[-1:] == "k":
factor = base
sfilter = sfilter[:-1]
if not sfilter:
return True
try:
sfilter = int(sfilter) * factor
except ValueError:
return True
operation = self.operators.get(op)
return operation(value, sfilter)
def check_filter(self, row):
filters = self.filters
if self.active_filter_count == 0:
return True
# "Included text"-filter, check full file path (located at index 12 in row)
if filters[0] and not filters[0].search(row[12].lower()):
return False
# "Excluded text"-filter, check full file path (located at index 12 in row)
if filters[1] and filters[1].search(row[12].lower()):
return False
if filters[2] and not self.check_digit(filters[2], row[14]):
return False
if filters[3] and not self.check_digit(filters[3], row[11], False):
return False
if filters[4] and row[3] != "Y":
return False
if filters[5]:
for cc in filters[5]:
if not cc:
continue
if row[13] is None:
return False
if cc[0] == "-":
if row[13].upper() == cc[1:].upper():
return False
elif cc.upper() != row[13].upper():
return False
return True
def set_filters(self, enable, f_in, f_out, size, bitrate, freeslot,
country):
self.filters = [None, None, None, None, freeslot, None]
self.active_filter_count = 0
if f_in:
try:
f_in = re.compile(f_in.lower())
self.filters[0] = f_in
except sre_constants.error:
set_widget_fg_bg_css(self.FilterInEntry, "red", "white")
else:
set_widget_fg_bg_css(self.FilterInEntry)
self.active_filter_count += 1
if f_out:
try:
f_out = re.compile(f_out.lower())
self.filters[1] = f_out
except sre_constants.error:
set_widget_fg_bg_css(self.FilterOutEntry, "red", "white")
else:
set_widget_fg_bg_css(self.FilterOutEntry)
self.active_filter_count += 1
if size:
self.filters[2] = size
self.active_filter_count += 1
if bitrate:
self.filters[3] = bitrate
self.active_filter_count += 1
if country:
self.filters[5] = country.upper().split(" ")
self.active_filter_count += 1
if freeslot:
self.active_filter_count += 1
self.usersiters.clear()
self.directoryiters.clear()
self.resultsmodel.clear()
self.numvisibleresults = 0
for row in self.all_data:
if self.numvisibleresults >= self.searches.maxdisplayedresults:
break
if self.check_filter(row):
self.add_row_to_model(row)
# Update number of visible results
self.update_result_counter()
self.update_filter_counter(self.active_filter_count)
def on_popup_menu_users(self, widget):
self.select_results()
self.popup_menu_users.clear()
if len(self.selected_users) > 0:
items = []
for user in self.selected_users:
popup = PopupMenu(self.frame, False)
popup.setup_user_menu(user)
popup.append_item(("", None))
popup.append_item(("#" + _("Select User's Transfers"),
self.on_select_user_results))
items.append((1, user, popup, self.on_popup_menu_user, popup))
self.popup_menu_users.setup(*items)
return True
def on_popup_menu_user(self, widget, popup=None):
if popup is None:
return
popup.toggle_user_items()
return True
def on_select_user_results(self, widget):
if len(self.selected_users) == 0:
return
selected_user = widget.get_parent().user
sel = self.ResultsList.get_selection()
fmodel = self.ResultsList.get_model()
sel.unselect_all()
iterator = fmodel.get_iter_first()
select_user_row_iter(fmodel, sel, 1, selected_user, iterator)
self.select_results()
def select_results(self):
self.selected_results = set()
self.selected_users = set()
self.ResultsList.get_selection().selected_foreach(
self.selected_results_callback)
def update_result_counter(self):
self.Counter.set_markup("<b>%d</b>" % self.numvisibleresults)
def update_visuals(self):
for widget in self.__dict__.values():
update_widget_visuals(widget, list_font_target="searchfont")
def save_columns(self):
save_columns("file_search", self.ResultsList.get_columns())
def selected_results_callback(self, model, path, iterator):
user = model.get_value(iterator, 1)
if user is None:
return
self.selected_users.add(user)
filepath = model.get_value(iterator, 12)
if filepath == "":
# Result is not a file or directory, don't add it
return
bitrate = model.get_value(iterator, 9)
length = model.get_value(iterator, 10)
size = model.get_value(iterator, 14)
self.selected_results.add((user, filepath, size, bitrate, length))
def on_list_clicked(self, widget, event):
if triggers_context_menu(event):
set_treeview_selected_row(widget, event)
return self.on_popup_menu(widget)
pathinfo = widget.get_path_at_pos(event.x, event.y)
if pathinfo is None:
widget.get_selection().unselect_all()
elif event.button == 1 and event.type == Gdk.EventType._2BUTTON_PRESS:
self.select_results()
self.on_download_files(widget)
self.ResultsList.get_selection().unselect_all()
return True
return False
def on_key_press_event(self, widget, event):
key = Gdk.keyval_name(event.keyval)
self.select_results()
if key in ("C", "c") and event.state in (
Gdk.ModifierType.CONTROL_MASK,
Gdk.ModifierType.LOCK_MASK | Gdk.ModifierType.CONTROL_MASK):
self.on_copy_file_path(widget)
else:
# No key match, continue event
return False
widget.stop_emission_by_name("key_press_event")
return True
def on_popup_menu(self, widget):
self.select_results()
items = self.popup_menu.get_items()
users = len(self.selected_users) > 0
files = len(self.selected_results) > 0
for i in (_("_Download File(s)"), _("Download File(s) _To..."),
_("File _Properties"), _("Copy _URL")):
items[i].set_sensitive(False)
for i in (_("Download _Folder(s)"), _("Download F_older(s) To..."),
_("_Browse Folder"), _("Copy _File Path"),
_("Copy Folder U_RL")):
items[i].set_sensitive(files)
items[_("User(s)")].set_sensitive(users)
for result in self.selected_results:
if not result[1].endswith('\\'):
# At least one selected result is a file, activate file-related items
for i in (_("_Download File(s)"), _("Download File(s) _To..."),
_("File _Properties"), _("Copy _URL")):
items[i].set_sensitive(True)
break
self.popup_menu.popup()
return True
def cell_data_func(self,
column,
cellrenderer,
model,
iterator,
dummy="dummy"):
imdl = model.get_value(iterator, 3)
color_id = imdl == "Y" and "search" or "searchq"
color = self.frame.np.config.sections["ui"][color_id]
if color:
cellrenderer.set_property("foreground", color)
else:
cellrenderer.set_property("foreground-set", False)
def on_browse_folder(self, widget):
requested_folders = set()
for file in self.selected_results:
user = file[0]
folder = file[1].rsplit('\\', 1)[0]
if folder not in requested_folders:
self.frame.browse_user(user, folder)
requested_folders.add(folder)
def selected_results_all_data(self, model, path, iterator, data):
filename = model.get_value(iterator, 7)
# We only want to see the metadata of files, not directories
if filename != "":
num = model.get_value(iterator, 0)
user = model.get_value(iterator, 1)
immediate = model.get_value(iterator, 3)
speed = model.get_value(iterator, 4)
queue = model.get_value(iterator, 5)
size = model.get_value(iterator, 8)
bitratestr = model.get_value(iterator, 9)
length = model.get_value(iterator, 10)
fn = model.get_value(iterator, 12)
directory = fn.rsplit('\\', 1)[0]
cc = model.get_value(iterator, 13)
country = "%s / %s" % (cc, code2name(cc))
data.append({
"user": user,
"fn": fn,
"position": num,
"filename": filename,
"directory": directory,
"size": size,
"speed": speed,
"queue": queue,
"immediate": immediate,
"bitrate": bitratestr,
"length": length,
"country": country
})
def on_file_properties(self, widget):
if not self.frame.np.transfers:
return
data = []
self.ResultsList.get_selection().selected_foreach(
self.selected_results_all_data, data)
if data:
FileProperties(self.frame, data).show()
def on_download_files(self, widget, prefix=""):
if not self.frame.np.transfers:
return
for file in self.selected_results:
# Make sure the selected result is not a directory
if not file[1].endswith('\\'):
self.frame.np.transfers.get_file(file[0],
file[1],
prefix,
size=file[2],
bitrate=file[3],
length=file[4],
checkduplicate=True)
def on_download_files_to(self, widget):
folder = choose_dir(
self.frame.MainWindow,
self.frame.np.config.sections["transfers"]["downloaddir"],
multichoice=False)
if folder is None:
return
for folders in folder:
self.on_download_files(widget, folders)
break
def on_download_folders(self, widget):
requested_folders = {}
for i in self.selected_results:
user = i[0]
folder = i[1].rsplit('\\', 1)[0]
if user not in requested_folders:
requested_folders[user] = []
if folder not in requested_folders[user]:
""" Ensure we don't send folder content requests for a folder more than once,
e.g. when several selected resuls belong to the same folder. """
self.frame.np.send_message_to_peer(
user, slskmessages.FolderContentsRequest(None, folder))
requested_folders[user].append(folder)
def on_download_folders_to(self, widget):
directories = choose_dir(
self.frame.MainWindow,
self.frame.np.config.sections["transfers"]["downloaddir"],
multichoice=False)
if directories is None or directories == []:
return
destination = directories[0]
for i in self.selected_results:
user = i[0]
folder = i[1].rsplit('\\', 1)[0]
if user not in self.frame.np.requested_folders:
self.frame.np.requested_folders[user] = {}
if folder not in self.frame.np.requested_folders[user]:
""" Ensure we don't send folder content requests for a folder more than once,
e.g. when several selected resuls belong to the same folder. """
self.frame.np.requested_folders[user][folder] = destination
self.frame.np.send_message_to_peer(
user, slskmessages.FolderContentsRequest(None, folder))
def on_copy_file_path(self, widget):
if not self.selected_results:
return
user, path = next(iter(self.selected_results))[:2]
self.frame.clip.set_text(path, -1)
def on_copy_url(self, widget):
user, path = next(iter(self.selected_results))[:2]
self.frame.set_clipboard_url(user, path)
def on_copy_dir_url(self, widget):
user, path = next(iter(self.selected_results))[:2]
path = "\\".join(path.split("\\")[:-1])
if path[:-1] != "/":
path += "/"
self.frame.set_clipboard_url(user, path)
def on_group(self, widget):
self.on_refilter(widget)
self.ResultsList.set_show_expanders(widget.get_active())
self.frame.np.config.sections["searches"][
"group_searches"] = widget.get_active()
if widget.get_active():
self.cols["id"].set_visible(False)
self.ExpandButton.show()
else:
self.cols["id"].set_visible(True)
self.ExpandButton.hide()
def on_toggle_expand_all(self, widget):
active = self.ExpandButton.get_active()
if active:
self.ResultsList.expand_all()
self.expand.set_from_icon_name("go-up-symbolic",
Gtk.IconSize.BUTTON)
else:
collapse_treeview(self.ResultsList,
self.ResultGrouping.get_active_id())
self.expand.set_from_icon_name("go-down-symbolic",
Gtk.IconSize.BUTTON)
self.frame.np.config.sections["searches"]["expand_searches"] = active
def on_toggle_filters(self, widget):
visible = widget.get_active()
self.FiltersContainer.set_visible(visible)
self.frame.np.config.sections["searches"]["filters_visible"] = visible
def on_clear(self, widget):
self.all_data = []
self.usersiters.clear()
self.directoryiters.clear()
self.resultsmodel.clear()
self.numvisibleresults = 0
# Update number of visible results
self.update_result_counter()
def on_close(self, widget):
self.searches.remove_tab(self)
def on_copy_search_term(self, widget):
self.frame.clip.set_text(self.text, -1)
def on_toggle_remember(self, widget):
self.remember = widget.get_active()
search = self.searches.searches[self.id]
if not self.remember:
self.searches.wish_list.remove_wish(search["term"])
else:
self.searches.wish_list.add_wish(search["term"])
def push_history(self, widget, title):
text = widget.get_child().get_text()
if not text.strip():
return None
text = text.strip()
history = self.frame.np.config.sections["searches"][title]
if text in history:
history.remove(text)
elif len(history) >= 5:
del history[-1]
history.insert(0, text)
self.frame.np.config.write_configuration()
self.add_combo(widget, text)
widget.get_child().set_text(text)
return text
def on_refilter(self, widget, *args):
if self.clearing_filters:
return
f_in = self.push_history(self.FilterIn, "filterin")
f_out = self.push_history(self.FilterOut, "filterout")
f_size = self.push_history(self.FilterSize, "filtersize")
f_br = self.push_history(self.FilterBitrate, "filterbr")
f_free = self.FilterFreeSlot.get_active()
f_country = self.push_history(self.FilterCountry, "filtercc")
self.ResultsList.set_model(None)
self.set_filters(1, f_in, f_out, f_size, f_br, f_free, f_country)
self.ResultsList.set_model(self.resultsmodel)
if self.ResultGrouping.get_active_id() != "ungrouped":
# Group by folder or user
if self.ExpandButton.get_active():
self.ResultsList.expand_all()
else:
collapse_treeview(self.ResultsList,
self.ResultGrouping.get_active_id())
def on_clear_filters(self, widget):
self.clearing_filters = True
self.FilterInEntry.set_text("")
self.FilterOutEntry.set_text("")
self.FilterSizeEntry.set_text("")
self.FilterBitrateEntry.set_text("")
self.FilterCountryEntry.set_text("")
self.FilterFreeSlot.set_active(False)
self.clearing_filters = False
self.FilterInEntry.grab_focus()
self.on_refilter(widget)
def on_about_filters(self, widget):
if not hasattr(self, "AboutSearchFiltersPopover"):
load_ui_elements(
self,
os.path.join(self.frame.gui_dir, "ui", "popovers",
"searchfilters.ui"))
self.AboutSearchFiltersPopover.set_relative_to(self.ShowChatHelp)
self.AboutSearchFiltersPopover.popup()
def update_filter_counter(self, count):
if count > 0:
self.FilterLabel.set_text(_("Result Filters") + " *")
else:
self.FilterLabel.set_text(_("Result Filters"))
self.FilterLabel.set_tooltip_text("%d active filter(s)" % count)
class Uploads(TransferList):
def __init__(self, frame):
TransferList.__init__(self, frame, frame.UploadList, type='uploads')
self.myvbox = self.frame.uploadsvbox
self.frame.UploadList.set_property("rules-hint", True)
self.popup_menu2 = popup2 = PopupMenu(frame)
popup2.setup(
("#" + _("Clear finished/erred"), self.OnClearFinishedErred),
("#" + _("Clear finished/aborted"), self.OnClearFinishedAborted),
("#" + _("Clear finished"), self.OnClearFinished),
("#" + _("Clear aborted"), self.OnClearAborted),
("#" + _("Clear queued"), self.OnClearQueued),
("#" + _("Clear failed"), self.OnClearFailed))
self.popup_menu_users = PopupMenu(frame)
self.popup_menu = popup = PopupMenu(frame)
popup.setup(
("#" + _("Copy _URL"), self.OnCopyURL),
("#" + _("Copy folder URL"), self.OnCopyDirURL),
("#" + _("Send to _player"), self.OnPlayFiles),
("#" + _("Open Directory"), self.OnOpenDirectory),
("#" + _("Search"), self.OnFileSearch),
(1, _("User(s)"), self.popup_menu_users, self.OnPopupMenuUsers),
("", None), ("#" + _("Abor_t"), self.OnAbortTransfer),
("#" + _("_Clear"), self.OnClearTransfer),
("#" + _("_Retry"), self.OnUploadTransfer), ("", None),
(1, _("Clear Groups"), self.popup_menu2, None))
frame.UploadList.connect("button_press_event", self.OnPopupMenu,
"mouse")
frame.UploadList.connect("key-press-event", self.on_key_press_event)
cols = frame.UploadList.get_columns()
for i in range(10):
parent = cols[i].get_widget().get_ancestor(gtk.Button)
if parent:
parent.connect("button_press_event", PressHeader)
# Read Show / Hide column settings from last session
cols[i].set_visible(
self.frame.np.config.sections["columns"]["uploads_columns"][i])
frame.clearUploadFinishedErredButton.connect("clicked",
self.OnClearFinishedErred)
frame.clearUploadQueueButton.connect("clicked", self.OnTryClearQueued)
frame.abortUploadButton.connect("clicked", self.OnAbortTransfer)
frame.abortUserUploadButton.connect("clicked", self.OnAbortUser)
frame.banUploadButton.connect("clicked", self.OnBan)
frame.UploadList.expand_all()
self.frame.ToggleAutoclear.set_active(
self.frame.np.config.sections["transfers"]["autoclear_uploads"])
frame.ToggleAutoclear.connect("toggled", self.OnToggleAutoclear)
self.frame.ToggleTreeUploads.set_active(
self.frame.np.config.sections["transfers"]["groupuploads"])
frame.ToggleTreeUploads.connect("toggled", self.OnToggleTree)
self.OnToggleTree(None)
self.frame.ExpandUploads.set_active(
self.frame.np.config.sections["transfers"]["uploadsexpanded"])
frame.ExpandUploads.connect("toggled", self.OnExpandUploads)
self.OnExpandUploads(None)
def saveColumns(self):
columns = []
widths = []
for column in self.frame.UploadList.get_columns():
columns.append(column.get_visible())
widths.append(column.get_width())
self.frame.np.config.sections["columns"]["uploads_columns"] = columns
self.frame.np.config.sections["columns"]["uploads_widths"] = widths
def OnTryClearQueued(self, widget):
direction = "up"
win = OptionDialog(self.frame,
_("Clear All Queued Uploads?"),
modal=True,
status=None,
option=False,
third="")
win.connect("response", self.frame.on_clear_response, direction)
win.set_title(_("Nicotine+") + ": " + _("Clear Queued Transfers"))
win.set_icon(self.frame.images["n"])
win.show()
def OnOpenDirectory(self, widget):
downloaddir = self.frame.np.config.sections["transfers"]["downloaddir"]
incompletedir = self.frame.np.config.sections["transfers"][
"incompletedir"]
if incompletedir == "":
incompletedir = downloaddir
filemanager = self.frame.np.config.sections["ui"]["filemanager"]
transfer = self.selected_transfers[0]
if os.path.exists(transfer.path):
executeCommand(filemanager, transfer.path)
else:
executeCommand(filemanager, incompletedir)
def OnFileSearch(self, widget):
self.select_transfers()
for transfer in self.selected_transfers:
self.frame.SearchEntry.set_text(
transfer.filename.rsplit("\\", 1)[1])
self.frame.ChangeMainPage(None, "search")
break
def expandcollapse(self, path):
if self.frame.ExpandUploads.get_active():
self.frame.UploadList.expand_row(path, True)
else:
self.frame.UploadList.collapse_row(path)
def OnExpandUploads(self, widget):
expanded = self.frame.ExpandUploads.get_active()
if expanded:
self.frame.UploadList.expand_all()
self.frame.ExpandUploadsImage.set_from_stock(gtk.STOCK_REMOVE, 4)
else:
self.frame.UploadList.collapse_all()
self.frame.ExpandUploadsImage.set_from_stock(gtk.STOCK_ADD, 4)
self.frame.np.config.sections["transfers"][
"uploadsexpanded"] = expanded
self.frame.np.config.writeConfiguration()
def OnToggleAutoclear(self, widget):
self.frame.np.config.sections["transfers"][
"autoclear_uploads"] = self.frame.ToggleAutoclear.get_active()
def OnToggleTree(self, widget):
self.TreeUsers = self.frame.ToggleTreeUploads.get_active()
self.frame.np.config.sections["transfers"][
"groupuploads"] = self.TreeUsers
self.RebuildTransfers()
if not self.TreeUsers:
self.frame.ExpandUploads.hide()
else:
self.frame.ExpandUploads.show()
def RebuildTransfers(self):
if self.frame.np.transfers is None:
return
self.Clear()
self.update()
def select_transfers(self):
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
def OnBan(self, widget):
self.select_transfers()
for user in self.selected_users:
self.frame.BanUser(user)
def OnAbortUser(self, widget):
self.select_transfers()
for user in self.selected_users:
for i in self.list[:]:
if i.user == user:
if i not in self.selected_transfers:
self.selected_transfers.append(i)
TransferList.OnAbortTransfer(self, widget, False, False)
self.frame.np.transfers.calcUploadQueueSizes()
self.frame.np.transfers.checkUploadQueue()
def OnUploadTransfer(self, widget):
self.select_transfers()
for transfer in self.selected_transfers:
filename = transfer.filename
path = transfer.path
user = transfer.user
if user in self.frame.np.transfers.getTransferringUsers():
continue
self.frame.np.ProcessRequestToPeer(
user, slskmessages.UploadQueueNotification(None))
self.frame.np.transfers.pushFile(user, filename, path, transfer)
self.frame.np.transfers.checkUploadQueue()
def OnSelectUserTransfer(self, widet):
if len(self.selected_users) != 1:
return
selected_user = self.selected_users[0]
sel = self.frame.UploadList.get_selection()
fmodel = self.frame.UploadList.get_model()
sel.unselect_all()
for item in self.transfers:
user_file, iter, transfer = item
user, filepath = user_file
if selected_user == user:
ix = fmodel.get_path(iter)
sel.select_path(ix, )
self.select_transfers()
def on_key_press_event(self, widget, event):
key = Gdk.keyval_name(event.keyval)
if key in ("P", "p"):
self.OnPopupMenu(widget, event, "keyboard")
else:
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
if key in ("T", "t"):
self.OnAbortTransfer(widget)
elif key == "Delete":
self.OnAbortTransfer(widget, False, True)
def OnPlayFiles(self, widget, prefix=""):
start_new_thread(self._OnPlayFiles, (widget, prefix))
def _OnPlayFiles(self, widget, prefix=""):
executable = self.frame.np.config.sections["players"]["default"]
if "$" not in executable:
return
for fn in self.selected_transfers:
file = fn.filename.replace("\\", os.sep)
if os.path.exists(file):
executeCommand(executable, file, background=False)
def OnPopupMenuUsers(self, widget):
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
self.popup_menu_users.clear()
if len(self.selected_users) > 0:
items = []
self.selected_users.sort(key=str.lower)
for user in self.selected_users:
popup = PopupMenu(self.frame)
popup.setup(
("#" + _("Send _message"), popup.OnSendMessage),
("#" + _("Show IP a_ddress"), popup.OnShowIPaddress),
("#" + _("Get user i_nfo"), popup.OnGetUserInfo),
("#" + _("Brow_se files"), popup.OnBrowseUser),
("#" + _("Gi_ve privileges"), popup.OnGivePrivileges),
("", None),
("$" + _("_Add user to list"), popup.OnAddToList),
("$" + _("_Ban this user"), popup.OnBanUser),
("$" + _("_Ignore this user"), popup.OnIgnoreUser),
("#" + _("Select User's Transfers"),
self.OnSelectUserTransfer))
popup.set_user(user)
items.append((1, user, popup, self.OnPopupMenuUser, popup))
self.popup_menu_users.setup(*items)
return True
def OnPopupMenuUser(self, widget, popup=None):
if popup is None:
return
menu = popup
user = menu.user
items = menu.get_children()
act = False
if len(self.selected_users) >= 1:
act = True
items[0].set_sensitive(act)
items[1].set_sensitive(act)
items[2].set_sensitive(act)
items[3].set_sensitive(act)
items[6].set_active(user in [
i[0] for i in self.frame.np.config.sections["server"]["userlist"]
])
items[7].set_active(
user in self.frame.np.config.sections["server"]["banlist"])
items[8].set_active(
user in self.frame.np.config.sections["server"]["ignorelist"])
for i in range(4, 9):
items[i].set_sensitive(act)
return True
def OnPopupMenu(self, widget, event, kind):
if kind == "mouse":
if event.button != 3:
if event.button == 1 and event.type == Gdk.EventType._2BUTTON_PRESS:
self.DoubleClick(event)
return False
self.selected_transfers = []
self.selected_users = []
self.widget.get_selection().selected_foreach(
self.SelectedTransfersCallback)
self.SelectCurrentRow(event, kind)
users = len(self.selected_users) > 0
multi_users = len(self.selected_users) > 1 # noqa: F841
files = len(self.selected_transfers) > 0
multi_files = len(self.selected_transfers) > 1
items = self.popup_menu.get_children()
if users:
items[5].set_sensitive(True) # Users Menu
else:
items[5].set_sensitive(False) # Users Menu
if files and not multi_files:
act = True
else:
act = False
items[0].set_sensitive(act)
items[1].set_sensitive(act)
if users and files:
act = True
else:
act = False
for i in list(range(3, 5)) + list(range(6, 10)):
items[i].set_sensitive(act)
items[2].set_sensitive(act) # send to player
self.popup_menu.popup(None, None, None, None, 3, event.time)
if kind == "keyboard":
widget.emit_stop_by_name("key_press_event")
elif kind == "mouse":
widget.emit_stop_by_name("button_press_event")
return True
def ClearByUser(self, user):
for i in self.list[:]:
if i.user == user:
if i.transfertimer is not None:
i.transfertimer.cancel()
self.list.remove(i)
self.frame.np.transfers.calcUploadQueueSizes()
self.frame.np.transfers.checkUploadQueue()
self.update()
def DoubleClick(self, event):
self.select_transfers()
dc = self.frame.np.config.sections["transfers"]["upload_doubleclick"]
if dc == 1: # Send to player
self.OnPlayFiles(None)
elif dc == 2: # File manager
self.OnOpenDirectory(None)
elif dc == 3: # Search
self.OnFileSearch(None)
elif dc == 4: # Abort
self.OnAbortTransfer(None, False)
elif dc == 5: # Clear
self.OnClearTransfer(None)
def OnAbortTransfer(self, widget, remove=False, clear=False):
self.select_transfers()
TransferList.OnAbortTransfer(self, widget, remove, clear)
self.frame.np.transfers.calcUploadQueueSizes()
self.frame.np.transfers.checkUploadQueue()
self.update()
def OnClearQueued(self, widget):
self.select_transfers()
TransferList.OnClearQueued(self, widget)
self.frame.np.transfers.calcUploadQueueSizes()
self.frame.np.transfers.checkUploadQueue()
self.update()
def OnClearFailed(self, widget):
TransferList.OnClearFailed(self, widget)
self.frame.np.transfers.calcUploadQueueSizes()
self.frame.np.transfers.checkUploadQueue()
self.update()
