1284 lines
49 KiB
Python
1284 lines
49 KiB
Python
|
import typing
|
||
|
|
|
||
|
|
from qtpy import QtCore as QC
|
||
|
|
from qtpy import QtWidgets as QW
|
||
|
|
|
||
|
|
from hydrus.core import HydrusConstants as HC
|
||
|
|
from hydrus.core import HydrusData
|
||
|
|
from hydrus.core import HydrusExceptions
|
||
|
|
from hydrus.client import ClientConstants as CC
|
||
|
|
from hydrus.client import ClientParsing
|
||
|
|
from hydrus.client.gui import ClientGUICommon
|
||
|
|
from hydrus.client.gui import ClientGUIDialogsQuick
|
||
|
|
from hydrus.client.gui import ClientGUIFunctions
|
||
|
|
from hydrus.client.gui import ClientGUIListBoxes
|
||
|
|
from hydrus.client.gui import ClientGUIListCtrl
|
||
|
|
from hydrus.client.gui import ClientGUIScrolledPanels
|
||
|
|
from hydrus.client.gui import ClientGUITopLevelWindowsPanels
|
||
|
|
from hydrus.client.gui import QtPorting as QP
|
||
|
|
|
||
|
|
class EditStringConverterPanel( ClientGUIScrolledPanels.EditPanel ):
|
||
|
|
|
||
|
|
def __init__( self, parent, string_converter, example_string_override = None ):
|
||
|
|
|
||
|
|
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
|
||
|
|
|
||
|
|
transformations_panel = ClientGUIListCtrl.BetterListCtrlPanel( self )
|
||
|
|
|
||
|
|
columns = [ ( '#', 3 ), ( 'transformation', 30 ), ( 'result', -1 ) ]
|
||
|
|
|
||
|
|
self._transformations = ClientGUIListCtrl.BetterListCtrl( transformations_panel, 'string_converter_transformations', 7, 35, columns, self._ConvertTransformationToListCtrlTuples, delete_key_callback = self._DeleteTransformation, activation_callback = self._EditTransformation )
|
||
|
|
|
||
|
|
transformations_panel.SetListCtrl( self._transformations )
|
||
|
|
|
||
|
|
transformations_panel.AddButton( 'add', self._AddTransformation )
|
||
|
|
transformations_panel.AddButton( 'edit', self._EditTransformation, enabled_only_on_selection = True )
|
||
|
|
transformations_panel.AddDeleteButton()
|
||
|
|
|
||
|
|
transformations_panel.AddSeparator()
|
||
|
|
|
||
|
|
transformations_panel.AddButton( 'move up', self._MoveUp, enabled_check_func = self._CanMoveUp )
|
||
|
|
transformations_panel.AddButton( 'move down', self._MoveDown, enabled_check_func = self._CanMoveDown )
|
||
|
|
|
||
|
|
self._example_string = QW.QLineEdit( self )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._transformations.AddDatas( [ ( i + 1, transformation_type, data ) for ( i, ( transformation_type, data ) ) in enumerate( string_converter.transformations ) ] )
|
||
|
|
|
||
|
|
if example_string_override is None:
|
||
|
|
|
||
|
|
self._example_string.setText( string_converter.example_string )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._example_string.setText( example_string_override )
|
||
|
|
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas() # to refresh, now they are all in the list
|
||
|
|
|
||
|
|
self._transformations.Sort( 0 )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
rows.append( ( 'example string: ', self._example_string ) )
|
||
|
|
|
||
|
|
gridbox = ClientGUICommon.WrapInGrid( self, rows )
|
||
|
|
|
||
|
|
vbox = QP.VBoxLayout()
|
||
|
|
|
||
|
|
QP.AddToLayout( vbox, transformations_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
QP.AddToLayout( vbox, gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
|
||
|
|
self.widget().setLayout( vbox )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_string.textChanged.connect( self.EventUpdate )
|
||
|
|
|
||
|
|
|
||
|
|
def _AddTransformation( self ):
|
||
|
|
|
||
|
|
transformation_type = ClientParsing.STRING_TRANSFORMATION_APPEND_TEXT
|
||
|
|
data = 'extra text'
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_converter = self._GetValue()
|
||
|
|
|
||
|
|
example_string_at_this_point = string_converter.Convert( self._example_string.text() )
|
||
|
|
|
||
|
|
except:
|
||
|
|
|
||
|
|
example_string_at_this_point = self._example_string.text()
|
||
|
|
|
||
|
|
|
||
|
|
with ClientGUITopLevelWindowsPanels.DialogEdit( self, 'edit transformation', frame_key = 'deeply_nested_dialog' ) as dlg:
|
||
|
|
|
||
|
|
panel = self._TransformationPanel( dlg, transformation_type, data, example_string_at_this_point )
|
||
|
|
|
||
|
|
dlg.SetPanel( panel )
|
||
|
|
|
||
|
|
if dlg.exec() == QW.QDialog.Accepted:
|
||
|
|
|
||
|
|
number = self._transformations.topLevelItemCount() + 1
|
||
|
|
|
||
|
|
( transformation_type, data ) = panel.GetValue()
|
||
|
|
|
||
|
|
enumerated_transformation = ( number, transformation_type, data )
|
||
|
|
|
||
|
|
self._transformations.AddDatas( ( enumerated_transformation, ) )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas() # need to refresh string after the insertion, so the new row can be included in the parsing calcs
|
||
|
|
|
||
|
|
self._transformations.Sort()
|
||
|
|
|
||
|
|
|
||
|
|
def _CanMoveDown( self ):
|
||
|
|
|
||
|
|
selected_data = self._transformations.GetData( only_selected = True )
|
||
|
|
|
||
|
|
if len( selected_data ) == 1:
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = selected_data[0]
|
||
|
|
|
||
|
|
if number < self._transformations.topLevelItemCount():
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def _CanMoveUp( self ):
|
||
|
|
|
||
|
|
selected_data = self._transformations.GetData( only_selected = True )
|
||
|
|
|
||
|
|
if len( selected_data ) == 1:
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = selected_data[0]
|
||
|
|
|
||
|
|
if number > 1:
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def _ConvertTransformationToListCtrlTuples( self, transformation ):
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = transformation
|
||
|
|
|
||
|
|
pretty_number = HydrusData.ToHumanInt( number )
|
||
|
|
pretty_transformation = ClientParsing.StringConverter.TransformationToString( ( transformation_type, data ) )
|
||
|
|
|
||
|
|
string_converter = self._GetValue()
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
pretty_result = ClientParsing.MakeParsedTextPretty( string_converter.Convert( self._example_string.text(), number ) )
|
||
|
|
|
||
|
|
except HydrusExceptions.StringConvertException as e:
|
||
|
|
|
||
|
|
pretty_result = str( e )
|
||
|
|
|
||
|
|
|
||
|
|
display_tuple = ( pretty_number, pretty_transformation, pretty_result )
|
||
|
|
sort_tuple = ( number, number, number )
|
||
|
|
|
||
|
|
return ( display_tuple, sort_tuple )
|
||
|
|
|
||
|
|
|
||
|
|
def _DeleteTransformation( self ):
|
||
|
|
|
||
|
|
if len( self._transformations.GetData( only_selected = True ) ) > 0:
|
||
|
|
|
||
|
|
text = 'Delete all selected?'
|
||
|
|
|
||
|
|
from hydrus.client.gui import ClientGUIDialogsQuick
|
||
|
|
|
||
|
|
result = ClientGUIDialogsQuick.GetYesNo( self, text )
|
||
|
|
|
||
|
|
if result == QW.QDialog.Accepted:
|
||
|
|
|
||
|
|
self._transformations.DeleteSelected()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
# now we need to shuffle up any missing numbers
|
||
|
|
|
||
|
|
num_rows = self._transformations.topLevelItemCount()
|
||
|
|
|
||
|
|
i = 1
|
||
|
|
search_i = i
|
||
|
|
|
||
|
|
while i <= num_rows:
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
transformation = self._GetTransformation( search_i )
|
||
|
|
|
||
|
|
if search_i != i:
|
||
|
|
|
||
|
|
self._transformations.DeleteDatas( ( transformation, ) )
|
||
|
|
|
||
|
|
( search_i, transformation_type, data ) = transformation
|
||
|
|
|
||
|
|
transformation = ( i, transformation_type, data )
|
||
|
|
|
||
|
|
self._transformations.AddDatas( ( transformation, ) )
|
||
|
|
|
||
|
|
|
||
|
|
i += 1
|
||
|
|
search_i = i
|
||
|
|
|
||
|
|
except HydrusExceptions.DataMissing:
|
||
|
|
|
||
|
|
search_i += 1
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas()
|
||
|
|
|
||
|
|
self._transformations.Sort()
|
||
|
|
|
||
|
|
|
||
|
|
def _EditTransformation( self ):
|
||
|
|
|
||
|
|
selected_data = self._transformations.GetData( only_selected = True )
|
||
|
|
|
||
|
|
for enumerated_transformation in selected_data:
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = enumerated_transformation
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_converter = self._GetValue()
|
||
|
|
|
||
|
|
example_string_at_this_point = string_converter.Convert( self._example_string.text(), number - 1 )
|
||
|
|
|
||
|
|
except:
|
||
|
|
|
||
|
|
example_string_at_this_point = self._example_string.text()
|
||
|
|
|
||
|
|
|
||
|
|
with ClientGUITopLevelWindowsPanels.DialogEdit( self, 'edit transformation', frame_key = 'deeply_nested_dialog' ) as dlg:
|
||
|
|
|
||
|
|
panel = self._TransformationPanel( dlg, transformation_type, data, example_string_at_this_point )
|
||
|
|
|
||
|
|
dlg.SetPanel( panel )
|
||
|
|
|
||
|
|
if dlg.exec() == QW.QDialog.Accepted:
|
||
|
|
|
||
|
|
self._transformations.DeleteDatas( ( enumerated_transformation, ) )
|
||
|
|
|
||
|
|
( transformation_type, data ) = panel.GetValue()
|
||
|
|
|
||
|
|
enumerated_transformation = ( number, transformation_type, data )
|
||
|
|
|
||
|
|
self._transformations.AddDatas( ( enumerated_transformation, ) )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
break
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas()
|
||
|
|
|
||
|
|
self._transformations.Sort()
|
||
|
|
|
||
|
|
|
||
|
|
def _GetTransformation( self, desired_number ):
|
||
|
|
|
||
|
|
for transformation in self._transformations.GetData():
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = transformation
|
||
|
|
|
||
|
|
if number == desired_number:
|
||
|
|
|
||
|
|
return transformation
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
raise HydrusExceptions.DataMissing()
|
||
|
|
|
||
|
|
|
||
|
|
def _GetValue( self ):
|
||
|
|
|
||
|
|
enumerated_transformations = sorted( self._transformations.GetData() )
|
||
|
|
|
||
|
|
transformations = [ ( transformation_type, data ) for ( number, transformation_type, data ) in enumerated_transformations ]
|
||
|
|
|
||
|
|
example_string = self._example_string.text()
|
||
|
|
|
||
|
|
string_converter = ClientParsing.StringConverter( transformations, example_string )
|
||
|
|
|
||
|
|
return string_converter
|
||
|
|
|
||
|
|
|
||
|
|
def _MoveDown( self ):
|
||
|
|
|
||
|
|
selected_transformation = self._transformations.GetData( only_selected = True )[0]
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = selected_transformation
|
||
|
|
|
||
|
|
swap_transformation = self._GetTransformation( number + 1 )
|
||
|
|
|
||
|
|
self._SwapTransformations( selected_transformation, swap_transformation )
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas()
|
||
|
|
|
||
|
|
self._transformations.Sort()
|
||
|
|
|
||
|
|
|
||
|
|
def _MoveUp( self ):
|
||
|
|
|
||
|
|
selected_transformation = self._transformations.GetData( only_selected = True )[0]
|
||
|
|
|
||
|
|
( number, transformation_type, data ) = selected_transformation
|
||
|
|
|
||
|
|
swap_transformation = self._GetTransformation( number - 1 )
|
||
|
|
|
||
|
|
self._SwapTransformations( selected_transformation, swap_transformation )
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas()
|
||
|
|
|
||
|
|
self._transformations.Sort()
|
||
|
|
|
||
|
|
|
||
|
|
def _SwapTransformations( self, one, two ):
|
||
|
|
|
||
|
|
selected_data = self._transformations.GetData( only_selected = True )
|
||
|
|
|
||
|
|
one_selected = one in selected_data
|
||
|
|
two_selected = two in selected_data
|
||
|
|
|
||
|
|
self._transformations.DeleteDatas( ( one, two ) )
|
||
|
|
|
||
|
|
( number_1, transformation_type_1, data_1 ) = one
|
||
|
|
( number_2, transformation_type_2, data_2 ) = two
|
||
|
|
|
||
|
|
one = ( number_2, transformation_type_1, data_1 )
|
||
|
|
two = ( number_1, transformation_type_2, data_2 )
|
||
|
|
|
||
|
|
self._transformations.AddDatas( ( one, two ) )
|
||
|
|
|
||
|
|
if one_selected:
|
||
|
|
|
||
|
|
self._transformations.SelectDatas( ( one, ) )
|
||
|
|
|
||
|
|
|
||
|
|
if two_selected:
|
||
|
|
|
||
|
|
self._transformations.SelectDatas( ( two, ) )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def EventUpdate( self, text ):
|
||
|
|
|
||
|
|
self._transformations.UpdateDatas()
|
||
|
|
|
||
|
|
|
||
|
|
def GetValue( self ):
|
||
|
|
|
||
|
|
string_converter = self._GetValue()
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_converter.Convert( self._example_string.text() )
|
||
|
|
|
||
|
|
except HydrusExceptions.StringConvertException:
|
||
|
|
|
||
|
|
raise HydrusExceptions.VetoException( 'Please enter an example text that can be converted!' )
|
||
|
|
|
||
|
|
|
||
|
|
return string_converter
|
||
|
|
|
||
|
|
|
||
|
|
class _TransformationPanel( ClientGUIScrolledPanels.EditPanel ):
|
||
|
|
|
||
|
|
def __init__( self, parent, transformation_type, data, example_text ):
|
||
|
|
|
||
|
|
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
|
||
|
|
|
||
|
|
self._transformation_type = ClientGUICommon.BetterChoice( self )
|
||
|
|
|
||
|
|
for t_type in ( ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_PREPEND_TEXT, ClientParsing.STRING_TRANSFORMATION_APPEND_TEXT, ClientParsing.STRING_TRANSFORMATION_ENCODE, ClientParsing.STRING_TRANSFORMATION_DECODE, ClientParsing.STRING_TRANSFORMATION_REVERSE, ClientParsing.STRING_TRANSFORMATION_REGEX_SUB, ClientParsing.STRING_TRANSFORMATION_DATE_DECODE, ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE, ClientParsing.STRING_TRANSFORMATION_INTEGER_ADDITION ):
|
||
|
|
|
||
|
|
self._transformation_type.addItem( ClientParsing.transformation_type_str_lookup[ t_type ], t_type )
|
||
|
|
|
||
|
|
|
||
|
|
self._example_string = QW.QLineEdit( self )
|
||
|
|
|
||
|
|
min_width = ClientGUIFunctions.ConvertTextToPixelWidth( self._example_string, 96 )
|
||
|
|
|
||
|
|
self._example_string.setMinimumWidth( min_width )
|
||
|
|
|
||
|
|
self._example_text = example_text
|
||
|
|
|
||
|
|
if isinstance( self._example_text, bytes ):
|
||
|
|
|
||
|
|
self._example_string.setText( repr( self._example_text ) )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._example_string.setText( self._example_text )
|
||
|
|
|
||
|
|
|
||
|
|
self._example_transformation = QW.QLineEdit( self )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_string.setReadOnly( True )
|
||
|
|
self._example_transformation.setReadOnly( True )
|
||
|
|
|
||
|
|
self._data_text = QW.QLineEdit( self )
|
||
|
|
self._data_number = QP.MakeQSpinBox( self, min=0, max=65535 )
|
||
|
|
self._data_encoding = ClientGUICommon.BetterChoice( self )
|
||
|
|
self._data_regex_repl = QW.QLineEdit( self )
|
||
|
|
self._data_date_link = ClientGUICommon.BetterHyperLink( self, 'link to date info', 'https://docs.python.org/3/library/datetime.html#strftime-strptime-behavior' )
|
||
|
|
self._data_timezone_decode = ClientGUICommon.BetterChoice( self )
|
||
|
|
self._data_timezone_encode = ClientGUICommon.BetterChoice( self )
|
||
|
|
self._data_timezone_offset = QP.MakeQSpinBox( self, min=-86400, max=86400 )
|
||
|
|
|
||
|
|
for e in ( 'hex', 'base64', 'url percent encoding' ):
|
||
|
|
|
||
|
|
self._data_encoding.addItem( e, e )
|
||
|
|
|
||
|
|
|
||
|
|
self._data_timezone_decode.addItem( 'UTC', HC.TIMEZONE_GMT )
|
||
|
|
self._data_timezone_decode.addItem( 'Local', HC.TIMEZONE_LOCAL )
|
||
|
|
self._data_timezone_decode.addItem( 'Offset', HC.TIMEZONE_OFFSET )
|
||
|
|
|
||
|
|
self._data_timezone_encode.addItem( 'UTC', HC.TIMEZONE_GMT )
|
||
|
|
self._data_timezone_encode.addItem( 'Local', HC.TIMEZONE_LOCAL )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._transformation_type.SetValue( transformation_type )
|
||
|
|
|
||
|
|
self._data_number.setValue( 1 )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
if transformation_type in ( ClientParsing.STRING_TRANSFORMATION_DECODE, ClientParsing.STRING_TRANSFORMATION_ENCODE ):
|
||
|
|
|
||
|
|
self._data_encoding.SetValue( data )
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_REGEX_SUB:
|
||
|
|
|
||
|
|
( pattern, repl ) = data
|
||
|
|
|
||
|
|
self._data_text.setText( pattern )
|
||
|
|
self._data_regex_repl.setText( repl )
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_DECODE:
|
||
|
|
|
||
|
|
( phrase, timezone_type, timezone_offset ) = data
|
||
|
|
|
||
|
|
self._data_text.setText( phrase )
|
||
|
|
self._data_timezone_decode.SetValue( timezone_type )
|
||
|
|
self._data_timezone_offset.setValue( timezone_offset )
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE:
|
||
|
|
|
||
|
|
( phrase, timezone_type ) = data
|
||
|
|
|
||
|
|
self._data_text.setText( phrase )
|
||
|
|
self._data_timezone_encode.SetValue( timezone_type )
|
||
|
|
|
||
|
|
elif data is not None:
|
||
|
|
|
||
|
|
if isinstance( data, int ):
|
||
|
|
|
||
|
|
self._data_number.setValue( data )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._data_text.setText( data )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
# This mess needs to be all replaced with a nice QFormLayout subclass that can do row hide/show
|
||
|
|
# or just a whole separate panel for each transformation type, but w/e
|
||
|
|
|
||
|
|
self._data_text_label = ClientGUICommon.BetterStaticText( self, 'string data: ' )
|
||
|
|
self._data_number_label = ClientGUICommon.BetterStaticText( self, 'number data: ' )
|
||
|
|
self._data_encoding_label = ClientGUICommon.BetterStaticText( self, 'encoding data: ' )
|
||
|
|
self._data_regex_repl_label = ClientGUICommon.BetterStaticText( self, 'regex replacement: ' )
|
||
|
|
self._data_date_link_label = ClientGUICommon.BetterStaticText( self, 'date info: ' )
|
||
|
|
self._data_timezone_decode_label = ClientGUICommon.BetterStaticText( self, 'date decode timezone: ' )
|
||
|
|
self._data_timezone_offset_label = ClientGUICommon.BetterStaticText( self, 'timezone offset: ' )
|
||
|
|
self._data_timezone_encode_label = ClientGUICommon.BetterStaticText( self, 'date encode timezone: ' )
|
||
|
|
|
||
|
|
rows.append( ( 'example string: ', self._example_string ) )
|
||
|
|
rows.append( ( 'transformed string: ', self._example_transformation ) )
|
||
|
|
rows.append( ( 'transformation type: ', self._transformation_type ) )
|
||
|
|
rows.append( ( self._data_text_label, self._data_text ) )
|
||
|
|
rows.append( ( self._data_number_label, self._data_number ) )
|
||
|
|
rows.append( ( self._data_encoding_label, self._data_encoding ) )
|
||
|
|
rows.append( ( self._data_regex_repl_label, self._data_regex_repl ) )
|
||
|
|
rows.append( ( self._data_date_link_label, self._data_date_link ) )
|
||
|
|
rows.append( ( self._data_timezone_decode_label, self._data_timezone_decode ) )
|
||
|
|
rows.append( ( self._data_timezone_offset_label, self._data_timezone_offset ) )
|
||
|
|
rows.append( ( self._data_timezone_encode_label, self._data_timezone_encode ) )
|
||
|
|
|
||
|
|
self._control_gridbox = ClientGUICommon.WrapInGrid( self, rows )
|
||
|
|
|
||
|
|
vbox = QP.VBoxLayout()
|
||
|
|
|
||
|
|
QP.AddToLayout( vbox, self._control_gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
QP.AddToLayout( vbox, QW.QWidget( self ), CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
|
||
|
|
self.widget().setLayout( vbox )
|
||
|
|
|
||
|
|
self._UpdateDataControls()
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._transformation_type.currentIndexChanged.connect( self._UpdateDataControls )
|
||
|
|
self._transformation_type.currentIndexChanged.connect( self._UpdateExampleText )
|
||
|
|
|
||
|
|
self._data_text.textEdited.connect( self._UpdateExampleText )
|
||
|
|
self._data_number.valueChanged.connect( self._UpdateExampleText )
|
||
|
|
self._data_encoding.currentIndexChanged.connect( self._UpdateExampleText )
|
||
|
|
self._data_regex_repl.textEdited.connect( self._UpdateExampleText )
|
||
|
|
self._data_timezone_decode.currentIndexChanged.connect( self._UpdateExampleText )
|
||
|
|
self._data_timezone_offset.valueChanged.connect( self._UpdateExampleText )
|
||
|
|
self._data_timezone_encode.currentIndexChanged.connect( self._UpdateExampleText )
|
||
|
|
|
||
|
|
self._data_timezone_decode.currentIndexChanged.connect( self._UpdateDataControls )
|
||
|
|
self._data_timezone_encode.currentIndexChanged.connect( self._UpdateDataControls )
|
||
|
|
|
||
|
|
self._UpdateExampleText()
|
||
|
|
|
||
|
|
|
||
|
|
def _UpdateDataControls( self ):
|
||
|
|
|
||
|
|
self._data_text_label.setVisible( False )
|
||
|
|
self._data_number_label.setVisible( False )
|
||
|
|
self._data_encoding_label.setVisible( False )
|
||
|
|
self._data_regex_repl_label.setVisible( False )
|
||
|
|
self._data_date_link_label.setVisible( False )
|
||
|
|
self._data_timezone_decode_label.setVisible( False )
|
||
|
|
self._data_timezone_offset_label.setVisible( False )
|
||
|
|
self._data_timezone_encode_label.setVisible( False )
|
||
|
|
|
||
|
|
self._data_text.setVisible( False )
|
||
|
|
self._data_number.setVisible( False )
|
||
|
|
self._data_encoding.setVisible( False )
|
||
|
|
self._data_regex_repl.setVisible( False )
|
||
|
|
self._data_date_link.setVisible( False )
|
||
|
|
self._data_timezone_decode.setVisible( False )
|
||
|
|
self._data_timezone_offset.setVisible( False )
|
||
|
|
self._data_timezone_encode.setVisible( False )
|
||
|
|
|
||
|
|
transformation_type = self._transformation_type.GetValue()
|
||
|
|
|
||
|
|
if transformation_type in ( ClientParsing.STRING_TRANSFORMATION_ENCODE, ClientParsing.STRING_TRANSFORMATION_DECODE ):
|
||
|
|
|
||
|
|
self._data_encoding_label.setVisible( True )
|
||
|
|
self._data_encoding.setVisible( True )
|
||
|
|
|
||
|
|
elif transformation_type in ( ClientParsing.STRING_TRANSFORMATION_PREPEND_TEXT, ClientParsing.STRING_TRANSFORMATION_APPEND_TEXT, ClientParsing.STRING_TRANSFORMATION_DATE_DECODE, ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE, ClientParsing.STRING_TRANSFORMATION_REGEX_SUB ):
|
||
|
|
|
||
|
|
self._data_text_label.setVisible( True )
|
||
|
|
self._data_text.setVisible( True )
|
||
|
|
|
||
|
|
data_text_label = 'string data: '
|
||
|
|
|
||
|
|
if transformation_type == ClientParsing.STRING_TRANSFORMATION_PREPEND_TEXT:
|
||
|
|
|
||
|
|
data_text_label = 'text to prepend: '
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_APPEND_TEXT:
|
||
|
|
|
||
|
|
data_text_label = 'text to append: '
|
||
|
|
|
||
|
|
elif transformation_type in ( ClientParsing.STRING_TRANSFORMATION_DATE_DECODE, ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE ):
|
||
|
|
|
||
|
|
self._data_date_link_label.setVisible( True )
|
||
|
|
self._data_date_link.setVisible( True )
|
||
|
|
|
||
|
|
if transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_DECODE:
|
||
|
|
|
||
|
|
data_text_label = 'date decode phrase: '
|
||
|
|
|
||
|
|
self._data_timezone_decode_label.setVisible( True )
|
||
|
|
self._data_timezone_decode.setVisible( True )
|
||
|
|
|
||
|
|
if self._data_timezone_decode.GetValue() == HC.TIMEZONE_OFFSET:
|
||
|
|
|
||
|
|
self._data_timezone_offset_label.setVisible( True )
|
||
|
|
self._data_timezone_offset.setVisible( True )
|
||
|
|
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE:
|
||
|
|
|
||
|
|
data_text_label = 'date encode phrase: '
|
||
|
|
|
||
|
|
self._data_timezone_encode_label.setVisible( True )
|
||
|
|
self._data_timezone_encode.setVisible( True )
|
||
|
|
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_REGEX_SUB:
|
||
|
|
|
||
|
|
data_text_label = 'regex pattern: '
|
||
|
|
|
||
|
|
self._data_regex_repl_label.setVisible( True )
|
||
|
|
self._data_regex_repl.setVisible( True )
|
||
|
|
|
||
|
|
|
||
|
|
self._data_text_label.setText( data_text_label )
|
||
|
|
|
||
|
|
elif transformation_type in ( ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_INTEGER_ADDITION ):
|
||
|
|
|
||
|
|
self._data_number_label.setVisible( True )
|
||
|
|
self._data_number.setVisible( True )
|
||
|
|
|
||
|
|
if transformation_type == ClientParsing.STRING_TRANSFORMATION_INTEGER_ADDITION:
|
||
|
|
|
||
|
|
self._data_number.setMinimum( -65535 )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._data_number.setMinimum( 0 )
|
||
|
|
|
||
|
|
|
||
|
|
data_number_label = 'number data: '
|
||
|
|
|
||
|
|
if transformation_type == ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_BEGINNING:
|
||
|
|
|
||
|
|
data_number_label = 'characters to remove: '
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_END:
|
||
|
|
|
||
|
|
data_number_label = 'characters to remove: '
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_BEGINNING:
|
||
|
|
|
||
|
|
data_number_label = 'characters to take: '
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_END:
|
||
|
|
|
||
|
|
data_number_label = 'characters to take: '
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_INTEGER_ADDITION:
|
||
|
|
|
||
|
|
data_number_label = 'number to add: '
|
||
|
|
|
||
|
|
|
||
|
|
self._data_number_label.setText( data_number_label )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def _UpdateExampleText( self ):
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
transformations = [ self.GetValue() ]
|
||
|
|
|
||
|
|
string_converter = ClientParsing.StringConverter( transformations, self._example_text )
|
||
|
|
|
||
|
|
example_transformation = string_converter.Convert( self._example_text )
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
self._example_transformation.setText( str( example_transformation ) )
|
||
|
|
|
||
|
|
except:
|
||
|
|
|
||
|
|
self._example_transformation.setText( repr( example_transformation ) )
|
||
|
|
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
|
||
|
|
self._example_transformation.setText( str( e ) )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def GetValue( self ):
|
||
|
|
|
||
|
|
transformation_type = self._transformation_type.GetValue()
|
||
|
|
|
||
|
|
if transformation_type in ( ClientParsing.STRING_TRANSFORMATION_ENCODE, ClientParsing.STRING_TRANSFORMATION_DECODE ):
|
||
|
|
|
||
|
|
data = self._data_encoding.GetValue()
|
||
|
|
|
||
|
|
elif transformation_type in ( ClientParsing.STRING_TRANSFORMATION_PREPEND_TEXT, ClientParsing.STRING_TRANSFORMATION_APPEND_TEXT ):
|
||
|
|
|
||
|
|
data = self._data_text.text()
|
||
|
|
|
||
|
|
elif transformation_type in ( ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_REMOVE_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_BEGINNING, ClientParsing.STRING_TRANSFORMATION_CLIP_TEXT_FROM_END, ClientParsing.STRING_TRANSFORMATION_INTEGER_ADDITION ):
|
||
|
|
|
||
|
|
data = self._data_number.value()
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_REGEX_SUB:
|
||
|
|
|
||
|
|
pattern = self._data_text.text()
|
||
|
|
repl = self._data_regex_repl.text()
|
||
|
|
|
||
|
|
data = ( pattern, repl )
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_DECODE:
|
||
|
|
|
||
|
|
phrase = self._data_text.text()
|
||
|
|
timezone_time = self._data_timezone_decode.GetValue()
|
||
|
|
timezone_offset = self._data_timezone_offset.value()
|
||
|
|
|
||
|
|
data = ( phrase, timezone_time, timezone_offset )
|
||
|
|
|
||
|
|
elif transformation_type == ClientParsing.STRING_TRANSFORMATION_DATE_ENCODE:
|
||
|
|
|
||
|
|
phrase = self._data_text.text()
|
||
|
|
timezone_time = self._data_timezone_encode.GetValue()
|
||
|
|
|
||
|
|
data = ( phrase, timezone_time )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
data = None
|
||
|
|
|
||
|
|
|
||
|
|
return ( transformation_type, data )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class EditStringMatchPanel( ClientGUIScrolledPanels.EditPanel ):
|
||
|
|
|
||
|
|
def __init__( self, parent, string_match = None ):
|
||
|
|
|
||
|
|
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
|
||
|
|
|
||
|
|
if string_match is None:
|
||
|
|
|
||
|
|
string_match = ClientParsing.StringMatch()
|
||
|
|
|
||
|
|
|
||
|
|
self._match_type = ClientGUICommon.BetterChoice( self )
|
||
|
|
|
||
|
|
self._match_type.addItem( 'any characters', ClientParsing.STRING_MATCH_ANY )
|
||
|
|
self._match_type.addItem( 'fixed characters', ClientParsing.STRING_MATCH_FIXED )
|
||
|
|
self._match_type.addItem( 'character set', ClientParsing.STRING_MATCH_FLEXIBLE )
|
||
|
|
self._match_type.addItem( 'regex', ClientParsing.STRING_MATCH_REGEX )
|
||
|
|
|
||
|
|
self._match_value_text_input = QW.QLineEdit( self )
|
||
|
|
|
||
|
|
self._match_value_flexible_input = ClientGUICommon.BetterChoice( self )
|
||
|
|
|
||
|
|
self._match_value_flexible_input.addItem( 'alphabetic characters (a-zA-Z)', ClientParsing.ALPHA )
|
||
|
|
self._match_value_flexible_input.addItem( 'alphanumeric characters (a-zA-Z0-9)', ClientParsing.ALPHANUMERIC )
|
||
|
|
self._match_value_flexible_input.addItem( 'numeric characters (0-9)', ClientParsing.NUMERIC )
|
||
|
|
|
||
|
|
self._min_chars = ClientGUICommon.NoneableSpinCtrl( self, min = 1, max = 65535, unit = 'characters', none_phrase = 'no limit' )
|
||
|
|
self._max_chars = ClientGUICommon.NoneableSpinCtrl( self, min = 1, max = 65535, unit = 'characters', none_phrase = 'no limit' )
|
||
|
|
|
||
|
|
self._example_string = QW.QLineEdit( self )
|
||
|
|
|
||
|
|
self._example_string_matches = ClientGUICommon.BetterStaticText( self )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self.SetValue( string_match )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
rows.append( ( 'match type: ', self._match_type ) )
|
||
|
|
rows.append( ( 'match text: ', self._match_value_text_input ) )
|
||
|
|
rows.append( ( 'match value (character set): ', self._match_value_flexible_input ) )
|
||
|
|
rows.append( ( 'minimum allowed number of characters: ', self._min_chars ) )
|
||
|
|
rows.append( ( 'maximum allowed number of characters: ', self._max_chars ) )
|
||
|
|
rows.append( ( 'example string: ', self._example_string ) )
|
||
|
|
|
||
|
|
gridbox = ClientGUICommon.WrapInGrid( self, rows )
|
||
|
|
|
||
|
|
vbox = QP.VBoxLayout()
|
||
|
|
|
||
|
|
QP.AddToLayout( vbox, gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
QP.AddToLayout( vbox, self._example_string_matches, CC.FLAGS_EXPAND_PERPENDICULAR )
|
||
|
|
|
||
|
|
self.widget().setLayout( vbox )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._match_type.currentIndexChanged.connect( self._UpdateControls )
|
||
|
|
self._match_value_text_input.textChanged.connect( self._UpdateControls )
|
||
|
|
self._match_value_flexible_input.currentIndexChanged.connect( self._UpdateControls )
|
||
|
|
self._min_chars.valueChanged.connect( self._UpdateControls )
|
||
|
|
self._max_chars.valueChanged.connect( self._UpdateControls )
|
||
|
|
self._example_string.textChanged.connect( self._UpdateControls )
|
||
|
|
|
||
|
|
|
||
|
|
def _GetValue( self ):
|
||
|
|
|
||
|
|
match_type = self._match_type.GetValue()
|
||
|
|
|
||
|
|
if match_type == ClientParsing.STRING_MATCH_ANY:
|
||
|
|
|
||
|
|
match_value = ''
|
||
|
|
|
||
|
|
elif match_type == ClientParsing.STRING_MATCH_FLEXIBLE:
|
||
|
|
|
||
|
|
match_value = self._match_value_flexible_input.GetValue()
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
match_value = self._match_value_text_input.text()
|
||
|
|
|
||
|
|
|
||
|
|
min_chars = self._min_chars.GetValue()
|
||
|
|
max_chars = self._max_chars.GetValue()
|
||
|
|
|
||
|
|
example_string = self._example_string.text()
|
||
|
|
|
||
|
|
string_match = ClientParsing.StringMatch( match_type = match_type, match_value = match_value, min_chars = min_chars, max_chars = max_chars, example_string = example_string )
|
||
|
|
|
||
|
|
return string_match
|
||
|
|
|
||
|
|
|
||
|
|
def _UpdateControls( self ):
|
||
|
|
|
||
|
|
match_type = self._match_type.GetValue()
|
||
|
|
|
||
|
|
if match_type == ClientParsing.STRING_MATCH_ANY:
|
||
|
|
|
||
|
|
self._match_value_text_input.setEnabled( False )
|
||
|
|
self._match_value_flexible_input.setEnabled( False )
|
||
|
|
|
||
|
|
elif match_type == ClientParsing.STRING_MATCH_FLEXIBLE:
|
||
|
|
|
||
|
|
self._match_value_text_input.setEnabled( False )
|
||
|
|
self._match_value_flexible_input.setEnabled( True )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._match_value_text_input.setEnabled( True )
|
||
|
|
self._match_value_flexible_input.setEnabled( False )
|
||
|
|
|
||
|
|
|
||
|
|
if match_type == ClientParsing.STRING_MATCH_FIXED:
|
||
|
|
|
||
|
|
self._min_chars.SetValue( None )
|
||
|
|
self._max_chars.SetValue( None )
|
||
|
|
|
||
|
|
self._min_chars.setEnabled( False )
|
||
|
|
self._max_chars.setEnabled( False )
|
||
|
|
|
||
|
|
self._example_string.blockSignals( True ) # Temporarily block the text changed signal here so we won't end up in infinite recursion
|
||
|
|
self._example_string.setText( self._match_value_text_input.text() )
|
||
|
|
self._example_string.blockSignals( False )
|
||
|
|
|
||
|
|
self._example_string_matches.setText( '' )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._min_chars.setEnabled( True )
|
||
|
|
self._max_chars.setEnabled( True )
|
||
|
|
|
||
|
|
string_match = self._GetValue()
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_match.Test( self._example_string.text() )
|
||
|
|
|
||
|
|
self._example_string_matches.setText( 'Example matches ok!' )
|
||
|
|
self._example_string_matches.setObjectName( 'HydrusValid' )
|
||
|
|
self._example_string_matches.style().polish( self._example_string_matches )
|
||
|
|
|
||
|
|
except HydrusExceptions.StringMatchException as e:
|
||
|
|
|
||
|
|
reason = str( e )
|
||
|
|
|
||
|
|
self._example_string_matches.setText( 'Example does not match - '+reason )
|
||
|
|
self._example_string_matches.setObjectName( 'HydrusInvalid' )
|
||
|
|
self._example_string_matches.style().polish( self._example_string_matches )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def GetValue( self ):
|
||
|
|
|
||
|
|
string_match = self._GetValue()
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_match.Test( self._example_string.text() )
|
||
|
|
|
||
|
|
except HydrusExceptions.StringMatchException:
|
||
|
|
|
||
|
|
raise HydrusExceptions.VetoException( 'Please enter an example text that matches the given rules!' )
|
||
|
|
|
||
|
|
|
||
|
|
return string_match
|
||
|
|
|
||
|
|
|
||
|
|
def SetValue( self, string_match: ClientParsing.StringMatch ):
|
||
|
|
|
||
|
|
( match_type, match_value, min_chars, max_chars, example_string ) = string_match.ToTuple()
|
||
|
|
|
||
|
|
self._match_type.SetValue( match_type )
|
||
|
|
|
||
|
|
if match_type == ClientParsing.STRING_MATCH_FLEXIBLE:
|
||
|
|
|
||
|
|
self._match_value_flexible_input.SetValue( match_value )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
self._match_value_flexible_input.SetValue( ClientParsing.ALPHA )
|
||
|
|
|
||
|
|
self._match_value_text_input.setText( match_value )
|
||
|
|
|
||
|
|
|
||
|
|
self._min_chars.SetValue( min_chars )
|
||
|
|
self._max_chars.SetValue( max_chars )
|
||
|
|
|
||
|
|
self._example_string.setText( example_string )
|
||
|
|
|
||
|
|
self._UpdateControls()
|
||
|
|
|
||
|
|
|
||
|
|
class EditStringSplitterPanel( ClientGUIScrolledPanels.EditPanel ):
|
||
|
|
|
||
|
|
def __init__( self, parent, string_splitter: ClientParsing.StringSplitter, example_string: str = '' ):
|
||
|
|
|
||
|
|
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._controls_panel = ClientGUICommon.StaticBox( self, 'splitter values' )
|
||
|
|
|
||
|
|
self._separator = QW.QLineEdit( self._controls_panel )
|
||
|
|
self._max_splits = ClientGUICommon.NoneableSpinCtrl( self._controls_panel, min = 1, max = 65535, unit = 'splits', none_phrase = 'no limit' )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_panel = ClientGUICommon.StaticBox( self, 'test results' )
|
||
|
|
|
||
|
|
self._example_string = QW.QLineEdit( self._example_panel )
|
||
|
|
|
||
|
|
self._example_string_splits = QW.QListWidget( self._example_panel )
|
||
|
|
self._example_string_splits.setSelectionMode( QW.QListWidget.NoSelection )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_string.setText( example_string )
|
||
|
|
|
||
|
|
self.SetValue( string_splitter )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
rows.append( ( 'separator: ', self._separator ) )
|
||
|
|
rows.append( ( 'max splits: ', self._max_splits ) )
|
||
|
|
|
||
|
|
gridbox = ClientGUICommon.WrapInGrid( self._controls_panel, rows )
|
||
|
|
|
||
|
|
self._controls_panel.Add( gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
rows.append( ( 'example string: ', self._example_string ) )
|
||
|
|
|
||
|
|
gridbox = ClientGUICommon.WrapInGrid( self._example_panel, rows )
|
||
|
|
|
||
|
|
self._example_panel.Add( gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
self._example_panel.Add( ClientGUICommon.BetterStaticText( self, label = 'result:' ), CC.FLAGS_EXPAND_PERPENDICULAR )
|
||
|
|
self._example_panel.Add( self._example_string_splits, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
|
||
|
|
vbox = QP.VBoxLayout()
|
||
|
|
|
||
|
|
QP.AddToLayout( vbox, self._controls_panel, CC.FLAGS_EXPAND_PERPENDICULAR )
|
||
|
|
QP.AddToLayout( vbox, self._example_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
|
||
|
|
self.widget().setLayout( vbox )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._separator.textChanged.connect( self._UpdateControls )
|
||
|
|
self._max_splits.valueChanged.connect( self._UpdateControls )
|
||
|
|
self._example_string.textChanged.connect( self._UpdateControls )
|
||
|
|
|
||
|
|
|
||
|
|
def _GetValue( self ):
|
||
|
|
|
||
|
|
separator = self._separator.text()
|
||
|
|
max_splits = self._max_splits.GetValue()
|
||
|
|
|
||
|
|
string_splitter = ClientParsing.StringSplitter( separator = separator, max_splits = max_splits )
|
||
|
|
|
||
|
|
return string_splitter
|
||
|
|
|
||
|
|
|
||
|
|
def _UpdateControls( self ):
|
||
|
|
|
||
|
|
string_splitter = self._GetValue()
|
||
|
|
|
||
|
|
results = string_splitter.Split( self._example_string.text() )
|
||
|
|
|
||
|
|
self._example_string_splits.clear()
|
||
|
|
|
||
|
|
for result in results:
|
||
|
|
|
||
|
|
self._example_string_splits.addItem( result )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def GetValue( self ):
|
||
|
|
|
||
|
|
string_match = self._GetValue()
|
||
|
|
|
||
|
|
return string_match
|
||
|
|
|
||
|
|
|
||
|
|
def SetValue( self, string_splitter: ClientParsing.StringSplitter ):
|
||
|
|
|
||
|
|
separator = string_splitter.GetSeparator()
|
||
|
|
max_splits = string_splitter.GetMaxSplits()
|
||
|
|
|
||
|
|
self._separator.setText( separator )
|
||
|
|
self._max_splits.SetValue( max_splits )
|
||
|
|
|
||
|
|
self._UpdateControls()
|
||
|
|
|
||
|
|
|
||
|
|
class EditStringProcessorPanel( ClientGUIScrolledPanels.EditPanel ):
|
||
|
|
|
||
|
|
NO_RESULTS_TEXT = 'no results'
|
||
|
|
|
||
|
|
def __init__( self, parent, string_processor: ClientParsing.StringProcessor, example_string: str = '' ):
|
||
|
|
|
||
|
|
ClientGUIScrolledPanels.EditPanel.__init__( self, parent )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._controls_panel = ClientGUICommon.StaticBox( self, 'processing steps' )
|
||
|
|
|
||
|
|
self._processing_steps = ClientGUIListBoxes.QueueListBox( self, 8, self._ConvertDataToListBoxString, add_callable = self._Add, edit_callable = self._Edit )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_panel = ClientGUICommon.StaticBox( self, 'test results' )
|
||
|
|
|
||
|
|
self._example_string = QW.QLineEdit( self._example_panel )
|
||
|
|
|
||
|
|
self._example_results = ClientGUICommon.BetterNotebook( self._example_panel )
|
||
|
|
|
||
|
|
( w, h ) = ClientGUIFunctions.ConvertTextToPixels( self._example_panel, ( 64, 24 ) )
|
||
|
|
|
||
|
|
self._example_panel.setMinimumSize( w, h )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._example_string.setText( example_string )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._controls_panel.Add( self._processing_steps, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
|
||
|
|
|
||
|
|
rows = []
|
||
|
|
|
||
|
|
rows.append( ( 'example string: ', self._example_string ) )
|
||
|
|
|
||
|
|
gridbox = ClientGUICommon.WrapInGrid( self._example_panel, rows )
|
||
|
|
|
||
|
|
self._example_panel.Add( gridbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
|
||
|
|
self._example_panel.Add( ClientGUICommon.BetterStaticText( self, label = 'result:' ), CC.FLAGS_EXPAND_PERPENDICULAR )
|
||
|
|
self._example_panel.Add( self._example_results, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
|
||
|
|
hbox = QP.VBoxLayout()
|
||
|
|
|
||
|
|
QP.AddToLayout( hbox, self._controls_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
QP.AddToLayout( hbox, self._example_panel, CC.FLAGS_EXPAND_BOTH_WAYS )
|
||
|
|
|
||
|
|
self.widget().setLayout( hbox )
|
||
|
|
|
||
|
|
#
|
||
|
|
|
||
|
|
self._processing_steps.listBoxChanged.connect( self._UpdateControls )
|
||
|
|
self._example_string.textChanged.connect( self._UpdateControls )
|
||
|
|
|
||
|
|
self.SetValue( string_processor )
|
||
|
|
|
||
|
|
|
||
|
|
def _Add( self ):
|
||
|
|
|
||
|
|
choice_tuples = [
|
||
|
|
( 'String Match', ClientParsing.StringMatch, 'An object that filters strings.' ),
|
||
|
|
( 'String Transformer', ClientParsing.StringConverter, 'An object that converts strings from one thing to another.' ),
|
||
|
|
( 'String Splitter', ClientParsing.StringSplitter, 'An object that breaks strings into smaller strings.' )
|
||
|
|
]
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
string_processing_step_type = ClientGUIDialogsQuick.SelectFromListButtons( self, 'Which type of processing step?', choice_tuples )
|
||
|
|
|
||
|
|
except HydrusExceptions.CancelledException:
|
||
|
|
|
||
|
|
raise HydrusExceptions.VetoException()
|
||
|
|
|
||
|
|
|
||
|
|
if string_processing_step_type == ClientParsing.StringMatch:
|
||
|
|
|
||
|
|
string_processing_step = ClientParsing.StringMatch( example_string = self._example_string.text() )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
string_processing_step = string_processing_step_type()
|
||
|
|
|
||
|
|
|
||
|
|
return self._Edit( string_processing_step )
|
||
|
|
|
||
|
|
|
||
|
|
def _Edit( self, string_processing_step: ClientParsing.StringProcessingStep ):
|
||
|
|
|
||
|
|
current_string_processor = self._GetValue()
|
||
|
|
|
||
|
|
current_string_processing_steps = current_string_processor.GetProcessingSteps()
|
||
|
|
|
||
|
|
if string_processing_step in current_string_processing_steps:
|
||
|
|
|
||
|
|
example_text_index = current_string_processing_steps.index( string_processing_step )
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
example_text_index = len( current_string_processing_steps )
|
||
|
|
|
||
|
|
|
||
|
|
example_text = self._example_string.text()
|
||
|
|
|
||
|
|
if 0 < example_text_index < self._example_results.count() + 1:
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
t = self._example_results.widget( example_text_index - 1 ).item( 0 ).text()
|
||
|
|
|
||
|
|
if t != self.NO_RESULTS_TEXT:
|
||
|
|
|
||
|
|
example_text = t
|
||
|
|
|
||
|
|
|
||
|
|
except:
|
||
|
|
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
with ClientGUITopLevelWindowsPanels.DialogEdit( self, 'edit processing step' ) as dlg:
|
||
|
|
|
||
|
|
if isinstance( string_processing_step, ClientParsing.StringMatch ):
|
||
|
|
|
||
|
|
panel = EditStringMatchPanel( dlg, string_processing_step )
|
||
|
|
|
||
|
|
elif isinstance( string_processing_step, ClientParsing.StringConverter ):
|
||
|
|
|
||
|
|
panel = EditStringConverterPanel( dlg, string_processing_step, example_string_override = example_text )
|
||
|
|
|
||
|
|
elif isinstance( string_processing_step, ClientParsing.StringSplitter ):
|
||
|
|
|
||
|
|
panel = EditStringSplitterPanel( dlg, string_processing_step, example_string = example_text )
|
||
|
|
|
||
|
|
|
||
|
|
dlg.SetPanel( panel )
|
||
|
|
|
||
|
|
if dlg.exec() == QW.QDialog.Accepted:
|
||
|
|
|
||
|
|
string_processing_step = panel.GetValue()
|
||
|
|
|
||
|
|
return string_processing_step
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
raise HydrusExceptions.VetoException()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def _ConvertDataToListBoxString( self, string_processing_step: ClientParsing.StringProcessingStep ):
|
||
|
|
|
||
|
|
return string_processing_step.ToString()
|
||
|
|
|
||
|
|
|
||
|
|
def _GetValue( self ):
|
||
|
|
|
||
|
|
processing_steps = self._processing_steps.GetData()
|
||
|
|
|
||
|
|
string_processor = ClientParsing.StringProcessor()
|
||
|
|
|
||
|
|
string_processor.SetProcessingSteps( processing_steps )
|
||
|
|
|
||
|
|
return string_processor
|
||
|
|
|
||
|
|
|
||
|
|
def _UpdateControls( self ):
|
||
|
|
|
||
|
|
string_processor = self._GetValue()
|
||
|
|
|
||
|
|
processing_steps = string_processor.GetProcessingSteps()
|
||
|
|
|
||
|
|
current_selected_index = self._example_results.currentIndex()
|
||
|
|
|
||
|
|
self._example_results.DeleteAllPages()
|
||
|
|
|
||
|
|
example_string = self._example_string.text()
|
||
|
|
|
||
|
|
stop_now = False
|
||
|
|
|
||
|
|
for i in range( len( processing_steps ) ):
|
||
|
|
|
||
|
|
try:
|
||
|
|
|
||
|
|
results = string_processor.ProcessStrings( [ example_string ], max_steps_allowed = i + 1 )
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
|
||
|
|
results = [ 'error: {}'.format( str( e ) ) ]
|
||
|
|
|
||
|
|
stop_now = True
|
||
|
|
|
||
|
|
|
||
|
|
results_list = QW.QListWidget( self._example_panel )
|
||
|
|
results_list.setSelectionMode( QW.QListWidget.NoSelection )
|
||
|
|
|
||
|
|
if len( results ) == 0:
|
||
|
|
|
||
|
|
results_list.addItem( self.NO_RESULTS_TEXT )
|
||
|
|
|
||
|
|
stop_now = True
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
for result in results:
|
||
|
|
|
||
|
|
results_list.addItem( result )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
tab_label = '{} ({})'.format( processing_steps[i].ToString( simple = True ), HydrusData.ToHumanInt( len( results ) ) )
|
||
|
|
|
||
|
|
self._example_results.addTab( results_list, tab_label )
|
||
|
|
|
||
|
|
if stop_now:
|
||
|
|
|
||
|
|
break
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if self._example_results.count() > current_selected_index:
|
||
|
|
|
||
|
|
self._example_results.setCurrentIndex( current_selected_index )
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def GetValue( self ):
|
||
|
|
|
||
|
|
string_processor = self._GetValue()
|
||
|
|
|
||
|
|
return string_processor
|
||
|
|
|
||
|
|
|
||
|
|
def SetValue( self, string_processor: ClientParsing.StringProcessor ):
|
||
|
|
|
||
|
|
processing_steps = string_processor.GetProcessingSteps()
|
||
|
|
|
||
|
|
self._processing_steps.AddDatas( processing_steps )
|
||
|
|
|
||
|
|
self._UpdateControls()
|
||
|
|
|
||
|
|
|