RepoMirrors
/
hydrus
mirror of https://github.com/hydrusnetwork/hydrus.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
hydrus/include/ClientGUIImport.py

1944 lines
68 KiB
Python
Raw Normal View History

Version 273
2017-09-13 20:50:41 +00:00
import ClientConstants as CC
Version 316
2018-08-01 20:44:57 +00:00
import ClientDefaults
import ClientDownloading
Version 282
2017-11-15 22:35:49 +00:00
import ClientGUIACDropdown
Version 273
2017-09-13 20:50:41 +00:00
import ClientGUICommon
Version 315
2018-07-18 21:07:15 +00:00
import ClientGUIControls
Version 316
2018-08-01 20:44:57 +00:00
import ClientGUIDialogs
Version 315
2018-07-18 21:07:15 +00:00
import ClientGUIFileSeedCache
import ClientGUIGallerySeedLog
Version 282
2017-11-15 22:35:49 +00:00
import ClientGUIListBoxes
import ClientGUIListCtrl
Version 314
2018-07-11 20:23:51 +00:00
import ClientGUIMenus
Version 282
2017-11-15 22:35:49 +00:00
import ClientGUIScrolledPanels
Version 273
2017-09-13 20:50:41 +00:00
import ClientGUIScrolledPanelsEdit
Version 315
2018-07-18 21:07:15 +00:00
import ClientGUIShortcuts
import ClientGUITime
Version 273
2017-09-13 20:50:41 +00:00
import ClientGUITopLevelWindows
Version 316
2018-08-01 20:44:57 +00:00
import ClientImportFileSeeds
import ClientImportGallerySeeds
Version 282
2017-11-15 22:35:49 +00:00
import ClientImporting
Version 303
2018-04-18 22:10:15 +00:00
import ClientImportOptions
Version 282
2017-11-15 22:35:49 +00:00
import collections
Version 273
2017-09-13 20:50:41 +00:00
import HydrusConstants as HC
Version 282
2017-11-15 22:35:49 +00:00
import HydrusData
Version 316
2018-08-01 20:44:57 +00:00
import HydrusExceptions
Version 282
2017-11-15 22:35:49 +00:00
import HydrusGlobals as HG
Version 314
2018-07-11 20:23:51 +00:00
import HydrusSerialisable
Version 282
2017-11-15 22:35:49 +00:00
import HydrusTags
Version 286
2017-12-13 22:33:07 +00:00
import HydrusText
Version 282
2017-11-15 22:35:49 +00:00
import os
import re
Version 273
2017-09-13 20:50:41 +00:00
import wx
Version 315
2018-07-18 21:07:15 +00:00
class CheckerOptionsButton( ClientGUICommon.BetterButton ):
def __init__( self, parent, checker_options, update_callable = None ):
ClientGUICommon.BetterButton.__init__( self, parent, 'checker options', self._EditOptions )
self._checker_options = checker_options
self._update_callable = update_callable
self._SetToolTip()
def _EditOptions( self ):
with ClientGUITopLevelWindows.DialogEdit( self, 'edit checker options' ) as dlg:
panel = ClientGUITime.EditCheckerOptions( dlg, self._checker_options )
dlg.SetPanel( panel )
if dlg.ShowModal() == wx.ID_OK:
checker_options = panel.GetValue()
self._SetValue( checker_options )
def _SetToolTip( self ):
self.SetToolTip( self._checker_options.GetSummary() )
def _SetValue( self, checker_options ):
self._checker_options = checker_options
self._SetToolTip()
if self._update_callable is not None:
self._update_callable( self._checker_options )
def GetValue( self ):
return self._checker_options
def SetValue( self, checker_options ):
self._SetValue( checker_options )
Version 274
2017-09-20 19:47:31 +00:00
class FileImportOptionsButton( ClientGUICommon.BetterButton ):
Version 273
2017-09-13 20:50:41 +00:00
Version 325
2018-10-03 21:00:15 +00:00
def __init__( self, parent, file_import_options, show_downloader_options, update_callable = None ):
Version 273
2017-09-13 20:50:41 +00:00
ClientGUICommon.BetterButton.__init__( self, parent, 'file import options', self._EditOptions )
self._file_import_options = file_import_options
Version 325
2018-10-03 21:00:15 +00:00
self._show_downloader_options = show_downloader_options
Version 273
2017-09-13 20:50:41 +00:00
self._update_callable = update_callable
self._SetToolTip()
def _EditOptions( self ):
with ClientGUITopLevelWindows.DialogEdit( self, 'edit file import options' ) as dlg:
Version 325
2018-10-03 21:00:15 +00:00
panel = ClientGUIScrolledPanelsEdit.EditFileImportOptions( dlg, self._file_import_options, self._show_downloader_options )
Version 273
2017-09-13 20:50:41 +00:00
dlg.SetPanel( panel )
if dlg.ShowModal() == wx.ID_OK:
file_import_options = panel.GetValue()
self._SetValue( file_import_options )
def _SetToolTip( self ):
Version 288
2018-01-03 22:37:30 +00:00
self.SetToolTip( self._file_import_options.GetSummary() )
Version 273
2017-09-13 20:50:41 +00:00
def _SetValue( self, file_import_options ):
self._file_import_options = file_import_options
self._SetToolTip()
if self._update_callable is not None:
self._update_callable( self._file_import_options )
def GetValue( self ):
return self._file_import_options
def SetValue( self, file_import_options ):
self._SetValue( file_import_options )
Version 282
2017-11-15 22:35:49 +00:00
class FilenameTaggingOptionsPanel( wx.Panel ):
def __init__( self, parent, service_key, tag_update_callable, filename_tagging_options = None, present_for_accompanying_file_list = False ):
if filename_tagging_options is None:
# pull from an options default
Version 303
2018-04-18 22:10:15 +00:00
filename_tagging_options = ClientImportOptions.FilenameTaggingOptions()
Version 282
2017-11-15 22:35:49 +00:00
wx.Panel.__init__( self, parent )
self._service_key = service_key
self._notebook = wx.Notebook( self )
# eventually these will take 'regexoptions' or whatever object and 'showspecificfiles' as well
self._simple_panel = self._SimplePanel( self._notebook, self._service_key, tag_update_callable, filename_tagging_options, present_for_accompanying_file_list )
self._advanced_panel = self._AdvancedPanel( self._notebook, self._service_key, tag_update_callable, filename_tagging_options, present_for_accompanying_file_list )
self._notebook.AddPage( self._simple_panel, 'simple', select = True )
self._notebook.AddPage( self._advanced_panel, 'advanced' )
vbox = wx.BoxSizer( wx.VERTICAL )
Version 288
2018-01-03 22:37:30 +00:00
vbox.Add( self._notebook, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
self.SetSizer( vbox )
def GetFilenameTaggingOptions( self ):
Version 303
2018-04-18 22:10:15 +00:00
filename_tagging_options = ClientImportOptions.FilenameTaggingOptions()
Version 282
2017-11-15 22:35:49 +00:00
self._advanced_panel.UpdateFilenameTaggingOptions( filename_tagging_options )
self._simple_panel.UpdateFilenameTaggingOptions( filename_tagging_options )
return filename_tagging_options
def GetTags( self, index, path ):
tags = set()
tags.update( self._simple_panel.GetTags( index, path ) )
tags.update( self._advanced_panel.GetTags( index, path ) )
tags = HydrusTags.CleanTags( tags )
siblings_manager = HG.client_controller.GetManager( 'tag_siblings' )
parents_manager = HG.client_controller.GetManager( 'tag_parents' )
tag_censorship_manager = HG.client_controller.GetManager( 'tag_censorship' )
tags = siblings_manager.CollapseTags( self._service_key, tags )
tags = parents_manager.ExpandTags( self._service_key, tags )
tags = tag_censorship_manager.FilterTags( self._service_key, tags )
return tags
def SetSelectedPaths( self, paths ):
self._simple_panel.SetSelectedPaths( paths )
class _AdvancedPanel( wx.Panel ):
def __init__( self, parent, service_key, refresh_callable, filename_tagging_options, present_for_accompanying_file_list ):
wx.Panel.__init__( self, parent )
self._service_key = service_key
self._refresh_callable = refresh_callable
self._present_for_accompanying_file_list = present_for_accompanying_file_list
#
self._quick_namespaces_panel = ClientGUICommon.StaticBox( self, 'quick namespaces' )
Version 316
2018-08-01 20:44:57 +00:00
columns = [ ( 'namespace', 12 ), ( 'regex', -1 ) ]
Version 326
2018-10-17 21:00:09 +00:00
self._quick_namespaces_list = ClientGUIListCtrl.BetterListCtrl( self._quick_namespaces_panel, 'quick_namespaces', 4, 20, columns, self._ConvertQuickRegexDataToListCtrlTuples, use_simple_delete = True, activation_callback = self.EditQuickNamespaces )
Version 282
2017-11-15 22:35:49 +00:00
self._add_quick_namespace_button = wx.Button( self._quick_namespaces_panel, label = 'add' )
self._add_quick_namespace_button.Bind( wx.EVT_BUTTON, self.EventAddQuickNamespace )
self._add_quick_namespace_button.SetMinSize( ( 20, -1 ) )
self._edit_quick_namespace_button = wx.Button( self._quick_namespaces_panel, label = 'edit' )
self._edit_quick_namespace_button.Bind( wx.EVT_BUTTON, self.EventEditQuickNamespace )
self._edit_quick_namespace_button.SetMinSize( ( 20, -1 ) )
self._delete_quick_namespace_button = wx.Button( self._quick_namespaces_panel, label = 'delete' )
self._delete_quick_namespace_button.Bind( wx.EVT_BUTTON, self.EventDeleteQuickNamespace )
self._delete_quick_namespace_button.SetMinSize( ( 20, -1 ) )
#
self._regexes_panel = ClientGUICommon.StaticBox( self, 'regexes' )
self._regexes = wx.ListBox( self._regexes_panel )
self._regexes.Bind( wx.EVT_LISTBOX_DCLICK, self.EventRemoveRegex )
self._regex_box = wx.TextCtrl( self._regexes_panel, style=wx.TE_PROCESS_ENTER )
self._regex_box.Bind( wx.EVT_TEXT_ENTER, self.EventAddRegex )
self._regex_shortcuts = ClientGUICommon.RegexButton( self._regexes_panel )
Version 320
2018-08-29 20:20:41 +00:00
self._regex_intro_link = ClientGUICommon.BetterHyperLink( self._regexes_panel, 'a good regex introduction', 'http://www.aivosto.com/vbtips/regex.html' )
self._regex_practise_link = ClientGUICommon.BetterHyperLink( self._regexes_panel, 'regex practise', 'http://regexr.com/3cvmf' )
Version 282
2017-11-15 22:35:49 +00:00
#
self._num_panel = ClientGUICommon.StaticBox( self, '#' )
self._num_base = wx.SpinCtrl( self._num_panel, min = -10000000, max = 10000000, size = ( 60, -1 ) )
self._num_base.SetValue( 1 )
self._num_base.Bind( wx.EVT_SPINCTRL, self.EventRecalcNum )
self._num_step = wx.SpinCtrl( self._num_panel, min = -1000000, max = 1000000, size = ( 60, -1 ) )
self._num_step.SetValue( 1 )
self._num_step.Bind( wx.EVT_SPINCTRL, self.EventRecalcNum )
self._num_namespace = wx.TextCtrl( self._num_panel, size = ( 100, -1 ) )
self._num_namespace.Bind( wx.EVT_TEXT, self.EventNumNamespaceChanged )
if not self._present_for_accompanying_file_list:
self._num_panel.Hide()
#
( quick_namespaces, regexes ) = filename_tagging_options.AdvancedToTuple()
self._quick_namespaces_list.AddDatas( quick_namespaces )
for regex in regexes:
self._regexes.Append( regex )
#
button_box = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
button_box.Add( self._add_quick_namespace_button, CC.FLAGS_EXPAND_BOTH_WAYS )
button_box.Add( self._edit_quick_namespace_button, CC.FLAGS_EXPAND_BOTH_WAYS )
button_box.Add( self._delete_quick_namespace_button, CC.FLAGS_EXPAND_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
Version 288
2018-01-03 22:37:30 +00:00
self._quick_namespaces_panel.Add( self._quick_namespaces_list, CC.FLAGS_EXPAND_BOTH_WAYS )
self._quick_namespaces_panel.Add( button_box, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
#
Version 288
2018-01-03 22:37:30 +00:00
self._regexes_panel.Add( self._regexes, CC.FLAGS_EXPAND_BOTH_WAYS )
self._regexes_panel.Add( self._regex_box, CC.FLAGS_EXPAND_PERPENDICULAR )
self._regexes_panel.Add( self._regex_shortcuts, CC.FLAGS_LONE_BUTTON )
self._regexes_panel.Add( self._regex_intro_link, CC.FLAGS_LONE_BUTTON )
self._regexes_panel.Add( self._regex_practise_link, CC.FLAGS_LONE_BUTTON )
Version 282
2017-11-15 22:35:49 +00:00
#
hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
hbox.Add( wx.StaticText( self._num_panel, label = '# base/step: ' ), CC.FLAGS_VCENTER )
hbox.Add( self._num_base, CC.FLAGS_VCENTER )
hbox.Add( self._num_step, CC.FLAGS_VCENTER )
Version 282
2017-11-15 22:35:49 +00:00
Version 288
2018-01-03 22:37:30 +00:00
self._num_panel.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
hbox.Add( wx.StaticText( self._num_panel, label = '# namespace: ' ), CC.FLAGS_VCENTER )
hbox.Add( self._num_namespace, CC.FLAGS_EXPAND_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
Version 288
2018-01-03 22:37:30 +00:00
self._num_panel.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
second_vbox = wx.BoxSizer( wx.VERTICAL )
Version 288
2018-01-03 22:37:30 +00:00
second_vbox.Add( self._regexes_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
second_vbox.Add( self._num_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
#
hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
hbox.Add( self._quick_namespaces_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
hbox.Add( second_vbox, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
self.SetSizer( hbox )
def _ConvertQuickRegexDataToListCtrlTuples( self, data ):
( namespace, regex ) = data
display_tuple = ( namespace, regex )
sort_tuple = ( namespace, regex )
return ( display_tuple, sort_tuple )
def EditQuickNamespaces( self ):
data_to_edit = self._quick_namespaces_list.GetData( only_selected = True )
for old_data in data_to_edit:
( namespace, regex ) = old_data
import ClientGUIDialogs
with ClientGUIDialogs.DialogInputNamespaceRegex( self, namespace = namespace, regex = regex ) as dlg:
if dlg.ShowModal() == wx.ID_OK:
( new_namespace, new_regex ) = dlg.GetInfo()
new_data = ( new_namespace, new_regex )
if new_data != old_data:
self._quick_namespaces_list.DeleteDatas( ( old_data, ) )
self._quick_namespaces_list.AddDatas( ( new_data, ) )
self._refresh_callable()
def EventAddRegex( self, event ):
regex = self._regex_box.GetValue()
if regex != '':
try:
re.compile( regex, flags = re.UNICODE )
except Exception as e:
text = 'That regex would not compile!'
text += os.linesep * 2
text += HydrusData.ToUnicode( e )
wx.MessageBox( text )
return
self._regexes.Append( regex )
self._regex_box.Clear()
self._refresh_callable()
def EventAddQuickNamespace( self, event ):
import ClientGUIDialogs
with ClientGUIDialogs.DialogInputNamespaceRegex( self ) as dlg:
if dlg.ShowModal() == wx.ID_OK:
( namespace, regex ) = dlg.GetInfo()
data = ( namespace, regex )
self._quick_namespaces_list.AddDatas( ( data, ) )
self._refresh_callable()
def EventDeleteQuickNamespace( self, event ):
self.DeleteQuickNamespaces()
def EventEditQuickNamespace( self, event ):
self.EditQuickNamespaces()
def EventNumNamespaceChanged( self, event ):
self._refresh_callable()
def EventRecalcNum( self, event ):
self._refresh_callable()
def EventRemoveRegex( self, event ):
selection = self._regexes.GetSelection()
if selection != wx.NOT_FOUND:
if len( self._regex_box.GetValue() ) == 0: self._regex_box.SetValue( self._regexes.GetString( selection ) )
self._regexes.Delete( selection )
self._refresh_callable()
def GetTags( self, index, path ):
tags = set()
num_namespace = self._num_namespace.GetValue()
if num_namespace != '':
num_base = self._num_base.GetValue()
num_step = self._num_step.GetValue()
tag_num = num_base + index * num_step
tags.add( num_namespace + ':' + str( tag_num ) )
return tags
def UpdateFilenameTaggingOptions( self, filename_tagging_options ):
quick_namespaces = self._quick_namespaces_list.GetData()
regexes = self._regexes.GetStrings()
filename_tagging_options.AdvancedSetTuple( quick_namespaces, regexes )
class _SimplePanel( wx.Panel ):
def __init__( self, parent, service_key, refresh_callable, filename_tagging_options, present_for_accompanying_file_list ):
wx.Panel.__init__( self, parent )
self._service_key = service_key
self._refresh_callable = refresh_callable
self._present_for_accompanying_file_list = present_for_accompanying_file_list
#
self._tags_panel = ClientGUICommon.StaticBox( self, 'tags for all' )
self._tags = ClientGUIListBoxes.ListBoxTagsStringsAddRemove( self._tags_panel, self._service_key, self.TagsRemoved )
expand_parents = True
self._tag_box = ClientGUIACDropdown.AutoCompleteDropdownTagsWrite( self._tags_panel, self.EnterTags, expand_parents, CC.LOCAL_FILE_SERVICE_KEY, service_key )
Version 284
2017-11-29 21:48:23 +00:00
self._tags_paste_button = ClientGUICommon.BetterButton( self._tags_panel, 'paste tags', self._PasteTags )
Version 282
2017-11-15 22:35:49 +00:00
#
self._single_tags_panel = ClientGUICommon.StaticBox( self, 'tags just for selected files' )
self._paths_to_single_tags = collections.defaultdict( set )
self._single_tags = ClientGUIListBoxes.ListBoxTagsStringsAddRemove( self._single_tags_panel, self._service_key, self.SingleTagsRemoved )
Version 284
2017-11-29 21:48:23 +00:00
self._single_tags_paste_button = ClientGUICommon.BetterButton( self._single_tags_panel, 'paste tags', self._PasteSingleTags )
Version 282
2017-11-15 22:35:49 +00:00
expand_parents = True
self._single_tag_box = ClientGUIACDropdown.AutoCompleteDropdownTagsWrite( self._single_tags_panel, self.EnterTagsSingle, expand_parents, CC.LOCAL_FILE_SERVICE_KEY, service_key )
self.SetSelectedPaths( [] )
if not self._present_for_accompanying_file_list:
self._single_tags_panel.Hide()
#
self._checkboxes_panel = ClientGUICommon.StaticBox( self, 'misc' )
self._load_from_txt_files_checkbox = wx.CheckBox( self._checkboxes_panel, label = 'try to load tags from neighbouring .txt files' )
txt_files_help_button = ClientGUICommon.BetterBitmapButton( self._checkboxes_panel, CC.GlobalBMPs.help, self._ShowTXTHelp )
Version 288
2018-01-03 22:37:30 +00:00
txt_files_help_button.SetToolTip( 'Show help regarding importing tags from .txt files.' )
Version 282
2017-11-15 22:35:49 +00:00
self._filename_namespace = wx.TextCtrl( self._checkboxes_panel )
self._filename_namespace.SetMinSize( ( 100, -1 ) )
self._filename_checkbox = wx.CheckBox( self._checkboxes_panel, label = 'add filename? [namespace]' )
Version 329
2018-11-07 23:09:40 +00:00
self._directory_namespace_controls = {}
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
directory_items = []
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
directory_items.append( ( 0, 'first' ) )
directory_items.append( ( 1, 'second' ) )
directory_items.append( ( 2, 'third' ) )
directory_items.append( ( -3, 'third last' ) )
directory_items.append( ( -2, 'second last' ) )
directory_items.append( ( -1, 'last' ) )
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
for ( index, phrase ) in directory_items:
dir_checkbox = wx.CheckBox( self._checkboxes_panel, label = 'add ' + phrase + ' directory? [namespace]' )
dir_namespace_textctrl = wx.TextCtrl( self._checkboxes_panel )
dir_namespace_textctrl.SetMinSize( ( 100, -1 ) )
self._directory_namespace_controls[ index ] = ( dir_checkbox, dir_namespace_textctrl )
Version 282
2017-11-15 22:35:49 +00:00
#
Version 329
2018-11-07 23:09:40 +00:00
( tags_for_all, load_from_neighbouring_txt_files, add_filename, directory_dict ) = filename_tagging_options.SimpleToTuple()
Version 282
2017-11-15 22:35:49 +00:00
self._tags.AddTags( tags_for_all )
self._load_from_txt_files_checkbox.SetValue( load_from_neighbouring_txt_files )
( add_filename_boolean, add_filename_namespace ) = add_filename
self._filename_checkbox.SetValue( add_filename_boolean )
self._filename_namespace.SetValue( add_filename_namespace )
Version 329
2018-11-07 23:09:40 +00:00
for ( index, ( dir_boolean, dir_namespace ) ) in directory_dict.items():
( dir_checkbox, dir_namespace_textctrl ) = self._directory_namespace_controls[ index ]
dir_checkbox.SetValue( dir_boolean )
dir_namespace_textctrl.SetValue( dir_namespace )
dir_checkbox.Bind( wx.EVT_CHECKBOX, self.EventRefresh )
dir_namespace_textctrl.Bind( wx.EVT_TEXT, self.EventRefresh )
Version 282
2017-11-15 22:35:49 +00:00
#
Version 288
2018-01-03 22:37:30 +00:00
self._tags_panel.Add( self._tags, CC.FLAGS_EXPAND_BOTH_WAYS )
self._tags_panel.Add( self._tag_box, CC.FLAGS_EXPAND_PERPENDICULAR )
self._tags_panel.Add( self._tags_paste_button, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
Version 288
2018-01-03 22:37:30 +00:00
self._single_tags_panel.Add( self._single_tags, CC.FLAGS_EXPAND_BOTH_WAYS )
self._single_tags_panel.Add( self._single_tag_box, CC.FLAGS_EXPAND_PERPENDICULAR )
self._single_tags_panel.Add( self._single_tags_paste_button, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
txt_hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
txt_hbox.Add( self._load_from_txt_files_checkbox, CC.FLAGS_EXPAND_BOTH_WAYS )
txt_hbox.Add( txt_files_help_button, CC.FLAGS_VCENTER )
Version 282
2017-11-15 22:35:49 +00:00
filename_hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
filename_hbox.Add( self._filename_checkbox, CC.FLAGS_VCENTER )
filename_hbox.Add( self._filename_namespace, CC.FLAGS_EXPAND_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
Version 288
2018-01-03 22:37:30 +00:00
self._checkboxes_panel.Add( txt_hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
self._checkboxes_panel.Add( filename_hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 329
2018-11-07 23:09:40 +00:00
for index in ( 0, 1, 2, -3, -2, -1 ):
hbox = wx.BoxSizer( wx.HORIZONTAL )
( dir_checkbox, dir_namespace_textctrl ) = self._directory_namespace_controls[ index ]
hbox.Add( dir_checkbox, CC.FLAGS_VCENTER )
hbox.Add( dir_namespace_textctrl, CC.FLAGS_EXPAND_BOTH_WAYS )
self._checkboxes_panel.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
hbox = wx.BoxSizer( wx.HORIZONTAL )
Version 288
2018-01-03 22:37:30 +00:00
hbox.Add( self._tags_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
hbox.Add( self._single_tags_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
hbox.Add( self._checkboxes_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 282
2017-11-15 22:35:49 +00:00
self.SetSizer( hbox )
#
self._load_from_txt_files_checkbox.Bind( wx.EVT_CHECKBOX, self.EventRefresh )
self._filename_namespace.Bind( wx.EVT_TEXT, self.EventRefresh )
self._filename_checkbox.Bind( wx.EVT_CHECKBOX, self.EventRefresh )
Version 284
2017-11-29 21:48:23 +00:00
def _GetTagsFromClipboard( self ):
Version 286
2017-12-13 22:33:07 +00:00
text = HG.client_controller.GetClipboardText()
try:
Version 284
2017-11-29 21:48:23 +00:00
Version 286
2017-12-13 22:33:07 +00:00
tags = HydrusText.DeserialiseNewlinedTexts( text )
Version 284
2017-11-29 21:48:23 +00:00
Version 286
2017-12-13 22:33:07 +00:00
tags = HydrusTags.CleanTags( tags )
Version 284
2017-11-29 21:48:23 +00:00
Version 286
2017-12-13 22:33:07 +00:00
return tags
Version 284
2017-11-29 21:48:23 +00:00
Version 286
2017-12-13 22:33:07 +00:00
except:
Version 284
2017-11-29 21:48:23 +00:00
Version 286
2017-12-13 22:33:07 +00:00
raise Exception( 'I could not understand what was in the clipboard' )
Version 284
2017-11-29 21:48:23 +00:00
def _PasteTags( self ):
try:
tags = self._GetTagsFromClipboard()
except Exception as e:
wx.MessageBox( HydrusData.ToUnicode( e ) )
return
self.EnterTags( tags )
def _PasteSingleTags( self ):
try:
tags = self._GetTagsFromClipboard()
except Exception as e:
wx.MessageBox( HydrusData.ToUnicode( e ) )
return
self.EnterTagsSingle( tags )
Version 282
2017-11-15 22:35:49 +00:00
def _ShowTXTHelp( self ):
message = 'If you would like to add custom tags with your files, add a .txt file beside the file like so:'
message += os.linesep * 2
message += 'my_file.jpg'
message += os.linesep
message += 'my_file.jpg.txt'
message += os.linesep * 2
message += 'And include your tags inside the .txt file in a newline-separated list (if you know how to script, generating these files automatically from another source of tags can save a lot of time!).'
message += os.linesep * 2
message += 'Make sure you preview the results in the table above to be certain everything is parsing correctly. Until you are comfortable with this, you should test it on just one or two files.'
wx.MessageBox( message )
def EnterTags( self, tags ):
tag_parents_manager = HG.client_controller.GetManager( 'tag_parents' )
parents = set()
for tag in tags:
some_parents = tag_parents_manager.GetParents( self._service_key, tag )
parents.update( some_parents )
if len( tags ) > 0:
self._tags.AddTags( tags )
self._tags.AddTags( parents )
self._refresh_callable()
def EnterTagsSingle( self, tags ):
tag_parents_manager = HG.client_controller.GetManager( 'tag_parents' )
parents = set()
for tag in tags:
some_parents = tag_parents_manager.GetParents( self._service_key, tag )
parents.update( some_parents )
if len( tags ) > 0:
self._single_tags.AddTags( tags )
self._single_tags.AddTags( parents )
for path in self._selected_paths:
current_tags = self._paths_to_single_tags[ path ]
current_tags.update( tags )
current_tags.update( parents )
self._refresh_callable()
def EventRefresh( self, event ):
self._refresh_callable()
def GetTags( self, index, path ):
tags = set()
if path in self._paths_to_single_tags:
tags.update( self._paths_to_single_tags[ path ] )
return tags
def SingleTagsRemoved( self, tags ):
for path in self._selected_paths:
current_tags = self._paths_to_single_tags[ path ]
current_tags.difference_update( tags )
self._refresh_callable()
def SetSelectedPaths( self, paths ):
self._selected_paths = paths
single_tags = set()
if len( paths ) > 0:
for path in self._selected_paths:
if path in self._paths_to_single_tags:
single_tags.update( self._paths_to_single_tags[ path ] )
self._single_tag_box.Enable()
Version 284
2017-11-29 21:48:23 +00:00
self._single_tags_paste_button.Enable()
Version 282
2017-11-15 22:35:49 +00:00
else:
self._single_tag_box.Disable()
Version 284
2017-11-29 21:48:23 +00:00
self._single_tags_paste_button.Disable()
Version 282
2017-11-15 22:35:49 +00:00
self._single_tags.SetTags( single_tags )
def TagsRemoved( self, tag ):
self._refresh_callable()
def UpdateFilenameTaggingOptions( self, filename_tagging_options ):
tags_for_all = self._tags.GetTags()
load_from_neighbouring_txt_files = self._load_from_txt_files_checkbox.GetValue()
Version 329
2018-11-07 23:09:40 +00:00
add_filename = ( self._filename_checkbox.GetValue(), self._filename_namespace.GetValue() )
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
directories_dict = {}
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
for ( index, ( dir_checkbox, dir_namespace_textctrl ) ) in self._directory_namespace_controls.items():
directories_dict[ index ] = ( dir_checkbox.GetValue(), dir_namespace_textctrl.GetValue() )
Version 282
2017-11-15 22:35:49 +00:00
Version 329
2018-11-07 23:09:40 +00:00
filename_tagging_options.SimpleSetTuple( tags_for_all, load_from_neighbouring_txt_files, add_filename, directories_dict )
Version 282
2017-11-15 22:35:49 +00:00
Version 283
2017-11-22 21:03:07 +00:00
class EditLocalImportFilenameTaggingPanel( ClientGUIScrolledPanels.EditPanel ):
def __init__( self, parent, paths ):
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
self._paths = paths
self._tag_repositories = ClientGUICommon.ListBook( self )
#
services = HG.client_controller.services_manager.GetServices( ( HC.TAG_REPOSITORY, ) )
for service in services:
if service.HasPermission( HC.CONTENT_TYPE_MAPPINGS, HC.PERMISSION_ACTION_CREATE ):
service_key = service.GetServiceKey()
name = service.GetName()
self._tag_repositories.AddPageArgs( name, service_key, self._Panel, ( self._tag_repositories, service_key, paths ), {} )
page = self._Panel( self._tag_repositories, CC.LOCAL_TAG_SERVICE_KEY, paths )
name = CC.LOCAL_TAG_SERVICE_KEY
self._tag_repositories.AddPage( name, name, page )
default_tag_repository_key = HC.options[ 'default_tag_repository' ]
self._tag_repositories.Select( default_tag_repository_key )
#
vbox = wx.BoxSizer( wx.VERTICAL )
Version 288
2018-01-03 22:37:30 +00:00
vbox.Add( self._tag_repositories, CC.FLAGS_EXPAND_BOTH_WAYS )
Version 283
2017-11-22 21:03:07 +00:00
self.SetSizer( vbox )
def GetValue( self ):
paths_to_tags = collections.defaultdict( dict )
for page in self._tag_repositories.GetActivePages():
( service_key, page_of_paths_to_tags ) = page.GetInfo()
for ( path, tags ) in page_of_paths_to_tags.items(): paths_to_tags[ path ][ service_key ] = tags
return paths_to_tags
class _Panel( wx.Panel ):
def __init__( self, parent, service_key, paths ):
wx.Panel.__init__( self, parent )
self._service_key = service_key
self._paths = paths
Version 316
2018-08-01 20:44:57 +00:00
columns = [ ( '#', 4 ), ( 'path', 40 ), ( 'tags', -1 ) ]
self._paths_list = ClientGUIListCtrl.BetterListCtrl( self, 'paths_to_tags', 25, 40, columns, self._ConvertDataToListCtrlTuples )
Version 283
2017-11-22 21:03:07 +00:00
self._paths_list.Bind( wx.EVT_LIST_ITEM_SELECTED, self.EventItemSelected )
self._paths_list.Bind( wx.EVT_LIST_ITEM_DESELECTED, self.EventItemSelected )
#
Version 309
2018-05-30 20:13:21 +00:00
self._filename_tagging_panel = FilenameTaggingOptionsPanel( self, self._service_key, self.ScheduleRefreshFileList, present_for_accompanying_file_list = True )
self._schedule_refresh_file_list_job = None
Version 283
2017-11-22 21:03:07 +00:00
#
# i.e. ( index, path )
self._paths_list.AddDatas( list( enumerate( self._paths ) ) )
#
vbox = wx.BoxSizer( wx.VERTICAL )
Version 288
2018-01-03 22:37:30 +00:00
vbox.Add( self._paths_list, CC.FLAGS_EXPAND_BOTH_WAYS )
vbox.Add( self._filename_tagging_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 283
2017-11-22 21:03:07 +00:00
self.SetSizer( vbox )
def _ConvertDataToListCtrlTuples( self, data ):
( index, path ) = data
tags = self._GetTags( index, path )
Version 313
2018-07-04 20:48:28 +00:00
pretty_index = HydrusData.ToHumanInt( index + 1 )
Version 283
2017-11-22 21:03:07 +00:00
pretty_path = path
pretty_tags = ', '.join( tags )
display_tuple = ( pretty_index, pretty_path, pretty_tags )
sort_tuple = ( index, path, tags )
return ( display_tuple, sort_tuple )
def _GetTags( self, index, path ):
filename_tagging_options = self._filename_tagging_panel.GetFilenameTaggingOptions()
tags = filename_tagging_options.GetTags( self._service_key, path )
tags.update( self._filename_tagging_panel.GetTags( index, path ) )
tags = list( tags )
tags.sort()
return tags
def EventItemSelected( self, event ):
paths = [ path for ( index, path ) in self._paths_list.GetData( only_selected = True ) ]
self._filename_tagging_panel.SetSelectedPaths( paths )
event.Skip()
def GetInfo( self ):
paths_to_tags = { path : self._GetTags( index, path ) for ( index, path ) in self._paths_list.GetData() }
return ( self._service_key, paths_to_tags )
def RefreshFileList( self ):
self._paths_list.UpdateDatas()
Version 309
2018-05-30 20:13:21 +00:00
def ScheduleRefreshFileList( self ):
if self._schedule_refresh_file_list_job is not None:
self._schedule_refresh_file_list_job.Cancel()
self._schedule_refresh_file_list_job = None
self._schedule_refresh_file_list_job = HG.client_controller.CallLaterWXSafe( self, 0.5, self.RefreshFileList )
Version 282
2017-11-15 22:35:49 +00:00
class EditFilenameTaggingOptionPanel( ClientGUIScrolledPanels.EditPanel ):
def __init__( self, parent, service_key, filename_tagging_options ):
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
self._service_key = service_key
self._example_path_input = wx.TextCtrl( self )
self._example_output = wx.TextCtrl( self )
Version 309
2018-05-30 20:13:21 +00:00
self._filename_tagging_options_panel = FilenameTaggingOptionsPanel( self, self._service_key, self.ScheduleRefreshTags, filename_tagging_options = filename_tagging_options, present_for_accompanying_file_list = False )
self._schedule_refresh_tags_job = None
Version 282
2017-11-15 22:35:49 +00:00
#
self._example_path_input.SetValue( 'enter example path here' )
self._example_output.Disable()
#
vbox = wx.BoxSizer( wx.VERTICAL )
Version 288
2018-01-03 22:37:30 +00:00
vbox.Add( self._example_path_input, CC.FLAGS_EXPAND_PERPENDICULAR )
vbox.Add( self._example_output, CC.FLAGS_EXPAND_PERPENDICULAR )
vbox.Add( self._filename_tagging_options_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
Version 282
2017-11-15 22:35:49 +00:00
self.SetSizer( vbox )
#
self._example_path_input.Bind( wx.EVT_TEXT, self.EventText )
def EventText( self, event ):
Version 309
2018-05-30 20:13:21 +00:00
self.ScheduleRefreshTags()
Version 282
2017-11-15 22:35:49 +00:00
def GetValue( self ):
return self._filename_tagging_options_panel.GetFilenameTaggingOptions()
def RefreshTags( self ):
example_path_input = self._example_path_input.GetValue()
filename_tagging_options = self.GetValue()
try:
tags = filename_tagging_options.GetTags( self._service_key, example_path_input )
except:
tags = [ 'could not parse' ]
self._example_output.SetValue( ', '.join( tags ) )
Version 309
2018-05-30 20:13:21 +00:00
def ScheduleRefreshTags( self ):
if self._schedule_refresh_tags_job is not None:
self._schedule_refresh_tags_job.Cancel()
self._schedule_refresh_tags_job = None
self._schedule_refresh_tags_job = HG.client_controller.CallLaterWXSafe( self, 0.5, self.RefreshTags )
Version 316
2018-08-01 20:44:57 +00:00
class GalleryImportPanel( ClientGUICommon.StaticBox ):
def __init__( self, parent, page_key, name = 'gallery query' ):
ClientGUICommon.StaticBox.__init__( self, parent, name )
self._page_key = page_key
self._gallery_import = None
#
self._query_text = wx.TextCtrl( self )
self._query_text.SetEditable( False )
self._import_queue_panel = ClientGUICommon.StaticBox( self, 'import queue' )
self._current_action = ClientGUICommon.BetterStaticText( self._import_queue_panel, style = wx.ST_ELLIPSIZE_END )
self._file_seed_cache_control = ClientGUIFileSeedCache.FileSeedCacheStatusControl( self._import_queue_panel, HG.client_controller, self._page_key )
self._file_download_control = ClientGUIControls.NetworkJobControl( self._import_queue_panel )
self._files_pause_button = wx.BitmapButton( self._import_queue_panel, bitmap = CC.GlobalBMPs.pause )
self._files_pause_button.Bind( wx.EVT_BUTTON, self.EventFilesPause )
self._gallery_panel = ClientGUICommon.StaticBox( self, 'gallery parser' )
self._gallery_status = ClientGUICommon.BetterStaticText( self._gallery_panel, style = wx.ST_ELLIPSIZE_END )
self._gallery_pause_button = wx.BitmapButton( self._gallery_panel, bitmap = CC.GlobalBMPs.pause )
self._gallery_pause_button.Bind( wx.EVT_BUTTON, self.EventGalleryPause )
Version 322
2018-09-12 21:36:26 +00:00
self._gallery_seed_log_control = ClientGUIGallerySeedLog.GallerySeedLogStatusControl( self._gallery_panel, HG.client_controller, False, True, page_key = self._page_key )
Version 316
2018-08-01 20:44:57 +00:00
self._gallery_download_control = ClientGUIControls.NetworkJobControl( self._gallery_panel )
self._file_limit = ClientGUICommon.NoneableSpinCtrl( self, 'stop after this many files', min = 1, none_phrase = 'no limit' )
self._file_limit.Bind( wx.EVT_SPINCTRL, self.EventFileLimit )
self._file_limit.SetToolTip( 'stop searching the gallery once this many files has been reached' )
file_import_options = ClientImportOptions.FileImportOptions()
tag_import_options = ClientImportOptions.TagImportOptions( is_default = True )
Version 325
2018-10-03 21:00:15 +00:00
show_downloader_options = True
self._file_import_options = FileImportOptionsButton( self, file_import_options, show_downloader_options, self._SetFileImportOptions )
self._tag_import_options = TagImportOptionsButton( self, tag_import_options, show_downloader_options, update_callable = self._SetTagImportOptions, allow_default_selection = True )
Version 316
2018-08-01 20:44:57 +00:00
#
hbox = wx.BoxSizer( wx.HORIZONTAL )
hbox.Add( self._gallery_status, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
hbox.Add( self._gallery_pause_button, CC.FLAGS_VCENTER )
self._gallery_panel.Add( hbox, CC.FLAGS_EXPAND_PERPENDICULAR )
self._gallery_panel.Add( self._gallery_seed_log_control, CC.FLAGS_EXPAND_PERPENDICULAR )
self._gallery_panel.Add( self._gallery_download_control, CC.FLAGS_EXPAND_PERPENDICULAR )
#
hbox = wx.BoxSizer( wx.HORIZONTAL )
hbox.Add( self._current_action, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
hbox.Add( self._files_pause_button, CC.FLAGS_VCENTER )
self._import_queue_panel.Add( hbox, CC.FLAGS_EXPAND_PERPENDICULAR )
self._import_queue_panel.Add( self._file_seed_cache_control, CC.FLAGS_EXPAND_PERPENDICULAR )
self._import_queue_panel.Add( self._file_download_control, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._query_text, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._import_queue_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._gallery_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._file_limit, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._file_import_options, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._tag_import_options, CC.FLAGS_EXPAND_PERPENDICULAR )
#
self._UpdateControlsForNewGalleryImport()
HG.client_controller.gui.RegisterUIUpdateWindow( self )
def _SetFileImportOptions( self, file_import_options ):
if self._gallery_import is not None:
self._gallery_import.SetFileImportOptions( file_import_options )
def _SetTagImportOptions( self, tag_import_options ):
if self._gallery_import is not None:
self._gallery_import.SetTagImportOptions( tag_import_options )
def _UpdateControlsForNewGalleryImport( self ):
if self._gallery_import is None:
self._import_queue_panel.Disable()
self._gallery_panel.Disable()
self._file_limit.Disable()
self._file_import_options.Disable()
self._tag_import_options.Disable()
self._query_text.SetValue( '' )
self._current_action.SetLabelText( '' )
self._gallery_status.SetLabelText( '' )
self._file_seed_cache_control.SetFileSeedCache( None )
self._gallery_seed_log_control.SetGallerySeedLog( None )
self._file_download_control.ClearNetworkJob()
self._gallery_download_control.ClearNetworkJob()
else:
self._import_queue_panel.Enable()
self._gallery_panel.Enable()
self._file_limit.Enable()
self._file_import_options.Enable()
self._tag_import_options.Enable()
query = self._gallery_import.GetQueryText()
self._query_text.SetValue( query )
file_limit = self._gallery_import.GetFileLimit()
self._file_limit.SetValue( file_limit )
file_import_options = self._gallery_import.GetFileImportOptions()
self._file_import_options.SetValue( file_import_options )
tag_import_options = self._gallery_import.GetTagImportOptions()
self._tag_import_options.SetValue( tag_import_options )
file_seed_cache = self._gallery_import.GetFileSeedCache()
self._file_seed_cache_control.SetFileSeedCache( file_seed_cache )
gallery_seed_log = self._gallery_import.GetGallerySeedLog()
self._gallery_seed_log_control.SetGallerySeedLog( gallery_seed_log )
def _UpdateStatus( self ):
if self._gallery_import is not None:
( gallery_status, current_action, files_paused, gallery_paused ) = self._gallery_import.GetStatus()
if files_paused:
ClientGUICommon.SetBitmapButtonBitmap( self._files_pause_button, CC.GlobalBMPs.play )
else:
ClientGUICommon.SetBitmapButtonBitmap( self._files_pause_button, CC.GlobalBMPs.pause )
if gallery_paused:
ClientGUICommon.SetBitmapButtonBitmap( self._gallery_pause_button, CC.GlobalBMPs.play )
else:
ClientGUICommon.SetBitmapButtonBitmap( self._gallery_pause_button, CC.GlobalBMPs.pause )
if gallery_paused:
if gallery_status == '':
gallery_status = 'paused'
else:
gallery_status = 'paused - ' + gallery_status
self._gallery_status.SetLabelText( gallery_status )
if files_paused:
if current_action == '':
current_action = 'paused'
else:
current_action = 'pausing - ' + current_action
self._current_action.SetLabelText( current_action )
( file_network_job, gallery_network_job ) = self._gallery_import.GetNetworkJobs()
self._file_download_control.SetNetworkJob( file_network_job )
self._gallery_download_control.SetNetworkJob( gallery_network_job )
def EventFileLimit( self, event ):
if self._gallery_import is not None:
self._gallery_import.SetFileLimit( self._file_limit.GetValue() )
event.Skip()
def EventFilesPause( self, event ):
if self._gallery_import is not None:
self._gallery_import.PausePlayFiles()
self._UpdateStatus()
def EventGalleryPause( self, event ):
if self._gallery_import is not None:
self._gallery_import.PausePlayGallery()
self._UpdateStatus()
def SetGalleryImport( self, gallery_import ):
self._gallery_import = gallery_import
self._UpdateControlsForNewGalleryImport()
def TIMERUIUpdate( self ):
if HG.client_controller.gui.IShouldRegularlyUpdate( self ):
self._UpdateStatus()
Version 321
2018-09-05 20:52:32 +00:00
class GUGKeyAndNameSelector( ClientGUICommon.BetterButton ):
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
def __init__( self, parent, gug_key_and_name, update_callable = None ):
Version 316
2018-08-01 20:44:57 +00:00
ClientGUICommon.BetterButton.__init__( self, parent, 'gallery selector', self._Edit )
Version 321
2018-09-05 20:52:32 +00:00
gug = HG.client_controller.network_engine.domain_manager.GetGUG( gug_key_and_name )
if gug is not None:
gug_key_and_name = gug.GetGUGKeyAndName()
self._gug_key_and_name = gug_key_and_name
Version 316
2018-08-01 20:44:57 +00:00
self._update_callable = update_callable
self._SetLabel()
def _Edit( self ):
Version 321
2018-09-05 20:52:32 +00:00
domain_manager = HG.client_controller.network_engine.domain_manager
# maybe relegate to hidden page and something like "(does not work)" if no gallery url class match
my_gug = domain_manager.GetGUG( self._gug_key_and_name )
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
gugs = domain_manager.GetGUGs()
gug_keys_to_display = domain_manager.GetGUGKeysToDisplay()
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
functional_gugs = []
non_functional_gugs = []
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
for gug in gugs:
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
if gug.IsFunctional():
functional_gugs.append( gug )
else:
non_functional_gugs.append( gug )
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
choice_tuples = [ ( gug.GetName(), gug ) for gug in functional_gugs if gug.GetGUGKey() in gug_keys_to_display ]
choice_tuples.sort()
second_choice_tuples = [ ( gug.GetName(), gug ) for gug in functional_gugs if gug.GetGUGKey() not in gug_keys_to_display ]
second_choice_tuples.sort()
if len( second_choice_tuples ) > 0:
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
choice_tuples.append( ( '--other galleries', -1 ) )
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
if len( non_functional_gugs ) > 0:
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
non_functional_choice_tuples = [ ( gug.GetName(), gug ) for gug in non_functional_gugs ]
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
non_functional_choice_tuples.sort()
choice_tuples.append( ( '--non-functional galleries', -2 ) )
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
with ClientGUIDialogs.DialogSelectFromList( self, 'select gallery', choice_tuples, value_to_select = my_gug, sort_tuples = False ) as dlg:
Version 316
2018-08-01 20:44:57 +00:00
if dlg.ShowModal() == wx.ID_OK:
Version 321
2018-09-05 20:52:32 +00:00
gug = dlg.GetChoice()
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
if gug == -1:
with ClientGUIDialogs.DialogSelectFromList( self, 'select gallery', second_choice_tuples, value_to_select = my_gug ) as dlg:
if dlg.ShowModal() == wx.ID_OK:
gug = dlg.GetChoice()
else:
return
elif gug == -2:
with ClientGUIDialogs.DialogSelectFromList( self, 'select gallery', non_functional_choice_tuples, value_to_select = my_gug ) as dlg:
if dlg.ShowModal() == wx.ID_OK:
gug = dlg.GetChoice()
else:
return
gug_key_and_name = gug.GetGUGKeyAndName()
self._SetValue( gug_key_and_name )
Version 316
2018-08-01 20:44:57 +00:00
def _SetLabel( self ):
Version 321
2018-09-05 20:52:32 +00:00
label = self._gug_key_and_name[1]
gug = HG.client_controller.network_engine.domain_manager.GetGUG( self._gug_key_and_name )
if gug is None:
label = 'not found: ' + label
self.SetLabelText( label )
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
def _SetValue( self, gug_key_and_name ):
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
self._gug_key_and_name = gug_key_and_name
Version 316
2018-08-01 20:44:57 +00:00
self._SetLabel()
if self._update_callable is not None:
Version 321
2018-09-05 20:52:32 +00:00
self._update_callable( gug_key_and_name )
Version 316
2018-08-01 20:44:57 +00:00
def GetValue( self ):
Version 321
2018-09-05 20:52:32 +00:00
return self._gug_key_and_name
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
def SetValue( self, gug_key_and_name ):
Version 316
2018-08-01 20:44:57 +00:00
Version 321
2018-09-05 20:52:32 +00:00
self._SetValue( gug_key_and_name )
Version 316
2018-08-01 20:44:57 +00:00
Version 274
2017-09-20 19:47:31 +00:00
class TagImportOptionsButton( ClientGUICommon.BetterButton ):
Version 325
2018-10-03 21:00:15 +00:00
def __init__( self, parent, tag_import_options, show_downloader_options, update_callable = None, allow_default_selection = False ):
Version 274
2017-09-20 19:47:31 +00:00
ClientGUICommon.BetterButton.__init__( self, parent, 'tag import options', self._EditOptions )
self._tag_import_options = tag_import_options
Version 312
2018-06-27 19:27:05 +00:00
self._show_downloader_options = show_downloader_options
Version 325
2018-10-03 21:00:15 +00:00
self._update_callable = update_callable
Version 314
2018-07-11 20:23:51 +00:00
self._allow_default_selection = allow_default_selection
Version 274
2017-09-20 19:47:31 +00:00
self._SetToolTip()
Version 314
2018-07-11 20:23:51 +00:00
#
self.Bind( wx.EVT_RIGHT_DOWN, self.EventShowMenu )
def _Copy( self ):
json_string = self._tag_import_options.DumpToString()
HG.client_controller.pub( 'clipboard', 'text', json_string )
Version 274
2017-09-20 19:47:31 +00:00
def _EditOptions( self ):
with ClientGUITopLevelWindows.DialogEdit( self, 'edit tag import options' ) as dlg:
Version 325
2018-10-03 21:00:15 +00:00
panel = ClientGUIScrolledPanelsEdit.EditTagImportOptionsPanel( dlg, self._tag_import_options, self._show_downloader_options, allow_default_selection = self._allow_default_selection )
Version 274
2017-09-20 19:47:31 +00:00
dlg.SetPanel( panel )
if dlg.ShowModal() == wx.ID_OK:
tag_import_options = panel.GetValue()
self._SetValue( tag_import_options )
Version 314
2018-07-11 20:23:51 +00:00
def _Paste( self ):
Version 315
2018-07-18 21:07:15 +00:00
Version 314
2018-07-11 20:23:51 +00:00
raw_text = HG.client_controller.GetClipboardText()
try:
tag_import_options = HydrusSerialisable.CreateFromString( raw_text )
Version 315
2018-07-18 21:07:15 +00:00
if not isinstance( tag_import_options, ClientImportOptions.TagImportOptions ):
raise Exception( 'Not a Tag Import Options!' )
Version 314
2018-07-11 20:23:51 +00:00
self._tag_import_options = tag_import_options
except Exception as e:
wx.MessageBox( 'I could not understand what was in the clipboard' )
HydrusData.ShowException( e )
def _SetDefault( self ):
self._tag_import_options.SetDefault()
Version 274
2017-09-20 19:47:31 +00:00
def _SetToolTip( self ):
Version 314
2018-07-11 20:23:51 +00:00
summary = self._tag_import_options.GetSummary( self._show_downloader_options )
self.SetToolTip( summary )
Version 274
2017-09-20 19:47:31 +00:00
def _SetValue( self, tag_import_options ):
self._tag_import_options = tag_import_options
self._SetToolTip()
if self._update_callable is not None:
self._update_callable( self._tag_import_options )
Version 314
2018-07-11 20:23:51 +00:00
def EventShowMenu( self, event ):
menu = wx.Menu()
ClientGUIMenus.AppendMenuItem( self, menu, 'copy to clipboard', 'Serialise this tag import options and copy it to clipboard.', self._Copy )
ClientGUIMenus.AppendSeparator( menu )
ClientGUIMenus.AppendMenuItem( self, menu, 'paste from clipboard', 'Try to import serialised tag import options from the clipboard.', self._Paste )
if not self._tag_import_options.IsDefault():
ClientGUIMenus.AppendSeparator( menu )
ClientGUIMenus.AppendMenuItem( self, menu, 'set to default', 'Set this tag import options to defer to the defaults.', self._SetDefault )
HG.client_controller.PopupMenu( self, menu )
Version 274
2017-09-20 19:47:31 +00:00
def GetValue( self ):
return self._tag_import_options
Version 280
2017-11-01 20:37:39 +00:00
def SetNamespaces( self, namespaces ):
self._namespaces = namespaces
Version 274
2017-09-20 19:47:31 +00:00
def SetValue( self, tag_import_options ):
self._SetValue( tag_import_options )
Version 315
2018-07-18 21:07:15 +00:00
class WatcherReviewPanel( ClientGUICommon.StaticBox ):
Version 316
2018-08-01 20:44:57 +00:00
def __init__( self, parent, page_key, name = 'watcher' ):
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
ClientGUICommon.StaticBox.__init__( self, parent, name )
Version 315
2018-07-18 21:07:15 +00:00
self._page_key = page_key
self._watcher = None
self._watcher_subject = ClientGUICommon.BetterStaticText( self, style = wx.ST_ELLIPSIZE_END )
Version 316
2018-08-01 20:44:57 +00:00
self._watcher_url = wx.TextCtrl( self )
self._watcher_url.SetEditable( False )
Version 315
2018-07-18 21:07:15 +00:00
self._options_panel = wx.Panel( self )
#
imports_panel = ClientGUICommon.StaticBox( self._options_panel, 'file imports' )
self._files_pause_button = wx.BitmapButton( imports_panel, bitmap = CC.GlobalBMPs.pause )
self._files_pause_button.Bind( wx.EVT_BUTTON, self.EventPauseFiles )
self._current_action = ClientGUICommon.BetterStaticText( imports_panel, style = wx.ST_ELLIPSIZE_END )
Version 316
2018-08-01 20:44:57 +00:00
self._file_seed_cache_control = ClientGUIFileSeedCache.FileSeedCacheStatusControl( imports_panel, HG.client_controller, self._page_key )
Version 315
2018-07-18 21:07:15 +00:00
self._file_download_control = ClientGUIControls.NetworkJobControl( imports_panel )
#
checker_panel = ClientGUICommon.StaticBox( self._options_panel, 'checker' )
self._file_velocity_status = ClientGUICommon.BetterStaticText( checker_panel, style = wx.ST_ELLIPSIZE_END )
self._checking_pause_button = wx.BitmapButton( checker_panel, bitmap = CC.GlobalBMPs.pause )
Version 316
2018-08-01 20:44:57 +00:00
self._checking_pause_button.Bind( wx.EVT_BUTTON, self.EventPauseChecking )
Version 315
2018-07-18 21:07:15 +00:00
self._watcher_status = ClientGUICommon.BetterStaticText( checker_panel, style = wx.ST_ELLIPSIZE_END )
self._check_now_button = wx.Button( checker_panel, label = 'check now' )
self._check_now_button.Bind( wx.EVT_BUTTON, self.EventCheckNow )
Version 322
2018-09-12 21:36:26 +00:00
self._gallery_seed_log_control = ClientGUIGallerySeedLog.GallerySeedLogStatusControl( checker_panel, HG.client_controller, True, False, page_key = self._page_key )
Version 315
2018-07-18 21:07:15 +00:00
checker_options = ClientImportOptions.CheckerOptions()
self._checker_options_button = CheckerOptionsButton( checker_panel, checker_options, update_callable = self._SetCheckerOptions )
self._checker_download_control = ClientGUIControls.NetworkJobControl( checker_panel )
file_import_options = ClientImportOptions.FileImportOptions()
tag_import_options = ClientImportOptions.TagImportOptions( is_default = True )
Version 325
2018-10-03 21:00:15 +00:00
show_downloader_options = True
self._file_import_options = FileImportOptionsButton( self, file_import_options, show_downloader_options, self._SetFileImportOptions )
self._tag_import_options = TagImportOptionsButton( self, tag_import_options, show_downloader_options, update_callable = self._SetTagImportOptions, allow_default_selection = True )
Version 315
2018-07-18 21:07:15 +00:00
#
hbox = wx.BoxSizer( wx.HORIZONTAL )
hbox.Add( self._current_action, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
hbox.Add( self._files_pause_button, CC.FLAGS_VCENTER )
imports_panel.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
imports_panel.Add( self._file_seed_cache_control, CC.FLAGS_EXPAND_PERPENDICULAR )
imports_panel.Add( self._file_download_control, CC.FLAGS_EXPAND_PERPENDICULAR )
#
Version 316
2018-08-01 20:44:57 +00:00
hbox_1 = wx.BoxSizer( wx.HORIZONTAL )
hbox_1.Add( self._file_velocity_status, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
hbox_1.Add( self._checking_pause_button, CC.FLAGS_VCENTER )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
hbox_2 = wx.BoxSizer( wx.HORIZONTAL )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
hbox_2.Add( self._watcher_status, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
hbox_2.Add( self._check_now_button, CC.FLAGS_VCENTER )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
checker_panel.Add( hbox_1, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
checker_panel.Add( hbox_2, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
Version 315
2018-07-18 21:07:15 +00:00
checker_panel.Add( self._gallery_seed_log_control, CC.FLAGS_EXPAND_PERPENDICULAR )
checker_panel.Add( self._checker_options_button, CC.FLAGS_EXPAND_PERPENDICULAR )
checker_panel.Add( self._checker_download_control, CC.FLAGS_EXPAND_PERPENDICULAR )
vbox = wx.BoxSizer( wx.VERTICAL )
vbox.Add( imports_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
vbox.Add( checker_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
self._options_panel.SetSizer( vbox )
Version 316
2018-08-01 20:44:57 +00:00
self.Add( self._watcher_subject, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._watcher_url, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._options_panel, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
self.Add( self._file_import_options, CC.FLAGS_EXPAND_PERPENDICULAR )
self.Add( self._tag_import_options, CC.FLAGS_EXPAND_PERPENDICULAR )
Version 315
2018-07-18 21:07:15 +00:00
#
Version 316
2018-08-01 20:44:57 +00:00
self._UpdateControlsForNewWatcher()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
HG.client_controller.gui.RegisterUIUpdateWindow( self )
Version 315
2018-07-18 21:07:15 +00:00
def _SetCheckerOptions( self, checker_options ):
if self._watcher is not None:
Version 316
2018-08-01 20:44:57 +00:00
self._watcher.SetCheckerOptions( checker_options )
Version 315
2018-07-18 21:07:15 +00:00
def _SetFileImportOptions( self, file_import_options ):
if self._watcher is not None:
self._watcher.SetFileImportOptions( file_import_options )
def _SetTagImportOptions( self, tag_import_options ):
if self._watcher is not None:
self._watcher.SetTagImportOptions( tag_import_options )
def _UpdateControlsForNewWatcher( self ):
if self._watcher is None:
Version 316
2018-08-01 20:44:57 +00:00
self._options_panel.Disable()
self._file_import_options.Disable()
self._tag_import_options.Disable()
self._watcher_subject.SetLabelText( '' )
self._watcher_url.SetValue( '' )
self._current_action.SetLabelText( '' )
self._file_velocity_status.SetLabelText( '' )
self._watcher_status.SetLabelText( '' )
self._file_seed_cache_control.SetFileSeedCache( None )
self._gallery_seed_log_control.SetGallerySeedLog( None )
self._file_download_control.ClearNetworkJob()
self._checker_download_control.ClearNetworkJob()
Version 315
2018-07-18 21:07:15 +00:00
else:
Version 316
2018-08-01 20:44:57 +00:00
self._options_panel.Enable()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._file_import_options.Enable()
self._tag_import_options.Enable()
if self._watcher.HasURL():
url = self._watcher.GetURL()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._watcher_url.SetValue( url )
Version 315
2018-07-18 21:07:15 +00:00
else:
Version 316
2018-08-01 20:44:57 +00:00
self._watcher_url.SetValue( '' )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
checker_options = self._watcher.GetCheckerOptions()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._checker_options_button.SetValue( checker_options )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
file_import_options = self._watcher.GetFileImportOptions()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._file_import_options.SetValue( file_import_options )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
tag_import_options = self._watcher.GetTagImportOptions()
self._tag_import_options.SetValue( tag_import_options )
file_seed_cache = self._watcher.GetFileSeedCache()
self._file_seed_cache_control.SetFileSeedCache( file_seed_cache )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
gallery_seed_log = self._watcher.GetGallerySeedLog()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._gallery_seed_log_control.SetGallerySeedLog( gallery_seed_log )
Version 315
2018-07-18 21:07:15 +00:00
def _UpdateStatus( self ):
Version 316
2018-08-01 20:44:57 +00:00
if self._watcher is not None:
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
( current_action, files_paused, file_velocity_status, next_check_time, watcher_status, subject, checking_status, check_now, checking_paused ) = self._watcher.GetStatus()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if files_paused:
if current_action == '':
current_action = 'paused'
else:
current_action = 'pausing, ' + current_action
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
ClientGUICommon.SetBitmapButtonBitmap( self._files_pause_button, CC.GlobalBMPs.play )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
else:
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
ClientGUICommon.SetBitmapButtonBitmap( self._files_pause_button, CC.GlobalBMPs.pause )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._current_action.SetLabelText( current_action )
self._file_velocity_status.SetLabelText( file_velocity_status )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if checking_paused:
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if watcher_status == '':
watcher_status = 'paused'
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
ClientGUICommon.SetBitmapButtonBitmap( self._checking_pause_button, CC.GlobalBMPs.play )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
else:
if watcher_status == '' and next_check_time is not None:
if HydrusData.TimeHasPassed( next_check_time ):
watcher_status = 'checking imminently'
else:
watcher_status = 'next check ' + HydrusData.TimestampToPrettyTimeDelta( next_check_time, just_now_threshold = 0 )
ClientGUICommon.SetBitmapButtonBitmap( self._checking_pause_button, CC.GlobalBMPs.pause )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._watcher_status.SetLabelText( watcher_status )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if checking_status == ClientImporting.CHECKER_STATUS_404:
self._checking_pause_button.Disable()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
elif checking_status == ClientImporting.CHECKER_STATUS_DEAD:
self._checking_pause_button.Disable()
Version 315
2018-07-18 21:07:15 +00:00
else:
Version 316
2018-08-01 20:44:57 +00:00
self._checking_pause_button.Enable()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if subject in ( '', 'unknown subject' ):
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
subject = 'no subject'
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._watcher_subject.SetLabelText( subject )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if check_now:
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._check_now_button.Disable()
else:
self._check_now_button.Enable()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
( file_network_job, checker_network_job ) = self._watcher.GetNetworkJobs()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._file_download_control.SetNetworkJob( file_network_job )
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._checker_download_control.SetNetworkJob( checker_network_job )
Version 315
2018-07-18 21:07:15 +00:00
def EventCheckNow( self, event ):
Version 316
2018-08-01 20:44:57 +00:00
if self._watcher is not None:
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._watcher.CheckNow()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
self._UpdateStatus()
Version 315
2018-07-18 21:07:15 +00:00
def EventPauseFiles( self, event ):
Version 316
2018-08-01 20:44:57 +00:00
if self._watcher is not None:
self._watcher.PausePlayFiles()
self._UpdateStatus()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
def EventPauseChecking( self, event ):
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if self._watcher is not None:
self._watcher.PausePlayChecking()
self._UpdateStatus()
Version 315
2018-07-18 21:07:15 +00:00
def SetWatcher( self, watcher ):
self._watcher = watcher
self._UpdateControlsForNewWatcher()
Version 316
2018-08-01 20:44:57 +00:00
def TIMERUIUpdate( self ):
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00
if HG.client_controller.gui.IShouldRegularlyUpdate( self ):
self._UpdateStatus()
Version 315
2018-07-18 21:07:15 +00:00
Version 316
2018-08-01 20:44:57 +00:00