import collections import itertools import os import random import time from qtpy import QtCore as QC from qtpy import QtWidgets as QW from qtpy import QtGui as QG from hydrus.core import HydrusConstants as HC from hydrus.core import HydrusData from hydrus.core import HydrusExceptions from hydrus.core import HydrusGlobals as HG from hydrus.core import HydrusPaths from hydrus.core.networking import HydrusNetwork from hydrus.client import ClientApplicationCommand as CAC from hydrus.client import ClientConstants as CC from hydrus.client import ClientData from hydrus.client import ClientFiles from hydrus.client import ClientLocation from hydrus.client import ClientPaths from hydrus.client import ClientSearch from hydrus.client.gui import ClientGUIDragDrop from hydrus.client.gui import ClientGUICore as CGC from hydrus.client.gui import ClientGUIDialogs from hydrus.client.gui import ClientGUIDialogsManage from hydrus.client.gui import ClientGUIDialogsQuick from hydrus.client.gui import ClientGUIDuplicates from hydrus.client.gui import ClientGUIFunctions from hydrus.client.gui import ClientGUIMedia from hydrus.client.gui import ClientGUIMediaActions from hydrus.client.gui import ClientGUIMediaMenus from hydrus.client.gui import ClientGUIMenus from hydrus.client.gui import ClientGUIScrolledPanelsEdit from hydrus.client.gui import ClientGUIScrolledPanelsManagement from hydrus.client.gui import ClientGUIShortcuts from hydrus.client.gui import ClientGUITags from hydrus.client.gui import ClientGUITopLevelWindowsPanels from hydrus.client.gui import QtPorting as QP from hydrus.client.gui.canvas import ClientGUICanvas from hydrus.client.gui.canvas import ClientGUICanvasFrame from hydrus.client.gui.exporting import ClientGUIExport from hydrus.client.gui.networking import ClientGUIHydrusNetwork from hydrus.client.media import ClientMedia from hydrus.client.metadata import ClientTags class MediaPanel( ClientMedia.ListeningMediaList, QW.QScrollArea, CAC.ApplicationCommandProcessorMixin ): selectedMediaTagPresentationChanged = QC.Signal( list, bool ) selectedMediaTagPresentationIncremented = QC.Signal( list ) statusTextChanged = QC.Signal( str ) focusMediaChanged = QC.Signal( ClientMedia.Media ) focusMediaCleared = QC.Signal() focusMediaPaused = QC.Signal() refreshQuery = QC.Signal() newMediaAdded = QC.Signal() def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results ): CAC.ApplicationCommandProcessorMixin.__init__( self ) QW.QScrollArea.__init__( self, parent ) self.setFrameStyle( QW.QFrame.Panel | QW.QFrame.Sunken ) self.setLineWidth( 2 ) self.resize( QC.QSize( 20, 20 ) ) self.setWidget( QW.QWidget() ) self.setWidgetResizable( True ) ClientMedia.ListeningMediaList.__init__( self, location_context, media_results ) self._UpdateBackgroundColour() self.verticalScrollBar().setSingleStep( 50 ) self._page_key = page_key self._focused_media = None self._last_hit_media = None self._next_best_media_if_focuses_removed = None self._shift_select_started_with_this_media = None self._media_added_in_current_shift_select = set() self._empty_page_status_override = None HG.client_controller.sub( self, 'AddMediaResults', 'add_media_results' ) HG.client_controller.sub( self, 'Collect', 'collect_media' ) HG.client_controller.sub( self, 'RemoveMedia', 'remove_media' ) HG.client_controller.sub( self, '_UpdateBackgroundColour', 'notify_new_colourset' ) HG.client_controller.sub( self, 'SelectByTags', 'select_files_with_tags' ) HG.client_controller.sub( self, 'LaunchMediaViewerOnFocus', 'launch_media_viewer' ) self._had_changes_to_tag_presentation_while_hidden = False self._my_shortcut_handler = ClientGUIShortcuts.ShortcutsHandler( self, [ 'media' ] ) def __bool__( self ): return QP.isValid( self ) def _Archive( self ): hashes = self._GetSelectedHashes( discriminant = CC.DISCRIMINANT_INBOX ) if len( hashes ) > 0: if HC.options[ 'confirm_archive' ]: if len( hashes ) > 1: message = 'Archive ' + HydrusData.ToHumanInt( len( hashes ) ) + ' files?' result = ClientGUIDialogsQuick.GetYesNo( self, message ) if result != QW.QDialog.Accepted: return HG.client_controller.Write( 'content_updates', { CC.COMBINED_LOCAL_FILE_SERVICE_KEY : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, hashes ) ] } ) def _ArchiveDeleteFilter( self ): if len( self._selected_media ) == 0: media_results = self.GenerateMediaResults( discriminant = CC.DISCRIMINANT_LOCAL_BUT_NOT_IN_TRASH, selected_media = set( self._sorted_media ), for_media_viewer = True ) else: media_results = self.GenerateMediaResults( discriminant = CC.DISCRIMINANT_LOCAL_BUT_NOT_IN_TRASH, selected_media = set( self._selected_media ), for_media_viewer = True ) if len( media_results ) > 0: self.SetFocusedMedia( None ) canvas_frame = ClientGUICanvasFrame.CanvasFrame( self.window() ) canvas_window = ClientGUICanvas.CanvasMediaListFilterArchiveDelete( canvas_frame, self._page_key, self._location_context, media_results ) canvas_frame.SetCanvas( canvas_window ) canvas_window.exitFocusMedia.connect( self.SetFocusedMedia ) def _ClearDeleteRecord( self ): media = self._GetSelectedFlatMedia() ClientGUIMediaActions.ClearDeleteRecord( self, media ) def _CopyBMPToClipboard( self ): copied = False if self._focused_media is not None: if self._HasFocusSingleton(): media = self._GetFocusSingleton() if media.GetMime() in HC.IMAGES: HG.client_controller.pub( 'clipboard', 'bmp', media ) copied = True return copied def _CopyFilesToClipboard( self ): client_files_manager = HG.client_controller.client_files_manager media = self._GetSelectedFlatMedia( discriminant = CC.DISCRIMINANT_LOCAL ) paths = [] for m in media: hash = m.GetHash() mime = m.GetMime() path = client_files_manager.GetFilePath( hash, mime, check_file_exists = False ) paths.append( path ) if len( paths ) > 0: HG.client_controller.pub( 'clipboard', 'paths', paths ) def _CopyHashToClipboard( self, hash_type ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() ClientGUIMedia.CopyHashesToClipboard( self, hash_type, [ media ] ) def _CopyHashesToClipboard( self, hash_type ): medias = self._GetSelectedMediaOrdered() ClientGUIMedia.CopyHashesToClipboard( self, hash_type, medias ) def _CopyPathToClipboard( self ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() client_files_manager = HG.client_controller.client_files_manager path = client_files_manager.GetFilePath( media.GetHash(), media.GetMime() ) HG.client_controller.pub( 'clipboard', 'text', path ) def _CopyPathsToClipboard( self ): media_results = self.GenerateMediaResults( discriminant = CC.DISCRIMINANT_LOCAL, selected_media = set( self._selected_media ) ) client_files_manager = HG.client_controller.client_files_manager paths = [] for media_result in media_results: paths.append( client_files_manager.GetFilePath( media_result.GetHash(), media_result.GetMime(), check_file_exists = False ) ) if len( paths ) > 0: text = os.linesep.join( paths ) HG.client_controller.pub( 'clipboard', 'text', text ) def _CopyServiceFilenameToClipboard( self, service_key ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() filename = media.GetLocationsManager().GetServiceFilename( service_key ) if filename is None: return service = HG.client_controller.services_manager.GetService( service_key ) if service.GetServiceType() == HC.IPFS: multihash_prefix = service.GetMultihashPrefix() filename = multihash_prefix + filename HG.client_controller.pub( 'clipboard', 'text', filename ) def _CopyServiceFilenamesToClipboard( self, service_key ): prefix = '' service = HG.client_controller.services_manager.GetService( service_key ) if service.GetServiceType() == HC.IPFS: prefix = service.GetMultihashPrefix() flat_media = self._GetSelectedFlatMedia( is_in_file_service_key = service_key ) if len( flat_media ) > 0: filenames_or_none = [ media.GetLocationsManager().GetServiceFilename( service_key ) for media in flat_media ] filenames = [ prefix + filename for filename in filenames_or_none if filename is not None ] if len( filenames ) > 0: copy_string = os.linesep.join( filenames ) HG.client_controller.pub( 'clipboard', 'text', copy_string ) else: HydrusData.ShowText( 'Could not find any service filenames for that selection!' ) else: HydrusData.ShowText( 'Could not find any files with the requested service!' ) def _Delete( self, file_service_key = None, only_those_in_file_service_key = None ): if file_service_key is None: if len( self._location_context.current_service_keys ) == 1: ( possible_suggested_file_service_key, ) = self._location_context.current_service_keys if HG.client_controller.services_manager.GetServiceType( possible_suggested_file_service_key ) in HC.SPECIFIC_LOCAL_FILE_SERVICES + ( HC.FILE_REPOSITORY, ): file_service_key = possible_suggested_file_service_key media_to_delete = ClientMedia.FlattenMedia( self._selected_media ) if only_those_in_file_service_key is not None: media_to_delete = ClientMedia.FlattenMedia( media_to_delete ) media_to_delete = [ m for m in media_to_delete if only_those_in_file_service_key in m.GetLocationsManager().GetCurrent() ] if file_service_key is None or HG.client_controller.services_manager.GetServiceType( file_service_key ) in HC.LOCAL_FILE_SERVICES: default_reason = 'Deleted from Media Page.' else: default_reason = 'admin' try: ( involves_physical_delete, jobs ) = ClientGUIDialogsQuick.GetDeleteFilesJobs( self, media_to_delete, default_reason, suggested_file_service_key = file_service_key ) except HydrusExceptions.CancelledException: return if involves_physical_delete: self._SetFocusedMedia( None ) def do_it( jobs ): for service_keys_to_content_updates in jobs: HG.client_controller.WriteSynchronous( 'content_updates', service_keys_to_content_updates ) HG.client_controller.CallToThread( do_it, jobs ) def _DeselectSelect( self, media_to_deselect, media_to_select ): if len( media_to_deselect ) > 0: for m in media_to_deselect: m.Deselect() self._RedrawMedia( media_to_deselect ) self._selected_media.difference_update( media_to_deselect ) if len( media_to_select ) > 0: for m in media_to_select: m.Select() self._RedrawMedia( media_to_select ) self._selected_media.update( media_to_select ) self._PublishSelectionChange() def _DownloadSelected( self ): hashes = self._GetSelectedHashes( discriminant = CC.DISCRIMINANT_NOT_LOCAL ) self._DownloadHashes( hashes ) def _DownloadHashes( self, hashes ): HG.client_controller.quick_download_manager.DownloadFiles( hashes ) def _EndShiftSelect( self ): self._shift_select_started_with_this_media = None self._media_added_in_current_shift_select = set() def _ExportFiles( self, do_export_and_then_quit = False ): if len( self._selected_media ) > 0: flat_media = [] for media in self._sorted_media: if media in self._selected_media: if media.IsCollection(): flat_media.extend( media.GetFlatMedia() ) else: flat_media.append( media ) frame = ClientGUITopLevelWindowsPanels.FrameThatTakesScrollablePanel( self, 'export files' ) panel = ClientGUIExport.ReviewExportFilesPanel( frame, flat_media, do_export_and_then_quit = do_export_and_then_quit ) frame.SetPanel( panel ) def _GetFocusSingleton( self ) -> ClientMedia.MediaSingleton: if self._focused_media is not None: media_singleton = self._focused_media.GetDisplayMedia() if media_singleton is not None: return media_singleton raise HydrusExceptions.DataMissing( 'No media singleton!' ) def _GetNumSelected( self ): return sum( [ media.GetNumFiles() for media in self._selected_media ] ) def _GetPrettyStatusForStatusBar( self ) -> str: num_files = len( self._hashes ) if self._empty_page_status_override is not None: if num_files == 0: return self._empty_page_status_override else: # user has dragged files onto this page or similar self._empty_page_status_override = None num_selected = self._GetNumSelected() ( num_files_descriptor, selected_files_descriptor ) = self._GetSortedSelectedMimeDescriptors() num_files_string = '{} {}'.format( HydrusData.ToHumanInt( num_files ), num_files_descriptor ) s = num_files_string # 23 files if num_selected == 0: if num_files > 0: pretty_total_size = self._GetPrettyTotalSize() s += ' - totalling ' + pretty_total_size pretty_total_duration = self._GetPrettyTotalDuration() if pretty_total_duration != '': s += ', {}'.format( pretty_total_duration ) else: s += ' - ' # if 1 selected, we show the whole mime string, so no need to specify if num_selected == 1 or selected_files_descriptor == num_files_descriptor: selected_files_string = HydrusData.ToHumanInt( num_selected ) else: selected_files_string = '{} {}'.format( HydrusData.ToHumanInt( num_selected ), selected_files_descriptor ) if num_selected == 1: # 23 files - 1 video selected, file_info ( selected_media, ) = self._selected_media pretty_info_lines = [ line for line in selected_media.GetPrettyInfoLines( only_interesting_lines = True ) if isinstance( line, str ) ] s += '{} selected, {}'.format( selected_files_string, ', '.join( pretty_info_lines ) ) else: # 23 files - 5 selected, selection_info num_inbox = sum( ( media.GetNumInbox() for media in self._selected_media ) ) if num_inbox == num_selected: inbox_phrase = 'all in inbox' elif num_inbox == 0: inbox_phrase = 'all archived' else: inbox_phrase = '{} in inbox and {} archived'.format( HydrusData.ToHumanInt( num_inbox ), HydrusData.ToHumanInt( num_selected - num_inbox ) ) pretty_total_size = self._GetPrettyTotalSize( only_selected = True ) s += '{} selected, {}, totalling {}'.format( selected_files_string, inbox_phrase, pretty_total_size ) pretty_total_duration = self._GetPrettyTotalDuration( only_selected = True ) if pretty_total_duration != '': s += ', {}'.format( pretty_total_duration ) return s def _GetPrettyTotalDuration( self, only_selected = False ): if only_selected: media_source = self._selected_media else: media_source = self._sorted_media if len( media_source ) == 0 or False in ( media.HasDuration() for media in media_source ): return '' total_duration = sum( ( media.GetDuration() for media in media_source ) ) return HydrusData.ConvertMillisecondsToPrettyTime( total_duration ) def _GetPrettyTotalSize( self, only_selected = False ): if only_selected: media_source = self._selected_media else: media_source = self._sorted_media total_size = sum( [ media.GetSize() for media in media_source ] ) unknown_size = False in ( media.IsSizeDefinite() for media in media_source ) if total_size == 0: if unknown_size: return 'unknown size' else: return HydrusData.ToHumanBytes( 0 ) else: if unknown_size: return HydrusData.ToHumanBytes( total_size ) + ' + some unknown size' else: return HydrusData.ToHumanBytes( total_size ) def _GetSelectedHashes( self, is_in_file_service_key = None, discriminant = None, is_not_in_file_service_key = None, ordered = False ): if ordered: result = [] for media in self._GetSelectedMediaOrdered(): result.extend( media.GetHashes( is_in_file_service_key, discriminant, is_not_in_file_service_key, ordered ) ) else: result = set() for media in self._selected_media: result.update( media.GetHashes( is_in_file_service_key, discriminant, is_not_in_file_service_key, ordered ) ) return result def _GetSelectedCollections( self ): sorted_selected_collections = [ media for media in self._sorted_media if media.IsCollection() and media in self._selected_media ] return sorted_selected_collections def _GetSelectedFlatMedia( self, is_in_file_service_key = None, discriminant = None, is_not_in_file_service_key = None ): # this now always delivers sorted results sorted_selected_media = [ media for media in self._sorted_media if media in self._selected_media ] flat_media = ClientMedia.FlattenMedia( sorted_selected_media ) flat_media = [ media for media in flat_media if media.MatchesDiscriminant( is_in_file_service_key = is_in_file_service_key, discriminant = discriminant, is_not_in_file_service_key = is_not_in_file_service_key ) ] return flat_media def _GetSelectedMediaOrdered( self ): medias = [] for media in self._sorted_media: if media in self._selected_media: medias.append( media ) return medias def _GetSimilarTo( self, max_hamming ): hashes = set() media = self._GetSelectedFlatMedia() for m in media: if m.GetMime() in HC.FILES_THAT_HAVE_PERCEPTUAL_HASH: hashes.add( m.GetHash() ) if len( hashes ) > 0: initial_predicates = [ ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIMILAR_TO, ( tuple( hashes ), max_hamming ) ) ] HG.client_controller.pub( 'new_page_query', self._location_context, initial_predicates = initial_predicates ) def _GetSortedSelectedMimeDescriptors( self ): def GetDescriptor( plural, classes, num_collections ): suffix = 's' if plural else '' if len( classes ) == 0: return 'file' + suffix if len( classes ) == 1: ( mime, ) = classes if mime == HC.APPLICATION_HYDRUS_CLIENT_COLLECTION: collections_suffix = 's' if num_collections > 1 else '' return 'file{} in {} collection{}'.format( suffix, HydrusData.ToHumanInt( num_collections ), collections_suffix ) else: return HC.mime_string_lookup[ mime ] + suffix if len( classes.difference( HC.IMAGES ) ) == 0: return 'image' + suffix elif len( classes.difference( HC.ANIMATIONS ) ) == 0: return 'animation' + suffix elif len( classes.difference( HC.VIDEO ) ) == 0: return 'video' + suffix elif len( classes.difference( HC.AUDIO ) ) == 0: return 'audio file' + suffix else: return 'file' + suffix if len( self._sorted_media ) > 1000: sorted_mime_descriptor = 'files' else: sorted_mimes = { media.GetMime() for media in self._sorted_media } if HC.APPLICATION_HYDRUS_CLIENT_COLLECTION in sorted_mimes: num_collections = len( [ media for media in self._sorted_media if isinstance( media, ClientMedia.MediaCollection ) ] ) else: num_collections = 0 plural = len( self._sorted_media ) > 1 or sum( ( m.GetNumFiles() for m in self._sorted_media ) ) > 1 sorted_mime_descriptor = GetDescriptor( plural, sorted_mimes, num_collections ) if len( self._selected_media ) > 1000: selected_mime_descriptor = 'files' else: selected_mimes = { media.GetMime() for media in self._selected_media } if HC.APPLICATION_HYDRUS_CLIENT_COLLECTION in selected_mimes: num_collections = len( [ media for media in self._selected_media if isinstance( media, ClientMedia.MediaCollection ) ] ) else: num_collections = 0 plural = len( self._selected_media ) > 1 or sum( ( m.GetNumFiles() for m in self._selected_media ) ) > 1 selected_mime_descriptor = GetDescriptor( plural, selected_mimes, num_collections ) return ( sorted_mime_descriptor, selected_mime_descriptor ) def _HasFocusSingleton( self ) -> bool: try: media = self._GetFocusSingleton() return True except HydrusExceptions.DataMissing: return False def _HitMedia( self, media, ctrl, shift ): if media is None: if not ctrl and not shift: self._Select( ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NONE ) ) self._SetFocusedMedia( None ) self._EndShiftSelect() else: if ctrl: if media.IsSelected(): self._DeselectSelect( ( media, ), () ) if self._focused_media == media: self._SetFocusedMedia( None ) self._EndShiftSelect() else: self._DeselectSelect( (), ( media, ) ) focus_it = False if HG.client_controller.new_options.GetBoolean( 'focus_preview_on_ctrl_click' ): if HG.client_controller.new_options.GetBoolean( 'focus_preview_on_ctrl_click_only_static' ): focus_it = media.GetDuration() is None else: focus_it = True if focus_it: self._SetFocusedMedia( media ) else: self._last_hit_media = media self._StartShiftSelect( media ) elif shift and self._shift_select_started_with_this_media is not None: start_index = self._sorted_media.index( self._shift_select_started_with_this_media ) end_index = self._sorted_media.index( media ) if start_index < end_index: media_from_start_of_shift_to_end = set( self._sorted_media[ start_index : end_index + 1 ] ) else: media_from_start_of_shift_to_end = set( self._sorted_media[ end_index : start_index + 1 ] ) media_to_deselect = [ m for m in self._media_added_in_current_shift_select if m not in media_from_start_of_shift_to_end ] media_to_select = [ m for m in media_from_start_of_shift_to_end if not m.IsSelected() ] self._media_added_in_current_shift_select.difference_update( media_to_deselect ) self._media_added_in_current_shift_select.update( media_to_select ) self._DeselectSelect( media_to_deselect, media_to_select ) focus_it = False if HG.client_controller.new_options.GetBoolean( 'focus_preview_on_shift_click' ): if HG.client_controller.new_options.GetBoolean( 'focus_preview_on_shift_click_only_static' ): focus_it = media.GetDuration() is None else: focus_it = True if focus_it: self._SetFocusedMedia( media ) else: self._last_hit_media = media else: if not media.IsSelected(): self._DeselectSelect( self._selected_media, ( media, ) ) else: self._PublishSelectionChange() self._SetFocusedMedia( media ) self._StartShiftSelect( media ) def _Inbox( self ): hashes = self._GetSelectedHashes( discriminant = CC.DISCRIMINANT_ARCHIVE, is_in_file_service_key = CC.COMBINED_LOCAL_FILE_SERVICE_KEY ) if len( hashes ) > 0: if HC.options[ 'confirm_archive' ]: if len( hashes ) > 1: message = 'Send {} files to inbox?'.format( HydrusData.ToHumanInt( len( hashes ) ) ) result = ClientGUIDialogsQuick.GetYesNo( self, message ) if result != QW.QDialog.Accepted: return HG.client_controller.Write( 'content_updates', { CC.COMBINED_LOCAL_FILE_SERVICE_KEY: [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_INBOX, hashes ) ] } ) def _LaunchMediaViewer( self, first_media = None ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() new_options = HG.client_controller.new_options ( media_show_action, media_start_paused, media_start_with_embed ) = new_options.GetMediaShowAction( media.GetMime() ) if media_show_action == CC.MEDIA_VIEWER_ACTION_DO_NOT_SHOW_ON_ACTIVATION_OPEN_EXTERNALLY: hash = media.GetHash() mime = media.GetMime() client_files_manager = HG.client_controller.client_files_manager path = client_files_manager.GetFilePath( hash, mime ) new_options = HG.client_controller.new_options launch_path = new_options.GetMimeLaunch( mime ) HydrusPaths.LaunchFile( path, launch_path ) return elif media_show_action == CC.MEDIA_VIEWER_ACTION_DO_NOT_SHOW: return media_results = self.GenerateMediaResults( discriminant = CC.DISCRIMINANT_LOCAL, for_media_viewer = True ) if len( media_results ) > 0: if first_media is None and self._focused_media is not None: first_media = self._focused_media if first_media is not None: first_media = first_media.GetDisplayMedia() if first_media is not None and first_media.GetLocationsManager().IsLocal(): first_hash = first_media.GetHash() else: first_hash = None self.SetFocusedMedia( None ) canvas_frame = ClientGUICanvasFrame.CanvasFrame( self.window() ) canvas_window = ClientGUICanvas.CanvasMediaListBrowser( canvas_frame, self._page_key, self._location_context, media_results, first_hash ) canvas_frame.SetCanvas( canvas_window ) canvas_window.exitFocusMedia.connect( self.SetFocusedMedia ) def _ManageNotes( self ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() ClientGUIMediaActions.EditFileNotes( self, media ) self.setFocus( QC.Qt.OtherFocusReason ) def _ManageRatings( self ): if len( self._selected_media ) > 0: if len( HG.client_controller.services_manager.GetServices( HC.RATINGS_SERVICES ) ) > 0: flat_media = self._GetSelectedFlatMedia() with ClientGUIDialogsManage.DialogManageRatings( self, flat_media ) as dlg: dlg.exec() self.setFocus( QC.Qt.OtherFocusReason ) def _ManageTags( self ): if len( self._selected_media ) > 0: num_files = self._GetNumSelected() title = 'manage tags for ' + HydrusData.ToHumanInt( num_files ) + ' files' frame_key = 'manage_tags_dialog' with ClientGUITopLevelWindowsPanels.DialogManage( self, title, frame_key ) as dlg: panel = ClientGUITags.ManageTagsPanel( dlg, self._location_context, self._selected_media ) dlg.SetPanel( panel ) dlg.exec() self.setFocus( QC.Qt.OtherFocusReason ) def _ManageURLs( self ): if len( self._selected_media ) > 0: num_files = self._GetNumSelected() title = 'manage urls for {} files'.format( num_files ) with ClientGUITopLevelWindowsPanels.DialogManage( self, title ) as dlg: panel = ClientGUIScrolledPanelsManagement.ManageURLsPanel( dlg, self._selected_media ) dlg.SetPanel( panel ) dlg.exec() self.setFocus( QC.Qt.OtherFocusReason ) def _MediaIsVisible( self, media ): return True def _ModifyUploaders( self, file_service_key ): hashes = self._GetSelectedHashes() contents = [ HydrusNetwork.Content( HC.CONTENT_TYPE_FILES, ( hash, ) ) for hash in hashes ] if len( contents ) > 0: subject_account_identifiers = [ HydrusNetwork.AccountIdentifier( content = content ) for content in contents ] frame = ClientGUITopLevelWindowsPanels.FrameThatTakesScrollablePanel( self, 'manage accounts' ) panel = ClientGUIHydrusNetwork.ModifyAccountsPanel( frame, file_service_key, subject_account_identifiers ) frame.SetPanel( panel ) def _OpenExternally( self ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() if media.GetLocationsManager().IsLocal(): self.SetFocusedMedia( None ) hash = media.GetHash() mime = media.GetMime() client_files_manager = HG.client_controller.client_files_manager path = client_files_manager.GetFilePath( hash, mime ) new_options = HG.client_controller.new_options launch_path = new_options.GetMimeLaunch( mime ) HydrusPaths.LaunchFile( path, launch_path ) def _OpenFileInWebBrowser( self ): if self._HasFocusSingleton(): focused_singleton = self._GetFocusSingleton() if focused_singleton.GetLocationsManager().IsLocal(): hash = focused_singleton.GetHash() mime = focused_singleton.GetMime() client_files_manager = HG.client_controller.client_files_manager path = client_files_manager.GetFilePath( hash, mime ) self._SetFocusedMedia( None ) ClientPaths.LaunchPathInWebBrowser( path ) def _OpenFileLocation( self ): if self._HasFocusSingleton(): focused_singleton = self._GetFocusSingleton() if focused_singleton.GetLocationsManager().IsLocal(): hash = focused_singleton.GetHash() mime = focused_singleton.GetMime() client_files_manager = HG.client_controller.client_files_manager path = client_files_manager.GetFilePath( hash, mime ) self._SetFocusedMedia( None ) HydrusPaths.OpenFileLocation( path ) def _OpenKnownURL( self ): if self._HasFocusSingleton(): focused_singleton = self._GetFocusSingleton() ClientGUIMedia.DoOpenKnownURLFromShortcut( self, focused_singleton ) def _PetitionFiles( self, remote_service_key ): hashes = self._GetSelectedHashes() if hashes is not None and len( hashes ) > 0: remote_service = HG.client_controller.services_manager.GetService( remote_service_key ) service_type = remote_service.GetServiceType() if service_type == HC.FILE_REPOSITORY: if len( hashes ) == 1: message = 'Enter a reason for this file to be removed from {}.'.format( remote_service.GetName() ) else: message = 'Enter a reason for these {} files to be removed from {}.'.format( HydrusData.ToHumanInt( len( hashes ) ), remote_service.GetName() ) with ClientGUIDialogs.DialogTextEntry( self, message ) as dlg: if dlg.exec() == QW.QDialog.Accepted: reason = dlg.GetValue() content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_PETITION, hashes, reason = reason ) service_keys_to_content_updates = { remote_service_key : ( content_update, ) } HG.client_controller.Write( 'content_updates', service_keys_to_content_updates ) self.setFocus( QC.Qt.OtherFocusReason ) elif service_type == HC.IPFS: content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_PETITION, hashes, reason = 'ipfs' ) service_keys_to_content_updates = { remote_service_key : ( content_update, ) } HG.client_controller.Write( 'content_updates', service_keys_to_content_updates ) def _PublishSelectionChange( self, tags_changed = False ): if HG.client_controller.gui.IsCurrentPage( self._page_key ): if len( self._selected_media ) == 0: tags_media = self._sorted_media else: tags_media = self._selected_media tags_media = list( tags_media ) tags_changed = tags_changed or self._had_changes_to_tag_presentation_while_hidden self.selectedMediaTagPresentationChanged.emit( tags_media, tags_changed ) self.statusTextChanged.emit( self._GetPrettyStatusForStatusBar() ) if tags_changed: self._had_changes_to_tag_presentation_while_hidden = False elif tags_changed: self._had_changes_to_tag_presentation_while_hidden = True def _PublishSelectionIncrement( self, medias ): if HG.client_controller.gui.IsCurrentPage( self._page_key ): medias = list( medias ) self.selectedMediaTagPresentationIncremented.emit( medias ) self.statusTextChanged.emit( self._GetPrettyStatusForStatusBar() ) else: self._had_changes_to_tag_presentation_while_hidden = True def _RecalculateVirtualSize( self, called_from_resize_event = False ): pass def _RedrawMedia( self, media ): pass def _Remove( self, file_filter ): hashes = self.GetFilteredHashes( file_filter ) if len( hashes ) > 0: self._RemoveMediaByHashes( hashes ) def _RegenerateFileData( self, job_type ): flat_media = self._GetSelectedFlatMedia() num_files = len( flat_media ) if num_files > 0: if job_type == ClientFiles.REGENERATE_FILE_DATA_JOB_FILE_METADATA: message = 'This will reparse the {} selected files\' metadata.'.format( HydrusData.ToHumanInt( num_files ) ) message += os.linesep * 2 message += 'If the files were imported before some more recent improvement in the parsing code (such as EXIF rotation or bad video resolution or duration or frame count calculation), this will update them.' elif job_type == ClientFiles.REGENERATE_FILE_DATA_JOB_FORCE_THUMBNAIL: message = 'This will force-regenerate the {} selected files\' thumbnails.'.format( HydrusData.ToHumanInt( num_files ) ) elif job_type == ClientFiles.REGENERATE_FILE_DATA_JOB_REFIT_THUMBNAIL: message = 'This will regenerate the {} selected files\' thumbnails, but only if they are the wrong size.'.format( HydrusData.ToHumanInt( num_files ) ) else: message = ClientFiles.regen_file_enum_to_description_lookup[ job_type ] do_it_now = True if num_files > 50: message += os.linesep * 2 message += 'You have selected {} files, so this job may take some time. You can run it all now or schedule it to the overall file maintenance queue for later spread-out processing.'.format( HydrusData.ToHumanInt( num_files ) ) yes_tuples = [] yes_tuples.append( ( 'do it now', 'now' ) ) yes_tuples.append( ( 'do it later', 'later' ) ) try: result = ClientGUIDialogsQuick.GetYesYesNo( self, message, yes_tuples = yes_tuples, no_label = 'forget it' ) except HydrusExceptions.CancelledException: return do_it_now = result == 'now' else: result = ClientGUIDialogsQuick.GetYesNo( self, message ) if result != QW.QDialog.Accepted: return if do_it_now: self._SetFocusedMedia( None ) time.sleep( 0.1 ) HG.client_controller.CallToThread( HG.client_controller.files_maintenance_manager.RunJobImmediately, flat_media, job_type ) else: hashes = { media.GetHash() for media in flat_media } HG.client_controller.CallToThread( HG.client_controller.files_maintenance_manager.ScheduleJob, hashes, job_type ) def _RescindDownloadSelected( self ): hashes = self._GetSelectedHashes( discriminant = CC.DISCRIMINANT_NOT_LOCAL ) HG.client_controller.Write( 'content_updates', { CC.COMBINED_LOCAL_FILE_SERVICE_KEY : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_RESCIND_PEND, hashes ) ] } ) def _RescindPetitionFiles( self, file_service_key ): hashes = self._GetSelectedHashes() if hashes is not None and len( hashes ) > 0: HG.client_controller.Write( 'content_updates', { file_service_key : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_RESCIND_PETITION, hashes ) ] } ) def _RescindUploadFiles( self, file_service_key ): hashes = self._GetSelectedHashes() if hashes is not None and len( hashes ) > 0: HG.client_controller.Write( 'content_updates', { file_service_key : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_RESCIND_PEND, hashes ) ] } ) def _Select( self, file_filter ): matching_media = self.GetFilteredMedia( file_filter ) media_to_deselect = self._selected_media.difference( matching_media ) media_to_select = matching_media.difference( self._selected_media ) move_focus = self._focused_media in media_to_deselect or self._focused_media is None if move_focus or self._shift_select_started_with_this_media in media_to_deselect: self._EndShiftSelect() self._DeselectSelect( media_to_deselect, media_to_select ) if move_focus: if len( self._selected_media ) == 0: self._SetFocusedMedia( None ) else: # let's not focus if one of the selectees is already visible media_visible = True in ( self._MediaIsVisible( media ) for media in self._selected_media ) if not media_visible: for m in self._sorted_media: if m in self._selected_media: ctrl = False shift = False self._HitMedia( m, ctrl, shift ) self._ScrollToMedia( m ) break def _SetCollectionsAsAlternate( self ): collections = self._GetSelectedCollections() if len( collections ) > 0: message = 'Are you sure you want to set files in the selected collections as alternates? Each collection will be considered a separate group of alternates.' message += os.linesep * 2 message += 'Be careful applying this to large groups--any more than a few dozen files, and the client could hang a long time.' result = ClientGUIDialogsQuick.GetYesNo( self, message ) if result == QW.QDialog.Accepted: for collection in collections: media_group = collection.GetFlatMedia() self._SetDuplicates( HC.DUPLICATE_ALTERNATE, media_group = media_group, silent = True ) def _SetDuplicates( self, duplicate_type, media_pairs = None, media_group = None, duplicate_content_merge_options = None, silent = False ): if duplicate_type == HC.DUPLICATE_POTENTIAL: yes_no_text = 'queue all possible and valid pair combinations into the duplicate filter' elif duplicate_content_merge_options is None: yes_no_text = 'apply "{}"'.format( HC.duplicate_type_string_lookup[ duplicate_type ] ) if duplicate_type in [ HC.DUPLICATE_BETTER, HC.DUPLICATE_SAME_QUALITY ] or ( HG.client_controller.new_options.GetBoolean( 'advanced_mode' ) and duplicate_type == HC.DUPLICATE_ALTERNATE ): yes_no_text += ' (with default duplicate metadata merge options)' new_options = HG.client_controller.new_options duplicate_content_merge_options = new_options.GetDuplicateContentMergeOptions( duplicate_type ) else: yes_no_text = 'apply "{}" (with custom duplicate metadata merge options)'.format( HC.duplicate_type_string_lookup[ duplicate_type ] ) file_deletion_reason = 'Deleted from duplicate action on Media Page ({}).'.format( yes_no_text ) if media_pairs is None: if media_group is None: flat_media = self._GetSelectedFlatMedia() else: flat_media = ClientMedia.FlattenMedia( media_group ) num_files_str = HydrusData.ToHumanInt( len( flat_media ) ) if len( flat_media ) < 2: return False if duplicate_type in ( HC.DUPLICATE_FALSE_POSITIVE, HC.DUPLICATE_ALTERNATE, HC.DUPLICATE_POTENTIAL ): media_pairs = list( itertools.combinations( flat_media, 2 ) ) else: first_media = flat_media[0] media_pairs = [ ( first_media, other_media ) for other_media in flat_media if other_media != first_media ] else: num_files_str = HydrusData.ToHumanInt( len( self._GetSelectedFlatMedia() ) ) if len( media_pairs ) == 0: return False if not silent: yes_label = 'yes' no_label = 'no' if len( media_pairs ) > 1 and duplicate_type in ( HC.DUPLICATE_FALSE_POSITIVE, HC.DUPLICATE_ALTERNATE ): media_pairs_str = HydrusData.ToHumanInt( len( media_pairs ) ) message = 'Are you sure you want to {} for the {} selected files? The relationship will be applied between every pair combination in the file selection ({} pairs).'.format( yes_no_text, num_files_str, media_pairs_str ) if len( media_pairs ) > 100: if duplicate_type == HC.DUPLICATE_FALSE_POSITIVE: message = 'False positive records are complicated, and setting that relationship for {} files ({} pairs) at once is likely a mistake.'.format( num_files_str, media_pairs_str ) message += os.linesep * 2 message += 'Are you sure all of these files are all potential duplicates and that they are all false positive matches with each other? If not, I recommend you step back for now.' yes_label = 'I know what I am doing' no_label = 'step back for now' elif duplicate_type == HC.DUPLICATE_ALTERNATE: message = 'Are you certain all these {} files are alternates with every other member of the selection, and that none are duplicates?'.format( num_files_str ) message += os.linesep * 2 message += 'If some of them may be duplicates, I recommend you either deselect the possible duplicates and try again, or just leave this group to be processed in the normal duplicate filter.' yes_label = 'they are all alternates' no_label = 'some may be duplicates' else: message = 'Are you sure you want to ' + yes_no_text + ' for the {} selected files?'.format( num_files_str ) result = ClientGUIDialogsQuick.GetYesNo( self, message, yes_label = yes_label, no_label = no_label ) if result != QW.QDialog.Accepted: return False pair_info = [] # there's an issue here in that one decision will affect the next. if we say 'copy tags both sides' and say A > B & C, then B's tags, merged with A, should soon merge with C # therefore, we need to update the media objects as we go here, which means we need duplicates to force content updates on # this is a little hacky, so maybe a big rewrite here would be nice # There's a second issue, wew, in that in order to propagate C back to B, we need to do the whole thing twice! wow! # some service_key_to_content_updates preservation gubbins is needed as a result hashes_to_duplicated_media = {} hash_pairs_to_list_of_service_keys_to_content_updates = collections.defaultdict( list ) for is_first_run in ( True, False ): for ( first_media, second_media ) in media_pairs: first_hash = first_media.GetHash() second_hash = second_media.GetHash() if first_hash not in hashes_to_duplicated_media: hashes_to_duplicated_media[ first_hash ] = first_media.Duplicate() first_duplicated_media = hashes_to_duplicated_media[ first_hash ] if second_hash not in hashes_to_duplicated_media: hashes_to_duplicated_media[ second_hash ] = second_media.Duplicate() second_duplicated_media = hashes_to_duplicated_media[ second_hash ] list_of_service_keys_to_content_updates = hash_pairs_to_list_of_service_keys_to_content_updates[ ( first_hash, second_hash ) ] if duplicate_content_merge_options is not None: do_not_do_deletes = is_first_run # so the important part of this mess is here. we send the duplicated media, which is keeping up with content updates, to the method here # original 'first_media' is not changed, and won't be until the database Write clears and publishes everything list_of_service_keys_to_content_updates.append( duplicate_content_merge_options.ProcessPairIntoContentUpdates( first_duplicated_media, second_duplicated_media, file_deletion_reason = file_deletion_reason, do_not_do_deletes = do_not_do_deletes ) ) for service_keys_to_content_updates in list_of_service_keys_to_content_updates: for ( service_key, content_updates ) in service_keys_to_content_updates.items(): for content_update in content_updates: hashes = content_update.GetHashes() if first_hash in hashes: first_duplicated_media.GetMediaResult().ProcessContentUpdate( service_key, content_update ) if second_hash in hashes: second_duplicated_media.GetMediaResult().ProcessContentUpdate( service_key, content_update ) if is_first_run: continue pair_info.append( ( duplicate_type, first_hash, second_hash, list_of_service_keys_to_content_updates ) ) if len( pair_info ) > 0: HG.client_controller.WriteSynchronous( 'duplicate_pair_status', pair_info ) return True return False def _SetDuplicatesCustom( self ): duplicate_types = [ HC.DUPLICATE_BETTER, HC.DUPLICATE_SAME_QUALITY ] if HG.client_controller.new_options.GetBoolean( 'advanced_mode' ): duplicate_types.append( HC.DUPLICATE_ALTERNATE ) choice_tuples = [ ( HC.duplicate_type_string_lookup[ duplicate_type ], duplicate_type ) for duplicate_type in duplicate_types ] try: duplicate_type = ClientGUIDialogsQuick.SelectFromList( self, 'select duplicate type', choice_tuples ) except HydrusExceptions.CancelledException: return new_options = HG.client_controller.new_options duplicate_content_merge_options = new_options.GetDuplicateContentMergeOptions( duplicate_type ) with ClientGUITopLevelWindowsPanels.DialogEdit( self, 'edit duplicate merge options' ) as dlg: panel = ClientGUIScrolledPanelsEdit.EditDuplicateContentMergeOptionsPanel( dlg, duplicate_type, duplicate_content_merge_options, for_custom_action = True ) dlg.SetPanel( panel ) if dlg.exec() == QW.QDialog.Accepted: duplicate_content_merge_options = panel.GetValue() if duplicate_type == HC.DUPLICATE_BETTER: self._SetDuplicatesFocusedBetter( duplicate_content_merge_options = duplicate_content_merge_options ) else: self._SetDuplicates( duplicate_type, duplicate_content_merge_options = duplicate_content_merge_options ) def _SetDuplicatesFocusedBetter( self, duplicate_content_merge_options = None ): if self._HasFocusSingleton(): focused_singleton = self._GetFocusSingleton() focused_hash = focused_singleton.GetHash() flat_media = self._GetSelectedFlatMedia() ( better_media, ) = [ media for media in flat_media if media.GetHash() == focused_hash ] worse_flat_media = [ media for media in flat_media if media.GetHash() != focused_hash ] if len( worse_flat_media ) == 0: return media_pairs = [ ( better_media, worse_media ) for worse_media in worse_flat_media ] message = 'Are you sure you want to set the focused file as better than the {} other files in the selection?'.format( HydrusData.ToHumanInt( len( worse_flat_media ) ) ) result = ClientGUIDialogsQuick.GetYesNo( self, message ) if result == QW.QDialog.Accepted: self._SetDuplicates( HC.DUPLICATE_BETTER, media_pairs = media_pairs, silent = True, duplicate_content_merge_options = duplicate_content_merge_options ) else: QW.QMessageBox.warning( self, 'Warning', 'No file is focused, so cannot set the focused file as better!' ) return def _SetDuplicatesFocusedKing( self ): if self._HasFocusSingleton(): media = self._GetFocusSingleton() focused_hash = media.GetHash() HG.client_controller.WriteSynchronous( 'duplicate_set_king', focused_hash ) else: QW.QMessageBox.warning( self, 'Warning', 'No file is focused, so cannot set the focused file as king!' ) return def _SetDuplicatesPotential( self ): media_group = self._GetSelectedFlatMedia() self._SetDuplicates( HC.DUPLICATE_POTENTIAL, media_group = media_group ) def _SetFocusedMedia( self, media ): if media is None and self._focused_media is not None: next_best_media = self._focused_media i = self._sorted_media.index( next_best_media ) while next_best_media in self._selected_media: if i == 0: next_best_media = None break i -= 1 next_best_media = self._sorted_media[ i ] self._next_best_media_if_focuses_removed = next_best_media else: self._next_best_media_if_focuses_removed = None publish_media = None self._focused_media = media self._last_hit_media = media if self._focused_media is not None: publish_media = self._focused_media.GetDisplayMedia() if publish_media is None: self.focusMediaCleared.emit() else: self.focusMediaChanged.emit( publish_media ) def _ScrollToMedia( self, media ): pass def _ShareOnLocalBooru( self ): if len( self._selected_media ) > 0: share_key = HydrusData.GenerateKey() name = '' text = '' timeout = HydrusData.GetNow() + 60 * 60 * 24 hashes = self._GetSelectedHashes() with ClientGUIDialogs.DialogInputLocalBooruShare( self, share_key, name, text, timeout, hashes, new_share = True ) as dlg: if dlg.exec() == QW.QDialog.Accepted: ( share_key, name, text, timeout, hashes ) = dlg.GetInfo() info = {} info[ 'name' ] = name info[ 'text' ] = text info[ 'timeout' ] = timeout info[ 'hashes' ] = hashes HG.client_controller.Write( 'local_booru_share', share_key, info ) self.setFocus( QC.Qt.OtherFocusReason ) def _ShowSelectionInNewPage( self ): hashes = self._GetSelectedHashes( ordered = True ) if hashes is not None and len( hashes ) > 0: HG.client_controller.pub( 'new_page_query', self._location_context, initial_hashes = hashes ) def _StartShiftSelect( self, media ): self._shift_select_started_with_this_media = media self._media_added_in_current_shift_select = set() def _Undelete( self ): media = self._GetSelectedFlatMedia() ClientGUIMediaActions.UndeleteMedia( self, media ) def _UpdateBackgroundColour( self ): self.widget().update() def _UploadDirectory( self, file_service_key ): hashes = self._GetSelectedHashes() if hashes is not None and len( hashes ) > 0: ipfs_service = HG.client_controller.services_manager.GetService( file_service_key ) with ClientGUIDialogs.DialogTextEntry( self, 'Enter a note to describe this directory.' ) as dlg: if dlg.exec() == QW.QDialog.Accepted: note = dlg.GetValue() HG.client_controller.CallToThread( ipfs_service.PinDirectory, hashes, note ) def _UploadFiles( self, file_service_key ): hashes = self._GetSelectedHashes( is_not_in_file_service_key = file_service_key ) if hashes is not None and len( hashes ) > 0: HG.client_controller.Write( 'content_updates', { file_service_key : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_PEND, hashes ) ] } ) def AddMediaResults( self, page_key, media_results ): if page_key == self._page_key: HG.client_controller.pub( 'refresh_page_name', self._page_key ) result = ClientMedia.ListeningMediaList.AddMediaResults( self, media_results ) self.newMediaAdded.emit() HG.client_controller.pub( 'notify_new_pages_count' ) return result def CleanBeforeDestroy( self ): self.Clear() def ClearPageKey( self ): self._page_key = b'dead media panel page key' def Collect( self, page_key, media_collect = None ): if page_key == self._page_key: self._Select( ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NONE ) ) ClientMedia.ListeningMediaList.Collect( self, media_collect ) self._RecalculateVirtualSize() # no refresh needed since the sort call that always comes after will do it def GetTotalFileSize( self ): return 0 def LaunchMediaViewerOnFocus( self, page_key ): if page_key == self._page_key: self._LaunchMediaViewer() def PageHidden( self ): pass def PageShown( self ): self._PublishSelectionChange() def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ): command_processed = True if command.IsSimpleCommand(): action = command.GetSimpleAction() if action == CAC.SIMPLE_COPY_BMP: self._CopyBMPToClipboard() elif action == CAC.SIMPLE_COPY_BMP_OR_FILE_IF_NOT_BMPABLE: copied = self._CopyBMPToClipboard() if not copied: self._CopyFilesToClipboard() elif action == CAC.SIMPLE_COPY_FILE: self._CopyFilesToClipboard() elif action == CAC.SIMPLE_COPY_PATH: self._CopyPathsToClipboard() elif action == CAC.SIMPLE_COPY_SHA256_HASH: self._CopyHashesToClipboard( 'sha256' ) elif action == CAC.SIMPLE_COPY_MD5_HASH: self._CopyHashesToClipboard( 'md5' ) elif action == CAC.SIMPLE_COPY_SHA1_HASH: self._CopyHashesToClipboard( 'sha1' ) elif action == CAC.SIMPLE_COPY_SHA512_HASH: self._CopyHashesToClipboard( 'sha512' ) elif action == CAC.SIMPLE_SHOW_DUPLICATES: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() duplicate_type = command.GetSimpleData() ClientGUIMedia.ShowDuplicatesInNewPage( self._location_context, hash, duplicate_type ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_CLEAR_FOCUSED_FALSE_POSITIVES: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.ClearFalsePositives( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_CLEAR_FALSE_POSITIVES: hashes = self._GetSelectedHashes() if len( hashes ) > 0: ClientGUIDuplicates.ClearFalsePositives( self, hashes ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_FOCUSED_ALTERNATE_GROUP: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.DissolveAlternateGroup( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_ALTERNATE_GROUP: hashes = self._GetSelectedHashes() if len( hashes ) > 0: ClientGUIDuplicates.DissolveAlternateGroup( self, hashes ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_FOCUSED_DUPLICATE_GROUP: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.DissolveDuplicateGroup( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_DUPLICATE_GROUP: hashes = self._GetSelectedHashes() if len( hashes ) > 0: ClientGUIDuplicates.DissolveDuplicateGroup( self, hashes ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_FROM_ALTERNATE_GROUP: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.RemoveFromAlternateGroup( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_FROM_DUPLICATE_GROUP: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.RemoveFromDuplicateGroup( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_RESET_FOCUSED_POTENTIAL_SEARCH: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.ResetPotentialSearch( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_RESET_POTENTIAL_SEARCH: hashes = self._GetSelectedHashes() if len( hashes ) > 0: ClientGUIDuplicates.ResetPotentialSearch( self, hashes ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_POTENTIALS: if self._HasFocusSingleton(): media = self._GetFocusSingleton() hash = media.GetHash() ClientGUIDuplicates.RemovePotentials( self, ( hash, ) ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_POTENTIALS: hashes = self._GetSelectedHashes() if len( hashes ) > 0: ClientGUIDuplicates.RemovePotentials( self, hashes ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_ALTERNATE: self._SetDuplicates( HC.DUPLICATE_ALTERNATE ) elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_ALTERNATE_COLLECTIONS: self._SetCollectionsAsAlternate() elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_CUSTOM: self._SetDuplicatesCustom() elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_FOCUSED_BETTER: self._SetDuplicatesFocusedBetter() elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_FOCUSED_KING: self._SetDuplicatesFocusedKing() elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_POTENTIAL: self._SetDuplicatesPotential() elif action == CAC.SIMPLE_DUPLICATE_MEDIA_SET_SAME_QUALITY: self._SetDuplicates( HC.DUPLICATE_SAME_QUALITY ) elif action == CAC.SIMPLE_EXPORT_FILES: self._ExportFiles() elif action == CAC.SIMPLE_EXPORT_FILES_QUICK_AUTO_EXPORT: self._ExportFiles( do_export_and_then_quit = True ) elif action == CAC.SIMPLE_MANAGE_FILE_RATINGS: self._ManageRatings() elif action == CAC.SIMPLE_MANAGE_FILE_TAGS: self._ManageTags() elif action == CAC.SIMPLE_MANAGE_FILE_URLS: self._ManageURLs() elif action == CAC.SIMPLE_MANAGE_FILE_NOTES: self._ManageNotes() elif action == CAC.SIMPLE_OPEN_KNOWN_URL: self._OpenKnownURL() elif action == CAC.SIMPLE_ARCHIVE_FILE: self._Archive() elif action == CAC.SIMPLE_DELETE_FILE: self._Delete() elif action == CAC.SIMPLE_UNDELETE_FILE: self._Undelete() elif action == CAC.SIMPLE_INBOX_FILE: self._Inbox() elif action == CAC.SIMPLE_REMOVE_FILE_FROM_VIEW: self._Remove( ClientMedia.FileFilter( ClientMedia.FILE_FILTER_SELECTED ) ) elif action == CAC.SIMPLE_GET_SIMILAR_TO_EXACT: self._GetSimilarTo( CC.HAMMING_EXACT_MATCH ) elif action == CAC.SIMPLE_GET_SIMILAR_TO_VERY_SIMILAR: self._GetSimilarTo( CC.HAMMING_VERY_SIMILAR ) elif action == CAC.SIMPLE_GET_SIMILAR_TO_SIMILAR: self._GetSimilarTo( CC.HAMMING_SIMILAR ) elif action == CAC.SIMPLE_GET_SIMILAR_TO_SPECULATIVE: self._GetSimilarTo( CC.HAMMING_SPECULATIVE ) elif action == CAC.SIMPLE_OPEN_FILE_IN_EXTERNAL_PROGRAM: self._OpenExternally() elif action == CAC.SIMPLE_OPEN_SELECTION_IN_NEW_PAGE: self._ShowSelectionInNewPage() elif action == CAC.SIMPLE_LAUNCH_THE_ARCHIVE_DELETE_FILTER: self._ArchiveDeleteFilter() else: command_processed = False elif command.IsContentCommand(): command_processed = ClientGUIMediaActions.ApplyContentApplicationCommandToMedia( self, command, self._GetSelectedFlatMedia() ) else: command_processed = False return command_processed def ProcessContentUpdates( self, service_keys_to_content_updates ): ClientMedia.ListeningMediaList.ProcessContentUpdates( self, service_keys_to_content_updates ) we_were_file_or_tag_affected = False for ( service_key, content_updates ) in service_keys_to_content_updates.items(): for content_update in content_updates: hashes = content_update.GetHashes() if self._HasHashes( hashes ): affected_media = self._GetMedia( hashes ) self._RedrawMedia( affected_media ) if content_update.GetDataType() in ( HC.CONTENT_TYPE_FILES, HC.CONTENT_TYPE_MAPPINGS ): we_were_file_or_tag_affected = True if we_were_file_or_tag_affected: self._PublishSelectionChange( tags_changed = True ) def ProcessServiceUpdates( self, service_keys_to_service_updates ): ClientMedia.ListeningMediaList.ProcessServiceUpdates( self, service_keys_to_service_updates ) for ( service_key, service_updates ) in list(service_keys_to_service_updates.items()): for service_update in service_updates: ( action, row ) = service_update.ToTuple() if action in ( HC.SERVICE_UPDATE_DELETE_PENDING, HC.SERVICE_UPDATE_RESET ): self._RecalculateVirtualSize() self._PublishSelectionChange( tags_changed = True ) def PublishSelectionChange( self ): self._PublishSelectionChange() def RemoveMedia( self, page_key, hashes ): if page_key == self._page_key: self._RemoveMediaByHashes( hashes ) def SelectByTags( self, page_key, tag_service_key, and_or_or, tags ): if page_key == self._page_key: self._Select( ClientMedia.FileFilter( ClientMedia.FILE_FILTER_TAGS, ( tag_service_key, and_or_or, tags ) ) ) self.setFocus( QC.Qt.OtherFocusReason ) def SetDuplicateStatusForAll( self, duplicate_type ): media_group = ClientMedia.FlattenMedia( self._sorted_media ) return self._SetDuplicates( duplicate_type, media_group = media_group ) def SetEmptyPageStatusOverride( self, value: str ): self._empty_page_status_override = value def SetFocusedMedia( self, media ): pass class MediaPanelLoading( MediaPanel ): def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext ): self._current = None self._max = None MediaPanel.__init__( self, parent, page_key, location_context, [] ) HG.client_controller.sub( self, 'SetNumQueryResults', 'set_num_query_results' ) def _GetPrettyStatusForStatusBar( self ): s = 'Loading\u2026' if self._current is not None: s += ' ' + HydrusData.ToHumanInt( self._current ) if self._max is not None: s += ' of ' + HydrusData.ToHumanInt( self._max ) return s def GetSortedMedia( self ): return [] def SetNumQueryResults( self, page_key, num_current, num_max ): if page_key == self._page_key: self._current = num_current self._max = num_max self._PublishSelectionChange() class MediaPanelThumbnails( MediaPanel ): def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results ): self._clean_canvas_pages = {} self._dirty_canvas_pages = [] self._num_rows_per_canvas_page = 1 self._num_rows_per_actual_page = 1 MediaPanel.__init__( self, parent, page_key, location_context, media_results ) self._last_size = QC.QSize( 20, 20 ) self._num_columns = 1 self._drag_init_coordinates = None self._drag_prefire_event_count = 0 self._thumbnails_being_faded_in = {} self._hashes_faded = set() self._last_device_pixel_ratio = self.devicePixelRatio() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() thumbnail_scroll_rate = float( HG.client_controller.new_options.GetString( 'thumbnail_scroll_rate' ) ) self.setWidget( MediaPanelThumbnails._InnerWidget( self ) ) self.setWidgetResizable( True ) self.verticalScrollBar().setSingleStep( int( round( thumbnail_span_height * thumbnail_scroll_rate ) ) ) self._widget_event_filter = QP.WidgetEventFilter( self.widget() ) self._widget_event_filter.EVT_LEFT_DCLICK( self.EventMouseFullScreen ) self._widget_event_filter.EVT_MIDDLE_DOWN( self.EventMouseFullScreen ) # notice this is on widget, not myself. fails to set up scrollbars if just moved up # there's a job in qt to-do to sort all this out and fix other scroll issues self._widget_event_filter.EVT_SIZE( self.EventResize ) self._widget_event_filter.EVT_KEY_DOWN( self.EventKeyDown ) self.widget().setMinimumSize( 50, 50 ) self.RefreshAcceleratorTable() self._UpdateScrollBars() HG.client_controller.sub( self, 'MaintainPageCache', 'memory_maintenance_pulse' ) HG.client_controller.sub( self, 'NotifyNewFileInfo', 'new_file_info' ) HG.client_controller.sub( self, 'NewThumbnails', 'new_thumbnails' ) HG.client_controller.sub( self, 'ThumbnailsReset', 'notify_complete_thumbnail_reset' ) HG.client_controller.sub( self, 'RedrawAllThumbnails', 'refresh_all_tag_presentation_gui' ) HG.client_controller.sub( self, 'RefreshAcceleratorTable', 'notify_new_options' ) HG.client_controller.sub( self, 'WaterfallThumbnails', 'waterfall_thumbnails' ) def _CalculateVisiblePageIndices( self ): y_start = self._GetYStart() earliest_y = y_start last_y = earliest_y + QP.ScrollAreaVisibleRect( self ).size().height() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() page_height = self._num_rows_per_canvas_page * thumbnail_span_height first_visible_page_index = earliest_y // page_height last_visible_page_index = last_y // page_height page_indices = list( range( first_visible_page_index, last_visible_page_index + 1 ) ) return page_indices def _CreateNewDirtyPage( self ): my_width = self.size().width() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() dpr = self.devicePixelRatio() canvas_width = int( my_width * dpr ) canvas_height = int( self._num_rows_per_canvas_page * thumbnail_span_height * dpr ) canvas_page = HG.client_controller.bitmap_manager.GetQtImage( canvas_width, canvas_height, 32 ) canvas_page.setDevicePixelRatio( dpr ) self._dirty_canvas_pages.append( canvas_page ) def _DeleteAllDirtyPages( self ): self._dirty_canvas_pages = [] def _DirtyAllPages( self ): clean_indices = list( self._clean_canvas_pages.keys() ) for clean_index in clean_indices: self._DirtyPage( clean_index ) def _DirtyPage( self, clean_index ): canvas_page = self._clean_canvas_pages[ clean_index ] del self._clean_canvas_pages[ clean_index ] thumbnails = [ thumbnail for ( thumbnail_index, thumbnail ) in self._GetThumbnailsFromPageIndex( clean_index ) ] if len( thumbnails ) > 0: HG.client_controller.GetCache( 'thumbnail' ).CancelWaterfall( self._page_key, thumbnails ) self._dirty_canvas_pages.append( canvas_page ) def _DrawCanvasPage( self, page_index, canvas_page ): painter = QG.QPainter( canvas_page ) new_options = HG.client_controller.new_options bg_colour = HG.client_controller.new_options.GetColour( CC.COLOUR_THUMBGRID_BACKGROUND ) if HG.thumbnail_debug_mode and page_index % 2 == 0: bg_colour = ClientGUIFunctions.GetLighterDarkerColour( bg_colour ) if new_options.GetNoneableString( 'media_background_bmp_path' ) is not None: comp_mode = painter.compositionMode() painter.setCompositionMode( QG.QPainter.CompositionMode_Source ) painter.setBackground( QG.QBrush( QC.Qt.transparent ) ) painter.eraseRect( painter.viewport() ) painter.setCompositionMode( comp_mode ) else: painter.setBackground( QG.QBrush( bg_colour ) ) painter.eraseRect( painter.viewport() ) # page_thumbnails = self._GetThumbnailsFromPageIndex( page_index ) ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() thumbnails_to_render_later = [] thumbnail_cache = HG.client_controller.GetCache( 'thumbnail' ) thumbnail_margin = HG.client_controller.new_options.GetInteger( 'thumbnail_margin' ) for ( thumbnail_index, thumbnail ) in page_thumbnails: display_media = thumbnail.GetDisplayMedia() if display_media is None: continue hash = display_media.GetHash() if hash in self._hashes_faded and thumbnail_cache.HasThumbnailCached( thumbnail ): self._StopFading( hash ) thumbnail_col = thumbnail_index % self._num_columns thumbnail_row = thumbnail_index // self._num_columns x = thumbnail_col * thumbnail_span_width + thumbnail_margin y = ( thumbnail_row - ( page_index * self._num_rows_per_canvas_page ) ) * thumbnail_span_height + thumbnail_margin painter.drawImage( x, y, thumbnail.GetQtImage( self.devicePixelRatio() ) ) else: thumbnails_to_render_later.append( thumbnail ) if len( thumbnails_to_render_later ) > 0: HG.client_controller.GetCache( 'thumbnail' ).Waterfall( self._page_key, thumbnails_to_render_later ) def _FadeThumbnails( self, thumbnails ): if len( thumbnails ) == 0: return if not HG.client_controller.gui.IsCurrentPage( self._page_key ): self._DirtyAllPages() return now_precise = HydrusData.GetNowPrecise() for thumbnail in thumbnails: display_media = thumbnail.GetDisplayMedia() if display_media is None: continue try: thumbnail_index = self._sorted_media.index( thumbnail ) except HydrusExceptions.DataMissing: # probably means a collect happened during an ongoing waterfall or whatever continue if self._GetPageIndexFromThumbnailIndex( thumbnail_index ) not in self._clean_canvas_pages: continue hash = display_media.GetHash() self._hashes_faded.add( hash ) self._StopFading( hash ) bmp = thumbnail.GetQtImage( self.devicePixelRatio() ) alpha_bmp = QP.AdjustOpacity( bmp, 0.20 ) self._thumbnails_being_faded_in[ hash ] = ( bmp, alpha_bmp, thumbnail_index, thumbnail, now_precise, 0 ) HG.client_controller.gui.RegisterAnimationUpdateWindow( self ) def _GenerateMediaCollection( self, media_results ): return ThumbnailMediaCollection( self._location_context, media_results ) def _GenerateMediaSingleton( self, media_result ): return ThumbnailMediaSingleton( media_result ) def _GetMediaCoordinates( self, media ): try: index = self._sorted_media.index( media ) except: return ( -1, -1 ) row = index // self._num_columns column = index % self._num_columns ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() thumbnail_margin = HG.client_controller.new_options.GetInteger( 'thumbnail_margin' ) ( x, y ) = ( column * thumbnail_span_width + thumbnail_margin, row * thumbnail_span_height + thumbnail_margin ) return ( x, y ) def _GetPageIndexFromThumbnailIndex( self, thumbnail_index ): thumbnails_per_page = self._num_columns * self._num_rows_per_canvas_page page_index = thumbnail_index // thumbnails_per_page return page_index def _GetThumbnailSpanDimensions( self ): thumbnail_border = HG.client_controller.new_options.GetInteger( 'thumbnail_border' ) thumbnail_margin = HG.client_controller.new_options.GetInteger( 'thumbnail_margin' ) return ClientData.AddPaddingToDimensions( HC.options[ 'thumbnail_dimensions' ], ( thumbnail_border + thumbnail_margin ) * 2 ) def _GetThumbnailUnderMouse( self, mouse_event ): pos = mouse_event.position().toPoint() x = pos.x() y = pos.y() ( t_span_x, t_span_y ) = self._GetThumbnailSpanDimensions() x_mod = x % t_span_x y_mod = y % t_span_y thumbnail_margin = HG.client_controller.new_options.GetInteger( 'thumbnail_margin' ) if x_mod <= thumbnail_margin or y_mod <= thumbnail_margin or x_mod > t_span_x - thumbnail_margin or y_mod > t_span_y - thumbnail_margin: return None column_index = x // t_span_x row_index = y // t_span_y if column_index >= self._num_columns: return None thumbnail_index = self._num_columns * row_index + column_index if thumbnail_index < 0: return None if thumbnail_index >= len( self._sorted_media ): return None return self._sorted_media[ thumbnail_index ] def _GetThumbnailsFromPageIndex( self, page_index ): num_thumbnails_per_page = self._num_columns * self._num_rows_per_canvas_page start_index = num_thumbnails_per_page * page_index if start_index <= len( self._sorted_media ): end_index = min( len( self._sorted_media ), start_index + num_thumbnails_per_page ) thumbnails = [ ( index, self._sorted_media[ index ] ) for index in range( start_index, end_index ) ] else: thumbnails = [] return thumbnails def _GetYStart( self ): visible_rect = QP.ScrollAreaVisibleRect( self ) visible_rect_y = visible_rect.y() visible_rect_height = visible_rect.height() my_virtual_size = self.widget().size() my_virtual_height = my_virtual_size.height() max_y = my_virtual_height - visible_rect_height y_start = max( 0, visible_rect_y ) y_start = min( y_start, max_y ) return y_start def _MediaIsInCleanPage( self, thumbnail ): try: index = self._sorted_media.index( thumbnail ) except HydrusExceptions.DataMissing: return False if self._GetPageIndexFromThumbnailIndex( index ) in self._clean_canvas_pages: return True else: return False def _MediaIsVisible( self, media ): if media is not None: ( x, y ) = self._GetMediaCoordinates( media ) visible_rect = QP.ScrollAreaVisibleRect( self ) visible_rect_y = visible_rect.y() visible_rect_height = visible_rect.height() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() bottom_edge_below_top_of_view = visible_rect_y < y + thumbnail_span_height top_edge_above_bottom_of_view = y < visible_rect_y + visible_rect_height is_visible = bottom_edge_below_top_of_view and top_edge_above_bottom_of_view return is_visible return True def _MoveFocusedThumbnail( self, rows, columns, shift ): if self._last_hit_media is not None: media_to_use = self._last_hit_media elif self._next_best_media_if_focuses_removed is not None: media_to_use = self._next_best_media_if_focuses_removed if columns == -1: # treat it as if the focused area is between this and the next columns = 0 elif len( self._sorted_media ) > 0: media_to_use = self._sorted_media[ 0 ] else: media_to_use = None if media_to_use is not None: try: current_position = self._sorted_media.index( media_to_use ) except HydrusExceptions.DataMissing: self._SetFocusedMedia( None ) return new_position = current_position + columns + ( self._num_columns * rows ) if new_position < 0: new_position = 0 elif new_position > len( self._sorted_media ) - 1: new_position = len( self._sorted_media ) - 1 new_media = self._sorted_media[ new_position ] self._HitMedia( new_media, False, shift ) self._ScrollToMedia( new_media ) def _RecalculateVirtualSize( self, called_from_resize_event = False ): my_size = QP.ScrollAreaVisibleRect( self ).size() my_width = my_size.width() my_height = my_size.height() if my_width > 0 and my_height > 0: ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() num_media = len( self._sorted_media ) num_rows = max( 1, num_media // self._num_columns ) if num_media % self._num_columns > 0: num_rows += 1 virtual_width = my_width virtual_height = num_rows * thumbnail_span_height yUnit = self.verticalScrollBar().singleStep() excess = virtual_height % yUnit if excess > 0: # we want virtual height to fit exactly into scroll units, even if that puts some padding below bottom row top_up = yUnit - excess virtual_height += top_up virtual_height = max( virtual_height, my_height ) virtual_size = QC.QSize( virtual_width, virtual_height ) if virtual_size != self.widget().size(): self.widget().resize( QC.QSize( virtual_width, virtual_height ) ) if not called_from_resize_event: self._UpdateScrollBars() # would lead to infinite recursion if called from a resize event def _RedrawMedia( self, thumbnails ): visible_thumbnails = [ thumbnail for thumbnail in thumbnails if self._MediaIsInCleanPage( thumbnail ) ] thumbnail_cache = HG.client_controller.GetCache( 'thumbnail' ) thumbnails_to_render_now = [] thumbnails_to_render_later = [] for thumbnail in visible_thumbnails: if thumbnail_cache.HasThumbnailCached( thumbnail ): thumbnails_to_render_now.append( thumbnail ) else: thumbnails_to_render_later.append( thumbnail ) if len( thumbnails_to_render_now ) > 0: self._FadeThumbnails( thumbnails_to_render_now ) if len( thumbnails_to_render_later ) > 0: HG.client_controller.GetCache( 'thumbnail' ).Waterfall( self._page_key, thumbnails_to_render_later ) def _ReinitialisePageCacheIfNeeded( self ): old_num_rows = self._num_rows_per_canvas_page old_num_columns = self._num_columns old_width = self._last_size.width() old_height = self._last_size.height() my_size = QP.ScrollAreaVisibleRect( self ).size() my_width = my_size.width() my_height = my_size.height() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() num_rows = ( my_height // thumbnail_span_height ) self._num_rows_per_actual_page = max( 1, num_rows ) self._num_rows_per_canvas_page = max( 1, num_rows // 2 ) self._num_columns = max( 1, my_width // thumbnail_span_width ) dimensions_changed = old_width != my_width or old_height != my_height thumb_layout_changed = old_num_columns != self._num_columns or old_num_rows != self._num_rows_per_canvas_page if dimensions_changed or thumb_layout_changed: width_got_bigger = old_width < my_width if thumb_layout_changed or width_got_bigger: self._DirtyAllPages() self._DeleteAllDirtyPages() self.widget().update() def _RemoveMediaDirectly( self, singleton_media, collected_media ): if self._focused_media is not None: if self._focused_media in singleton_media or self._focused_media in collected_media: self._SetFocusedMedia( None ) MediaPanel._RemoveMediaDirectly( self, singleton_media, collected_media ) self._EndShiftSelect() self._RecalculateVirtualSize() self._DirtyAllPages() self._PublishSelectionChange() HG.client_controller.pub( 'refresh_page_name', self._page_key ) HG.client_controller.pub( 'notify_new_pages_count' ) self.widget().update() def _ScrollEnd( self, shift = False ): if len( self._sorted_media ) > 0: end_media = self._sorted_media[ -1 ] self._HitMedia( end_media, False, shift ) self._ScrollToMedia( end_media ) def _ScrollHome( self, shift = False ): if len( self._sorted_media ) > 0: home_media = self._sorted_media[ 0 ] self._HitMedia( home_media, False, shift ) self._ScrollToMedia( home_media ) def _ScrollToMedia( self, media ): if media is not None: ( x, y ) = self._GetMediaCoordinates( media ) visible_rect = QP.ScrollAreaVisibleRect( self ) visible_rect_y = visible_rect.y() visible_rect_height = visible_rect.height() ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() new_options = HG.client_controller.new_options percent_visible = new_options.GetInteger( 'thumbnail_visibility_scroll_percent' ) / 100 if y < visible_rect_y: self.ensureVisible( 0, y, 0, 0 ) elif y > visible_rect_y + visible_rect_height - ( thumbnail_span_height * percent_visible ): self.ensureVisible( 0, y + thumbnail_span_height ) def _StopFading( self, hash ): if hash in self._thumbnails_being_faded_in: ( bmp, alpha_bmp, thumbnail_index, thumbnail, animation_started, num_frames ) = self._thumbnails_being_faded_in[ hash ] del self._thumbnails_being_faded_in[ hash ] def _UpdateBackgroundColour( self ): MediaPanel._UpdateBackgroundColour( self ) self._DirtyAllPages() self._DeleteAllDirtyPages() self.widget().update() def _UpdateScrollBars( self ): # The following call is officially a no-op since this property is already true, but it also triggers an update # of the scroll area's scrollbars which we need. # We need this since we are intercepting & doing work in resize events which causes # event propagation between the scroll area and the scrolled widget to not work properly (since we are suppressing resize events of the scrolled widget - otherwise we would get an infinite loop). # Probably the best would be to change how this work and not intercept any resize events. # Originally this was wx event handling which got ported to Qt more or less unchanged, hence the hackiness. self.setWidgetResizable( True ) def AddMediaResults( self, page_key, media_results ): if page_key == self._page_key: thumbnails = MediaPanel.AddMediaResults( self, page_key, media_results ) if len( thumbnails ) > 0: self._RecalculateVirtualSize() HG.client_controller.GetCache( 'thumbnail' ).Waterfall( self._page_key, thumbnails ) if len( self._selected_media ) == 0: self._PublishSelectionIncrement( thumbnails ) else: self.statusTextChanged.emit( self._GetPrettyStatusForStatusBar() ) def mouseMoveEvent( self, event ): if event.buttons() & QC.Qt.LeftButton: we_started_dragging_on_this_panel = self._drag_init_coordinates is not None if we_started_dragging_on_this_panel: old_drag_pos = self._drag_init_coordinates global_mouse_pos = QG.QCursor.pos() delta_pos = global_mouse_pos - old_drag_pos total_absolute_pixels_moved = delta_pos.manhattanLength() we_moved = total_absolute_pixels_moved > 0 if we_moved: self._drag_prefire_event_count += 1 # prefire deal here is mpv lags on initial click, which can cause a drag (and hence an immediate pause) event by accident when mouserelease isn't processed quick # so now we'll say we can't start a drag unless we get a smooth ramp to our pixel delta threshold clean_drag_started = self._drag_prefire_event_count >= 10 moved_a_decent_bit_from_start = total_absolute_pixels_moved > 20 if clean_drag_started and moved_a_decent_bit_from_start: media = self._GetSelectedFlatMedia( discriminant = CC.DISCRIMINANT_LOCAL ) if len( media ) > 0: alt_down = event.modifiers() & QC.Qt.AltModifier result = ClientGUIDragDrop.DoFileExportDragDrop( self, self._page_key, media, alt_down ) if result not in ( QC.Qt.IgnoreAction, ): self.focusMediaPaused.emit() else: self._drag_init_coordinates = None self._drag_prefire_event_count = 0 event.ignore() def EventKeyDown( self, event ): # accelerator tables can't handle escape key in windows, gg if event.key() == QC.Qt.Key_Escape: self._Select( ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NONE ) ) elif event.key() in ( QC.Qt.Key_PageUp, QC.Qt.Key_PageDown ): if event.key() == QC.Qt.Key_PageUp: direction = -1 elif event.key() == QC.Qt.Key_PageDown: direction = 1 shift = event.modifiers() & QC.Qt.ShiftModifier self._MoveFocusedThumbnail( self._num_rows_per_actual_page * direction, 0, shift ) else: return True # was: event.ignore() def EventMouseFullScreen( self, event ): t = self._GetThumbnailUnderMouse( event ) if t is not None: locations_manager = t.GetLocationsManager() if locations_manager.IsLocal(): self._LaunchMediaViewer( t ) else: can_download = not locations_manager.GetCurrent().isdisjoint( HG.client_controller.services_manager.GetRemoteFileServiceKeys() ) if can_download: self._DownloadHashes( t.GetHashes() ) def showEvent( self, event ): self._UpdateScrollBars() class _InnerWidget( QW.QWidget ): def __init__( self, parent ): QW.QWidget.__init__( self, parent ) self._parent = parent def mousePressEvent( self, event ): self._parent._drag_init_coordinates = QG.QCursor.pos() thumb = self._parent._GetThumbnailUnderMouse( event ) right_on_whitespace = event.button() == QC.Qt.RightButton and thumb is None if not right_on_whitespace: self._parent._HitMedia( thumb, event.modifiers() & QC.Qt.ControlModifier, event.modifiers() & QC.Qt.ShiftModifier ) # this specifically does not scroll to media, as for clicking (esp. double-clicking attempts), the scroll can be jarring def paintEvent( self, event ): if self._parent.devicePixelRatio() != self._parent._last_device_pixel_ratio: self._parent._last_device_pixel_ratio = self._parent.devicePixelRatio() self._parent._DirtyAllPages() self._parent._DeleteAllDirtyPages() painter = QG.QPainter( self ) ( thumbnail_span_width, thumbnail_span_height ) = self._parent._GetThumbnailSpanDimensions() page_height = self._parent._num_rows_per_canvas_page * thumbnail_span_height page_indices_to_display = self._parent._CalculateVisiblePageIndices() earliest_page_index_to_display = min( page_indices_to_display ) last_page_index_to_display = max( page_indices_to_display ) page_indices_to_draw = list( page_indices_to_display ) if earliest_page_index_to_display > 0: page_indices_to_draw.append( earliest_page_index_to_display - 1 ) page_indices_to_draw.append( last_page_index_to_display + 1 ) page_indices_to_draw.sort() potential_clean_indices_to_steal = [ page_index for page_index in self._parent._clean_canvas_pages.keys() if page_index not in page_indices_to_draw ] random.shuffle( potential_clean_indices_to_steal ) y_start = self._parent._GetYStart() earliest_y = y_start bg_colour = HG.client_controller.new_options.GetColour( CC.COLOUR_THUMBGRID_BACKGROUND ) painter.setBackground( QG.QBrush( bg_colour ) ) painter.eraseRect( painter.viewport() ) background_pixmap = HG.client_controller.bitmap_manager.GetMediaBackgroundPixmap() if background_pixmap is not None: my_size = QP.ScrollAreaVisibleRect( self._parent ).size() pixmap_size = background_pixmap.size() painter.drawPixmap( my_size.width() - pixmap_size.width(), my_size.height() - pixmap_size.height(), background_pixmap ) for page_index in page_indices_to_draw: if page_index not in self._parent._clean_canvas_pages: if len( self._parent._dirty_canvas_pages ) == 0: if len( potential_clean_indices_to_steal ) > 0: index_to_steal = potential_clean_indices_to_steal.pop() self._parent._DirtyPage( index_to_steal ) else: self._parent._CreateNewDirtyPage() canvas_page = self._parent._dirty_canvas_pages.pop() self._parent._DrawCanvasPage( page_index, canvas_page ) self._parent._clean_canvas_pages[ page_index ] = canvas_page if page_index in page_indices_to_display: canvas_page = self._parent._clean_canvas_pages[ page_index ] page_virtual_y = page_height * page_index painter.drawImage( 0, page_virtual_y, canvas_page ) def EventResize( self, event ): self._ReinitialisePageCacheIfNeeded() self._RecalculateVirtualSize( called_from_resize_event = True ) self._last_size = QP.ScrollAreaVisibleRect( self ).size() def contextMenuEvent( self, event ): if event.reason() == QG.QContextMenuEvent.Keyboard: self.ShowMenu() def mouseReleaseEvent( self, event ): if event.button() != QC.Qt.RightButton: QW.QScrollArea.mouseReleaseEvent( self, event ) return self.ShowMenu() def GetTotalFileSize( self ): return sum( ( m.GetSize() for m in self._sorted_media ) ) def MaintainPageCache( self ): if not HG.client_controller.gui.IsCurrentPage( self._page_key ): self._DirtyAllPages() self._DeleteAllDirtyPages() def NewThumbnails( self, hashes ): affected_thumbnails = self._GetMedia( hashes ) if len( affected_thumbnails ) > 0: self._RedrawMedia( affected_thumbnails ) def NotifyNewFileInfo( self, hashes ): def qt_do_update( hashes_to_media_results ): affected_media = self._GetMedia( set( hashes_to_media_results.keys() ) ) for media in affected_media: media.UpdateFileInfo( hashes_to_media_results ) self._RedrawMedia( affected_media ) def do_it( win, callable, affected_hashes ): media_results = HG.client_controller.Read( 'media_results', affected_hashes ) hashes_to_media_results = { media_result.GetHash() : media_result for media_result in media_results } HG.client_controller.CallAfterQtSafe( win, 'new file info notification', qt_do_update, hashes_to_media_results ) affected_hashes = self._hashes.intersection( hashes ) HG.client_controller.CallToThread( do_it, self, do_it, affected_hashes ) def RedrawAllThumbnails( self ): self._DirtyAllPages() for m in self._collected_media: m.RecalcInternals() for thumbnail in self._sorted_media: thumbnail.ClearTagSummaryCaches() self.widget().update() def RefreshAcceleratorTable( self ): if not self or not QP.isValid( self ): return # Remove old shortcuts for child in self.children(): if isinstance( child, QW.QShortcut ): child.setParent( None ) child.deleteLater() def ctrl_space_callback( self ): if self._focused_media is not None: self._HitMedia( self._focused_media, True, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Home, self._ScrollHome, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Home, self._ScrollHome, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_End, self._ScrollEnd, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_End, self._ScrollEnd, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Return, self._LaunchMediaViewer ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Enter, self._LaunchMediaViewer ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Up, self._MoveFocusedThumbnail, -1, 0, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Up, self._MoveFocusedThumbnail, -1, 0, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Down, self._MoveFocusedThumbnail, 1, 0, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Down, self._MoveFocusedThumbnail, 1, 0, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Left, self._MoveFocusedThumbnail, 0, -1, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Left, self._MoveFocusedThumbnail, 0, -1, False ) QP.AddShortcut( self, QC.Qt.NoModifier, QC.Qt.Key_Right, self._MoveFocusedThumbnail, 0, 1, False ) QP.AddShortcut( self, QC.Qt.KeypadModifier, QC.Qt.Key_Right, self._MoveFocusedThumbnail, 0, 1, False ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_Home, self._ScrollHome, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_Home, self._ScrollHome, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_End, self._ScrollEnd, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_End, self._ScrollEnd, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_Up, self._MoveFocusedThumbnail, -1, 0, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_Up, self._MoveFocusedThumbnail, -1, 0, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_Down, self._MoveFocusedThumbnail, 1, 0, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_Down, self._MoveFocusedThumbnail, 1, 0, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_Left, self._MoveFocusedThumbnail, 0, -1, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_Left, self._MoveFocusedThumbnail, 0, -1, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier, QC.Qt.Key_Right, self._MoveFocusedThumbnail, 0, 1, True ) QP.AddShortcut( self, QC.Qt.ShiftModifier | QC.Qt.KeypadModifier, QC.Qt.Key_Right, self._MoveFocusedThumbnail, 0, 1, True ) QP.AddShortcut( self, QC.Qt.ControlModifier, QC.Qt.Key_A, self._Select, ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ALL ) ) QP.AddShortcut( self, QC.Qt.ControlModifier, QC.Qt.Key_Space, ctrl_space_callback, self ) def SetFocusedMedia( self, media ): MediaPanel.SetFocusedMedia( self, media ) if media is None: self._SetFocusedMedia( None ) else: try: my_media = self._GetMedia( media.GetHashes() )[0] self._HitMedia( my_media, False, False ) self._ScrollToMedia( self._focused_media ) except: pass def ShowMenu( self, do_not_show_just_return = False ): new_options = HG.client_controller.new_options advanced_mode = new_options.GetBoolean( 'advanced_mode' ) services_manager = HG.client_controller.services_manager flat_selected_medias = ClientMedia.FlattenMedia( self._selected_media ) all_locations_managers = [ media.GetLocationsManager() for media in ClientMedia.FlattenMedia( self._sorted_media ) ] selected_locations_managers = [ media.GetLocationsManager() for media in flat_selected_medias ] selection_has_local = True in ( locations_manager.IsLocal() for locations_manager in selected_locations_managers ) selection_has_local_file_domain = True in ( locations_manager.IsLocal() and not locations_manager.IsTrashed() for locations_manager in selected_locations_managers ) selection_has_trash = True in ( locations_manager.IsTrashed() for locations_manager in selected_locations_managers ) selection_has_inbox = True in ( media.HasInbox() for media in self._selected_media ) selection_has_archive = True in ( media.HasArchive() and media.GetLocationsManager().IsLocal() for media in self._selected_media ) selection_has_deletion_record = True in ( CC.COMBINED_LOCAL_FILE_SERVICE_KEY in locations_manager.GetDeleted() for locations_manager in selected_locations_managers ) all_file_domains = HydrusData.MassUnion( locations_manager.GetCurrent() for locations_manager in all_locations_managers ) all_specific_file_domains = all_file_domains.difference( { CC.COMBINED_FILE_SERVICE_KEY, CC.COMBINED_LOCAL_FILE_SERVICE_KEY } ) all_local_file_domains = services_manager.Filter( all_specific_file_domains, ( HC.LOCAL_FILE_DOMAIN, ) ) all_local_file_domains_sorted = sorted( all_local_file_domains, key = HG.client_controller.services_manager.GetName ) all_file_repos = services_manager.Filter( all_specific_file_domains, ( HC.FILE_REPOSITORY, ) ) has_local = True in ( locations_manager.IsLocal() for locations_manager in all_locations_managers ) has_remote = True in ( locations_manager.IsRemote() for locations_manager in all_locations_managers ) num_files = self.GetNumFiles() num_selected = self._GetNumSelected() num_inbox = self.GetNumInbox() num_archive = self.GetNumArchive() multiple_selected = num_selected > 1 media_has_inbox = num_inbox > 0 media_has_archive = num_archive > 0 menu = QW.QMenu( self.window() ) if self._HasFocusSingleton(): focus_singleton = self._GetFocusSingleton() # variables collections_selected = True in ( media.IsCollection() for media in self._selected_media ) services_manager = HG.client_controller.services_manager services = services_manager.GetServices() service_keys_to_names = { service.GetServiceKey() : service.GetName() for service in services } file_repositories = [ service for service in services if service.GetServiceType() == HC.FILE_REPOSITORY ] ipfs_services = [ service for service in services if service.GetServiceType() == HC.IPFS ] local_ratings_services = [ service for service in services if service.GetServiceType() in ( HC.LOCAL_RATING_LIKE, HC.LOCAL_RATING_NUMERICAL ) ] local_booru_service = [ service for service in services if service.GetServiceType() == HC.LOCAL_BOORU ][0] local_booru_is_running = local_booru_service.GetPort() is not None i_can_post_ratings = len( local_ratings_services ) > 0 focused_is_local = CC.COMBINED_LOCAL_FILE_SERVICE_KEY in self._focused_media.GetLocationsManager().GetCurrent() local_media_file_service_keys = { service.GetServiceKey() for service in services if service.GetServiceType() == HC.LOCAL_FILE_DOMAIN } file_repository_service_keys = { repository.GetServiceKey() for repository in file_repositories } upload_permission_file_service_keys = { repository.GetServiceKey() for repository in file_repositories if repository.HasPermission( HC.CONTENT_TYPE_FILES, HC.PERMISSION_ACTION_CREATE ) } petition_resolve_permission_file_service_keys = { repository.GetServiceKey() for repository in file_repositories if repository.HasPermission( HC.CONTENT_TYPE_FILES, HC.PERMISSION_ACTION_MODERATE ) } petition_permission_file_service_keys = { repository.GetServiceKey() for repository in file_repositories if repository.HasPermission( HC.CONTENT_TYPE_FILES, HC.PERMISSION_ACTION_PETITION ) } - petition_resolve_permission_file_service_keys user_manage_permission_file_service_keys = { repository.GetServiceKey() for repository in file_repositories if repository.HasPermission( HC.CONTENT_TYPE_ACCOUNTS, HC.PERMISSION_ACTION_MODERATE ) } ipfs_service_keys = { service.GetServiceKey() for service in ipfs_services } focused_is_ipfs = not self._focused_media.GetLocationsManager().GetCurrent().isdisjoint( ipfs_service_keys ) if multiple_selected: download_phrase = 'download all possible selected' rescind_download_phrase = 'cancel downloads for all possible selected' upload_phrase = 'upload all possible selected to' rescind_upload_phrase = 'rescind pending selected uploads to' petition_phrase = 'petition all possible selected for removal from' rescind_petition_phrase = 'rescind selected petitions for' remote_delete_phrase = 'delete all possible selected from' modify_account_phrase = 'modify the accounts that uploaded selected to' pin_phrase = 'pin all to' rescind_pin_phrase = 'rescind pin to' unpin_phrase = 'unpin all from' rescind_unpin_phrase = 'rescind unpin from' archive_phrase = 'archive selected' inbox_phrase = 'return selected to inbox' local_delete_phrase = 'delete selected' delete_physically_phrase = 'delete selected physically now' undelete_phrase = 'undelete selected' clear_deletion_phrase = 'clear deletion record for selected' export_phrase = 'files' copy_phrase = 'files' else: download_phrase = 'download' rescind_download_phrase = 'cancel download' upload_phrase = 'upload to' rescind_upload_phrase = 'rescind pending upload to' petition_phrase = 'petition for removal from' rescind_petition_phrase = 'rescind petition for' remote_delete_phrase = 'delete from' modify_account_phrase = 'modify the account that uploaded this to' pin_phrase = 'pin to' rescind_pin_phrase = 'rescind pin to' unpin_phrase = 'unpin from' rescind_unpin_phrase = 'rescind unpin from' archive_phrase = 'archive' inbox_phrase = 'return to inbox' local_delete_phrase = 'delete' delete_physically_phrase = 'delete physically now' undelete_phrase = 'undelete' clear_deletion_phrase = 'clear deletion record' export_phrase = 'file' copy_phrase = 'file' # info about the files remote_service_keys = HG.client_controller.services_manager.GetRemoteFileServiceKeys() groups_of_current_remote_service_keys = [ locations_manager.GetCurrent().intersection( remote_service_keys ) for locations_manager in selected_locations_managers ] groups_of_pending_remote_service_keys = [ locations_manager.GetPending().intersection( remote_service_keys ) for locations_manager in selected_locations_managers ] groups_of_petitioned_remote_service_keys = [ locations_manager.GetPetitioned().intersection( remote_service_keys ) for locations_manager in selected_locations_managers ] groups_of_deleted_remote_service_keys = [ locations_manager.GetDeleted().intersection( remote_service_keys ) for locations_manager in selected_locations_managers ] current_remote_service_keys = HydrusData.MassUnion( groups_of_current_remote_service_keys ) pending_remote_service_keys = HydrusData.MassUnion( groups_of_pending_remote_service_keys ) petitioned_remote_service_keys = HydrusData.MassUnion( groups_of_petitioned_remote_service_keys ) deleted_remote_service_keys = HydrusData.MassUnion( groups_of_deleted_remote_service_keys ) common_current_remote_service_keys = HydrusData.IntelligentMassIntersect( groups_of_current_remote_service_keys ) common_pending_remote_service_keys = HydrusData.IntelligentMassIntersect( groups_of_pending_remote_service_keys ) common_petitioned_remote_service_keys = HydrusData.IntelligentMassIntersect( groups_of_petitioned_remote_service_keys ) common_deleted_remote_service_keys = HydrusData.IntelligentMassIntersect( groups_of_deleted_remote_service_keys ) disparate_current_remote_service_keys = current_remote_service_keys - common_current_remote_service_keys disparate_pending_remote_service_keys = pending_remote_service_keys - common_pending_remote_service_keys disparate_petitioned_remote_service_keys = petitioned_remote_service_keys - common_petitioned_remote_service_keys disparate_deleted_remote_service_keys = deleted_remote_service_keys - common_deleted_remote_service_keys some_downloading = True in ( locations_manager.IsDownloading() for locations_manager in selected_locations_managers ) pending_file_service_keys = pending_remote_service_keys.intersection( file_repository_service_keys ) petitioned_file_service_keys = petitioned_remote_service_keys.intersection( file_repository_service_keys ) common_current_file_service_keys = common_current_remote_service_keys.intersection( file_repository_service_keys ) common_pending_file_service_keys = common_pending_remote_service_keys.intersection( file_repository_service_keys ) common_petitioned_file_service_keys = common_petitioned_remote_service_keys.intersection( file_repository_service_keys ) common_deleted_file_service_keys = common_deleted_remote_service_keys.intersection( file_repository_service_keys ) disparate_current_file_service_keys = disparate_current_remote_service_keys.intersection( file_repository_service_keys ) disparate_pending_file_service_keys = disparate_pending_remote_service_keys.intersection( file_repository_service_keys ) disparate_petitioned_file_service_keys = disparate_petitioned_remote_service_keys.intersection( file_repository_service_keys ) disparate_deleted_file_service_keys = disparate_deleted_remote_service_keys.intersection( file_repository_service_keys ) pending_ipfs_service_keys = pending_remote_service_keys.intersection( ipfs_service_keys ) petitioned_ipfs_service_keys = petitioned_remote_service_keys.intersection( ipfs_service_keys ) common_current_ipfs_service_keys = common_current_remote_service_keys.intersection( ipfs_service_keys ) common_pending_ipfs_service_keys = common_pending_file_service_keys.intersection( ipfs_service_keys ) common_petitioned_ipfs_service_keys = common_petitioned_remote_service_keys.intersection( ipfs_service_keys ) disparate_current_ipfs_service_keys = disparate_current_remote_service_keys.intersection( ipfs_service_keys ) disparate_pending_ipfs_service_keys = disparate_pending_remote_service_keys.intersection( ipfs_service_keys ) disparate_petitioned_ipfs_service_keys = disparate_petitioned_remote_service_keys.intersection( ipfs_service_keys ) # valid commands for the files current_file_service_keys = set() uploadable_file_service_keys = set() downloadable_file_service_keys = set() petitionable_file_service_keys = set() deletable_file_service_keys = set() modifyable_file_service_keys = set() pinnable_ipfs_service_keys = set() unpinnable_ipfs_service_keys = set() remote_file_service_keys = ipfs_service_keys.union( file_repository_service_keys ) for locations_manager in selected_locations_managers: current = locations_manager.GetCurrent() deleted = locations_manager.GetDeleted() pending = locations_manager.GetPending() petitioned = locations_manager.GetPetitioned() # ALL current_file_service_keys.update( current ) # FILE REPOS # we can upload (set pending) to a repo_id when we have permission, a file is local, not current, not pending, and either ( not deleted or we_can_overrule ) if locations_manager.IsLocal(): cannot_upload_to = current.union( pending ).union( deleted.difference( petition_resolve_permission_file_service_keys ) ) can_upload_to = upload_permission_file_service_keys.difference( cannot_upload_to ) uploadable_file_service_keys.update( can_upload_to ) # we can download (set pending to local) when we have permission, a file is not local and not already downloading and current if not locations_manager.IsLocal() and not locations_manager.IsDownloading(): downloadable_file_service_keys.update( remote_file_service_keys.intersection( current ) ) # we can petition when we have permission and a file is current and it is not already petitioned petitionable_file_service_keys.update( ( petition_permission_file_service_keys & current ) - petitioned ) # we can delete remote when we have permission and a file is current and it is not already petitioned deletable_file_service_keys.update( ( petition_resolve_permission_file_service_keys & current ) - petitioned ) # we can modify users when we have permission and the file is current or deleted modifyable_file_service_keys.update( user_manage_permission_file_service_keys & ( current | deleted ) ) # IPFS # we can pin if a file is local, not current, not pending if locations_manager.IsLocal(): pinnable_ipfs_service_keys.update( ipfs_service_keys - current - pending ) # we can unpin a file if it is current and not petitioned unpinnable_ipfs_service_keys.update( ( ipfs_service_keys & current ) - petitioned ) # do the actual menu selection_info_menu = QW.QMenu( menu ) if multiple_selected: ( num_files_descriptor, selected_files_descriptor ) = self._GetSortedSelectedMimeDescriptors() selection_info_menu_label = '{} {}, {}'.format( HydrusData.ToHumanInt( num_selected ), selected_files_descriptor, self._GetPrettyTotalSize( only_selected = True ) ) pretty_total_duration = self._GetPrettyTotalDuration( only_selected = True ) if pretty_total_duration != '': selection_info_menu_label += ', {}'.format( pretty_total_duration ) else: pretty_info_lines = list( focus_singleton.GetPrettyInfoLines() ) top_line = pretty_info_lines.pop( 0 ) selection_info_menu_label = top_line ClientGUIMediaMenus.AddPrettyInfoLines( selection_info_menu, pretty_info_lines ) ClientGUIMediaMenus.AddFileViewingStatsMenu( selection_info_menu, self._selected_media ) if len( disparate_current_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_current_file_service_keys, 'some uploaded to' ) if multiple_selected and len( common_current_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_current_file_service_keys, 'selected uploaded to' ) if len( disparate_pending_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_pending_file_service_keys, 'some pending to' ) if len( common_pending_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_pending_file_service_keys, 'pending to' ) if len( disparate_petitioned_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_petitioned_file_service_keys, 'some petitioned for removal from' ) if len( common_petitioned_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_petitioned_file_service_keys, 'petitioned for removal from' ) if len( disparate_deleted_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_deleted_file_service_keys, 'some deleted from' ) if len( common_deleted_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_deleted_file_service_keys, 'deleted from' ) if len( disparate_current_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_current_ipfs_service_keys, 'some pinned to' ) if multiple_selected and len( common_current_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_current_ipfs_service_keys, 'selected pinned to' ) if len( disparate_pending_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_pending_ipfs_service_keys, 'some to be pinned to' ) if len( common_pending_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_pending_ipfs_service_keys, 'to be pinned to' ) if len( disparate_petitioned_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, disparate_petitioned_ipfs_service_keys, 'some to be unpinned from' ) if len( common_petitioned_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeyLabelsToMenu( selection_info_menu, common_petitioned_ipfs_service_keys, unpin_phrase ) if len( selection_info_menu.actions() ) == 0: selection_info_menu.deleteLater() ClientGUIMenus.AppendMenuLabel( menu, selection_info_menu_label ) else: ClientGUIMenus.AppendMenu( menu, selection_info_menu, selection_info_menu_label ) ClientGUIMenus.AppendSeparator( menu ) ClientGUIMenus.AppendMenuItem( menu, 'refresh', 'Refresh the current search.', self.refreshQuery.emit ) if len( self._sorted_media ) > 0: ClientGUIMenus.AppendSeparator( menu ) filter_counts = {} filter_counts[ ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ALL ) ] = num_files filter_counts[ ClientMedia.FileFilter( ClientMedia.FILE_FILTER_INBOX ) ] = num_inbox filter_counts[ ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ARCHIVE ) ] = num_archive filter_counts[ ClientMedia.FileFilter( ClientMedia.FILE_FILTER_SELECTED ) ] = num_selected has_local_and_remote = has_local and has_remote AddSelectMenu( self, menu, filter_counts, all_specific_file_domains, has_local_and_remote ) AddRemoveMenu( self, menu, filter_counts, all_specific_file_domains, has_local_and_remote ) ClientGUIMenus.AppendSeparator( menu ) if has_local: ClientGUIMenus.AppendMenuItem( menu, 'archive/delete filter', 'Launch a special media viewer that will quickly archive (left-click) and delete (right-click) the selected media.', self._ArchiveDeleteFilter ) if self._HasFocusSingleton(): focus_singleton = self._GetFocusSingleton() if selection_has_inbox: ClientGUIMenus.AppendMenuItem( menu, archive_phrase, 'Archive the selected files.', self._Archive ) if selection_has_archive: ClientGUIMenus.AppendMenuItem( menu, inbox_phrase, 'Put the selected files back in the inbox.', self._Inbox ) ClientGUIMenus.AppendSeparator( menu ) local_file_service_keys_we_are_in = sorted( current_file_service_keys.intersection( local_media_file_service_keys ), key = HG.client_controller.services_manager.GetName ) for file_service_key in local_file_service_keys_we_are_in: ClientGUIMenus.AppendMenuItem( menu, '{} from {}'.format( local_delete_phrase, HG.client_controller.services_manager.GetName( file_service_key ) ), 'Delete the selected files.', self._Delete, file_service_key ) if selection_has_trash: if selection_has_local_file_domain: ClientGUIMenus.AppendMenuItem( menu, 'delete trash physically now', 'Completely delete the selected trashed files, forcing an immediate physical delete from your hard drive.', self._Delete, CC.COMBINED_LOCAL_FILE_SERVICE_KEY, only_those_in_file_service_key = CC.TRASH_SERVICE_KEY ) ClientGUIMenus.AppendMenuItem( menu, delete_physically_phrase, 'Completely delete the selected files, forcing an immediate physical delete from your hard drive.', self._Delete, CC.COMBINED_LOCAL_FILE_SERVICE_KEY ) ClientGUIMenus.AppendMenuItem( menu, undelete_phrase, 'Restore the selected files back to \'my files\'.', self._Undelete ) if selection_has_deletion_record: ClientGUIMenus.AppendMenuItem( menu, clear_deletion_phrase, 'Clear the deletion record for these files, allowing them to reimport even if previously deleted files are set to be discarded.', self._ClearDeleteRecord ) # ClientGUIMenus.AppendSeparator( menu ) manage_menu = QW.QMenu( menu ) ClientGUIMenus.AppendMenuItem( manage_menu, 'tags', 'Manage tags for the selected files.', self._ManageTags ) if i_can_post_ratings: ClientGUIMenus.AppendMenuItem( manage_menu, 'ratings', 'Manage ratings for the selected files.', self._ManageRatings ) ClientGUIMenus.AppendMenuItem( manage_menu, 'urls', 'Manage urls for the selected files.', self._ManageURLs ) num_notes = focus_singleton.GetNotesManager().GetNumNotes() notes_str = 'notes' if num_notes > 0: notes_str = '{} ({})'.format( notes_str, HydrusData.ToHumanInt( num_notes ) ) ClientGUIMenus.AppendMenuItem( manage_menu, notes_str, 'Manage notes for the focused file.', self._ManageNotes ) ClientGUIMediaMenus.AddDuplicatesMenu( self, manage_menu, self._location_context, focus_singleton, num_selected, collections_selected ) ClientGUIMediaMenus.AddManageFileViewingStatsMenu( self, manage_menu, flat_selected_medias ) regen_menu = QW.QMenu( manage_menu ) ClientGUIMenus.AppendMenuItem( regen_menu, 'thumbnails, but only if wrong size', 'Regenerate the selected files\' thumbnails, but only if they are the wrong size.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_REFIT_THUMBNAIL ) ClientGUIMenus.AppendMenuItem( regen_menu, 'thumbnails', 'Regenerate the selected files\'s thumbnails.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_FORCE_THUMBNAIL ) ClientGUIMenus.AppendMenuItem( regen_menu, 'file metadata', 'Regenerated the selected files\' metadata and thumbnails.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_FILE_METADATA ) ClientGUIMenus.AppendMenuItem( regen_menu, 'similar files data', 'Regenerated the selected files\' perceptual hashes.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_SIMILAR_FILES_METADATA ) ClientGUIMenus.AppendMenuItem( regen_menu, 'duplicate pixel data', 'Regenerated the selected files\' pixel hashes.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_PIXEL_HASH ) ClientGUIMenus.AppendMenuItem( regen_menu, 'full presence check', 'Check file presence and try to fix.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_FILE_INTEGRITY_PRESENCE_TRY_URL_ELSE_REMOVE_RECORD ) ClientGUIMenus.AppendMenuItem( regen_menu, 'full integrity check', 'Check file integrity and try to fix.', self._RegenerateFileData, ClientFiles.REGENERATE_FILE_DATA_JOB_FILE_INTEGRITY_DATA_TRY_URL_ELSE_REMOVE_RECORD ) ClientGUIMenus.AppendMenu( manage_menu, regen_menu, 'regenerate' ) ClientGUIMenus.AppendMenu( menu, manage_menu, 'manage' ) ( local_duplicable_to_file_service_keys, local_moveable_from_and_to_file_service_keys ) = ClientGUIMediaActions.GetLocalFileActionServiceKeys( flat_selected_medias ) len_interesting_local_service_keys = 0 len_interesting_local_service_keys += len( local_duplicable_to_file_service_keys ) len_interesting_local_service_keys += len( local_moveable_from_and_to_file_service_keys ) # len_interesting_remote_service_keys = 0 len_interesting_remote_service_keys += len( downloadable_file_service_keys ) len_interesting_remote_service_keys += len( uploadable_file_service_keys ) len_interesting_remote_service_keys += len( pending_file_service_keys ) len_interesting_remote_service_keys += len( petitionable_file_service_keys ) len_interesting_remote_service_keys += len( petitioned_file_service_keys ) len_interesting_remote_service_keys += len( deletable_file_service_keys ) len_interesting_remote_service_keys += len( modifyable_file_service_keys ) len_interesting_remote_service_keys += len( pinnable_ipfs_service_keys ) len_interesting_remote_service_keys += len( pending_ipfs_service_keys ) len_interesting_remote_service_keys += len( unpinnable_ipfs_service_keys ) len_interesting_remote_service_keys += len( petitioned_ipfs_service_keys ) if multiple_selected: len_interesting_remote_service_keys += len( ipfs_service_keys ) if len_interesting_local_service_keys > 0 and len_interesting_remote_service_keys > 0: files_parent_menu = QW.QMenu( menu ) ClientGUIMenus.AppendMenu( menu, files_parent_menu, 'files' ) else: files_parent_menu = menu if len_interesting_local_service_keys > 0: ClientGUIMediaMenus.AddLocalFilesMoveAddToMenu( self, files_parent_menu, local_duplicable_to_file_service_keys, local_moveable_from_and_to_file_service_keys, multiple_selected, self.ProcessApplicationCommand ) if len_interesting_remote_service_keys > 0: remote_action_menu = QW.QMenu( files_parent_menu ) if len( downloadable_file_service_keys ) > 0: ClientGUIMenus.AppendMenuItem( remote_action_menu, download_phrase, 'Download all possible selected files.', self._DownloadSelected ) if some_downloading: ClientGUIMenus.AppendMenuItem( remote_action_menu, rescind_download_phrase, 'Stop downloading any of the selected files.', self._RescindDownloadSelected ) if len( uploadable_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, uploadable_file_service_keys, upload_phrase, 'Upload all selected files to the file repository.', self._UploadFiles ) if len( pending_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, pending_file_service_keys, rescind_upload_phrase, 'Rescind the pending upload to the file repository.', self._RescindUploadFiles ) if len( petitionable_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, petitionable_file_service_keys, petition_phrase, 'Petition these files for deletion from the file repository.', self._PetitionFiles ) if len( petitioned_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, petitioned_file_service_keys, rescind_petition_phrase, 'Rescind the petition to delete these files from the file repository.', self._RescindPetitionFiles ) if len( deletable_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, deletable_file_service_keys, remote_delete_phrase, 'Delete these files from the file repository.', self._Delete ) if len( modifyable_file_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, modifyable_file_service_keys, modify_account_phrase, 'Modify the account(s) that uploaded these files to the file repository.', self._ModifyUploaders ) if len( pinnable_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, pinnable_ipfs_service_keys, pin_phrase, 'Pin these files to the ipfs service.', self._UploadFiles ) if len( pending_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, pending_ipfs_service_keys, rescind_pin_phrase, 'Rescind the pending pin to the ipfs service.', self._RescindUploadFiles ) if len( unpinnable_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, unpinnable_ipfs_service_keys, unpin_phrase, 'Unpin these files from the ipfs service.', self._PetitionFiles ) if len( petitioned_ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, petitioned_ipfs_service_keys, rescind_unpin_phrase, 'Rescind the pending unpin from the ipfs service.', self._RescindPetitionFiles ) if multiple_selected and len( ipfs_service_keys ) > 0: ClientGUIMediaMenus.AddServiceKeysToMenu( remote_action_menu, ipfs_service_keys, 'pin new directory to', 'Pin these files as a directory to the ipfs service.', self._UploadDirectory ) ClientGUIMenus.AppendMenu( files_parent_menu, remote_action_menu, 'remote services' ) # ClientGUIMediaMenus.AddKnownURLsViewCopyMenu( self, menu, self._focused_media, selected_media = self._selected_media ) # open_menu = QW.QMenu( menu ) ClientGUIMenus.AppendMenuItem( open_menu, 'in a new page', 'Copy your current selection into a simple new page.', self._ShowSelectionInNewPage ) if self._focused_media.HasImages(): similar_menu = QW.QMenu( open_menu ) ClientGUIMenus.AppendMenuItem( similar_menu, 'exact match', 'Search the database for files that look precisely like those selected.', self._GetSimilarTo, CC.HAMMING_EXACT_MATCH ) ClientGUIMenus.AppendMenuItem( similar_menu, 'very similar', 'Search the database for files that look just like those selected.', self._GetSimilarTo, CC.HAMMING_VERY_SIMILAR ) ClientGUIMenus.AppendMenuItem( similar_menu, 'similar', 'Search the database for files that look generally like those selected.', self._GetSimilarTo, CC.HAMMING_SIMILAR ) ClientGUIMenus.AppendMenuItem( similar_menu, 'speculative', 'Search the database for files that probably look like those selected. This is sometimes useful for symbols with sharp edges or lines.', self._GetSimilarTo, CC.HAMMING_SPECULATIVE ) ClientGUIMenus.AppendMenu( open_menu, similar_menu, 'similar-looking files' ) ClientGUIMenus.AppendSeparator( open_menu ) ClientGUIMenus.AppendMenuItem( open_menu, 'in external program', 'Launch this file with your OS\'s default program for it.', self._OpenExternally ) ClientGUIMenus.AppendMenuItem( open_menu, 'in web browser', 'Show this file in your OS\'s web browser.', self._OpenFileInWebBrowser ) if focused_is_local: show_open_in_explorer = advanced_mode and ( HC.PLATFORM_WINDOWS or HC.PLATFORM_MACOS ) if show_open_in_explorer: ClientGUIMenus.AppendMenuItem( open_menu, 'in file browser', 'Show this file in your OS\'s file browser.', self._OpenFileLocation ) ClientGUIMenus.AppendMenu( menu, open_menu, 'open' ) # share share_menu = QW.QMenu( menu ) # copy_menu = QW.QMenu( share_menu ) if selection_has_local: ClientGUIMenus.AppendMenuItem( copy_menu, copy_phrase, 'Copy the selected files to the clipboard.', self._CopyFilesToClipboard ) copy_hash_menu = QW.QMenu( copy_menu ) if self._HasFocusSingleton(): focus_singleton = self._GetFocusSingleton() ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha256 ({})'.format( focus_singleton.GetHash().hex() ), 'Copy the selected file\'s SHA256 hash to the clipboard.', self._CopyHashToClipboard, 'sha256' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'md5', 'Copy the selected file\'s MD5 hash to the clipboard.', self._CopyHashToClipboard, 'md5' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha1', 'Copy the selected file\'s SHA1 hash to the clipboard.', self._CopyHashToClipboard, 'sha1' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha512', 'Copy the selected file\'s SHA512 hash to the clipboard.', self._CopyHashToClipboard, 'sha512' ) ClientGUIMenus.AppendMenu( copy_menu, copy_hash_menu, 'hash' ) if multiple_selected: copy_hash_menu = QW.QMenu( copy_menu ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha256 (hydrus default)', 'Copy the selected files\' SHA256 hashes to the clipboard.', self._CopyHashesToClipboard, 'sha256' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'md5', 'Copy the selected files\' MD5 hashes to the clipboard.', self._CopyHashesToClipboard, 'md5' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha1', 'Copy the selected files\' SHA1 hashes to the clipboard.', self._CopyHashesToClipboard, 'sha1' ) ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha512', 'Copy the selected files\' SHA512 hashes to the clipboard.', self._CopyHashesToClipboard, 'sha512' ) ClientGUIMenus.AppendMenu( copy_menu, copy_hash_menu, 'hashes' ) else: ClientGUIMenus.AppendMenuItem( copy_menu, 'sha256 hash', 'Copy the selected file\'s SHA256 hash to the clipboard.', self._CopyHashToClipboard, 'sha256' ) if multiple_selected: ClientGUIMenus.AppendMenuItem( copy_menu, 'sha256 hashes', 'Copy the selected files\' SHA256 hash to the clipboard.', self._CopyHashesToClipboard, 'sha256' ) if advanced_mode: hash_id_str = str( focus_singleton.GetHashId() ) ClientGUIMenus.AppendMenuItem( copy_menu, 'file_id ({})'.format( hash_id_str ), 'Copy this file\'s internal file/hash_id.', HG.client_controller.pub, 'clipboard', 'text', hash_id_str ) for ipfs_service_key in self._focused_media.GetLocationsManager().GetCurrent().intersection( ipfs_service_keys ): name = service_keys_to_names[ ipfs_service_key ] ClientGUIMenus.AppendMenuItem( copy_menu, name + ' multihash', 'Copy the selected file\'s multihash to the clipboard.', self._CopyServiceFilenameToClipboard, ipfs_service_key ) if multiple_selected: for ipfs_service_key in disparate_current_ipfs_service_keys.union( common_current_ipfs_service_keys ): name = service_keys_to_names[ ipfs_service_key ] ClientGUIMenus.AppendMenuItem( copy_menu, name + ' multihashes', 'Copy the selected files\' multihashes to the clipboard.', self._CopyServiceFilenamesToClipboard, ipfs_service_key ) if focused_is_local: if self._focused_media.GetMime() in HC.IMAGES: ClientGUIMenus.AppendMenuItem( copy_menu, 'image (bitmap)', 'Copy the selected file\'s image data to the clipboard (as a bmp).', self._CopyBMPToClipboard ) ClientGUIMenus.AppendMenuItem( copy_menu, 'path', 'Copy the selected file\'s path to the clipboard.', self._CopyPathToClipboard ) if multiple_selected and selection_has_local: ClientGUIMenus.AppendMenuItem( copy_menu, 'paths', 'Copy the selected files\' paths to the clipboard.', self._CopyPathsToClipboard ) ClientGUIMenus.AppendMenu( share_menu, copy_menu, 'copy' ) # export_menu = QW.QMenu( share_menu ) ClientGUIMenus.AppendMenuItem( export_menu, export_phrase, 'Export the selected files to an external folder.', self._ExportFiles ) ClientGUIMenus.AppendMenu( share_menu, export_menu, 'export' ) # if local_booru_is_running: ClientGUIMenus.AppendMenuItem( share_menu, 'on local booru', 'Share the selected files on your client\'s local booru.', self._ShareOnLocalBooru ) # ClientGUIMenus.AppendMenu( menu, share_menu, 'share' ) if not do_not_show_just_return: CGC.core().PopupMenu( self, menu ) else: return menu def Sort( self, media_sort = None ): MediaPanel.Sort( self, media_sort ) self._DirtyAllPages() self.widget().update() def ThumbnailsReset( self ): ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() thumbnail_scroll_rate = float( HG.client_controller.new_options.GetString( 'thumbnail_scroll_rate' ) ) self.verticalScrollBar().setSingleStep( int( round( thumbnail_span_height * thumbnail_scroll_rate ) ) ) self._thumbnails_being_faded_in = {} self._hashes_faded = set() self._ReinitialisePageCacheIfNeeded() self._RecalculateVirtualSize() self.RedrawAllThumbnails() def TIMERAnimationUpdate( self ): FRAME_DURATION = 1.0 / 60 NUM_FRAMES_TO_FILL_IN = 15 loop_started = HydrusData.GetNowPrecise() loop_should_break_time = loop_started + ( FRAME_DURATION / 2 ) ( thumbnail_span_width, thumbnail_span_height ) = self._GetThumbnailSpanDimensions() thumbnail_margin = HG.client_controller.new_options.GetInteger( 'thumbnail_margin' ) hashes = list( self._thumbnails_being_faded_in.keys() ) random.shuffle( hashes ) dcs = {} y_start = self._GetYStart() earliest_y = y_start page_height = self._num_rows_per_canvas_page * thumbnail_span_height for hash in hashes: ( original_bmp, alpha_bmp, thumbnail_index, thumbnail, animation_started, num_frames_rendered ) = self._thumbnails_being_faded_in[ hash ] num_frames_supposed_to_be_rendered = int( ( loop_started - animation_started ) / FRAME_DURATION ) num_frames_to_render = num_frames_supposed_to_be_rendered - num_frames_rendered if num_frames_to_render == 0: continue delete_entry = False try: expected_thumbnail = self._sorted_media[ thumbnail_index ] except: expected_thumbnail = None page_index = self._GetPageIndexFromThumbnailIndex( thumbnail_index ) if expected_thumbnail != thumbnail: delete_entry = True elif page_index not in self._clean_canvas_pages: delete_entry = True else: times_to_draw = 1 if num_frames_supposed_to_be_rendered >= NUM_FRAMES_TO_FILL_IN: bmp_to_use = original_bmp delete_entry = True else: times_to_draw = num_frames_to_render bmp_to_use = alpha_bmp num_frames_rendered += times_to_draw self._thumbnails_being_faded_in[ hash ] = ( original_bmp, alpha_bmp, thumbnail_index, thumbnail, animation_started, num_frames_rendered ) thumbnail_col = thumbnail_index % self._num_columns thumbnail_row = thumbnail_index // self._num_columns x = thumbnail_col * thumbnail_span_width + thumbnail_margin y = ( thumbnail_row - ( page_index * self._num_rows_per_canvas_page ) ) * thumbnail_span_height + thumbnail_margin if page_index not in dcs: canvas_page = self._clean_canvas_pages[ page_index ] painter = QG.QPainter( canvas_page ) dcs[ page_index ] = painter painter = dcs[ page_index ] for i in range( times_to_draw ): painter.drawImage( x, y, bmp_to_use ) # page_virtual_y = page_height * page_index self.widget().update( QC.QRect( x, page_virtual_y + y, thumbnail_span_width - thumbnail_margin, thumbnail_span_height - thumbnail_margin ) ) if delete_entry: del self._thumbnails_being_faded_in[ hash ] if HydrusData.TimeHasPassedPrecise( loop_should_break_time ): break if len( self._thumbnails_being_faded_in ) == 0: HG.client_controller.gui.UnregisterAnimationUpdateWindow( self ) def WaterfallThumbnails( self, page_key, thumbnails ): if self._page_key == page_key: self._FadeThumbnails( thumbnails ) def AddRemoveMenu( win: MediaPanel, menu, filter_counts, all_specific_file_domains, has_local_and_remote ): file_filter_all = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ALL ) if file_filter_all.GetCount( win, filter_counts ) > 0: remove_menu = QW.QMenu( menu ) # file_filter_selected = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_SELECTED ) file_filter_inbox = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_INBOX ) file_filter_archive = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ARCHIVE ) file_filter_not_selected = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NOT_SELECTED ) # selected_count = file_filter_selected.GetCount( win, filter_counts ) if 0 < selected_count < file_filter_all.GetCount( win, filter_counts ): ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_selected.ToString( win, filter_counts ), 'Remove all the selected files from the current view.', win._Remove, file_filter_selected ) if file_filter_all.GetCount( win, filter_counts ) > 0: ClientGUIMenus.AppendSeparator( remove_menu ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_all.ToString( win, filter_counts ), 'Remove all the files from the current view.', win._Remove, file_filter_all ) if file_filter_inbox.GetCount( win, filter_counts ) > 0 and file_filter_archive.GetCount( win, filter_counts ) > 0: ClientGUIMenus.AppendSeparator( remove_menu ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_inbox.ToString( win, filter_counts ), 'Remove all the inbox files from the current view.', win._Remove, file_filter_inbox ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_archive.ToString( win, filter_counts ), 'Remove all the archived files from the current view.', win._Remove, file_filter_archive ) if len( all_specific_file_domains ) > 1: ClientGUIMenus.AppendSeparator( remove_menu ) all_specific_file_domains = ClientLocation.SortFileServiceKeysNicely( all_specific_file_domains ) all_specific_file_domains = ClientLocation.FilterOutRedundantMetaServices( all_specific_file_domains ) for file_service_key in all_specific_file_domains: file_filter = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_FILE_SERVICE, file_service_key ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter.ToString( win, filter_counts ), 'Remove all the files that are in this file domain.', win._Remove, file_filter ) if has_local_and_remote: file_filter_local = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_LOCAL ) file_filter_remote = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_REMOTE ) ClientGUIMenus.AppendSeparator( remove_menu ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_local.ToString( win, filter_counts ), 'Remove all the files that are in this client.', win._Remove, file_filter_local ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_remote.ToString( win, filter_counts ), 'Remove all the files that are not in this client.', win._Remove, file_filter_remote ) not_selected_count = file_filter_not_selected.GetCount( win, filter_counts ) if not_selected_count > 0 and selected_count > 0: ClientGUIMenus.AppendSeparator( remove_menu ) ClientGUIMenus.AppendMenuItem( remove_menu, file_filter_not_selected.ToString( win, filter_counts ), 'Remove all the not selected files from the current view.', win._Remove, file_filter_not_selected ) ClientGUIMenus.AppendMenu( menu, remove_menu, 'remove' ) def AddSelectMenu( win: MediaPanel, menu, filter_counts, all_specific_file_domains, has_local_and_remote ): file_filter_all = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ALL ) if file_filter_all.GetCount( win, filter_counts ) > 0: select_menu = QW.QMenu( menu ) # file_filter_inbox = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_INBOX ) file_filter_archive = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_ARCHIVE ) file_filter_not_selected = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NOT_SELECTED ) file_filter_none = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_NONE ) # if file_filter_all.GetCount( win, filter_counts ) > 0: ClientGUIMenus.AppendSeparator( select_menu ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_all.ToString( win, filter_counts ), 'Select all the files in the current view.', win._Select, file_filter_all ) if file_filter_inbox.GetCount( win, filter_counts ) > 0 and file_filter_archive.GetCount( win, filter_counts ) > 0: ClientGUIMenus.AppendSeparator( select_menu ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_inbox.ToString( win, filter_counts ), 'Select all the inbox files in the current view.', win._Select, file_filter_inbox ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_archive.ToString( win, filter_counts ), 'Select all the archived files in the current view.', win._Select, file_filter_archive ) if len( all_specific_file_domains ) > 1: ClientGUIMenus.AppendSeparator( select_menu ) all_specific_file_domains = ClientLocation.SortFileServiceKeysNicely( all_specific_file_domains ) all_specific_file_domains = ClientLocation.FilterOutRedundantMetaServices( all_specific_file_domains ) for file_service_key in all_specific_file_domains: file_filter = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_FILE_SERVICE, file_service_key ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter.ToString( win, filter_counts ), 'Select all the files in this file domain.', win._Select, file_filter ) if has_local_and_remote: file_filter_local = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_LOCAL ) file_filter_remote = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_REMOTE ) ClientGUIMenus.AppendSeparator( select_menu ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_local.ToString( win, filter_counts ), 'Remove all the files that are in this client.', win._Select, file_filter_local ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_remote.ToString( win, filter_counts ), 'Remove all the files that are not in this client.', win._Select, file_filter_remote ) file_filter_selected = ClientMedia.FileFilter( ClientMedia.FILE_FILTER_SELECTED ) selected_count = file_filter_selected.GetCount( win, filter_counts ) not_selected_count = file_filter_not_selected.GetCount( win, filter_counts ) if selected_count > 0: if not_selected_count > 0: ClientGUIMenus.AppendSeparator( select_menu ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_not_selected.ToString( win, filter_counts ), 'Swap what is and is not selected.', win._Select, file_filter_not_selected ) ClientGUIMenus.AppendSeparator( select_menu ) ClientGUIMenus.AppendMenuItem( select_menu, file_filter_none.ToString( win, filter_counts ), 'Deselect everything selected.', win._Select, file_filter_none ) ClientGUIMenus.AppendMenu( menu, select_menu, 'select' ) class Selectable( object ): def __init__( self ): self._selected = False def Deselect( self ): self._selected = False def IsSelected( self ): return self._selected def Select( self ): self._selected = True class Thumbnail( Selectable ): def __init__( self ): Selectable.__init__( self ) self._last_tags = None self._last_upper_summary = None self._last_lower_summary = None def ClearTagSummaryCaches( self ): self._last_tags = None self._last_upper_summary = None self._last_lower_summary = None def GetQtImage( self, device_pixel_ratio ) -> QG.QImage: # we probably don't really want to say DPR as a param here, but instead ask for a qt_image in a certain resolution? # or just give the qt_image to be drawn to? # or just give a painter and a rect and draw to that or something # we don't really want to mess around with DPR here, we just want to draw thumbs # that said, this works after a medium-high headache getting it there, so let's not get ahead of ourselves thumbnail_hydrus_bmp = HG.client_controller.GetCache( 'thumbnail' ).GetThumbnail( self ) thumbnail_border = HG.client_controller.new_options.GetInteger( 'thumbnail_border' ) ( width, height ) = ClientData.AddPaddingToDimensions( HC.options[ 'thumbnail_dimensions' ], thumbnail_border * 2 ) qt_image_width = int( width * device_pixel_ratio ) qt_image_height = int( height * device_pixel_ratio ) qt_image = HG.client_controller.bitmap_manager.GetQtImage( qt_image_width, qt_image_height, 24 ) qt_image.setDevicePixelRatio( device_pixel_ratio ) inbox = self.HasInbox() local = self.GetLocationsManager().IsLocal() # # BAD FONT QUALITY AT 100% UI Scale (semi fixed now, look at the bottom) # # Ok I have spent hours on this now trying to figure it out and can't, so I'll just write about it for when I come back # So, if you boot with two monitors at 100% UI scale, the text here on a QImage is ugly, but on QWidget it is fine # If you boot with one monitor at 125%, the text is beautiful on QImage both screens # My current assumption is booting Qt with unusual UI scales triggers some extra init and that spills over to QImage QPainter initialisation # # I checked painter hints, font stuff, fontinfo and fontmetrics, and the only difference was with fontmetrics, on all-100% vs one >100%: # minLeftBearing: -1, -7 # minRightBearing: -1, -8 # xHeight: 3, 6 # # The fontmetric produced a text size one pixel less wide on the both-100% run, so it is calculating different # However these differences are global to the program so don't explain why painting on a QImage specifically has bad font rather than QWidget # The ugly font is anti-aliased, but it looks like not drawn with sub-pixel calculations, like ClearType isn't kicking in or something # If I blow the font size up to 72, there is still a difference in screenshots between the all-100% and some >100% boot. # So, maybe if the program boots with any weird UI scale going on, Qt kicks in a different renderer for all QImages, the same renderer for QWidgets, perhaps more expensively # Or this is just some weird bug # Or I am still missing some flag # # bit like this https://stackoverflow.com/questions/31043332/qt-antialiasing-of-vertical-text-rendered-using-qpainter # # EDIT: OK, I 'fixed' it with setStyleStrategy( preferantialias ), which has no change in 125%, but in all-100% it draws something different but overall better quality # Note you can't setStyleStrategy on the font when it is in the QPainter. either it gets set read only or there is some other voodoo going on # It does look very slightly weird, but it is a step up so I won't complain. it really seems like the isolated QPainter of only-100% world has some different initialisation. it just can't find the nice font renderer # # EDIT 2: I think it may only look weird when the thumb banner has opacity. Maybe I need to learn about CompositionModes # # EDIT 3: Appalently Qt 6.4.0 may fix the basic 100% UI scale QImage init bug! # # UPDATE 3a: Qt 6.4.x did not magically fix it. It draws much nicer, but still a different font weight/metrics compared to media viewer background, say. # The PreferAntialias flag on 6.4.x seems to draw very very close to our ideal, so let's be happy with it for now. painter = QG.QPainter( qt_image ) painter.setRenderHint( QG.QPainter.TextAntialiasing, True ) # is true already in tests, is supposed to be 'the way' to fix the ugly text issue painter.setRenderHint( QG.QPainter.Antialiasing, True ) # seems to do nothing, it only affects primitives? painter.setRenderHint( QG.QPainter.SmoothPixmapTransform, True ) # makes the thumb QImage scale up and down prettily when we need it, either because it is too small or DPR gubbins new_options = HG.client_controller.new_options if not local: if self._selected: background_colour_type = CC.COLOUR_THUMB_BACKGROUND_REMOTE_SELECTED else: background_colour_type = CC.COLOUR_THUMB_BACKGROUND_REMOTE else: if self._selected: background_colour_type = CC.COLOUR_THUMB_BACKGROUND_SELECTED else: background_colour_type = CC.COLOUR_THUMB_BACKGROUND # the painter isn't getting QSS style from the qt_image, we need to set the font explitly to get font size changes from QSS etc.. f = QG.QFont( HG.client_controller.gui.font() ) # this line magically fixes the bad text, as above f.setStyleStrategy( QG.QFont.PreferAntialias ) painter.setFont( f ) painter.fillRect( thumbnail_border, thumbnail_border, width - ( thumbnail_border * 2 ), height - ( thumbnail_border * 2 ), new_options.GetColour( background_colour_type ) ) raw_thumbnail_qt_image = thumbnail_hydrus_bmp.GetQtImage() thumbnail_dpr_percent = HG.client_controller.new_options.GetInteger( 'thumbnail_dpr_percent' ) if thumbnail_dpr_percent != 100: thumbnail_dpr = thumbnail_dpr_percent / 100 raw_thumbnail_qt_image.setDevicePixelRatio( thumbnail_dpr ) # qt_image.deviceIndepedentSize isn't supported in Qt5 lmao device_independent_thumb_size = raw_thumbnail_qt_image.size() / thumbnail_dpr else: device_independent_thumb_size = raw_thumbnail_qt_image.size() x_offset = ( width - device_independent_thumb_size.width() ) // 2 y_offset = ( height - device_independent_thumb_size.height() ) // 2 painter.drawImage( x_offset, y_offset, raw_thumbnail_qt_image ) TEXT_BORDER = 1 new_options = HG.client_controller.new_options tags = self.GetTagsManager().GetCurrentAndPending( CC.COMBINED_TAG_SERVICE_KEY, ClientTags.TAG_DISPLAY_SINGLE_MEDIA ) if len( tags ) > 0: upper_tag_summary_generator = new_options.GetTagSummaryGenerator( 'thumbnail_top' ) lower_tag_summary_generator = new_options.GetTagSummaryGenerator( 'thumbnail_bottom_right' ) if self._last_tags is not None and self._last_tags == tags: upper_summary = self._last_upper_summary lower_summary = self._last_lower_summary else: upper_summary = upper_tag_summary_generator.GenerateSummary( tags ) lower_summary = lower_tag_summary_generator.GenerateSummary( tags ) self._last_tags = set( tags ) self._last_upper_summary = upper_summary self._last_lower_summary = lower_summary if len( upper_summary ) > 0 or len( lower_summary ) > 0: if len( upper_summary ) > 0: text_colour_with_alpha = upper_tag_summary_generator.GetTextColour() background_colour_with_alpha = upper_tag_summary_generator.GetBackgroundColour() ( text_size, upper_summary ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, upper_summary ) box_x = thumbnail_border box_y = thumbnail_border box_width = width - ( thumbnail_border * 2 ) box_height = text_size.height() + 2 painter.fillRect( box_x, box_y, box_width, box_height, background_colour_with_alpha ) text_x = ( width - text_size.width() ) // 2 text_y = box_y + TEXT_BORDER painter.setPen( QG.QPen( text_colour_with_alpha ) ) ClientGUIFunctions.DrawText( painter, text_x, text_y, upper_summary ) if len( lower_summary ) > 0: text_colour_with_alpha = lower_tag_summary_generator.GetTextColour() background_colour_with_alpha = lower_tag_summary_generator.GetBackgroundColour() ( text_size, lower_summary ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, lower_summary ) text_width = text_size.width() text_height = text_size.height() box_width = text_width + ( TEXT_BORDER * 2 ) box_height = text_height + ( TEXT_BORDER * 2 ) box_x = width - box_width - thumbnail_border box_y = height - text_height - thumbnail_border painter.fillRect( box_x, box_y, box_width, box_height, background_colour_with_alpha ) text_x = box_x + TEXT_BORDER text_y = box_y + TEXT_BORDER painter.setPen( QG.QPen( text_colour_with_alpha ) ) ClientGUIFunctions.DrawText( painter, text_x, text_y, lower_summary ) if thumbnail_border > 0: if not local: if self._selected: border_colour_type = CC.COLOUR_THUMB_BORDER_REMOTE_SELECTED else: border_colour_type = CC.COLOUR_THUMB_BORDER_REMOTE else: if self._selected: border_colour_type = CC.COLOUR_THUMB_BORDER_SELECTED else: border_colour_type = CC.COLOUR_THUMB_BORDER # I had a hell of a time getting a transparent box to draw right with a pen border without crazy +1px in the params for reasons I did not understand # so I just decided four rects is neater and fine and actually prob faster in some cases # _____ ______ _____ ______ ________________ # ___________(_)___ _________ /_______ _______ ______ __ /______ ___ /_________ /__ /__ / # ___ __ \_ /__ |/_/ _ \_ /__ ___/ __ __ `/ __ \ _ __/ __ \ __ __ \ _ \_ /__ /__ / # __ /_/ / / __> < / __/ / _(__ ) _ /_/ // /_/ / / /_ / /_/ / _ / / / __/ / _ / /_/ # _ .___//_/ /_/|_| \___//_/ /____/ _\__, / \____/ \__/ \____/ /_/ /_/\___//_/ /_/ (_) # /_/ /____/ painter.setBrush( QG.QBrush( new_options.GetColour( border_colour_type ) ) ) painter.setPen( QG.QPen( QC.Qt.NoPen ) ) rectangles = [] side_height = height - ( thumbnail_border * 2 ) rectangles.append( QC.QRectF( 0, 0, width, thumbnail_border ) ) # top rectangles.append( QC.QRectF( 0, height - thumbnail_border, width, thumbnail_border ) ) # bottom rectangles.append( QC.QRectF( 0, thumbnail_border, thumbnail_border, side_height ) ) # left rectangles.append( QC.QRectF( width - thumbnail_border, thumbnail_border, thumbnail_border, side_height ) ) # right painter.drawRects( rectangles ) ICON_MARGIN = 1 locations_manager = self.GetLocationsManager() icons_to_draw = [] if locations_manager.IsDownloading(): icons_to_draw.append( CC.global_pixmaps().downloading ) if self.HasNotes(): icons_to_draw.append( CC.global_pixmaps().notes ) if locations_manager.IsTrashed() or CC.COMBINED_LOCAL_FILE_SERVICE_KEY in locations_manager.GetDeleted(): icons_to_draw.append( CC.global_pixmaps().trash ) if inbox: icons_to_draw.append( CC.global_pixmaps().inbox ) if len( icons_to_draw ) > 0: icon_x = - ( thumbnail_border + ICON_MARGIN ) for icon in icons_to_draw: icon_x -= icon.width() painter.drawPixmap( width + icon_x, thumbnail_border, icon ) icon_x -= 2 * ICON_MARGIN if self.IsCollection(): icon = CC.global_pixmaps().collection icon_x = thumbnail_border + ICON_MARGIN icon_y = ( height - 1 ) - thumbnail_border - ICON_MARGIN - icon.height() painter.drawPixmap( icon_x, icon_y, icon ) num_files_str = HydrusData.ToHumanInt( self.GetNumFiles() ) ( text_size, num_files_str ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, num_files_str ) text_width = text_size.width() text_height = text_size.height() box_width = text_width + ( ICON_MARGIN * 2 ) box_x = icon_x + icon.width() + ICON_MARGIN box_height = text_height + ( ICON_MARGIN * 2 ) box_y = ( height - 1 ) - box_height painter.fillRect( box_x, height - text_height - 3, box_width, box_height, CC.COLOUR_UNSELECTED ) painter.setPen( QG.QPen( CC.COLOUR_SELECTED_DARK ) ) text_x = box_x + ICON_MARGIN text_y = box_y + ICON_MARGIN ClientGUIFunctions.DrawText( painter, text_x, text_y, num_files_str ) # top left icons icons_to_draw = [] if self.HasAudio(): icons_to_draw.append( CC.global_pixmaps().sound ) elif self.HasDuration(): icons_to_draw.append( CC.global_pixmaps().play ) services_manager = HG.client_controller.services_manager remote_file_service_keys = HG.client_controller.services_manager.GetRemoteFileServiceKeys() current = locations_manager.GetCurrent().intersection( remote_file_service_keys ) pending = locations_manager.GetPending().intersection( remote_file_service_keys ) petitioned = locations_manager.GetPetitioned().intersection( remote_file_service_keys ) current_to_display = current.difference( petitioned ) # service_types = [ services_manager.GetService( service_key ).GetServiceType() for service_key in current_to_display ] if HC.FILE_REPOSITORY in service_types: icons_to_draw.append( CC.global_pixmaps().file_repository ) if HC.IPFS in service_types: icons_to_draw.append( CC.global_pixmaps().ipfs ) # service_types = [ services_manager.GetService( service_key ).GetServiceType() for service_key in pending ] if HC.FILE_REPOSITORY in service_types: icons_to_draw.append( CC.global_pixmaps().file_repository_pending ) if HC.IPFS in service_types: icons_to_draw.append( CC.global_pixmaps().ipfs_pending ) # service_types = [ services_manager.GetService( service_key ).GetServiceType() for service_key in petitioned ] if HC.FILE_REPOSITORY in service_types: icons_to_draw.append( CC.global_pixmaps().file_repository_petitioned ) if HC.IPFS in service_types: icons_to_draw.append( CC.global_pixmaps().ipfs_petitioned ) top_left_x = thumbnail_border + ICON_MARGIN for icon_to_draw in icons_to_draw: painter.drawPixmap( top_left_x, thumbnail_border + ICON_MARGIN, icon_to_draw ) top_left_x += icon_to_draw.width() + ( ICON_MARGIN * 2 ) return qt_image class ThumbnailMediaCollection( Thumbnail, ClientMedia.MediaCollection ): def __init__( self, location_context, media_results ): ClientMedia.MediaCollection.__init__( self, location_context, media_results ) Thumbnail.__init__( self ) class ThumbnailMediaSingleton( Thumbnail, ClientMedia.MediaSingleton ): def __init__( self, media_result ): ClientMedia.MediaSingleton.__init__( self, media_result ) Thumbnail.__init__( self )