2620 lines
96 KiB
Python
2620 lines
96 KiB
Python
import ClientConstants as CC
|
|
import ClientData
|
|
import ClientDefaults
|
|
import ClientExporting
|
|
import ClientGUICommon
|
|
import ClientGUIControls
|
|
import ClientGUIDialogs
|
|
import ClientGUIFrames
|
|
import ClientGUIListBoxes
|
|
import ClientGUIListCtrl
|
|
import ClientGUIScrolledPanels
|
|
import ClientGUIScrolledPanelsEdit
|
|
import ClientGUIPanels
|
|
import ClientGUIPopupMessages
|
|
import ClientGUITime
|
|
import ClientGUITopLevelWindows
|
|
import ClientNetworking
|
|
import ClientNetworkingContexts
|
|
import ClientPaths
|
|
import ClientTags
|
|
import ClientThreading
|
|
import collections
|
|
import cookielib
|
|
import HydrusConstants as HC
|
|
import HydrusData
|
|
import HydrusGlobals as HG
|
|
import HydrusNATPunch
|
|
import HydrusPaths
|
|
import HydrusSerialisable
|
|
import os
|
|
import stat
|
|
import sys
|
|
import threading
|
|
import time
|
|
import traceback
|
|
import wx
|
|
|
|
try:
|
|
|
|
import ClientGUIMatPlotLib
|
|
|
|
MATPLOTLIB_OK = True
|
|
|
|
except ImportError:
|
|
|
|
MATPLOTLIB_OK = False
|
|
|
|
|
|
class AdvancedContentUpdatePanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
COPY = 0
|
|
DELETE = 1
|
|
DELETE_DELETED = 2
|
|
DELETE_FOR_DELETED_FILES = 3
|
|
|
|
ALL_MAPPINGS = 0
|
|
SPECIFIC_MAPPINGS = 1
|
|
SPECIFIC_NAMESPACE = 2
|
|
NAMESPACED = 3
|
|
UNNAMESPACED = 4
|
|
|
|
def __init__( self, parent, service_key, hashes = None ):
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._service_key = service_key
|
|
self._hashes = hashes
|
|
|
|
service = HG.client_controller.services_manager.GetService( self._service_key )
|
|
|
|
self._service_name = service.GetName()
|
|
|
|
self._command_panel = ClientGUICommon.StaticBox( self, 'database commands' )
|
|
|
|
self._action_dropdown = ClientGUICommon.BetterChoice( self._command_panel )
|
|
self._action_dropdown.Bind( wx.EVT_CHOICE, self.EventChoice )
|
|
self._tag_type_dropdown = ClientGUICommon.BetterChoice( self._command_panel )
|
|
self._action_text = wx.StaticText( self._command_panel, label = 'initialising' )
|
|
self._service_key_dropdown = ClientGUICommon.BetterChoice( self._command_panel )
|
|
|
|
self._go = ClientGUICommon.BetterButton( self._command_panel, 'Go!', self.Go )
|
|
|
|
#
|
|
|
|
self._hta_panel = ClientGUICommon.StaticBox( self, 'hydrus tag archives' )
|
|
|
|
self._import_from_hta = ClientGUICommon.BetterButton( self._hta_panel, 'one-time mass import or delete using a hydrus tag archive', self.ImportFromHTA )
|
|
self._export_to_hta = ClientGUICommon.BetterButton( self._hta_panel, 'export to hydrus tag archive', self.ExportToHTA )
|
|
|
|
#
|
|
|
|
services = [ service for service in HG.client_controller.services_manager.GetServices( HC.TAG_SERVICES ) if service.GetServiceKey() != self._service_key ]
|
|
|
|
if len( services ) > 0:
|
|
|
|
self._action_dropdown.Append( 'copy', self.COPY )
|
|
|
|
|
|
if self._service_key == CC.LOCAL_TAG_SERVICE_KEY:
|
|
|
|
self._action_dropdown.Append( 'delete', self.DELETE )
|
|
self._action_dropdown.Append( 'clear deleted record', self.DELETE_DELETED )
|
|
self._action_dropdown.Append( 'delete from deleted files', self.DELETE_FOR_DELETED_FILES )
|
|
|
|
|
|
self._action_dropdown.Select( 0 )
|
|
|
|
#
|
|
|
|
self._tag_type_dropdown.Append( 'all mappings', self.ALL_MAPPINGS )
|
|
self._tag_type_dropdown.Append( 'all namespaced mappings', self.NAMESPACED )
|
|
self._tag_type_dropdown.Append( 'all unnamespaced mappings', self.UNNAMESPACED )
|
|
self._tag_type_dropdown.Append( 'specific tag\'s mappings', self.SPECIFIC_MAPPINGS )
|
|
self._tag_type_dropdown.Append( 'specific namespace\'s mappings', self.SPECIFIC_NAMESPACE )
|
|
|
|
self._tag_type_dropdown.Select( 0 )
|
|
|
|
#
|
|
|
|
for service in services:
|
|
|
|
self._service_key_dropdown.Append( service.GetName(), service.GetServiceKey() )
|
|
|
|
|
|
self._service_key_dropdown.Select( 0 )
|
|
|
|
#
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( self._action_dropdown, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._tag_type_dropdown, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._action_text, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._service_key_dropdown, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._go, CC.FLAGS_VCENTER )
|
|
|
|
self._command_panel.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
|
|
#
|
|
|
|
self._hta_panel.Add( self._import_from_hta, CC.FLAGS_LONE_BUTTON )
|
|
self._hta_panel.Add( self._export_to_hta, CC.FLAGS_LONE_BUTTON )
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
message = 'Regarding '
|
|
|
|
if self._hashes is None:
|
|
|
|
message += 'all'
|
|
|
|
else:
|
|
|
|
message += HydrusData.ConvertIntToPrettyString( len( self._hashes ) )
|
|
|
|
|
|
message += ' files on ' + self._service_name
|
|
|
|
title_st = ClientGUICommon.BetterStaticText( self, message )
|
|
|
|
title_st.Wrap( 540 )
|
|
|
|
message = 'These advanced operations are powerful, so think before you click. They can lock up your client for a _long_ time, and are not undoable.'
|
|
message += os.linesep * 2
|
|
message += 'You may need to refresh your existing searches to see their effect.'
|
|
|
|
st = ClientGUICommon.BetterStaticText( self, message )
|
|
|
|
st.Wrap( 540 )
|
|
|
|
vbox.Add( title_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( self._command_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( self._hta_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
self.EventChoice( None )
|
|
|
|
|
|
def EventChoice( self, event ):
|
|
|
|
data = self._action_dropdown.GetChoice()
|
|
|
|
if data in ( self.DELETE, self.DELETE_DELETED, self.DELETE_FOR_DELETED_FILES ):
|
|
|
|
self._action_text.SetLabelText( 'from ' + self._service_name )
|
|
|
|
self._service_key_dropdown.Hide()
|
|
|
|
else:
|
|
|
|
self._action_text.SetLabelText( 'from ' + self._service_name + ' to')
|
|
|
|
self._service_key_dropdown.Show()
|
|
|
|
|
|
self.Layout()
|
|
|
|
|
|
def ExportToHTA( self ):
|
|
|
|
ClientTags.ExportToHTA( self, self._service_key, self._hashes )
|
|
|
|
|
|
def Go( self ):
|
|
|
|
# at some point, rewrite this to cope with multiple tags. setsometag is ready to go on that front
|
|
# this should prob be with a listbox so people can enter their new multiple tags in several separate goes, rather than overwriting every time
|
|
|
|
action = self._action_dropdown.GetChoice()
|
|
|
|
tag_type = self._tag_type_dropdown.GetChoice()
|
|
|
|
if tag_type == self.ALL_MAPPINGS:
|
|
|
|
tag = None
|
|
|
|
elif tag_type == self.SPECIFIC_MAPPINGS:
|
|
|
|
with ClientGUIDialogs.DialogTextEntry( self, 'Enter tag' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
entry = dlg.GetValue()
|
|
|
|
tag = ( 'tag', entry )
|
|
|
|
else:
|
|
|
|
return
|
|
|
|
|
|
|
|
elif tag_type == self.SPECIFIC_NAMESPACE:
|
|
|
|
with ClientGUIDialogs.DialogTextEntry( self, 'Enter namespace' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
entry = dlg.GetValue()
|
|
|
|
if entry.endswith( ':' ): entry = entry[:-1]
|
|
|
|
tag = ( 'namespace', entry )
|
|
|
|
else:
|
|
|
|
return
|
|
|
|
|
|
|
|
elif tag_type == self.NAMESPACED:
|
|
|
|
tag = ( 'namespaced', None )
|
|
|
|
elif tag_type == self.UNNAMESPACED:
|
|
|
|
tag = ( 'unnamespaced', None )
|
|
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Are you sure?' ) as dlg:
|
|
|
|
if dlg.ShowModal() != wx.ID_YES:
|
|
|
|
return
|
|
|
|
|
|
|
|
if action == self.COPY:
|
|
|
|
service_key_target = self._service_key_dropdown.GetChoice()
|
|
|
|
content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_MAPPINGS, HC.CONTENT_UPDATE_ADVANCED, ( 'copy', ( tag, self._hashes, service_key_target ) ) )
|
|
|
|
elif action == self.DELETE:
|
|
|
|
content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_MAPPINGS, HC.CONTENT_UPDATE_ADVANCED, ( 'delete', ( tag, self._hashes ) ) )
|
|
|
|
elif action == self.DELETE_DELETED:
|
|
|
|
content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_MAPPINGS, HC.CONTENT_UPDATE_ADVANCED, ( 'delete_deleted', ( tag, self._hashes ) ) )
|
|
|
|
elif action == self.DELETE_FOR_DELETED_FILES:
|
|
|
|
content_update = HydrusData.ContentUpdate( HC.CONTENT_TYPE_MAPPINGS, HC.CONTENT_UPDATE_ADVANCED, ( 'delete_for_deleted_files', ( tag, self._hashes ) ) )
|
|
|
|
|
|
service_keys_to_content_updates = { self._service_key : [ content_update ] }
|
|
|
|
HG.client_controller.Write( 'content_updates', service_keys_to_content_updates )
|
|
|
|
|
|
def ImportFromHTA( self ):
|
|
|
|
text = 'Select the Hydrus Tag Archive\'s location.'
|
|
|
|
with wx.FileDialog( self, message = text, style = wx.FD_OPEN ) as dlg_file:
|
|
|
|
if dlg_file.ShowModal() == wx.ID_OK:
|
|
|
|
path = HydrusData.ToUnicode( dlg_file.GetPath() )
|
|
|
|
ClientTags.ImportFromHTA( self, path, self._service_key, self._hashes )
|
|
|
|
|
|
|
|
|
|
class MigrateDatabasePanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
RESIZED_RATIO = 0.012
|
|
FULLSIZE_RATIO = 0.016
|
|
|
|
def __init__( self, parent, controller ):
|
|
|
|
self._controller = controller
|
|
|
|
self._new_options = self._controller.new_options
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
service_info = HG.client_controller.Read( 'service_info', CC.COMBINED_LOCAL_FILE_SERVICE_KEY )
|
|
|
|
self._all_local_files_total_size = service_info[ HC.SERVICE_INFO_TOTAL_SIZE ]
|
|
|
|
menu_items = []
|
|
|
|
page_func = HydrusData.Call( ClientPaths.LaunchPathInWebBrowser, os.path.join( HC.HELP_DIR, 'database_migration.html' ) )
|
|
|
|
menu_items.append( ( 'normal', 'open the html migration help', 'Open the help page for database migration in your web browesr.', page_func ) )
|
|
|
|
help_button = ClientGUICommon.MenuBitmapButton( self, CC.GlobalBMPs.help, menu_items )
|
|
|
|
help_hbox = ClientGUICommon.WrapInText( help_button, self, 'help for this panel -->', wx.Colour( 0, 0, 255 ) )
|
|
|
|
#
|
|
|
|
info_panel = ClientGUICommon.StaticBox( self, 'locations' )
|
|
|
|
self._current_install_path_st = ClientGUICommon.BetterStaticText( info_panel )
|
|
self._current_db_path_st = ClientGUICommon.BetterStaticText( info_panel )
|
|
self._current_media_paths_st = ClientGUICommon.BetterStaticText( info_panel )
|
|
|
|
current_media_locations_listctrl_panel = ClientGUIListCtrl.BetterListCtrlPanel( info_panel )
|
|
|
|
self._current_media_locations_listctrl = ClientGUIListCtrl.BetterListCtrl( current_media_locations_listctrl_panel, 'db_migration_locations', 6, 36, [ ( 'location', -1 ), ( 'portable?', 12 ), ( 'free space', 12 ), ( 'weight', 10 ), ( 'ideal usage', 24 ), ( 'current usage', 24 ) ], self._ConvertLocationToListCtrlTuples )
|
|
|
|
self._current_media_locations_listctrl.Sort()
|
|
|
|
current_media_locations_listctrl_panel.SetListCtrl( self._current_media_locations_listctrl )
|
|
|
|
current_media_locations_listctrl_panel.AddButton( 'add location', self._AddPath )
|
|
current_media_locations_listctrl_panel.AddButton( 'empty/remove location', self._RemovePaths, enabled_check_func = self._FileLocationSelected )
|
|
current_media_locations_listctrl_panel.AddButton( 'increase weight', self._IncreaseWeight, enabled_check_func = self._FileLocationSelected )
|
|
current_media_locations_listctrl_panel.AddButton( 'decrease weight', self._DecreaseWeight, enabled_check_func = self._FileLocationSelected )
|
|
|
|
self._resized_thumbs_location = wx.TextCtrl( info_panel )
|
|
self._resized_thumbs_location.Disable()
|
|
|
|
self._fullsize_thumbs_location = wx.TextCtrl( info_panel )
|
|
self._fullsize_thumbs_location.Disable()
|
|
|
|
self._resized_thumbs_location_set = ClientGUICommon.BetterButton( info_panel, 'set', self._SetResizedThumbnailLocation )
|
|
self._fullsize_thumbs_location_set = ClientGUICommon.BetterButton( info_panel, 'set', self._SetFullsizeThumbnailLocation )
|
|
|
|
self._resized_thumbs_location_clear = ClientGUICommon.BetterButton( info_panel, 'clear', self._ClearResizedThumbnailLocation )
|
|
self._fullsize_thumbs_location_clear = ClientGUICommon.BetterButton( info_panel, 'clear', self._ClearFullsizeThumbnailLocation )
|
|
|
|
self._rebalance_status_st = ClientGUICommon.BetterStaticText( info_panel, style = wx.ALIGN_RIGHT | wx.ST_NO_AUTORESIZE )
|
|
|
|
self._rebalance_button = ClientGUICommon.BetterButton( info_panel, 'move files now', self._Rebalance )
|
|
|
|
#
|
|
|
|
migration_panel = ClientGUICommon.StaticBox( self, 'migrate entire database' )
|
|
|
|
self._migrate_db_button = ClientGUICommon.BetterButton( migration_panel, 'move entire database and all portable paths', self._MigrateDatabase )
|
|
|
|
#
|
|
|
|
r_hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
r_hbox.Add( ClientGUICommon.BetterStaticText( info_panel, 'resized thumbnail location' ), CC.FLAGS_VCENTER )
|
|
r_hbox.Add( self._resized_thumbs_location, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
|
|
r_hbox.Add( self._resized_thumbs_location_set, CC.FLAGS_VCENTER )
|
|
r_hbox.Add( self._resized_thumbs_location_clear, CC.FLAGS_VCENTER )
|
|
|
|
t_hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
t_hbox.Add( ClientGUICommon.BetterStaticText( info_panel, 'full-size thumbnail location' ), CC.FLAGS_VCENTER )
|
|
t_hbox.Add( self._fullsize_thumbs_location, CC.FLAGS_VCENTER_EXPAND_DEPTH_ONLY )
|
|
t_hbox.Add( self._fullsize_thumbs_location_set, CC.FLAGS_VCENTER )
|
|
t_hbox.Add( self._fullsize_thumbs_location_clear, CC.FLAGS_VCENTER )
|
|
|
|
rebalance_hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
rebalance_hbox.Add( self._rebalance_status_st, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
rebalance_hbox.Add( self._rebalance_button, CC.FLAGS_VCENTER )
|
|
|
|
info_panel.Add( self._current_install_path_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( self._current_db_path_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( self._current_media_paths_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( current_media_locations_listctrl_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
info_panel.Add( r_hbox, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( t_hbox, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( rebalance_hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
|
|
#
|
|
|
|
migration_panel.Add( self._migrate_db_button, CC.FLAGS_LONE_BUTTON )
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( help_hbox, CC.FLAGS_BUTTON_SIZER )
|
|
vbox.Add( info_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( migration_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
min_width = ClientGUICommon.ConvertTextToPixelWidth( self, 100 )
|
|
|
|
self.SetMinSize( ( min_width, -1 ) )
|
|
|
|
self._Update()
|
|
|
|
|
|
def _AddPath( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
with wx.DirDialog( self, 'Select the location' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
path = HydrusData.ToUnicode( dlg.GetPath() )
|
|
|
|
if path in locations_to_ideal_weights:
|
|
|
|
wx.MessageBox( 'You already have that location entered!' )
|
|
|
|
return
|
|
|
|
|
|
if path == resized_thumbnail_override or path == full_size_thumbnail_override:
|
|
|
|
wx.MessageBox( 'That path is already used as a special thumbnail location--please choose another.' )
|
|
|
|
return
|
|
|
|
|
|
self._new_options.SetClientFilesLocation( path, 1 )
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
|
|
def _AdjustWeight( self, amount ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
adjustees = set()
|
|
|
|
for location in self._current_media_locations_listctrl.GetData( only_selected = True ):
|
|
|
|
if location in locations_to_ideal_weights:
|
|
|
|
adjustees.add( location )
|
|
|
|
|
|
|
|
if len( adjustees ) > 0:
|
|
|
|
for location in adjustees:
|
|
|
|
current_weight = locations_to_ideal_weights[ location ]
|
|
|
|
new_amount = current_weight + amount
|
|
|
|
if new_amount > 0:
|
|
|
|
self._new_options.SetClientFilesLocation( location, new_amount )
|
|
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
def _ClearFullsizeThumbnailLocation( self ):
|
|
|
|
self._new_options.SetFullsizeThumbnailOverride( None )
|
|
|
|
self._Update()
|
|
|
|
|
|
def _ClearResizedThumbnailLocation( self ):
|
|
|
|
self._new_options.SetResizedThumbnailOverride( None )
|
|
|
|
self._Update()
|
|
|
|
|
|
def _ConvertLocationToListCtrlTuples( self, location ):
|
|
|
|
f_space = self._all_local_files_total_size
|
|
r_space = self._all_local_files_total_size * self.RESIZED_RATIO
|
|
t_space = self._all_local_files_total_size * self.FULLSIZE_RATIO
|
|
|
|
# ideal
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
# current
|
|
|
|
( locations_to_file_weights, locations_to_fs_thumb_weights, locations_to_r_thumb_weights ) = self._GetLocationsToCurrentWeights()
|
|
|
|
#
|
|
|
|
pretty_location = location
|
|
|
|
portable_location = HydrusPaths.ConvertAbsPathToPortablePath( location )
|
|
portable = not os.path.isabs( portable_location )
|
|
|
|
if portable:
|
|
|
|
pretty_portable = 'yes'
|
|
|
|
else:
|
|
|
|
pretty_portable = 'no'
|
|
|
|
|
|
try:
|
|
|
|
free_space = HydrusPaths.GetFreeSpace( location )
|
|
pretty_free_space = HydrusData.ConvertIntToBytes( free_space )
|
|
|
|
except Exception as e:
|
|
|
|
HydrusData.ShowException( e )
|
|
|
|
message = 'There was a problem finding the free space for "' + location + '"! Perhaps this location does not exist?'
|
|
|
|
wx.MessageBox( message )
|
|
HydrusData.ShowText( message )
|
|
|
|
free_space = 0
|
|
pretty_free_space = 'problem finding free space'
|
|
|
|
|
|
fp = locations_to_file_weights[ location ] / 256.0
|
|
tp = locations_to_fs_thumb_weights[ location ] / 256.0
|
|
rp = locations_to_r_thumb_weights[ location ] / 256.0
|
|
|
|
p = HydrusData.ConvertFloatToPercentage
|
|
|
|
current_bytes = fp * f_space + tp * t_space + rp * r_space
|
|
|
|
current_usage = ( fp, tp, rp )
|
|
|
|
usages = []
|
|
|
|
if fp > 0:
|
|
|
|
usages.append( p( fp ) + ' files' )
|
|
|
|
|
|
if tp > 0:
|
|
|
|
usages.append( p( tp ) + ' full-size thumbnails' )
|
|
|
|
|
|
if rp > 0:
|
|
|
|
usages.append( p( rp ) + ' resized thumbnails' )
|
|
|
|
|
|
if len( usages ) > 0:
|
|
|
|
if fp == tp and tp == rp:
|
|
|
|
usages = [ p( fp ) + ' everything' ]
|
|
|
|
|
|
pretty_current_usage = HydrusData.ConvertIntToBytes( current_bytes ) + ' - ' + ','.join( usages )
|
|
|
|
else:
|
|
|
|
pretty_current_usage = 'nothing'
|
|
|
|
|
|
#
|
|
|
|
if location in locations_to_ideal_weights:
|
|
|
|
ideal_weight = locations_to_ideal_weights[ location ]
|
|
|
|
pretty_ideal_weight = str( int( ideal_weight ) )
|
|
|
|
else:
|
|
|
|
ideal_weight = 0
|
|
|
|
pretty_ideal_weight = 'n/a'
|
|
|
|
|
|
if location in locations_to_ideal_weights:
|
|
|
|
total_ideal_weight = sum( locations_to_ideal_weights.values() )
|
|
|
|
ideal_fp = locations_to_ideal_weights[ location ] / float( total_ideal_weight )
|
|
|
|
else:
|
|
|
|
ideal_fp = 0.0
|
|
|
|
|
|
if full_size_thumbnail_override is None:
|
|
|
|
ideal_tp = ideal_fp
|
|
|
|
else:
|
|
|
|
if location == full_size_thumbnail_override:
|
|
|
|
ideal_tp = 1.0
|
|
|
|
else:
|
|
|
|
ideal_tp = 0.0
|
|
|
|
|
|
|
|
if resized_thumbnail_override is None:
|
|
|
|
ideal_rp = ideal_fp
|
|
|
|
else:
|
|
|
|
if location == resized_thumbnail_override:
|
|
|
|
ideal_rp = 1.0
|
|
|
|
else:
|
|
|
|
ideal_rp = 0.0
|
|
|
|
|
|
|
|
ideal_bytes = ideal_fp * f_space + ideal_tp * t_space + ideal_rp * r_space
|
|
|
|
ideal_usage = ( ideal_fp, ideal_tp, ideal_rp )
|
|
|
|
usages = []
|
|
|
|
if ideal_fp > 0:
|
|
|
|
usages.append( p( ideal_fp ) + ' files' )
|
|
|
|
|
|
if ideal_tp > 0:
|
|
|
|
usages.append( p( ideal_tp ) + ' full-size thumbnails' )
|
|
|
|
|
|
if ideal_rp > 0:
|
|
|
|
usages.append( p( ideal_rp ) + ' resized thumbnails' )
|
|
|
|
|
|
if len( usages ) > 0:
|
|
|
|
if ideal_fp == ideal_tp and ideal_tp == ideal_rp:
|
|
|
|
usages = [ p( ideal_fp ) + ' everything' ]
|
|
|
|
|
|
pretty_ideal_usage = HydrusData.ConvertIntToBytes( ideal_bytes ) + ' - ' + ','.join( usages )
|
|
|
|
else:
|
|
|
|
pretty_ideal_usage = 'nothing'
|
|
|
|
|
|
display_tuple = ( pretty_location, pretty_portable, pretty_free_space, pretty_ideal_weight, pretty_ideal_usage, pretty_current_usage )
|
|
sort_tuple = ( location, portable, free_space, ideal_weight, ideal_usage, current_usage )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _DecreaseWeight( self ):
|
|
|
|
self._AdjustWeight( -1 )
|
|
|
|
|
|
def _FileLocationSelected( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
( locations_to_file_weights, locations_to_fs_thumb_weights, locations_to_r_thumb_weights ) = self._GetLocationsToCurrentWeights()
|
|
|
|
locations = self._current_media_locations_listctrl.GetData( only_selected = True )
|
|
|
|
for location in locations:
|
|
|
|
if location in locations_to_file_weights or location in locations_to_ideal_weights:
|
|
|
|
return True
|
|
|
|
|
|
|
|
return False
|
|
|
|
|
|
def _GetLocationsToCurrentWeights( self ):
|
|
|
|
prefixes_to_locations = HG.client_controller.Read( 'client_files_locations' )
|
|
|
|
locations_to_file_weights = collections.Counter()
|
|
locations_to_fs_thumb_weights = collections.Counter()
|
|
locations_to_r_thumb_weights = collections.Counter()
|
|
|
|
for ( prefix, location ) in prefixes_to_locations.items():
|
|
|
|
if prefix.startswith( 'f' ):
|
|
|
|
locations_to_file_weights[ location ] += 1
|
|
|
|
|
|
if prefix.startswith( 't' ):
|
|
|
|
locations_to_fs_thumb_weights[ location ] += 1
|
|
|
|
|
|
if prefix.startswith( 'r' ):
|
|
|
|
locations_to_r_thumb_weights[ location ] += 1
|
|
|
|
|
|
|
|
return ( locations_to_file_weights, locations_to_fs_thumb_weights, locations_to_r_thumb_weights )
|
|
|
|
|
|
def _GetListCtrlLocations( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
# current
|
|
|
|
( locations_to_file_weights, locations_to_fs_thumb_weights, locations_to_r_thumb_weights ) = self._GetLocationsToCurrentWeights()
|
|
|
|
#
|
|
|
|
all_locations = set()
|
|
|
|
all_locations.update( locations_to_ideal_weights.keys() )
|
|
|
|
if resized_thumbnail_override is not None:
|
|
|
|
all_locations.add( resized_thumbnail_override )
|
|
|
|
|
|
if full_size_thumbnail_override is not None:
|
|
|
|
all_locations.add( full_size_thumbnail_override )
|
|
|
|
|
|
all_locations.update( locations_to_file_weights.keys() )
|
|
all_locations.update( locations_to_fs_thumb_weights.keys() )
|
|
all_locations.update( locations_to_r_thumb_weights.keys() )
|
|
|
|
all_locations = list( all_locations )
|
|
|
|
return all_locations
|
|
|
|
|
|
def _IncreaseWeight( self ):
|
|
|
|
self._AdjustWeight( 1 )
|
|
|
|
|
|
def _MigrateDatabase( self ):
|
|
|
|
message = 'This operation will move your database files and any \'portable\' paths. It is a big job that will require a client shutdown and need you to create a new shortcut before you can launch it again.'
|
|
message += os.linesep * 2
|
|
message += 'If you have not read the database migration help or otherwise do not know what is going on here, turn back now!'
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, message, yes_label = 'do it', no_label = 'forget it' ) as dlg_1:
|
|
|
|
if dlg_1.ShowModal() == wx.ID_YES:
|
|
|
|
source = self._controller.GetDBDir()
|
|
|
|
with wx.DirDialog( self, message = 'Choose new database location.' ) as dlg_2:
|
|
|
|
dlg_2.SetPath( source )
|
|
|
|
if dlg_2.ShowModal() == wx.ID_OK:
|
|
|
|
dest = dlg_2.GetPath()
|
|
|
|
if source == dest:
|
|
|
|
wx.MessageBox( 'That is the same location!' )
|
|
|
|
return
|
|
|
|
|
|
if len( os.listdir( dest ) ) > 0:
|
|
|
|
message = dest + ' is not empty! Please select an empty destination--if your situation is more complicated, please do this move manually! Feel free to ask hydrus dev for help.'
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, message ) as dlg_not_empty:
|
|
|
|
if dlg_not_empty.ShowModal() != wx.ID_YES:
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
message = 'Here is the client\'s best guess at your new launch command. Make sure it looks correct and copy it to your clipboard. Update your program shortcut when the transfer is complete.'
|
|
message += os.linesep * 2
|
|
message += 'Hit ok to close the client and start the transfer, cancel to back out.'
|
|
|
|
me = sys.argv[0]
|
|
|
|
shortcut = '"' + me + '" -d="' + dest + '"'
|
|
|
|
with ClientGUIDialogs.DialogTextEntry( self, message, default = shortcut ) as dlg_3:
|
|
|
|
if dlg_3.ShowModal() == wx.ID_OK:
|
|
|
|
# careful with this stuff!
|
|
# the app's mainloop didn't want to exit for me, for a while, because this dialog didn't have time to exit before the thread's dialog laid a new event loop on top
|
|
# the confused event loops lead to problems at a C++ level in ShowModal not being able to do the Destroy because parent stuff had already died
|
|
# this works, so leave it alone if you can
|
|
|
|
wx.CallAfter( self.GetParent().DoOK )
|
|
|
|
prefixes_to_locations = self._controller.Read( 'client_files_locations' )
|
|
|
|
portable_locations = []
|
|
|
|
for location in set( prefixes_to_locations.values() ):
|
|
|
|
if not os.path.exists( location ):
|
|
|
|
continue
|
|
|
|
|
|
portable_location = HydrusPaths.ConvertAbsPathToPortablePath( location )
|
|
portable = not os.path.isabs( portable_location )
|
|
|
|
if portable:
|
|
|
|
portable_locations.append( portable_location )
|
|
|
|
|
|
|
|
HG.client_controller.CallToThreadLongRunning( THREADMigrateDatabase, self._controller, source, portable_locations, dest )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _Rebalance( self ):
|
|
|
|
job_key = ClientThreading.JobKey( cancellable = True )
|
|
|
|
job_key.SetVariable( 'popup_title', 'rebalancing files' )
|
|
|
|
self._controller.CallToThread( self._controller.client_files_manager.Rebalance, job_key )
|
|
|
|
with ClientGUITopLevelWindows.DialogNullipotentVetoable( self, 'migrating files' ) as dlg:
|
|
|
|
panel = ClientGUIPopupMessages.PopupMessageDialogPanel( dlg, job_key )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
dlg.ShowModal()
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
def _RemovePaths( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
removees = set()
|
|
|
|
for location in self._current_media_locations_listctrl.GetData( only_selected = True ):
|
|
|
|
if location in locations_to_ideal_weights:
|
|
|
|
removees.add( location )
|
|
|
|
|
|
|
|
# eventually have a check and veto if not enough size on the destination partition
|
|
|
|
if len( removees ) == 0:
|
|
|
|
wx.MessageBox( 'Please select some locations with weight.' )
|
|
|
|
elif len( removees ) == len( locations_to_ideal_weights ):
|
|
|
|
wx.MessageBox( 'You cannot empty every single location--please add a new place for the files to be moved to and then try again.' )
|
|
|
|
else:
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Are you sure? This will schedule all the selected locations to have all their current files removed.' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
for location in removees:
|
|
|
|
self._new_options.RemoveClientFilesLocation( location )
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
|
|
|
|
def _SetFullsizeThumbnailLocation( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
with wx.DirDialog( self ) as dlg:
|
|
|
|
if full_size_thumbnail_override is not None:
|
|
|
|
dlg.SetPath( full_size_thumbnail_override )
|
|
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
path = dlg.GetPath()
|
|
|
|
if path in locations_to_ideal_weights:
|
|
|
|
wx.MessageBox( 'That path already exists as a regular file location! Please choose another.' )
|
|
|
|
else:
|
|
|
|
self._new_options.SetFullsizeThumbnailOverride( path )
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
|
|
|
|
def _SetResizedThumbnailLocation( self ):
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
with wx.DirDialog( self ) as dlg:
|
|
|
|
if resized_thumbnail_override is not None:
|
|
|
|
dlg.SetPath( resized_thumbnail_override )
|
|
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
path = dlg.GetPath()
|
|
|
|
if path in locations_to_ideal_weights:
|
|
|
|
wx.MessageBox( 'That path already exists as a regular file location! Please choose another.' )
|
|
|
|
else:
|
|
|
|
self._new_options.SetResizedThumbnailOverride( path )
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
|
|
|
|
def _Update( self ):
|
|
|
|
approx_total_db_size = self._controller.db.GetApproxTotalFileSize()
|
|
|
|
self._current_db_path_st.SetLabelText( 'database (about ' + HydrusData.ConvertIntToBytes( approx_total_db_size ) + '): ' + self._controller.GetDBDir() )
|
|
self._current_install_path_st.SetLabelText( 'install: ' + HC.BASE_DIR )
|
|
|
|
approx_total_client_files = self._all_local_files_total_size
|
|
approx_total_resized_thumbs = self._all_local_files_total_size * self.RESIZED_RATIO
|
|
approx_total_fullsize_thumbs = self._all_local_files_total_size * self.FULLSIZE_RATIO
|
|
|
|
label_components = []
|
|
|
|
label_components.append( 'media (about ' + HydrusData.ConvertIntToBytes( approx_total_client_files ) + ')' )
|
|
label_components.append( 'resized thumbnails (about ' + HydrusData.ConvertIntToBytes( approx_total_resized_thumbs ) + ')' )
|
|
label_components.append( 'full-size thumbnails (about ' + HydrusData.ConvertIntToBytes( approx_total_fullsize_thumbs ) + ')' )
|
|
|
|
label = ', '.join( label_components ) + ':'
|
|
|
|
self._current_media_paths_st.SetLabelText( label )
|
|
|
|
locations = self._GetListCtrlLocations()
|
|
|
|
self._current_media_locations_listctrl.SetData( locations )
|
|
|
|
#
|
|
|
|
( locations_to_ideal_weights, resized_thumbnail_override, full_size_thumbnail_override ) = self._new_options.GetClientFilesLocationsToIdealWeights()
|
|
|
|
if resized_thumbnail_override is None:
|
|
|
|
self._resized_thumbs_location.SetValue( 'none set' )
|
|
|
|
self._resized_thumbs_location_set.Enable()
|
|
self._resized_thumbs_location_clear.Disable()
|
|
|
|
else:
|
|
|
|
self._resized_thumbs_location.SetValue( resized_thumbnail_override )
|
|
|
|
self._resized_thumbs_location_set.Disable()
|
|
self._resized_thumbs_location_clear.Enable()
|
|
|
|
|
|
if full_size_thumbnail_override is None:
|
|
|
|
self._fullsize_thumbs_location.SetValue( 'none set' )
|
|
|
|
self._fullsize_thumbs_location_set.Enable()
|
|
self._fullsize_thumbs_location_clear.Disable()
|
|
|
|
else:
|
|
|
|
self._fullsize_thumbs_location.SetValue( full_size_thumbnail_override )
|
|
|
|
self._fullsize_thumbs_location_set.Disable()
|
|
self._fullsize_thumbs_location_clear.Enable()
|
|
|
|
|
|
#
|
|
|
|
if self._controller.client_files_manager.RebalanceWorkToDo():
|
|
|
|
self._rebalance_button.Enable()
|
|
|
|
self._rebalance_status_st.SetForegroundColour( ( 128, 0, 0 ) )
|
|
|
|
self._rebalance_status_st.SetLabelText( 'files need to be moved' )
|
|
|
|
else:
|
|
|
|
self._rebalance_button.Disable()
|
|
|
|
self._rebalance_status_st.SetForegroundColour( ( 0, 128, 0 ) )
|
|
|
|
self._rebalance_status_st.SetLabelText( 'all files are in their ideal locations' )
|
|
|
|
|
|
|
|
def THREADMigrateDatabase( controller, source, portable_locations, dest ):
|
|
|
|
time.sleep( 2 ) # important to have this, so the migrate dialog can close itself and clean its event loop, wew
|
|
|
|
def wx_code( job_key ):
|
|
|
|
HG.client_controller.CallLaterWXSafe( controller.gui, 3.0, controller.gui.Exit )
|
|
|
|
# no parent because this has to outlive the gui, obvs
|
|
|
|
style_override = wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER | wx.DIALOG_NO_PARENT
|
|
|
|
with ClientGUITopLevelWindows.DialogNullipotentVetoable( None, 'migrating files', style_override = style_override ) as dlg:
|
|
|
|
panel = ClientGUIPopupMessages.PopupMessageDialogPanel( dlg, job_key )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
dlg.ShowModal()
|
|
|
|
|
|
|
|
db = controller.db
|
|
|
|
job_key = ClientThreading.JobKey( cancel_on_shutdown = False )
|
|
|
|
job_key.SetVariable( 'popup_title', 'migrating database' )
|
|
|
|
wx.CallAfter( wx_code, job_key )
|
|
|
|
try:
|
|
|
|
job_key.SetVariable( 'popup_text_1', 'waiting for db shutdown' )
|
|
|
|
while not db.LoopIsFinished():
|
|
|
|
time.sleep( 1 )
|
|
|
|
|
|
job_key.SetVariable( 'popup_text_1', 'doing the move' )
|
|
|
|
def text_update_hook( text ):
|
|
|
|
job_key.SetVariable( 'popup_text_1', text )
|
|
|
|
|
|
for filename in os.listdir( source ):
|
|
|
|
if filename.startswith( 'client' ) and filename.endswith( '.db' ):
|
|
|
|
job_key.SetVariable( 'popup_text_1', 'moving ' + filename )
|
|
|
|
source_path = os.path.join( source, filename )
|
|
dest_path = os.path.join( dest, filename )
|
|
|
|
HydrusPaths.MergeFile( source_path, dest_path )
|
|
|
|
|
|
|
|
for portable_location in portable_locations:
|
|
|
|
source_path = os.path.join( source, portable_location )
|
|
dest_path = os.path.join( dest, portable_location )
|
|
|
|
HydrusPaths.MergeTree( source_path, dest_path, text_update_hook = text_update_hook )
|
|
|
|
|
|
job_key.SetVariable( 'popup_text_1', 'done!' )
|
|
|
|
except:
|
|
|
|
wx.CallAfter( wx.MessageBox, traceback.format_exc() )
|
|
|
|
job_key.SetVariable( 'popup_text_1', 'error!' )
|
|
|
|
finally:
|
|
|
|
time.sleep( 3 )
|
|
|
|
job_key.Finish()
|
|
|
|
|
|
class ReviewAllBandwidthPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, controller ):
|
|
|
|
self._controller = controller
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._history_time_delta_threshold = ClientGUITime.TimeDeltaButton( self, days = True, hours = True, minutes = True, seconds = True )
|
|
self._history_time_delta_threshold.Bind( ClientGUITime.EVT_TIME_DELTA, self.EventTimeDeltaChanged )
|
|
|
|
self._history_time_delta_none = wx.CheckBox( self, label = 'show all' )
|
|
self._history_time_delta_none.Bind( wx.EVT_CHECKBOX, self.EventTimeDeltaChanged )
|
|
|
|
self._bandwidths = ClientGUIListCtrl.BetterListCtrl( self, 'bandwidth review', 20, 30, [ ( 'name', -1 ), ( 'type', 14 ), ( 'current usage', 14 ), ( 'past 24 hours', 15 ), ( 'search distance', 17 ), ( 'this month', 12 ), ( 'has specific rules', 18 ), ( 'blocked?', 10 ) ], self._ConvertNetworkContextsToListCtrlTuples, activation_callback = self.ShowNetworkContext )
|
|
|
|
self._edit_default_bandwidth_rules_button = ClientGUICommon.BetterButton( self, 'edit default bandwidth rules', self._EditDefaultBandwidthRules )
|
|
|
|
self._reset_default_bandwidth_rules_button = ClientGUICommon.BetterButton( self, 'reset default bandwidth rules', self._ResetDefaultBandwidthRules )
|
|
|
|
default_rules_help_button = ClientGUICommon.BetterBitmapButton( self, CC.GlobalBMPs.help, self._ShowDefaultRulesHelp )
|
|
default_rules_help_button.SetToolTip( 'Show help regarding default bandwidth rules.' )
|
|
|
|
self._delete_record_button = ClientGUICommon.BetterButton( self, 'delete selected history', self._DeleteNetworkContexts )
|
|
|
|
#
|
|
|
|
last_review_bandwidth_search_distance = self._controller.new_options.GetNoneableInteger( 'last_review_bandwidth_search_distance' )
|
|
|
|
if last_review_bandwidth_search_distance is None:
|
|
|
|
self._history_time_delta_threshold.SetValue( 86400 * 7 )
|
|
self._history_time_delta_threshold.Disable()
|
|
|
|
self._history_time_delta_none.SetValue( True )
|
|
|
|
else:
|
|
|
|
self._history_time_delta_threshold.SetValue( last_review_bandwidth_search_distance )
|
|
|
|
|
|
self._bandwidths.Sort( 0 )
|
|
|
|
self._update_job = HG.client_controller.CallRepeatingWXSafe( self, 0.5, 5.0, self._Update )
|
|
|
|
#
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( ClientGUICommon.BetterStaticText( self, 'Show network contexts with usage in the past: ' ), CC.FLAGS_VCENTER )
|
|
hbox.Add( self._history_time_delta_threshold, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
hbox.Add( self._history_time_delta_none, CC.FLAGS_VCENTER )
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
button_hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
button_hbox.Add( self._edit_default_bandwidth_rules_button, CC.FLAGS_VCENTER )
|
|
button_hbox.Add( self._reset_default_bandwidth_rules_button, CC.FLAGS_VCENTER )
|
|
button_hbox.Add( default_rules_help_button, CC.FLAGS_VCENTER )
|
|
button_hbox.Add( self._delete_record_button, CC.FLAGS_VCENTER )
|
|
|
|
vbox.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
vbox.Add( self._bandwidths, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
vbox.Add( button_hbox, CC.FLAGS_BUTTON_SIZER )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
|
|
def _ConvertNetworkContextsToListCtrlTuples( self, network_context ):
|
|
|
|
bandwidth_tracker = self._controller.network_engine.bandwidth_manager.GetTracker( network_context )
|
|
|
|
has_rules = not self._controller.network_engine.bandwidth_manager.UsesDefaultRules( network_context )
|
|
|
|
sortable_network_context = ( network_context.context_type, network_context.context_data )
|
|
sortable_context_type = CC.network_context_type_string_lookup[ network_context.context_type ]
|
|
current_usage = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_DATA, 1, for_user = True )
|
|
|
|
day_usage_requests = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_REQUESTS, 86400 )
|
|
day_usage_data = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_DATA, 86400 )
|
|
|
|
day_usage = ( day_usage_data, day_usage_requests )
|
|
|
|
month_usage_requests = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_REQUESTS, None )
|
|
month_usage_data = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_DATA, None )
|
|
|
|
month_usage = ( month_usage_data, month_usage_requests )
|
|
|
|
if self._history_time_delta_none.GetValue():
|
|
|
|
search_usage = 0
|
|
pretty_search_usage = ''
|
|
|
|
else:
|
|
|
|
search_delta = self._history_time_delta_threshold.GetValue()
|
|
|
|
search_usage_requests = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_REQUESTS, search_delta )
|
|
search_usage_data = bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_DATA, search_delta )
|
|
|
|
search_usage = ( search_usage_data, search_usage_requests )
|
|
|
|
pretty_search_usage = HydrusData.ConvertIntToBytes( search_usage_data ) + ' in ' + HydrusData.ConvertIntToPrettyString( search_usage_requests ) + ' requests'
|
|
|
|
|
|
pretty_network_context = network_context.ToUnicode()
|
|
pretty_context_type = CC.network_context_type_string_lookup[ network_context.context_type ]
|
|
|
|
if current_usage == 0:
|
|
|
|
pretty_current_usage = ''
|
|
|
|
else:
|
|
|
|
pretty_current_usage = HydrusData.ConvertIntToBytes( current_usage ) + '/s'
|
|
|
|
|
|
pretty_day_usage = HydrusData.ConvertIntToBytes( day_usage_data ) + ' in ' + HydrusData.ConvertIntToPrettyString( day_usage_requests ) + ' requests'
|
|
pretty_month_usage = HydrusData.ConvertIntToBytes( month_usage_data ) + ' in ' + HydrusData.ConvertIntToPrettyString( month_usage_requests ) + ' requests'
|
|
|
|
if has_rules:
|
|
|
|
pretty_has_rules = 'yes'
|
|
|
|
else:
|
|
|
|
pretty_has_rules = ''
|
|
|
|
|
|
blocked = not self._controller.network_engine.bandwidth_manager.CanDoWork( [ network_context ] )
|
|
|
|
if blocked:
|
|
|
|
pretty_blocked = 'yes'
|
|
|
|
else:
|
|
|
|
pretty_blocked = ''
|
|
|
|
|
|
display_tuple = ( pretty_network_context, pretty_context_type, pretty_current_usage, pretty_day_usage, pretty_search_usage, pretty_month_usage, pretty_has_rules, pretty_blocked )
|
|
sort_tuple = ( sortable_network_context, sortable_context_type, current_usage, day_usage, search_usage, month_usage, has_rules, blocked )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _DeleteNetworkContexts( self ):
|
|
|
|
selected_network_contexts = self._bandwidths.GetData( only_selected = True )
|
|
|
|
if len( selected_network_contexts ) > 0:
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Are you sure? This will delete all bandwidth record for the selected network contexts.' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
self._controller.network_engine.bandwidth_manager.DeleteHistory( selected_network_contexts )
|
|
|
|
self._update_job.Wake()
|
|
|
|
|
|
|
|
|
|
def _EditDefaultBandwidthRules( self ):
|
|
|
|
network_contexts_and_bandwidth_rules = self._controller.network_engine.bandwidth_manager.GetDefaultRules()
|
|
|
|
choice_tuples = [ ( network_context.ToUnicode() + ' (' + str( len( bandwidth_rules.GetRules() ) ) + ' rules)', ( network_context, bandwidth_rules ) ) for ( network_context, bandwidth_rules ) in network_contexts_and_bandwidth_rules ]
|
|
|
|
with ClientGUIDialogs.DialogSelectFromList( self, 'select network context', choice_tuples ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
( network_context, bandwidth_rules ) = dlg.GetChoice()
|
|
|
|
with ClientGUITopLevelWindows.DialogEdit( self, 'edit bandwidth rules for ' + network_context.ToUnicode() ) as dlg_2:
|
|
|
|
summary = network_context.GetSummary()
|
|
|
|
panel = ClientGUIScrolledPanelsEdit.EditBandwidthRulesPanel( dlg_2, bandwidth_rules, summary )
|
|
|
|
dlg_2.SetPanel( panel )
|
|
|
|
if dlg_2.ShowModal() == wx.ID_OK:
|
|
|
|
bandwidth_rules = panel.GetValue()
|
|
|
|
self._controller.network_engine.bandwidth_manager.SetRules( network_context, bandwidth_rules )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _ResetDefaultBandwidthRules( self ):
|
|
|
|
message = 'Reset your \'default\' and \'global\' bandwidth rules to default?'
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, message ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
ClientDefaults.SetDefaultBandwidthManagerRules( self._controller.network_engine.bandwidth_manager )
|
|
|
|
|
|
|
|
|
|
def _ShowDefaultRulesHelp( self ):
|
|
|
|
help = 'Network requests act in multiple contexts. Most use the \'global\' and \'web domain\' network contexts, but a hydrus server request, for instance, will add its own service-specific context, and a subscription will add both itself and its downloader.'
|
|
help += os.linesep * 2
|
|
help += 'If a network context does not have some specific rules set up, it will use its respective default, which may or may not have rules of its own. If you want to set general policy, like "Never download more than 1GB/day from any individual website," or "Limit the entire client to 2MB/s," do it through \'global\' and these defaults.'
|
|
help += os.linesep * 2
|
|
help += 'All contexts\' rules are consulted and have to pass before a request can do work. If you set a 200KB/s limit on a website domain and a 50KB/s limit on global, your download will only ever run at 50KB/s. To make sense, network contexts with broader scope should have more lenient rules.'
|
|
help += os.linesep * 2
|
|
help += 'There are two special \'instance\' contexts, for downloaders and threads. These represent individual queries, either a single gallery search or a single watched thread. It is useful to set rules for these so your searches will gather a fast initial sample of results in the first few minutes--so you can make sure you are happy with them--but otherwise trickle the rest in over time. This keeps your CPU and other bandwidth limits less hammered and helps to avoid accidental downloads of many thousands of small bad files or a few hundred gigantic files all in one go.'
|
|
help += os.linesep * 2
|
|
help += 'Please note that this system bases its calendar dates on UTC/GMT time (it helps servers and clients around the world stay in sync a bit easier). This has no bearing on what, for instance, the \'past 24 hours\' means, but monthly transitions may occur a few hours off whatever your midnight is.'
|
|
help += os.linesep * 2
|
|
help += 'If you do not understand what is going on here, you can safely leave it alone. The default settings make for a _reasonable_ and polite profile that will not accidentally cause you to download way too much in one go or piss off servers by being too aggressive. If you want to throttle your client, the simplest way is to add a simple rule like \'500MB per day\' to the global context.'
|
|
wx.MessageBox( help )
|
|
|
|
|
|
def _Update( self ):
|
|
|
|
if self._history_time_delta_none.GetValue() == True:
|
|
|
|
history_time_delta_threshold = None
|
|
|
|
else:
|
|
|
|
history_time_delta_threshold = self._history_time_delta_threshold.GetValue()
|
|
|
|
|
|
network_contexts = self._controller.network_engine.bandwidth_manager.GetNetworkContextsForUser( history_time_delta_threshold )
|
|
|
|
self._bandwidths.SetData( network_contexts )
|
|
|
|
timer_duration_s = max( len( network_contexts ), 20 )
|
|
|
|
|
|
def EventTimeDeltaChanged( self, event ):
|
|
|
|
if self._history_time_delta_none.GetValue() == True:
|
|
|
|
self._history_time_delta_threshold.Disable()
|
|
|
|
last_review_bandwidth_search_distance = None
|
|
|
|
else:
|
|
|
|
self._history_time_delta_threshold.Enable()
|
|
|
|
last_review_bandwidth_search_distance = self._history_time_delta_threshold.GetValue()
|
|
|
|
|
|
self._controller.new_options.SetNoneableInteger( 'last_review_bandwidth_search_distance', last_review_bandwidth_search_distance )
|
|
|
|
self._update_job.Wake()
|
|
|
|
|
|
def ShowNetworkContext( self ):
|
|
|
|
for network_context in self._bandwidths.GetData( only_selected = True ):
|
|
|
|
parent = self.GetTopLevelParent().GetParent()
|
|
|
|
frame = ClientGUITopLevelWindows.FrameThatTakesScrollablePanel( parent, 'review bandwidth for ' + network_context.ToUnicode() )
|
|
|
|
panel = ReviewNetworkContextBandwidthPanel( frame, self._controller, network_context )
|
|
|
|
frame.SetPanel( panel )
|
|
|
|
|
|
|
|
class ReviewExportFilesPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, flat_media ):
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
new_options = HG.client_controller.new_options
|
|
|
|
self._media_to_paths = {}
|
|
self._existing_filenames = set()
|
|
self._last_phrase_used = ''
|
|
self._last_dir_used = ''
|
|
|
|
self._tags_box = ClientGUICommon.StaticBoxSorterForListBoxTags( self, 'files\' tags' )
|
|
|
|
services_manager = HG.client_controller.services_manager
|
|
|
|
self._neighbouring_txt_tag_service_keys = services_manager.FilterValidServiceKeys( new_options.GetKeyList( 'default_neighbouring_txt_tag_service_keys' ) )
|
|
|
|
t = ClientGUIListBoxes.ListBoxTagsSelection( self._tags_box, include_counts = True, collapse_siblings = True )
|
|
|
|
self._tags_box.SetTagsBox( t )
|
|
|
|
self._tags_box.SetMinSize( ( 220, 300 ) )
|
|
|
|
self._paths = ClientGUIListCtrl.BetterListCtrl( self, 'export_files', 24, 64, [ ( 'number', 8 ), ( 'mime', 20 ), ( 'expected path', -1 ) ], self._ConvertDataToListCtrlTuples, delete_key_callback = self.DeletePaths )
|
|
|
|
self._paths.Sort( 0 )
|
|
|
|
self._export_path_box = ClientGUICommon.StaticBox( self, 'export path' )
|
|
|
|
self._directory_picker = wx.DirPickerCtrl( self._export_path_box )
|
|
self._directory_picker.Bind( wx.EVT_DIRPICKER_CHANGED, self.EventRecalcPaths )
|
|
|
|
self._open_location = wx.Button( self._export_path_box, label = 'open this location' )
|
|
self._open_location.Bind( wx.EVT_BUTTON, self.EventOpenLocation )
|
|
|
|
self._filenames_box = ClientGUICommon.StaticBox( self, 'filenames' )
|
|
|
|
self._pattern = wx.TextCtrl( self._filenames_box )
|
|
|
|
self._update = wx.Button( self._filenames_box, label = 'update' )
|
|
self._update.Bind( wx.EVT_BUTTON, self.EventRecalcPaths )
|
|
|
|
self._examples = ClientGUICommon.ExportPatternButton( self._filenames_box )
|
|
|
|
text = 'This will export all the files\' tags, newline separated, into .txts beside the files themselves.'
|
|
|
|
self._export_tag_txts = wx.CheckBox( self, label = 'export tags to .txt files?' )
|
|
self._export_tag_txts.SetToolTip( text )
|
|
self._export_tag_txts.Bind( wx.EVT_CHECKBOX, self.EventExportTagTxtsChanged )
|
|
|
|
self._export = wx.Button( self, label = 'export' )
|
|
self._export.Bind( wx.EVT_BUTTON, self.EventExport )
|
|
|
|
#
|
|
|
|
export_path = ClientExporting.GetExportPath()
|
|
|
|
self._directory_picker.SetPath( export_path )
|
|
|
|
phrase = new_options.GetString( 'export_phrase' )
|
|
|
|
self._pattern.SetValue( phrase )
|
|
|
|
if len( self._neighbouring_txt_tag_service_keys ) > 0:
|
|
|
|
self._export_tag_txts.SetValue( True )
|
|
|
|
|
|
self._paths.SetData( list( enumerate( flat_media ) ) )
|
|
|
|
#
|
|
|
|
top_hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
top_hbox.Add( self._tags_box, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
top_hbox.Add( self._paths, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( self._directory_picker, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
hbox.Add( self._open_location, CC.FLAGS_VCENTER )
|
|
|
|
self._export_path_box.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( self._pattern, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
hbox.Add( self._update, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._examples, CC.FLAGS_VCENTER )
|
|
|
|
self._filenames_box.Add( hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( top_hbox, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
|
|
vbox.Add( self._export_path_box, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( self._filenames_box, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( self._export_tag_txts, CC.FLAGS_LONE_BUTTON )
|
|
vbox.Add( self._export, CC.FLAGS_LONE_BUTTON )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
self._RefreshTags()
|
|
|
|
wx.CallAfter( self._export.SetFocus )
|
|
|
|
self._paths.Bind( wx.EVT_LIST_ITEM_SELECTED, self.EventSelectPath )
|
|
self._paths.Bind( wx.EVT_LIST_ITEM_DESELECTED, self.EventSelectPath )
|
|
|
|
|
|
def _ConvertDataToListCtrlTuples( self, data ):
|
|
|
|
( ordering_index, media ) = data
|
|
|
|
number = ordering_index
|
|
mime = media.GetMime()
|
|
path = self._GetPath( media )
|
|
|
|
pretty_number = HydrusData.ConvertIntToPrettyString( ordering_index + 1 )
|
|
pretty_mime = HC.mime_string_lookup[ mime ]
|
|
pretty_path = path
|
|
|
|
display_tuple = ( pretty_number, pretty_mime, pretty_path )
|
|
sort_tuple = ( number, pretty_mime, path )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _GetPath( self, media ):
|
|
|
|
if media in self._media_to_paths:
|
|
|
|
return self._media_to_paths[ media ]
|
|
|
|
|
|
directory = HydrusData.ToUnicode( self._directory_picker.GetPath() )
|
|
|
|
pattern = self._pattern.GetValue()
|
|
|
|
terms = ClientExporting.ParseExportPhrase( pattern )
|
|
|
|
filename = ClientExporting.GenerateExportFilename( directory, media, terms )
|
|
|
|
i = 1
|
|
|
|
while filename in self._existing_filenames:
|
|
|
|
filename = ClientExporting.GenerateExportFilename( directory, media, terms + [ ( 'string', ' (' + str( i ) + ')' ) ] )
|
|
|
|
i += 1
|
|
|
|
|
|
path = os.path.join( directory, filename )
|
|
|
|
self._existing_filenames.add( filename )
|
|
self._media_to_paths[ media ] = path
|
|
|
|
return path
|
|
|
|
|
|
def _RefreshPaths( self ):
|
|
|
|
pattern = self._pattern.GetValue()
|
|
dir_path = self._directory_picker.GetPath()
|
|
|
|
if pattern == self._last_phrase_used and dir_path == self._last_dir_used:
|
|
|
|
return
|
|
|
|
|
|
self._last_phrase_used = pattern
|
|
self._last_dir_used = dir_path
|
|
|
|
HG.client_controller.new_options.SetString( 'export_phrase', pattern )
|
|
|
|
self._existing_filenames = set()
|
|
self._media_to_paths = {}
|
|
|
|
self._paths.UpdateDatas()
|
|
|
|
|
|
def _RefreshTags( self ):
|
|
|
|
data = self._paths.GetData( only_selected = True )
|
|
|
|
if len( data ) == 0:
|
|
|
|
data = self._paths.GetData()
|
|
|
|
|
|
all_media = [ media for ( ordering_index, media ) in data ]
|
|
|
|
self._tags_box.SetTagsByMedia( all_media )
|
|
|
|
|
|
def DeletePaths( self ):
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Remove all selected?' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
self._paths.DeleteSelected()
|
|
|
|
|
|
|
|
|
|
def EventExport( self, event ):
|
|
|
|
self._RefreshPaths()
|
|
|
|
export_tag_txts = self._export_tag_txts.GetValue()
|
|
|
|
directory = self._directory_picker.GetPath()
|
|
|
|
HydrusPaths.MakeSureDirectoryExists( directory )
|
|
|
|
pattern = self._pattern.GetValue()
|
|
|
|
new_options = HG.client_controller.new_options
|
|
|
|
new_options.SetKeyList( 'default_neighbouring_txt_tag_service_keys', self._neighbouring_txt_tag_service_keys )
|
|
|
|
new_options.SetString( 'export_phrase', pattern )
|
|
|
|
terms = ClientExporting.ParseExportPhrase( pattern )
|
|
|
|
client_files_manager = HG.client_controller.client_files_manager
|
|
|
|
self._export.Disable()
|
|
|
|
to_do = self._paths.GetData()
|
|
|
|
num_to_do = len( to_do )
|
|
|
|
def wx_update_label( text ):
|
|
|
|
if not self or not self._export:
|
|
|
|
return
|
|
|
|
|
|
self._export.SetLabel( text )
|
|
|
|
|
|
def wx_done():
|
|
|
|
if not self or not self._export:
|
|
|
|
return
|
|
|
|
|
|
self._export.Enable()
|
|
|
|
|
|
def do_it( neighbouring_txt_tag_service_keys ):
|
|
|
|
for ( index, ( ordering_index, media ) ) in enumerate( to_do ):
|
|
|
|
try:
|
|
|
|
wx.CallAfter( wx_update_label, HydrusData.ConvertValueRangeToPrettyString( index + 1, num_to_do ) )
|
|
|
|
hash = media.GetHash()
|
|
mime = media.GetMime()
|
|
|
|
path = self._GetPath( media )
|
|
|
|
if export_tag_txts:
|
|
|
|
tags_manager = media.GetTagsManager()
|
|
|
|
tags = set()
|
|
|
|
siblings_manager = HG.controller.GetManager( 'tag_siblings' )
|
|
|
|
tag_censorship_manager = HG.client_controller.GetManager( 'tag_censorship' )
|
|
|
|
for service_key in neighbouring_txt_tag_service_keys:
|
|
|
|
current_tags = tags_manager.GetCurrent( service_key )
|
|
|
|
current_tags = siblings_manager.CollapseTags( service_key, current_tags )
|
|
|
|
current_tags = tag_censorship_manager.FilterTags( service_key, current_tags )
|
|
|
|
tags.update( current_tags )
|
|
|
|
|
|
tags = list( tags )
|
|
|
|
tags.sort()
|
|
|
|
txt_path = path + '.txt'
|
|
|
|
with open( txt_path, 'wb' ) as f:
|
|
|
|
f.write( HydrusData.ToByteString( os.linesep.join( tags ) ) )
|
|
|
|
|
|
|
|
source_path = client_files_manager.GetFilePath( hash, mime )
|
|
|
|
HydrusPaths.MirrorFile( source_path, path )
|
|
|
|
try: os.chmod( path, stat.S_IWRITE | stat.S_IREAD )
|
|
except: pass
|
|
|
|
except:
|
|
|
|
wx.CallAfter( wx.MessageBox, 'Encountered a problem while attempting to export file with index ' + str( ordering_index + 1 ) + ':' + os.linesep * 2 + traceback.format_exc() )
|
|
|
|
break
|
|
|
|
|
|
|
|
wx.CallAfter( wx_update_label, 'done!' )
|
|
|
|
time.sleep( 1 )
|
|
|
|
wx.CallAfter( wx_update_label, 'export' )
|
|
|
|
wx.CallAfter( wx_done )
|
|
|
|
|
|
HG.client_controller.CallToThread( do_it, self._neighbouring_txt_tag_service_keys )
|
|
|
|
|
|
def EventExportTagTxtsChanged( self, event ):
|
|
|
|
if self._export_tag_txts.GetValue() == True:
|
|
|
|
services_manager = HG.client_controller.services_manager
|
|
|
|
tag_services = services_manager.GetServices( HC.TAG_SERVICES )
|
|
|
|
list_of_tuples = [ ( service.GetName(), service.GetServiceKey(), service.GetServiceKey() in self._neighbouring_txt_tag_service_keys ) for service in tag_services ]
|
|
|
|
list_of_tuples.sort()
|
|
|
|
with ClientGUIDialogs.DialogCheckFromList( self, 'select tag services', list_of_tuples ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
self._neighbouring_txt_tag_service_keys = dlg.GetChecked()
|
|
|
|
if len( self._neighbouring_txt_tag_service_keys ) == 0:
|
|
|
|
self._export_tag_txts.SetValue( False )
|
|
|
|
else:
|
|
|
|
self._export_tag_txts.SetValue( True )
|
|
|
|
|
|
else:
|
|
|
|
self._export_tag_txts.SetValue( False )
|
|
|
|
|
|
|
|
else:
|
|
|
|
self._neighbouring_txt_tag_service_keys = []
|
|
|
|
|
|
|
|
def EventOpenLocation( self, event ):
|
|
|
|
directory = self._directory_picker.GetPath()
|
|
|
|
if directory is not None and directory != '':
|
|
|
|
HydrusPaths.LaunchDirectory( directory )
|
|
|
|
|
|
|
|
def EventRecalcPaths( self, event ):
|
|
|
|
self._RefreshPaths()
|
|
|
|
|
|
def EventSelectPath( self, event ):
|
|
|
|
self._RefreshTags()
|
|
|
|
|
|
class ReviewNetworkContextBandwidthPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, controller, network_context ):
|
|
|
|
self._controller = controller
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._network_context = network_context
|
|
|
|
self._bandwidth_rules = self._controller.network_engine.bandwidth_manager.GetRules( self._network_context )
|
|
self._bandwidth_tracker = self._controller.network_engine.bandwidth_manager.GetTracker( self._network_context )
|
|
|
|
self._last_fetched_rule_rows = set()
|
|
|
|
#
|
|
|
|
info_panel = ClientGUICommon.StaticBox( self, 'description' )
|
|
|
|
description = CC.network_context_type_description_lookup[ self._network_context.context_type ]
|
|
|
|
self._name = ClientGUICommon.BetterStaticText( info_panel, label = self._network_context.ToUnicode() )
|
|
self._description = ClientGUICommon.BetterStaticText( info_panel, label = description )
|
|
|
|
#
|
|
|
|
usage_panel = ClientGUICommon.StaticBox( self, 'usage' )
|
|
|
|
self._current_usage_st = ClientGUICommon.BetterStaticText( usage_panel )
|
|
|
|
self._time_delta_usage_bandwidth_type = ClientGUICommon.BetterChoice( usage_panel )
|
|
self._time_delta_usage_time_delta = ClientGUITime.TimeDeltaButton( usage_panel, days = True, hours = True, minutes = True, seconds = True )
|
|
self._time_delta_usage_st = ClientGUICommon.BetterStaticText( usage_panel )
|
|
|
|
#
|
|
|
|
rules_panel = ClientGUICommon.StaticBox( self, 'rules' )
|
|
|
|
self._uses_default_rules_st = ClientGUICommon.BetterStaticText( rules_panel, style = wx.ALIGN_CENTER )
|
|
|
|
self._rules_rows_panel = wx.Panel( rules_panel )
|
|
|
|
self._use_default_rules_button = ClientGUICommon.BetterButton( rules_panel, 'use default rules', self._UseDefaultRules )
|
|
self._edit_rules_button = ClientGUICommon.BetterButton( rules_panel, 'edit rules', self._EditRules )
|
|
|
|
#
|
|
|
|
self._time_delta_usage_time_delta.SetValue( 86400 )
|
|
|
|
for bandwidth_type in ( HC.BANDWIDTH_TYPE_DATA, HC.BANDWIDTH_TYPE_REQUESTS ):
|
|
|
|
self._time_delta_usage_bandwidth_type.Append( HC.bandwidth_type_string_lookup[ bandwidth_type ], bandwidth_type )
|
|
|
|
|
|
self._time_delta_usage_bandwidth_type.SelectClientData( HC.BANDWIDTH_TYPE_DATA )
|
|
|
|
monthly_usage = self._bandwidth_tracker.GetMonthlyDataUsage()
|
|
|
|
if len( monthly_usage ) > 0:
|
|
|
|
if MATPLOTLIB_OK:
|
|
|
|
self._barchart_canvas = ClientGUIMatPlotLib.BarChartBandwidthHistory( usage_panel, monthly_usage )
|
|
|
|
else:
|
|
|
|
self._barchart_canvas = ClientGUICommon.BetterStaticText( usage_panel, 'Could not find matplotlib, so cannot display bar chart here.' )
|
|
|
|
|
|
else:
|
|
|
|
self._barchart_canvas = ClientGUICommon.BetterStaticText( usage_panel, 'No usage yet, so no usage history to show.' )
|
|
|
|
|
|
#
|
|
|
|
info_panel.Add( self._name, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
info_panel.Add( self._description, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
#
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( self._time_delta_usage_bandwidth_type, CC.FLAGS_VCENTER )
|
|
hbox.Add( ClientGUICommon.BetterStaticText( usage_panel, ' in the past ' ), CC.FLAGS_VCENTER )
|
|
hbox.Add( self._time_delta_usage_time_delta, CC.FLAGS_VCENTER )
|
|
hbox.Add( self._time_delta_usage_st, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
usage_panel.Add( self._current_usage_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
usage_panel.Add( hbox, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
usage_panel.Add( self._barchart_canvas, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
#
|
|
|
|
hbox = wx.BoxSizer( wx.HORIZONTAL )
|
|
|
|
hbox.Add( self._edit_rules_button, CC.FLAGS_SIZER_VCENTER )
|
|
hbox.Add( self._use_default_rules_button, CC.FLAGS_SIZER_VCENTER )
|
|
|
|
rules_panel.Add( self._uses_default_rules_st, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
rules_panel.Add( self._rules_rows_panel, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
rules_panel.Add( hbox, CC.FLAGS_BUTTON_SIZER )
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( info_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( usage_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
vbox.Add( rules_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
#
|
|
|
|
self._rules_job = HG.client_controller.CallRepeatingWXSafe( self, 0.5, 5.0, self._UpdateRules )
|
|
|
|
self._update_job = HG.client_controller.CallRepeatingWXSafe( self, 0.5, 1.0, self._Update )
|
|
|
|
|
|
def _EditRules( self ):
|
|
|
|
with ClientGUITopLevelWindows.DialogEdit( self, 'edit bandwidth rules for ' + self._network_context.ToUnicode() ) as dlg:
|
|
|
|
summary = self._network_context.GetSummary()
|
|
|
|
panel = ClientGUIScrolledPanelsEdit.EditBandwidthRulesPanel( dlg, self._bandwidth_rules, summary )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
self._bandwidth_rules = panel.GetValue()
|
|
|
|
self._controller.network_engine.bandwidth_manager.SetRules( self._network_context, self._bandwidth_rules )
|
|
|
|
self._UpdateRules()
|
|
|
|
|
|
|
|
|
|
def _Update( self ):
|
|
|
|
current_usage = self._bandwidth_tracker.GetUsage( HC.BANDWIDTH_TYPE_DATA, 1, for_user = True )
|
|
|
|
pretty_current_usage = 'current usage: ' + HydrusData.ConvertIntToBytes( current_usage ) + '/s'
|
|
|
|
self._current_usage_st.SetLabelText( pretty_current_usage )
|
|
|
|
#
|
|
|
|
bandwidth_type = self._time_delta_usage_bandwidth_type.GetChoice()
|
|
time_delta = self._time_delta_usage_time_delta.GetValue()
|
|
|
|
time_delta_usage = self._bandwidth_tracker.GetUsage( bandwidth_type, time_delta )
|
|
|
|
if bandwidth_type == HC.BANDWIDTH_TYPE_DATA:
|
|
|
|
converter = HydrusData.ConvertIntToBytes
|
|
|
|
elif bandwidth_type == HC.BANDWIDTH_TYPE_REQUESTS:
|
|
|
|
converter = HydrusData.ConvertIntToPrettyString
|
|
|
|
|
|
pretty_time_delta_usage = ': ' + converter( time_delta_usage )
|
|
|
|
self._time_delta_usage_st.SetLabelText( pretty_time_delta_usage )
|
|
|
|
|
|
def _UpdateRules( self ):
|
|
|
|
changes_made = False
|
|
|
|
if self._network_context.IsDefault() or self._network_context == ClientNetworkingContexts.GLOBAL_NETWORK_CONTEXT:
|
|
|
|
if self._use_default_rules_button.IsShown():
|
|
|
|
self._uses_default_rules_st.Hide()
|
|
self._use_default_rules_button.Hide()
|
|
|
|
changes_made = True
|
|
|
|
|
|
else:
|
|
|
|
if self._controller.network_engine.bandwidth_manager.UsesDefaultRules( self._network_context ):
|
|
|
|
self._uses_default_rules_st.SetLabelText( 'uses default rules' )
|
|
|
|
self._edit_rules_button.SetLabel( 'set specific rules' )
|
|
|
|
if self._use_default_rules_button.IsShown():
|
|
|
|
self._use_default_rules_button.Hide()
|
|
|
|
changes_made = True
|
|
|
|
|
|
else:
|
|
|
|
self._uses_default_rules_st.SetLabelText( 'has its own rules' )
|
|
|
|
self._edit_rules_button.SetLabel( 'edit rules' )
|
|
|
|
if not self._use_default_rules_button.IsShown():
|
|
|
|
self._use_default_rules_button.Show()
|
|
|
|
changes_made = True
|
|
|
|
|
|
|
|
|
|
rule_rows = self._bandwidth_rules.GetBandwidthStringsAndGaugeTuples( self._bandwidth_tracker, threshold = 0 )
|
|
|
|
if rule_rows != self._last_fetched_rule_rows:
|
|
|
|
self._last_fetched_rule_rows = rule_rows
|
|
|
|
self._rules_rows_panel.DestroyChildren()
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
for ( status, ( v, r ) ) in rule_rows:
|
|
|
|
tg = ClientGUICommon.TextAndGauge( self._rules_rows_panel )
|
|
|
|
tg.SetValue( status, v, r )
|
|
|
|
vbox.Add( tg, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
|
|
self._rules_rows_panel.SetSizer( vbox )
|
|
|
|
changes_made = True
|
|
|
|
|
|
if changes_made:
|
|
|
|
self.Layout()
|
|
|
|
ClientGUITopLevelWindows.PostSizeChangedEvent( self )
|
|
|
|
|
|
|
|
def _UseDefaultRules( self ):
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Are you sure you want to revert to using the default rules for this context?' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
self._controller.network_engine.bandwidth_manager.DeleteRules( self._network_context )
|
|
|
|
self._rules_job.Wake()
|
|
|
|
|
|
|
|
|
|
class ReviewNetworkJobs( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, controller ):
|
|
|
|
self._controller = controller
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._list_ctrl_panel = ClientGUIListCtrl.BetterListCtrlPanel( self )
|
|
|
|
self._list_ctrl = ClientGUIListCtrl.BetterListCtrl( self._list_ctrl_panel, 'network jobs review', 20, 30, [ ( 'position', 20 ), ( 'url', -1 ), ( 'status', 40 ), ( 'current speed', 8 ), ( 'progress', 12 ) ], self._ConvertDataToListCtrlTuples )
|
|
|
|
self._list_ctrl_panel.SetListCtrl( self._list_ctrl )
|
|
|
|
# button to stop jobs en-masse
|
|
|
|
self._list_ctrl_panel.AddButton( 'refresh snapshot', self._RefreshSnapshot )
|
|
|
|
#
|
|
|
|
self._list_ctrl.Sort( 0 )
|
|
|
|
self._RefreshSnapshot()
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( self._list_ctrl_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
|
|
def _ConvertDataToListCtrlTuples( self, job_row ):
|
|
|
|
( network_engine_status, job ) = job_row
|
|
|
|
position = network_engine_status
|
|
url = job.GetURL()
|
|
( status, current_speed, num_bytes_read, num_bytes_to_read ) = job.GetStatus()
|
|
progress = ( num_bytes_read, num_bytes_to_read )
|
|
|
|
pretty_position = ClientNetworking.job_status_str_lookup[ position ]
|
|
pretty_url = url
|
|
pretty_status = status
|
|
pretty_current_speed = HydrusData.ConvertIntToBytes( current_speed ) + '/s'
|
|
pretty_progress = HydrusData.ConvertValueRangeToBytes( num_bytes_read, num_bytes_to_read )
|
|
|
|
display_tuple = ( pretty_position, pretty_url, pretty_status, pretty_current_speed, pretty_progress )
|
|
sort_tuple = ( position, url, status, current_speed, progress )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _RefreshSnapshot( self ):
|
|
|
|
job_rows = self._controller.network_engine.GetJobsSnapshot()
|
|
|
|
self._list_ctrl.SetData( job_rows )
|
|
|
|
|
|
class ReviewNetworkSessionsPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, session_manager ):
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._session_manager = session_manager
|
|
|
|
listctrl_panel = ClientGUIListCtrl.BetterListCtrlPanel( self )
|
|
|
|
self._listctrl = ClientGUIListCtrl.BetterListCtrl( listctrl_panel, 'review_network_sessions', 32, 34, [ ( 'network context', -1 ), ( 'cookies', 9 ), ( 'expires', 28 ) ], self._ConvertNetworkContextToListCtrlTuple, delete_key_callback = self._Clear, activation_callback = self._Review )
|
|
|
|
self._listctrl.Sort()
|
|
|
|
listctrl_panel.SetListCtrl( self._listctrl )
|
|
|
|
listctrl_panel.AddButton( 'create new', self._Add )
|
|
listctrl_panel.AddButton( 'review', self._Review, enabled_only_on_selection = True )
|
|
listctrl_panel.AddButton( 'clear', self._Clear, enabled_only_on_selection = True )
|
|
listctrl_panel.AddSeparator()
|
|
listctrl_panel.AddButton( 'refresh', self._Update )
|
|
|
|
#
|
|
|
|
self._Update()
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( listctrl_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
|
|
def _Add( self ):
|
|
|
|
with ClientGUITopLevelWindows.DialogEdit( self, 'enter new network context' ) as dlg:
|
|
|
|
network_context = ClientNetworkingContexts.NetworkContext( CC.NETWORK_CONTEXT_DOMAIN, 'example.com' )
|
|
|
|
panel = ClientGUIScrolledPanelsEdit.EditNetworkContextPanel( dlg, network_context, limited_types = ( CC.NETWORK_CONTEXT_DOMAIN, CC.NETWORK_CONTEXT_HYDRUS ), allow_default = False )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
network_context = panel.GetValue()
|
|
|
|
self._AddNetworkContext( network_context )
|
|
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
def _AddNetworkContext( self, network_context ):
|
|
|
|
# this establishes a bare session
|
|
|
|
self._session_manager.GetSession( network_context )
|
|
|
|
|
|
def _Clear( self ):
|
|
|
|
for network_context in self._listctrl.GetData( only_selected = True ):
|
|
|
|
self._session_manager.ClearSession( network_context )
|
|
|
|
self._AddNetworkContext( network_context )
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
def _ConvertNetworkContextToListCtrlTuple( self, network_context ):
|
|
|
|
session = self._session_manager.GetSession( network_context )
|
|
|
|
pretty_network_context = network_context.ToUnicode()
|
|
|
|
number_of_cookies = len( session.cookies )
|
|
pretty_number_of_cookies = HydrusData.ConvertIntToPrettyString( number_of_cookies )
|
|
|
|
expires_numbers = [ c.expires for c in session.cookies if c.expires is not None ]
|
|
|
|
if len( expires_numbers ) == 0:
|
|
|
|
if number_of_cookies > 0:
|
|
|
|
expiry = 0
|
|
pretty_expiry = 'session'
|
|
|
|
else:
|
|
|
|
expiry = -1
|
|
pretty_expiry = ''
|
|
|
|
|
|
else:
|
|
|
|
expiry = max( expires_numbers )
|
|
pretty_expiry = HydrusData.ConvertTimestampToPrettyExpires( expiry )
|
|
|
|
|
|
display_tuple = ( pretty_network_context, pretty_number_of_cookies, pretty_expiry )
|
|
sort_tuple = ( pretty_network_context, number_of_cookies, expiry )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _Review( self ):
|
|
|
|
for network_context in self._listctrl.GetData( only_selected = True ):
|
|
|
|
parent = self.GetTopLevelParent().GetParent()
|
|
|
|
frame = ClientGUITopLevelWindows.FrameThatTakesScrollablePanel( parent, 'review session for ' + network_context.ToUnicode() )
|
|
|
|
panel = ReviewNetworkSessionPanel( frame, self._session_manager, network_context )
|
|
|
|
frame.SetPanel( panel )
|
|
|
|
|
|
|
|
def _Update( self ):
|
|
|
|
network_contexts = [ network_context for network_context in self._session_manager.GetNetworkContexts() if network_context.context_type in ( CC.NETWORK_CONTEXT_DOMAIN, CC.NETWORK_CONTEXT_HYDRUS ) ]
|
|
|
|
self._listctrl.SetData( network_contexts )
|
|
|
|
|
|
class ReviewNetworkSessionPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, session_manager, network_context ):
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._session_manager = session_manager
|
|
self._network_context = network_context
|
|
|
|
self._session = self._session_manager.GetSession( self._network_context )
|
|
|
|
self._description = ClientGUICommon.BetterStaticText( self, network_context.ToUnicode() )
|
|
|
|
listctrl_panel = ClientGUIListCtrl.BetterListCtrlPanel( self )
|
|
|
|
self._listctrl = ClientGUIListCtrl.BetterListCtrl( listctrl_panel, 'review_network_session', 8, 18, [ ( 'name', -1 ), ( 'value', 32 ), ( 'domain', 20 ), ( 'path', 8 ), ( 'expires', 28 ) ], self._ConvertCookieToListCtrlTuple, delete_key_callback = self._Delete, activation_callback = self._Edit )
|
|
|
|
self._listctrl.Sort()
|
|
|
|
listctrl_panel.SetListCtrl( self._listctrl )
|
|
|
|
listctrl_panel.AddButton( 'add', self._Add )
|
|
listctrl_panel.AddButton( 'edit', self._Edit, enabled_only_on_selection = True )
|
|
listctrl_panel.AddButton( 'delete', self._Delete, enabled_only_on_selection = True )
|
|
listctrl_panel.AddSeparator()
|
|
listctrl_panel.AddButton( 'refresh', self._Update )
|
|
|
|
#
|
|
|
|
self._Update()
|
|
|
|
#
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( self._description, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.Add( listctrl_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
|
|
def _Add( self ):
|
|
|
|
with ClientGUITopLevelWindows.DialogEdit( self, 'edit cookie' ) as dlg:
|
|
|
|
name = 'name'
|
|
value = '123'
|
|
|
|
if self._network_context.context_type == CC.NETWORK_CONTEXT_DOMAIN:
|
|
|
|
domain = '.' + self._network_context.context_data
|
|
|
|
else:
|
|
|
|
domain = 'service domain'
|
|
|
|
|
|
path = '/'
|
|
expires = HydrusData.GetNow() + 30 * 86400
|
|
|
|
panel = ClientGUIScrolledPanelsEdit.EditCookiePanel( dlg, name, value, domain, path, expires )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
( name, value, domain, path, expires ) = panel.GetValue()
|
|
|
|
self._SetCookie( name, value, domain, path, expires )
|
|
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
def _ConvertCookieToListCtrlTuple( self, cookie ):
|
|
|
|
name = cookie.name
|
|
pretty_name = name
|
|
|
|
value = cookie.value
|
|
pretty_value = value
|
|
|
|
domain = cookie.domain
|
|
pretty_domain = domain
|
|
|
|
path = cookie.path
|
|
pretty_path = path
|
|
|
|
expiry = cookie.expires
|
|
|
|
if expiry is None:
|
|
|
|
expiry = -1
|
|
pretty_expiry = 'session'
|
|
|
|
else:
|
|
|
|
pretty_expiry = HydrusData.ConvertTimestampToPrettyExpires( expiry )
|
|
|
|
|
|
display_tuple = ( pretty_name, pretty_value, pretty_domain, pretty_path, pretty_expiry )
|
|
sort_tuple = ( name, value, domain, path, expiry )
|
|
|
|
return ( display_tuple, sort_tuple )
|
|
|
|
|
|
def _Delete( self ):
|
|
|
|
with ClientGUIDialogs.DialogYesNo( self, 'Delete all selected cookies?' ) as dlg:
|
|
|
|
if dlg.ShowModal() == wx.ID_YES:
|
|
|
|
for cookie in self._listctrl.GetData( only_selected = True ):
|
|
|
|
domain = cookie.domain
|
|
path = cookie.path
|
|
name = cookie.name
|
|
|
|
self._session.cookies.clear( domain, path, name )
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
|
|
|
|
def _Edit( self ):
|
|
|
|
for cookie in self._listctrl.GetData( only_selected = True ):
|
|
|
|
with ClientGUITopLevelWindows.DialogEdit( self, 'edit cookie' ) as dlg:
|
|
|
|
name = cookie.name
|
|
value = cookie.value
|
|
domain = cookie.domain
|
|
path = cookie.path
|
|
expires = cookie.expires
|
|
|
|
panel = ClientGUIScrolledPanelsEdit.EditCookiePanel( dlg, name, value, domain, path, expires )
|
|
|
|
dlg.SetPanel( panel )
|
|
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
|
|
( name, value, domain, path, expires ) = panel.GetValue()
|
|
|
|
self._SetCookie( name, value, domain, path, expires )
|
|
|
|
else:
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
self._Update()
|
|
|
|
|
|
def _SetCookie( self, name, value, domain, path, expires ):
|
|
|
|
version = 0
|
|
port = None
|
|
port_specified = False
|
|
domain_specified = True
|
|
domain_initial_dot = domain.startswith( '.' )
|
|
path_specified = True
|
|
secure = False
|
|
discard = False
|
|
comment = None
|
|
comment_url = None
|
|
rest = {}
|
|
|
|
cookie = cookielib.Cookie( version, name, value, port, port_specified, domain, domain_specified, domain_initial_dot, path, path_specified, secure, expires, discard, comment, comment_url, rest )
|
|
|
|
self._session.cookies.set_cookie( cookie )
|
|
|
|
|
|
def _Update( self ):
|
|
|
|
self._session = self._session_manager.GetSession( self._network_context )
|
|
|
|
cookies = list( self._session.cookies )
|
|
|
|
self._listctrl.SetData( cookies )
|
|
|
|
|
|
class ReviewServicesPanel( ClientGUIScrolledPanels.ReviewPanel ):
|
|
|
|
def __init__( self, parent, controller ):
|
|
|
|
self._controller = controller
|
|
|
|
ClientGUIScrolledPanels.ReviewPanel.__init__( self, parent )
|
|
|
|
self._notebook = wx.Notebook( self )
|
|
|
|
self._local_listbook = ClientGUICommon.ListBook( self._notebook )
|
|
self._remote_listbook = ClientGUICommon.ListBook( self._notebook )
|
|
|
|
self._notebook.AddPage( self._local_listbook, 'local' )
|
|
self._notebook.AddPage( self._remote_listbook, 'remote' )
|
|
|
|
self._InitialiseServices()
|
|
|
|
self.Bind( wx.EVT_NOTEBOOK_PAGE_CHANGED, self.EventPageChanged )
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.Add( self._notebook, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
self._controller.sub( self, 'RefreshServices', 'notify_new_services_gui' )
|
|
|
|
|
|
def _InitialiseServices( self ):
|
|
|
|
lb = self._notebook.GetCurrentPage()
|
|
|
|
if lb.GetPageCount() == 0:
|
|
|
|
previous_service_key = CC.LOCAL_FILE_SERVICE_KEY
|
|
|
|
else:
|
|
|
|
page = lb.GetCurrentPage().GetCurrentPage()
|
|
|
|
previous_service_key = page.GetServiceKey()
|
|
|
|
|
|
self._local_listbook.DeleteAllPages()
|
|
self._remote_listbook.DeleteAllPages()
|
|
|
|
listbook_dict = {}
|
|
|
|
services = self._controller.services_manager.GetServices( randomised = False )
|
|
|
|
lb_to_select = None
|
|
service_type_name_to_select = None
|
|
service_type_lb = None
|
|
service_name_to_select = None
|
|
|
|
for service in services:
|
|
|
|
service_type = service.GetServiceType()
|
|
|
|
if service_type in HC.LOCAL_SERVICES: parent_listbook = self._local_listbook
|
|
else: parent_listbook = self._remote_listbook
|
|
|
|
if service_type == HC.TAG_REPOSITORY: service_type_name = 'tag repositories'
|
|
elif service_type == HC.FILE_REPOSITORY: service_type_name = 'file repositories'
|
|
elif service_type == HC.MESSAGE_DEPOT: service_type_name = 'message depots'
|
|
elif service_type == HC.SERVER_ADMIN: service_type_name = 'administrative servers'
|
|
elif service_type in HC.LOCAL_FILE_SERVICES: service_type_name = 'files'
|
|
elif service_type == HC.LOCAL_TAG: service_type_name = 'tags'
|
|
elif service_type == HC.LOCAL_RATING_LIKE: service_type_name = 'like/dislike ratings'
|
|
elif service_type == HC.LOCAL_RATING_NUMERICAL: service_type_name = 'numerical ratings'
|
|
elif service_type == HC.LOCAL_BOORU: service_type_name = 'booru'
|
|
elif service_type == HC.IPFS: service_type_name = 'ipfs'
|
|
else: continue
|
|
|
|
if service_type_name not in listbook_dict:
|
|
|
|
listbook = ClientGUICommon.ListBook( parent_listbook )
|
|
|
|
listbook_dict[ service_type_name ] = listbook
|
|
|
|
parent_listbook.AddPage( service_type_name, service_type_name, listbook )
|
|
|
|
|
|
listbook = listbook_dict[ service_type_name ]
|
|
|
|
name = service.GetName()
|
|
|
|
panel_class = ClientGUIPanels.ReviewServicePanel
|
|
|
|
listbook.AddPageArgs( name, name, panel_class, ( listbook, service ), {} )
|
|
|
|
if service.GetServiceKey() == previous_service_key:
|
|
|
|
lb_to_select = parent_listbook
|
|
service_type_name_to_select = service_name_to_select
|
|
service_type_lb = listbook
|
|
name_to_select = name
|
|
|
|
|
|
|
|
if lb_to_select is not None:
|
|
|
|
if self._notebook.GetCurrentPage() != lb_to_select:
|
|
|
|
selection = self._notebook.GetSelection()
|
|
|
|
if selection == 0:
|
|
|
|
self._notebook.SetSelection( 1 )
|
|
|
|
else:
|
|
|
|
self._notebook.SetSelection( 0 )
|
|
|
|
|
|
|
|
lb_to_select.Select( service_name_to_select )
|
|
|
|
service_type_lb.Select( name_to_select )
|
|
|
|
|
|
|
|
def EventPageChanged( self, event ):
|
|
|
|
ClientGUITopLevelWindows.PostSizeChangedEvent( self )
|
|
|
|
|
|
def DoGetBestSize( self ):
|
|
|
|
# this overrides the py stub in ScrolledPanel, which allows for unusual scroll behaviour driven by whatever this returns
|
|
|
|
# wx.Notebook isn't expanding on page change and hence increasing min/virtual size and so on to the scrollable panel above, nullifying the neat expand-on-change-page event
|
|
# so, until I write my own or figure out a clever solution, let's just force it
|
|
|
|
if hasattr( self, '_notebook' ):
|
|
|
|
current_page = self._notebook.GetCurrentPage()
|
|
|
|
( notebook_width, notebook_height ) = self._notebook.GetSize()
|
|
( page_width, page_height ) = current_page.GetSize()
|
|
|
|
extra_width = notebook_width - page_width
|
|
extra_height = notebook_height - page_height
|
|
|
|
( page_best_width, page_best_height ) = current_page.GetBestSize()
|
|
|
|
best_size = ( page_best_width + extra_width, page_best_height + extra_height )
|
|
|
|
return best_size
|
|
|
|
else:
|
|
|
|
return ( -1, -1 )
|
|
|
|
|
|
|
|
def RefreshServices( self ):
|
|
|
|
self._InitialiseServices()
|
|
|
|
|