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()