360 lines
14 KiB
Python
360 lines
14 KiB
Python
import ClientGUICollapsible
|
|
import ClientConstants as CC
|
|
import ClientGUICommon
|
|
import ClientCaches
|
|
import HydrusConstants as HC
|
|
import wx
|
|
|
|
class OptionsPanel( wx.Panel ):
|
|
|
|
def GetInfo( self ): raise NotImplementedError()
|
|
|
|
def SetInfo( self, info ): raise NotImplementedError()
|
|
|
|
class OptionsPanelHentaiFoundry( OptionsPanel ):
|
|
|
|
def __init__( self, parent ):
|
|
|
|
OptionsPanel.__init__( self, parent )
|
|
|
|
def offensive_choice():
|
|
|
|
c = wx.Choice( self )
|
|
|
|
c.Append( 'none', 0 )
|
|
c.Append( 'mild', 1 )
|
|
c.Append( 'moderate', 2 )
|
|
c.Append( 'strong', 3 )
|
|
|
|
c.SetSelection( 3 )
|
|
|
|
return c
|
|
|
|
|
|
self._rating_nudity = offensive_choice()
|
|
self._rating_violence = offensive_choice()
|
|
self._rating_profanity = offensive_choice()
|
|
self._rating_racism = offensive_choice()
|
|
self._rating_sex = offensive_choice()
|
|
self._rating_spoilers = offensive_choice()
|
|
|
|
self._rating_yaoi = wx.CheckBox( self )
|
|
self._rating_yuri = wx.CheckBox( self )
|
|
self._rating_teen = wx.CheckBox( self )
|
|
self._rating_guro = wx.CheckBox( self )
|
|
self._rating_furry = wx.CheckBox( self )
|
|
self._rating_beast = wx.CheckBox( self )
|
|
self._rating_male = wx.CheckBox( self )
|
|
self._rating_female = wx.CheckBox( self )
|
|
self._rating_futa = wx.CheckBox( self )
|
|
self._rating_other = wx.CheckBox( self )
|
|
|
|
self._rating_yaoi.SetValue( True )
|
|
self._rating_yuri.SetValue( True )
|
|
self._rating_teen.SetValue( True )
|
|
self._rating_guro.SetValue( True )
|
|
self._rating_furry.SetValue( True )
|
|
self._rating_beast.SetValue( True )
|
|
self._rating_male.SetValue( True )
|
|
self._rating_female.SetValue( True )
|
|
self._rating_futa.SetValue( True )
|
|
self._rating_other.SetValue( True )
|
|
|
|
self._filter_order = wx.Choice( self )
|
|
|
|
self._filter_order.Append( 'newest first', 'date_new' )
|
|
self._filter_order.Append( 'oldest first', 'date_old' )
|
|
self._filter_order.Append( 'most views first', 'views most' ) # no underscore
|
|
self._filter_order.Append( 'highest rating first', 'rating highest' ) # no underscore
|
|
self._filter_order.Append( 'most favourites first', 'faves most' ) # no underscore
|
|
self._filter_order.Append( 'most popular first', 'popularity most' ) # no underscore
|
|
|
|
self._filter_order.SetSelection( 0 )
|
|
|
|
gridbox = wx.FlexGridSizer( 0, 2 )
|
|
|
|
gridbox.AddGrowableCol( 1, 1 )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'nudity' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_nudity, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'violence' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_violence, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'profanity' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_profanity, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'racism' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_racism, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'sex' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_sex, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'spoilers' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_spoilers, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'yaoi' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_yaoi, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'yuri' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_yuri, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'teen' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_teen, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'guro' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_guro, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'furry' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_furry, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'beast' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_beast, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'male' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_male, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'female' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_female, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'futa' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_futa, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'other' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._rating_other, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
gridbox.AddF( wx.StaticText( self, label = 'order' ), CC.FLAGS_MIXED )
|
|
gridbox.AddF( self._filter_order, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
self.SetSizer( gridbox )
|
|
|
|
|
|
def GetInfo( self ):
|
|
|
|
info = {}
|
|
|
|
info[ 'rating_nudity' ] = self._rating_nudity.GetClientData( self._rating_nudity.GetSelection() )
|
|
info[ 'rating_violence' ] = self._rating_violence.GetClientData( self._rating_violence.GetSelection() )
|
|
info[ 'rating_profanity' ] = self._rating_profanity.GetClientData( self._rating_profanity.GetSelection() )
|
|
info[ 'rating_racism' ] = self._rating_racism.GetClientData( self._rating_racism.GetSelection() )
|
|
info[ 'rating_sex' ] = self._rating_sex.GetClientData( self._rating_sex.GetSelection() )
|
|
info[ 'rating_spoilers' ] = self._rating_spoilers.GetClientData( self._rating_spoilers.GetSelection() )
|
|
|
|
info[ 'rating_yaoi' ] = int( self._rating_yaoi.GetValue() )
|
|
info[ 'rating_yuri' ] = int( self._rating_yuri.GetValue() )
|
|
info[ 'rating_teen' ] = int( self._rating_teen.GetValue() )
|
|
info[ 'rating_guro' ] = int( self._rating_guro.GetValue() )
|
|
info[ 'rating_furry' ] = int( self._rating_furry.GetValue() )
|
|
info[ 'rating_beast' ] = int( self._rating_beast.GetValue() )
|
|
info[ 'rating_male' ] = int( self._rating_male.GetValue() )
|
|
info[ 'rating_female' ] = int( self._rating_female.GetValue() )
|
|
info[ 'rating_futa' ] = int( self._rating_futa.GetValue() )
|
|
info[ 'rating_other' ] = int( self._rating_other.GetValue() )
|
|
|
|
info[ 'filter_media' ] = 'A'
|
|
info[ 'filter_order' ] = self._filter_order.GetClientData( self._filter_order.GetSelection() )
|
|
info[ 'filter_type' ] = 0
|
|
|
|
return info
|
|
|
|
|
|
def SetInfo( self, info ):
|
|
|
|
self._rating_nudity.SetSelection( info[ 'rating_nudity' ] )
|
|
self._rating_violence.SetSelection( info[ 'rating_violence' ] )
|
|
self._rating_profanity.SetSelection( info[ 'rating_profanity' ] )
|
|
self._rating_racism.SetSelection( info[ 'rating_racism' ] )
|
|
self._rating_sex.SetSelection( info[ 'rating_sex' ] )
|
|
self._rating_spoilers.SetSelection( info[ 'rating_spoilers' ] )
|
|
|
|
self._rating_yaoi.SetValue( bool( info[ 'rating_yaoi' ] ) )
|
|
self._rating_yuri.SetValue( bool( info[ 'rating_yuri' ] ) )
|
|
self._rating_teen.SetValue( bool( info[ 'rating_teen' ] ) )
|
|
self._rating_guro.SetValue( bool( info[ 'rating_guro' ] ) )
|
|
self._rating_furry.SetValue( bool( info[ 'rating_furry' ] ) )
|
|
self._rating_beast.SetValue( bool( info[ 'rating_beast' ] ) )
|
|
self._rating_male.SetValue( bool( info[ 'rating_male' ] ) )
|
|
self._rating_female.SetValue( bool( info[ 'rating_female' ] ) )
|
|
self._rating_futa.SetValue( bool( info[ 'rating_futa' ] ) )
|
|
self._rating_other.SetValue( bool( info[ 'rating_other' ] ) )
|
|
|
|
#info[ 'filter_media' ] = 'A'
|
|
self._filter_order.SetSelection( info[ 'filter_order' ] )
|
|
#info[ 'filter_type' ] = 0
|
|
|
|
|
|
class OptionsPanelImport( OptionsPanel ):
|
|
|
|
def __init__( self, parent ):
|
|
|
|
OptionsPanel.__init__( self, parent )
|
|
|
|
self._auto_archive = wx.CheckBox( self, label = 'archive all imports' )
|
|
|
|
self._exclude_deleted = wx.CheckBox( self, label = 'exclude already deleted files' )
|
|
|
|
self._min_size = ClientGUICommon.NoneableSpinCtrl( self, 'minimum size (KB): ', multiplier = 1024 )
|
|
self._min_size.SetValue( 5120 )
|
|
|
|
self._min_resolution = ClientGUICommon.NoneableSpinCtrl( self, 'minimum resolution: ', num_dimensions = 2 )
|
|
self._min_resolution.SetValue( ( 50, 50 ) )
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
vbox.AddF( self._auto_archive, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.AddF( self._exclude_deleted, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.AddF( self._min_size, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
vbox.AddF( self._min_resolution, CC.FLAGS_EXPAND_PERPENDICULAR )
|
|
|
|
self.SetSizer( vbox )
|
|
|
|
|
|
def GetInfo( self ):
|
|
|
|
info = {}
|
|
|
|
if self._auto_archive.GetValue(): info[ 'auto_archive' ] = True
|
|
|
|
if self._exclude_deleted.GetValue(): info[ 'exclude_deleted_files' ] = True
|
|
|
|
min_size = self._min_size.GetValue()
|
|
|
|
if min_size is not None: info[ 'min_size' ] = min_size
|
|
|
|
min_resolution = self._min_resolution.GetValue()
|
|
|
|
if min_resolution is not None: info[ 'min_resolution' ] = min_resolution
|
|
|
|
return info
|
|
|
|
|
|
def SetInfo( self, info ):
|
|
|
|
if 'auto_archive' in info: self._auto_archive.SetValue( info[ 'auto_archive' ] )
|
|
else: self._auto_archive.SetValue( False )
|
|
|
|
if 'exclude_deleted_files' in info: self._exclude_deleted.SetValue( info[ 'exclude_deleted_files' ] )
|
|
else: self._exclude_deleted.SetValue( HC.options[ 'exclude_deleted_files' ] )
|
|
|
|
if 'min_size' in info: self._min_size.SetValue( info[ 'min_size' ] )
|
|
else: self._min_size.SetValue( None )
|
|
|
|
if 'min_resolution' in info: self._min_resolution.SetValue( info[ 'min_resolution' ] )
|
|
else: self._min_resolution.SetValue( None )
|
|
|
|
|
|
class OptionsPanelTags( OptionsPanel ):
|
|
|
|
def __init__( self, parent ):
|
|
|
|
OptionsPanel.__init__( self, parent )
|
|
|
|
self._service_keys_to_checkbox_info = {}
|
|
|
|
self._vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
self.SetSizer( self._vbox )
|
|
|
|
|
|
def EventChecked( self, event ):
|
|
|
|
wx.PostEvent( self, wx.CommandEvent( commandType = wx.wxEVT_COMMAND_MENU_SELECTED, winid = ClientCaches.MENU_EVENT_ID_TO_ACTION_CACHE.GetId( 'advanced_tag_options_changed' ) ) )
|
|
|
|
event.Skip()
|
|
|
|
|
|
def GetInfo( self ):
|
|
|
|
result = {}
|
|
|
|
for ( service_key, checkbox_info ) in self._service_keys_to_checkbox_info.items():
|
|
|
|
namespaces = [ namespace for ( namespace, checkbox ) in checkbox_info if checkbox.GetValue() == True ]
|
|
|
|
result[ service_key ] = namespaces
|
|
|
|
|
|
return result
|
|
|
|
|
|
def SetNamespaces( self, namespaces ):
|
|
|
|
self._vbox.Clear( True )
|
|
|
|
self._service_keys_to_checkbox_info = {}
|
|
|
|
services = HC.app.GetManager( 'services' ).GetServices( ( HC.TAG_REPOSITORY, HC.LOCAL_TAG ) )
|
|
|
|
if len( services ) > 0:
|
|
|
|
outer_gridbox = wx.FlexGridSizer( 0, 2 )
|
|
|
|
outer_gridbox.AddGrowableCol( 1, 1 )
|
|
|
|
for service in services:
|
|
|
|
service_key = service.GetServiceKey()
|
|
|
|
self._service_keys_to_checkbox_info[ service_key ] = []
|
|
|
|
outer_gridbox.AddF( wx.StaticText( self, label = service.GetName() ), CC.FLAGS_MIXED )
|
|
|
|
vbox = wx.BoxSizer( wx.VERTICAL )
|
|
|
|
for namespace in namespaces:
|
|
|
|
if namespace == '': label = 'no namespace'
|
|
else: label = namespace
|
|
|
|
namespace_checkbox = wx.CheckBox( self, label = label )
|
|
|
|
namespace_checkbox.Bind( wx.EVT_CHECKBOX, self.EventChecked )
|
|
|
|
self._service_keys_to_checkbox_info[ service_key ].append( ( namespace, namespace_checkbox ) )
|
|
|
|
vbox.AddF( namespace_checkbox, CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
|
|
outer_gridbox.AddF( vbox, CC.FLAGS_MIXED )
|
|
|
|
|
|
self._vbox.AddF( outer_gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
|
|
|
else:
|
|
|
|
self._vbox.AddF( wx.StaticText( self, label = 'no tag repositories' ), CC.FLAGS_EXPAND_BOTH_WAYS )
|
|
|
|
|
|
|
|
def SetInfo( self, new_service_keys_to_namespaces_info ):
|
|
|
|
for ( service_key, checkbox_info ) in self._service_keys_to_checkbox_info.items():
|
|
|
|
if service_key in new_service_keys_to_namespaces_info:
|
|
|
|
new_namespaces_info = new_service_keys_to_namespaces_info[ service_key ]
|
|
|
|
for ( namespace, checkbox ) in checkbox_info:
|
|
|
|
if type( new_namespaces_info ) == bool:
|
|
|
|
value = new_namespaces_info
|
|
|
|
checkbox.SetValue( value )
|
|
|
|
else:
|
|
|
|
new_namespaces = new_namespaces_info
|
|
|
|
if namespace in new_namespaces: checkbox.SetValue( True )
|
|
else: checkbox.SetValue( False )
|
|
|
|
|
|
|
|
else:
|
|
|
|
for ( namespace, checkbox ) in checkbox_info: checkbox.SetValue( False )
|
|
|
|
|
|
|
|
|