RepoMirrors
/
hydrus
mirror of https://github.com/hydrusnetwork/hydrus.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
hydrus/hydrus/client/gui/search/ClientGUIPredicatesSingle.py

2542 lines
84 KiB
Python
Raw Blame History

import os
import re
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.core import HydrusGlobals as HG
from hydrus.core.files import HydrusFileHandling
from hydrus.core.files.images import HydrusImageHandling
from hydrus.client import ClientConstants as CC
from hydrus.client import ClientGlobals as CG
from hydrus.client import ClientImageHandling
from hydrus.client.gui import ClientGUIDialogsMessage
from hydrus.client.gui import ClientGUIFunctions
from hydrus.client.gui import ClientGUIOptionsPanels
from hydrus.client.gui import ClientGUITime
from hydrus.client.gui import QtPorting as QP
from hydrus.client.gui.widgets import ClientGUICommon
from hydrus.client.gui.widgets import ClientGUIControls
from hydrus.client.search import ClientSearch
class StaticSystemPredicateButton( QW.QWidget ):
predicatesChosen = QC.Signal( QW.QWidget )
predicatesRemoved = QC.Signal( QW.QWidget )
def __init__( self, parent, predicates, forced_label = None, show_remove_button = True ):
QW.QWidget.__init__( self, parent )
self._predicates = predicates
self._forced_label = forced_label
if forced_label is None:
label = ', '.join( ( predicate.ToString() for predicate in self._predicates ) )
else:
label = forced_label
self._predicates_button = ClientGUICommon.BetterButton( self, label, self._DoPredicatesChoose )
self._remove_button = ClientGUICommon.BetterBitmapButton( self, CC.global_pixmaps().trash_delete, self._DoPredicatesRemove )
hbox = QP.HBoxLayout()
if show_remove_button:
flag = CC.FLAGS_EXPAND_BOTH_WAYS
else:
flag = CC.FLAGS_EXPAND_SIZER_BOTH_WAYS
self._remove_button.hide()
QP.AddToLayout( hbox, self._predicates_button, flag )
QP.AddToLayout( hbox, self._remove_button, CC.FLAGS_CENTER )
self.setFocusProxy( self._predicates_button )
self.setLayout( hbox )
def _DoPredicatesChoose( self ):
self.predicatesChosen.emit( self )
def _DoPredicatesRemove( self ):
self.predicatesRemoved.emit( self )
def GetPredicates( self ) -> typing.List[ ClientSearch.Predicate ]:
return self._predicates
class TimeDateOperator( QP.DataRadioBox ):
def __init__( self, parent ):
choice_tuples = [
( 'before', '<' ),
( 'since', '>' ),
( 'the day of', '=' ),
( '+/- a month of', HC.UNICODE_APPROX_EQUAL )
]
QP.DataRadioBox.__init__( self, parent, choice_tuples, vertical = True )
class TimeDeltaOperator( QP.DataRadioBox ):
def __init__( self, parent ):
choice_tuples = [
( 'before', '>' ),
( 'since', '<' ),
( '+/- 15% of', HC.UNICODE_APPROX_EQUAL )
]
QP.DataRadioBox.__init__( self, parent, choice_tuples, vertical = True )
class InvertiblePredicateButton( ClientGUICommon.BetterButton ):
def __init__( self, parent: QW.QWidget, predicate: ClientSearch.Predicate ):
self._predicate = predicate
ClientGUICommon.BetterButton.__init__( self, parent, 'predicate', self._ButtonHit )
self._UpdateLabel()
def _ButtonHit( self ):
inverse_predicate = self._predicate.GetInverseCopy()
if inverse_predicate is not None:
self._predicate = inverse_predicate
self._UpdateLabel()
def _UpdateLabel( self ):
s = self._predicate.ToString( with_count = False )
self.setText( s )
def GetPredicate( self ):
return self._predicate
def GetPredicateFromSimpleTagText( simple_tag_text: str ):
inclusive = True
while simple_tag_text.startswith( '-' ):
inclusive = False
simple_tag_text = simple_tag_text[1:]
if simple_tag_text == '':
raise Exception( 'Please enter some tag, namespace, or wildcard text!' )
if simple_tag_text.count( ':' ) == 1 and ( simple_tag_text.endswith( ':' ) or simple_tag_text.endswith( ':*' ) ):
if simple_tag_text.endswith( ':' ):
namespace = simple_tag_text[:-1]
else:
namespace = simple_tag_text[:-2]
if namespace == '':
raise Exception( 'Please enter some namespace text!' )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_NAMESPACE, namespace, inclusive = inclusive )
elif '*' in simple_tag_text:
wildcard = simple_tag_text
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_WILDCARD, wildcard, inclusive = inclusive )
else:
tag = simple_tag_text
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_TAG, tag, inclusive = inclusive )
class PanelPredicateSimpleTagTypes( QW.QWidget ):
def __init__( self, parent: QW.QWidget, predicate: ClientSearch.Predicate ):
QW.QWidget.__init__( self, parent )
if predicate.GetType() not in ( ClientSearch.PREDICATE_TYPE_TAG, ClientSearch.PREDICATE_TYPE_NAMESPACE, ClientSearch.PREDICATE_TYPE_WILDCARD ):
raise Exception( 'Launched a SimpleTextPredicateControl without a Tag, Namespace, or Wildcard Pred!' )
self._simple_tag_text = QW.QLineEdit()
inclusive = predicate.IsInclusive()
if predicate.GetType() == ClientSearch.PREDICATE_TYPE_NAMESPACE:
namespace = predicate.GetValue()
text = '{}{}:*'.format( '' if inclusive else '-', namespace )
else:
inclusive = predicate.IsInclusive()
wildcard_or_tag = predicate.GetValue()
text = '{}{}'.format( '' if inclusive else '-', wildcard_or_tag )
self._simple_tag_text.setText( text )
vbox = QP.VBoxLayout()
QP.AddToLayout( vbox, self._simple_tag_text, CC.FLAGS_EXPAND_BOTH_WAYS )
self.setLayout( vbox )
ClientGUIFunctions.SetFocusLater( self._simple_tag_text )
def CheckValid( self ):
try:
predicates = self.GetPredicates()
except Exception as e:
raise HydrusExceptions.VetoException( str( e ) )
def GetPredicates( self ):
simple_tag_text = self._simple_tag_text.text()
simple_tag_text_predicate = GetPredicateFromSimpleTagText( simple_tag_text )
return [ simple_tag_text_predicate ]
class PanelPredicateSystem( QW.QWidget ):
def __init__( self, parent ):
QW.QWidget.__init__( self, parent )
def CheckValid( self ):
try:
predicates = self.GetPredicates()
except Exception as e:
raise HydrusExceptions.VetoException( str( e ) )
def ClearCustomDefault( self ):
raise NotImplementedError()
def GetPredicates( self ):
raise NotImplementedError()
def SaveCustomDefault( self ):
raise NotImplementedError()
def UsesCustomDefault( self ) -> bool:
raise NotImplementedError()
class PanelPredicateSystemSingle( PanelPredicateSystem ):
def _GetPredicateToInitialisePanelWith( self, predicate: ClientSearch.Predicate ) -> ClientSearch.Predicate:
default_predicate = self.GetDefaultPredicate()
if predicate.IsUIEditable( default_predicate ):
return predicate
custom_defaults = CG.client_controller.new_options.GetCustomDefaultSystemPredicates( comparable_predicate = default_predicate )
if len( custom_defaults ) > 0:
return list( custom_defaults )[0]
return default_predicate
def ClearCustomDefault( self ):
default_predicate = self.GetDefaultPredicate()
CG.client_controller.new_options.ClearCustomDefaultSystemPredicates( comparable_predicate = default_predicate )
def GetDefaultPredicate( self ) -> ClientSearch.Predicate:
raise NotImplementedError()
def GetPredicates( self ):
raise NotImplementedError()
def SaveCustomDefault( self ):
predicates = self.GetPredicates()
CG.client_controller.new_options.SetCustomDefaultSystemPredicates( comparable_predicates = predicates )
def UsesCustomDefault( self ) -> bool:
default_predicate = self.GetDefaultPredicate()
custom_defaults = CG.client_controller.new_options.GetCustomDefaultSystemPredicates( comparable_predicate = default_predicate )
return len( custom_defaults ) > 0
class PanelPredicateSystemDate( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = TimeDateOperator( self )
self._date = QW.QCalendarWidget( self )
self._time = QW.QTimeEdit( self )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, age_type, ( years, months, days, hours, minutes ) ) = predicate.GetValue()
self._sign.SetValue( sign )
qt_dt = QC.QDate( years, months, days )
self._date.setSelectedDate( qt_dt )
qt_t = QC.QTime( hours, minutes )
self._time.setTime( qt_t )
#
hbox = QP.HBoxLayout()
system_pred_label = self._GetSystemPredicateLabel()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self, system_pred_label ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._date, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._time, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def _GetSystemPredicateLabel( self ) -> str:
raise NotImplementedError()
def _GetPredicateType( self ) -> int:
raise NotImplementedError()
def GetDefaultPredicate( self ) -> ClientSearch.Predicate:
qt_dt = QC.QDate.currentDate()
qt_dt.addDays( -7 )
year = qt_dt.year()
month = qt_dt.month()
day = qt_dt.day()
hour = 0
minute = 0
predicate_type = self._GetPredicateType()
return ClientSearch.Predicate( predicate_type, ( '>', 'date', ( year, month, day, hour, minute ) ) )
def GetPredicates( self ):
qt_dt = self._date.selectedDate()
year = qt_dt.year()
month = qt_dt.month()
day = qt_dt.day()
qt_t = self._time.time()
hour = qt_t.hour()
minute = qt_t.minute()
predicate_type = self._GetPredicateType()
predicates = ( ClientSearch.Predicate( predicate_type, ( self._sign.GetValue(), 'date', ( year, month, day, hour, minute ) ) ), )
return predicates
class PanelPredicateSystemAgeDate( PanelPredicateSystemDate ):
def _GetSystemPredicateLabel( self ) -> str:
return 'system:import time'
def _GetPredicateType( self ) -> int:
return ClientSearch.PREDICATE_TYPE_SYSTEM_AGE
def GetDefaultPredicate( self ) -> ClientSearch.Predicate:
qt_dt = QC.QDate.currentDate()
qt_dt.addDays( -7 )
year = qt_dt.year()
month = qt_dt.month()
day = qt_dt.day()
hour = 0
minute = 0
predicate_type = self._GetPredicateType()
return ClientSearch.Predicate( predicate_type, ( '<', 'date', ( year, month, day, hour, minute ) ) )
class PanelPredicateSystemLastViewedDate( PanelPredicateSystemDate ):
def _GetSystemPredicateLabel( self ) -> str:
return 'system:last viewed time'
def _GetPredicateType( self ) -> int:
return ClientSearch.PREDICATE_TYPE_SYSTEM_LAST_VIEWED_TIME
class PanelPredicateSystemArchivedDate( PanelPredicateSystemDate ):
def _GetSystemPredicateLabel( self ) -> str:
return 'system:archived time'
def _GetPredicateType( self ) -> int:
return ClientSearch.PREDICATE_TYPE_SYSTEM_ARCHIVED_TIME
class PanelPredicateSystemModifiedDate( PanelPredicateSystemDate ):
def _GetSystemPredicateLabel( self ) -> str:
return 'system:modified time'
def _GetPredicateType( self ) -> int:
return ClientSearch.PREDICATE_TYPE_SYSTEM_MODIFIED_TIME
class PanelPredicateSystemAgeDelta( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = TimeDeltaOperator( self )
self._years = ClientGUICommon.BetterSpinBox( self, max = 50000 )
self._months = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._days = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._hours = ClientGUICommon.BetterSpinBox( self, max = 10000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, age_type, ( years, months, days, hours ) ) = predicate.GetValue()
self._sign.SetValue( sign )
self._years.setValue( years )
self._months.setValue( months )
self._days.setValue( days )
self._hours.setValue( hours )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:import time'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._years, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'years'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._months, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'months'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._days, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'days'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hours, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'hours ago'), CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_AGE, ( '<', 'delta', ( 0, 0, 7, 0 ) ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_AGE, ( self._sign.GetValue(), 'delta', (self._years.value(), self._months.value(), self._days.value(), self._hours.value() ) ) ), )
return predicates
# TODO: Merge all this gubbins together, like date above
class PanelPredicateSystemLastViewedDelta( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = TimeDeltaOperator( self )
self._years = ClientGUICommon.BetterSpinBox( self, max = 50000 )
self._months = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._days = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._hours = ClientGUICommon.BetterSpinBox( self, max = 10000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, age_type, ( years, months, days, hours ) ) = predicate.GetValue()
self._sign.SetValue( sign )
self._years.setValue( years )
self._months.setValue( months )
self._days.setValue( days )
self._hours.setValue( hours )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:last viewed time'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._years, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'years'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._months, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'months'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._days, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'days'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hours, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'hours ago'), CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_LAST_VIEWED_TIME, ( '<', 'delta', ( 0, 0, 7, 0 ) ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_LAST_VIEWED_TIME, ( self._sign.GetValue(), 'delta', ( self._years.value(), self._months.value(), self._days.value(), self._hours.value() ) ) ), )
return predicates
class PanelPredicateSystemArchivedDelta( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = TimeDeltaOperator( self )
self._years = ClientGUICommon.BetterSpinBox( self, max = 50000 )
self._months = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._days = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._hours = ClientGUICommon.BetterSpinBox( self, max = 10000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, age_type, ( years, months, days, hours ) ) = predicate.GetValue()
self._sign.SetValue( sign )
self._years.setValue( years )
self._months.setValue( months )
self._days.setValue( days )
self._hours.setValue( hours )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:archived time'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._years, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'years'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._months, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'months'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._days, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'days'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hours, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'hours ago'), CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_ARCHIVED_TIME, ( '<', 'delta', ( 0, 0, 7, 0 ) ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_ARCHIVED_TIME, ( self._sign.GetValue(), 'delta', ( self._years.value(), self._months.value(), self._days.value(), self._hours.value() ) ) ), )
return predicates
class PanelPredicateSystemModifiedDelta( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = TimeDeltaOperator( self )
self._years = ClientGUICommon.BetterSpinBox( self, max = 50000 )
self._months = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._days = ClientGUICommon.BetterSpinBox( self, max = 1000 )
self._hours = ClientGUICommon.BetterSpinBox( self, max = 10000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, age_type, ( years, months, days, hours ) ) = predicate.GetValue()
self._sign.SetValue( sign )
self._years.setValue( years )
self._months.setValue( months )
self._days.setValue( days )
self._hours.setValue( hours )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:modified time'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._years, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'years'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._months, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'months'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._days, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'days'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hours, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'hours ago'), CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_MODIFIED_TIME, ( '<', 'delta', ( 0, 0, 7, 0 ) ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_MODIFIED_TIME, ( self._sign.GetValue(), 'delta', ( self._years.value(), self._months.value(), self._days.value(), self._hours.value() ) ) ), )
return predicates
class PanelPredicateSystemDuplicateRelationships( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
choices = [ '<', HC.UNICODE_APPROX_EQUAL, '=', '>' ]
self._sign = QP.RadioBox( self, choices = choices )
self._num = ClientGUICommon.BetterSpinBox( self, min=0, max=65535 )
choices = [ ( HC.duplicate_type_string_lookup[ status ], status ) for status in ( HC.DUPLICATE_MEMBER, HC.DUPLICATE_ALTERNATE, HC.DUPLICATE_FALSE_POSITIVE, HC.DUPLICATE_POTENTIAL ) ]
self._dupe_type = ClientGUICommon.BetterRadioBox( self, choices = choices, vertical = True )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, num, dupe_type ) = predicate.GetValue()
self._sign.SetStringSelection( sign )
self._num.setValue( num )
self._dupe_type.SetValue( dupe_type )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:num file relationships'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._num, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._dupe_type, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
sign = '>'
num = 0
dupe_type = HC.DUPLICATE_MEMBER
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_RELATIONSHIPS_COUNT, ( sign, num, dupe_type ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_RELATIONSHIPS_COUNT, ( self._sign.GetStringSelection(), self._num.value(), self._dupe_type.GetValue() ) ), )
return predicates
class PanelPredicateSystemDuration( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUITime.NumberTestWidgetDuration( self, allowed_operators = allowed_operators, appropriate_absolute_plus_or_minus_default = 30000, appropriate_percentage_plus_or_minus_default = 5 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:duration'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN, value = 0 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_DURATION, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_DURATION, number_test ), )
return predicates
class PanelPredicateSystemFileService( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = ClientGUICommon.BetterRadioBox( self, choices = [ ( 'is', True ), ( 'is not', False ) ], vertical = True )
choices = [
( 'currently in', HC.CONTENT_STATUS_CURRENT ),
( 'deleted from', HC.CONTENT_STATUS_DELETED ),
( 'pending to', HC.CONTENT_STATUS_PENDING ),
( 'petitioned from', HC.CONTENT_STATUS_PETITIONED )
]
self._status = ClientGUICommon.BetterRadioBox( self, choices = choices, vertical = True )
services = CG.client_controller.services_manager.GetServices( HC.REAL_FILE_SERVICES )
choices = [ ( service.GetName(), service.GetServiceKey() ) for service in services ]
self._file_service_key = ClientGUICommon.BetterRadioBox( self, choices = choices, vertical = True )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, current_pending, file_service_key ) = predicate.GetValue()
self._sign.SetValue( sign )
self._status.SetValue( current_pending )
self._file_service_key.SetValue( file_service_key )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:file service:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._status, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._file_service_key, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
sign = True
status = HC.CONTENT_STATUS_CURRENT
file_service_key = bytes()
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_SERVICE, ( sign, status, file_service_key ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_SERVICE, ( self._sign.GetValue(), self._status.GetValue(), self._file_service_key.GetValue() ) ), )
return predicates
class PanelPredicateSystemFileViewingStatsViews( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._viewing_locations = ClientGUICommon.BetterCheckBoxList( self )
self._viewing_locations.Append( 'media views', 'media' )
self._viewing_locations.Append( 'preview views', 'preview' )
self._sign = QP.RadioBox( self, choices=['<',HC.UNICODE_APPROX_EQUAL,'=','>'] )
self._num = ClientGUICommon.BetterSpinBox( self, min=0, max=1000000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( view_type, viewing_locations, sign, num ) = predicate.GetValue()
self._viewing_locations.SetValue( viewing_locations )
( width, height ) = ClientGUIFunctions.ConvertTextToPixels( self._viewing_locations, ( 10, 3 ) )
self._viewing_locations.setMaximumHeight( height )
self._sign.SetStringSelection( sign )
self._num.setValue( num )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._viewing_locations, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._num, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
viewing_locations = ( 'media', )
sign = '>'
num = 10
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_VIEWING_STATS, ( 'views', tuple( viewing_locations ), sign, num ) )
def GetPredicates( self ):
viewing_locations = self._viewing_locations.GetValue()
if len( viewing_locations ) == 0:
viewing_locations = [ 'media' ]
sign = self._sign.GetStringSelection()
num = self._num.value()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_VIEWING_STATS, ( 'views', tuple( viewing_locations ), sign, num ) ), )
return predicates
class PanelPredicateSystemFileViewingStatsViewtime( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._viewing_locations = ClientGUICommon.BetterCheckBoxList( self )
self._viewing_locations.Append( 'media viewtime', 'media' )
self._viewing_locations.Append( 'preview viewtime', 'preview' )
self._sign = QP.RadioBox( self, choices=['<',HC.UNICODE_APPROX_EQUAL,'=','>'] )
self._time_delta = ClientGUITime.TimeDeltaCtrl( self, min = 0, days = True, hours = True, minutes = True, seconds = True )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( view_type, viewing_locations, sign, time_delta ) = predicate.GetValue()
self._viewing_locations.SetValue( viewing_locations )
( width, height ) = ClientGUIFunctions.ConvertTextToPixels( self._viewing_locations, ( 10, 3 ) )
self._viewing_locations.setMaximumHeight( height )
self._sign.SetStringSelection( sign )
self._time_delta.SetValue( time_delta )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._viewing_locations, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._time_delta, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
viewing_locations = ( 'media', )
sign = '>'
time_delta = 600
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_VIEWING_STATS, ( 'viewtime', tuple( viewing_locations ), sign, time_delta ) )
def GetPredicates( self ):
viewing_locations = self._viewing_locations.GetValue()
if len( viewing_locations ) == 0:
viewing_locations = [ 'media' ]
sign = self._sign.GetStringSelection()
time_delta = self._time_delta.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FILE_VIEWING_STATS, ( 'viewtime', tuple( viewing_locations ), sign, time_delta ) ), )
return predicates
class PanelPredicateSystemFramerate( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators, max = 1000000, unit_string = 'fps', appropriate_absolute_plus_or_minus_default = 1, appropriate_percentage_plus_or_minus_default = 5 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self, 'system:framerate' ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT, value = 60, extra_value = 0.05 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FRAMERATE, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_FRAMERATE, number_test ), )
return predicates
class PanelPredicateSystemHash( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = ClientGUICommon.BetterRadioBox( self, choices = [ ( 'is', True ), ( 'is not', False ) ], vertical = True )
choices = [ 'sha256', 'md5', 'sha1', 'sha512' ]
self._hash_type = QP.RadioBox( self, choices = choices, vertical = True )
self._hashes = QW.QPlainTextEdit( self )
self._hashes.setPlaceholderText( 'enter hash (paste newline-separated for multiple hashes)' )
( init_width, init_height ) = ClientGUIFunctions.ConvertTextToPixels( self._hashes, ( 66, 10 ) )
self._hashes.setMinimumSize( QC.QSize( init_width, init_height ) )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( hashes, hash_type ) = predicate.GetValue()
self._sign.SetValue( predicate.IsInclusive() )
hashes_text = os.linesep.join( [ hash.hex() for hash in hashes ] )
self._hashes.setPlainText( hashes_text )
self._hash_type.SetStringSelection( hash_type )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:hash'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hashes, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hash_type, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
hashes = tuple()
hash_type = 'sha256'
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HASH, ( hashes, hash_type ) )
def GetPredicates( self ):
inclusive = self._sign.GetValue()
hash_type = self._hash_type.GetStringSelection()
hex_hashes_raw = self._hashes.toPlainText()
hashes = HydrusData.ParseHashesFromRawHexText( hash_type, hex_hashes_raw )
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HASH, ( hashes, hash_type ), inclusive = inclusive ), )
return predicates
class PanelPredicateSystemHasNoteName( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._operator = ClientGUICommon.BetterChoice( self )
self._operator.addItem( 'has note with name ', True )
self._operator.addItem( 'does not have note with name', False )
self._name = QW.QLineEdit( self )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( operator, name ) = predicate.GetValue()
self._operator.SetValue( operator )
self._name.setText( name )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:note name'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._operator, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._name, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
operator = True
name = ''
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HAS_NOTE_NAME, ( operator, name ) )
def GetPredicates( self ):
name = self._name.text()
if name == '':
name = 'notes'
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HAS_NOTE_NAME, ( self._operator.GetValue(), name ) ), )
return predicates
class PanelPredicateSystemHeight( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators, unit_string = 'px', appropriate_absolute_plus_or_minus_default = 200 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:height'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_EQUAL, value = 1080 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HEIGHT, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_HEIGHT, number_test ), )
return predicates
class PanelPredicateSystemKnownURLsExactURL( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._operator = ClientGUICommon.BetterChoice( self )
self._operator.addItem( 'has', True )
self._operator.addItem( 'does not have', False )
self._exact_url = QW.QLineEdit( self )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( operator, rule_type, rule, description ) = predicate.GetValue()
self._operator.SetValue( operator )
self._exact_url.setText( rule )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:known url'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._operator, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'exact url:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._exact_url, CC.FLAGS_EXPAND_BOTH_WAYS )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
operator = True
rule_type = 'exact_match'
rule = ''
description = ''
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) )
def GetPredicates( self ):
operator = self._operator.GetValue()
if operator:
operator_description = 'has url: '
else:
operator_description = 'does not have url: '
rule_type = 'exact_match'
exact_url = self._exact_url.text()
rule = exact_url
description = operator_description + exact_url
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) ), )
return predicates
class PanelPredicateSystemKnownURLsDomain( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._operator = ClientGUICommon.BetterChoice( self )
self._operator.addItem( 'has', True )
self._operator.addItem( 'does not have', False )
self._domain = QW.QLineEdit( self )
self._domain.setPlaceholderText( 'example.com' )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( operator, rule_type, rule, description ) = predicate.GetValue()
self._operator.SetValue( operator )
self._domain.setText( rule )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:known url'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._operator, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'a url with domain:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._domain, CC.FLAGS_EXPAND_BOTH_WAYS )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
operator = True
rule_type = 'domain'
rule = ''
description = ''
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) )
def GetPredicates( self ):
operator = self._operator.GetValue()
if operator:
operator_description = 'has a url with domain: '
else:
operator_description = 'does not have a url with domain: '
rule_type = 'domain'
domain = self._domain.text()
rule = domain
description = operator_description + domain
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) ), )
return predicates
class PanelPredicateSystemKnownURLsRegex( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._operator = ClientGUICommon.BetterChoice( self )
self._operator.addItem( 'has', True )
self._operator.addItem( 'does not have', False )
self._regex = QW.QLineEdit( self )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( operator, rule_type, rule, description ) = predicate.GetValue()
self._operator.SetValue( operator )
self._regex.setText( rule )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:known url'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._operator, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'a url that matches this regex:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._regex, CC.FLAGS_EXPAND_BOTH_WAYS )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
operator = True
rule_type = 'regex'
rule = ''
description = ''
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) )
def CheckValid( self ):
regex = self._regex.text()
try:
re.compile( regex )
except Exception as e:
raise HydrusExceptions.VetoException( 'Cannot compile that regex: {}'.format( e ) )
def GetPredicates( self ):
operator = self._operator.GetValue()
if operator:
operator_description = 'has a url matching regex: '
else:
operator_description = 'does not have a url matching regex: '
rule_type = 'regex'
regex = self._regex.text()
rule = regex
description = operator_description + regex
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) ), )
return predicates
class PanelPredicateSystemKnownURLsURLClass( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._operator = ClientGUICommon.BetterChoice( self )
self._operator.addItem( 'has', True )
self._operator.addItem( 'does not have', False )
self._url_classes = ClientGUICommon.BetterChoice( self )
for url_class in CG.client_controller.network_engine.domain_manager.GetURLClasses():
if url_class.ShouldAssociateWithFiles():
self._url_classes.addItem( url_class.GetName(), url_class )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( operator, rule_type, rule, description ) = predicate.GetValue()
self._operator.SetValue( operator )
self._url_classes.SetValue( rule )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:known url'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._operator, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'url matching this class:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._url_classes, CC.FLAGS_EXPAND_BOTH_WAYS )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
operator = True
rule_type = 'regex'
rule = None
description = ''
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) )
def GetPredicates( self ):
operator = self._operator.GetValue()
if operator:
operator_description = 'has '
else:
operator_description = 'does not have '
rule_type = 'url_class'
url_class = self._url_classes.GetValue()
rule = url_class
description = operator_description + url_class.GetName() + ' url'
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_KNOWN_URLS, ( operator, rule_type, rule, description ) ), )
return predicates
class PanelPredicateSystemLimit( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._limit = ClientGUICommon.BetterSpinBox( self, min = 1, max=1000000, width = 60 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
limit = predicate.GetValue()
self._limit.setValue( limit )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:limit='), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._limit, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
limit = 256
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_LIMIT, limit )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_LIMIT, self._limit.value() ), )
return predicates
class PanelPredicateSystemMime( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._mimes = ClientGUIOptionsPanels.OptionsPanelMimesTree( self, HC.SEARCHABLE_MIMES )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
summary_mimes = predicate.GetValue()
if isinstance( summary_mimes, int ):
summary_mimes = ( summary_mimes, )
specific_mimes = ClientSearch.ConvertSummaryFiletypesToSpecific( summary_mimes )
self._mimes.SetValue( specific_mimes )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self, 'system:filetype' ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._mimes, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
specific_mimes = tuple()
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_MIME, specific_mimes )
def GetPredicates( self ):
specific_mimes = self._mimes.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_MIME, specific_mimes ), )
return predicates
class PanelPredicateSystemNumPixels( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = QP.RadioBox( self, choices=[ '<', HC.UNICODE_APPROX_EQUAL, '=', HC.UNICODE_NOT_EQUAL, '>' ] )
self._num_pixels = ClientGUICommon.BetterSpinBox( self, max=1048576, width = 60 )
self._unit = QP.RadioBox( self, choices=['pixels','kilopixels','megapixels'] )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, num_pixels, unit ) = predicate.GetValue()
self._sign.SetStringSelection( sign )
self._num_pixels.setValue( num_pixels )
self._unit.SetStringSelection( HydrusData.ConvertIntToPixels( unit ) )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:num_pixels'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._num_pixels, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._unit, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
sign = HC.UNICODE_APPROX_EQUAL
num_pixels = 2
unit = 1000000
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_PIXELS, ( sign, num_pixels, unit ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_PIXELS, ( self._sign.GetStringSelection(), self._num_pixels.value(), HydrusData.ConvertPixelsToInt( self._unit.GetStringSelection() ) ) ), )
return predicates
class PanelPredicateSystemNumFrames( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators, max = 1000000, appropriate_absolute_plus_or_minus_default = 300 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self, 'system:number of frames' ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN, value = 600 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_FRAMES, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_FRAMES, number_test ), )
return predicates
class PanelPredicateSystemNumTags( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._namespace = QW.QLineEdit( self )
self._namespace.setPlaceholderText( 'Leave empty for unnamespaced, \'*\' for all namespaces' )
self._namespace.setToolTip( 'Leave empty for unnamespaced, \'*\' for all namespaces. Other wildcards also supported.' )
self._sign = QP.RadioBox( self, choices=['<',HC.UNICODE_APPROX_EQUAL,'=','>'] )
self._num_tags = ClientGUICommon.BetterSpinBox( self, max=2000, width = 60 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( namespace, sign, num_tags ) = predicate.GetValue()
if namespace is None:
namespace = '*'
self._namespace.setText( namespace )
self._sign.SetStringSelection( sign )
self._num_tags.setValue( num_tags )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:number of tags: namespace:'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._namespace, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._num_tags, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
namespace = '*'
sign = '>'
num_tags = 4
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_TAGS, ( namespace, sign, num_tags ) )
def GetPredicates( self ):
( namespace, operator, value ) = ( self._namespace.text(), self._sign.GetStringSelection(), self._num_tags.value() )
predicate = None
# swap num character tags > 0 with character:*anything*
if namespace != '*':
number_test = ClientSearch.NumberTest.STATICCreateFromCharacters( operator, value )
if number_test.IsZero():
predicate = ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_NAMESPACE, namespace, inclusive = False )
elif number_test.IsAnythingButZero():
predicate = ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_NAMESPACE, namespace )
if predicate is None:
predicate = ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_TAGS, ( namespace, operator, value ) )
predicates = ( predicate, )
return predicates
class PanelPredicateSystemNumNotes( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self,'system:number of notes' ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_EQUAL, value = 2 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_NOTES, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_NOTES, number_test ), )
return predicates
class PanelPredicateSystemNumURLs( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText( self,'system:number of urls' ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN, value = 0 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_URLS, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_URLS, number_test ), )
return predicates
class PanelPredicateSystemNumWords( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators, max = 100000000, appropriate_absolute_plus_or_minus_default = 5000 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:number of words'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN, value = 30000 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_WORDS, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_NUM_WORDS, number_test ), )
return predicates
class PanelPredicateSystemRatio( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = QP.RadioBox( self, choices=['=','wider than','taller than',HC.UNICODE_APPROX_EQUAL,HC.UNICODE_NOT_EQUAL] )
self._width = ClientGUICommon.BetterSpinBox( self, max=50000, width = 60 )
self._height = ClientGUICommon.BetterSpinBox( self, max=50000, width = 60 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, width, height ) = predicate.GetValue()
self._sign.SetStringSelection( sign )
self._width.setValue( width )
self._height.setValue( height )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:ratio'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._width, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,':'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._height, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
sign = 'wider than'
width = 16
height = 9
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_RATIO, ( sign, width, height ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_RATIO, ( self._sign.GetStringSelection(), self._width.value(), self._height.value() ) ), )
return predicates
class PanelPredicateSystemSimilarToData( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._clear_button = ClientGUICommon.BetterButton( self, 'clear', self._Clear )
self._paste_button = ClientGUICommon.BetterBitmapButton( self, CC.global_pixmaps().paste, self._Paste )
self._paste_button.setText( 'Paste image!')
self._pixel_hashes = QW.QPlainTextEdit( self )
self._perceptual_hashes = QW.QPlainTextEdit( self )
( init_width, init_height ) = ClientGUIFunctions.ConvertTextToPixels( self._pixel_hashes, ( 66, 4 ) )
self._pixel_hashes.setMinimumWidth( init_width )
self._pixel_hashes.setMaximumHeight( init_height )
self._perceptual_hashes.setMinimumWidth( init_width )
self._perceptual_hashes.setMaximumHeight( init_height )
self._max_hamming = ClientGUICommon.BetterSpinBox( self, max=256, width = 60 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
self._pixel_hashes.setPlaceholderText( 'enter pixel hash (64 chars each, paste newline-separated for multiple)' )
self._perceptual_hashes.setPlaceholderText( 'enter perceptual hash (16 chars each, paste newline-separated for multiple)' )
( pixel_hashes, perceptual_hashes, hamming_distance ) = predicate.GetValue()
hashes_text = os.linesep.join( [ hash.hex() for hash in pixel_hashes ] )
self._pixel_hashes.setPlainText( hashes_text )
hashes_text = os.linesep.join( [ hash.hex() for hash in perceptual_hashes ] )
self._perceptual_hashes.setPlainText( hashes_text )
self._max_hamming.setValue( hamming_distance )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:similar to'), CC.FLAGS_CENTER_PERPENDICULAR )
vbox = QP.VBoxLayout()
QP.AddToLayout( vbox, self._pixel_hashes, CC.FLAGS_EXPAND_BOTH_WAYS )
QP.AddToLayout( vbox, self._perceptual_hashes, CC.FLAGS_EXPAND_BOTH_WAYS )
QP.AddToLayout( hbox, vbox, CC.FLAGS_EXPAND_SIZER_BOTH_WAYS )
QP.AddToLayout( hbox, QW.QLabel( HC.UNICODE_APPROX_EQUAL, self ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._max_hamming, CC.FLAGS_CENTER_PERPENDICULAR )
big_vbox = QP.VBoxLayout()
st = ClientGUICommon.BetterStaticText( self, label = 'Use this if you want to look up a file without importing it. Just copy its file path or image data to your clipboard and paste.' )
st.setWordWrap( True )
st.setAlignment( QC.Qt.AlignCenter )
QP.AddToLayout( big_vbox, st, CC.FLAGS_EXPAND_PERPENDICULAR )
button_hbox = QP.HBoxLayout()
QP.AddToLayout( button_hbox, self._paste_button, CC.FLAGS_CENTER )
QP.AddToLayout( button_hbox, self._clear_button, CC.FLAGS_CENTER )
QP.AddToLayout( big_vbox, button_hbox, CC.FLAGS_CENTER )
QP.AddToLayout( big_vbox, hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
big_vbox.addStretch( 1 )
self.setLayout( big_vbox )
def _Clear( self ):
self._pixel_hashes.setPlainText( '' )
self._perceptual_hashes.setPlainText( '' )
def _Paste( self ):
if CG.client_controller.ClipboardHasImage():
try:
qt_image = CG.client_controller.GetClipboardImage()
numpy_image = ClientGUIFunctions.ConvertQtImageToNumPy( qt_image )
pixel_hash = HydrusImageHandling.GetImagePixelHashNumPy( numpy_image )
perceptual_hashes = ClientImageHandling.GenerateShapePerceptualHashesNumPy( numpy_image )
except Exception as e:
HydrusData.PrintException( e )
ClientGUIDialogsMessage.ShowCritical( self, 'Problem pasting!', f'Sorry, seemed to be a problem: {e}' )
return
else:
try:
raw_text = CG.client_controller.GetClipboardText()
except HydrusExceptions.DataMissing as e:
ClientGUIDialogsMessage.ShowWarning( self, 'Did not see an image bitmap or a file path in the clipboard!' )
return
try:
path = raw_text
if os.path.exists( path ) and os.path.isfile( path ):
mime = HydrusFileHandling.GetMime( path )
if mime in HC.FILES_THAT_HAVE_PERCEPTUAL_HASH:
pixel_hash = HydrusImageHandling.GetImagePixelHash( path, mime )
perceptual_hashes = ClientImageHandling.GenerateShapePerceptualHashes( path, mime )
else:
ClientGUIDialogsMessage.ShowWarning( self, 'Sorry, "{}" files are not compatible with the similar file search system!'.format( HC.mime_string_lookup[ mime ] ) )
return
else:
ClientGUIDialogsMessage.ShowWarning( self, 'Sorry, that clipboard text did not look like a valid file path!' )
return
except Exception as e:
HydrusData.PrintException( e )
ClientGUIDialogsMessage.ShowCritical( self, 'Problem pasting!', f'Sorry, seemed to be a problem: {e}' )
return
new_text_lines = self._pixel_hashes.toPlainText().splitlines()
new_text_lines.append( pixel_hash.hex() )
new_text_lines = HydrusData.DedupeList( new_text_lines )
self._pixel_hashes.setPlainText( '\n'.join( new_text_lines ) )
new_text_lines = self._perceptual_hashes.toPlainText().splitlines()
new_text_lines.extend( [ perceptual_hash.hex() for perceptual_hash in perceptual_hashes ] )
new_text_lines = HydrusData.DedupeList( new_text_lines )
self._perceptual_hashes.setPlainText( '\n'.join( new_text_lines ) )
def GetDefaultPredicate( self ):
pixel_hashes = tuple()
perceptual_hashes = tuple()
max_hamming = 8
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIMILAR_TO_DATA, ( pixel_hashes, perceptual_hashes, max_hamming ) )
def GetPredicates( self ):
hex_pixel_hashes_raw = self._pixel_hashes.toPlainText()
pixel_hashes = HydrusData.ParseHashesFromRawHexText( 'pixel', hex_pixel_hashes_raw )
hex_perceptual_hashes_raw = self._perceptual_hashes.toPlainText()
perceptual_hashes = HydrusData.ParseHashesFromRawHexText( 'perceptual', hex_perceptual_hashes_raw )
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIMILAR_TO_DATA, ( pixel_hashes, perceptual_hashes, self._max_hamming.value() ) ), )
return predicates
class PanelPredicateSystemSimilarToFiles( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._hashes = QW.QPlainTextEdit( self )
( init_width, init_height ) = ClientGUIFunctions.ConvertTextToPixels( self._hashes, ( 66, 6 ) )
self._hashes.setMinimumWidth( init_width )
self._hashes.setMaximumHeight( init_height )
self._max_hamming = ClientGUICommon.BetterSpinBox( self, max=256, width = 60 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
self._hashes.setPlaceholderText( 'enter file hash (64 chars each, paste newline-separated for multiple)' )
( hashes, hamming_distance ) = predicate.GetValue()
hashes_text = os.linesep.join( [ hash.hex() for hash in hashes ] )
self._hashes.setPlainText( hashes_text )
self._max_hamming.setValue( hamming_distance )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:similar to'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._hashes, CC.FLAGS_EXPAND_BOTH_WAYS )
QP.AddToLayout( hbox, QW.QLabel( HC.UNICODE_APPROX_EQUAL, self ), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._max_hamming, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
st = ClientGUICommon.BetterStaticText( self, label = 'This searches for files that look like each other, just like in the duplicates system. Paste the files\' hash(es) here, and the results will look like any of them.' )
st.setWordWrap( True )
st.setAlignment( QC.Qt.AlignCenter )
vbox = QP.VBoxLayout()
QP.AddToLayout( vbox, st, CC.FLAGS_EXPAND_PERPENDICULAR )
QP.AddToLayout( vbox, hbox, CC.FLAGS_EXPAND_SIZER_PERPENDICULAR )
vbox.addStretch( 1 )
self.setLayout( vbox )
def GetDefaultPredicate( self ):
hashes = tuple()
max_hamming = 4
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIMILAR_TO_FILES, ( hashes, max_hamming ) )
def GetPredicates( self ):
hex_hashes_raw = self._hashes.toPlainText()
hashes = HydrusData.ParseHashesFromRawHexText( 'sha256', hex_hashes_raw )
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIMILAR_TO_FILES, ( hashes, self._max_hamming.value() ) ), )
return predicates
class PanelPredicateSystemSize( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._sign = QP.RadioBox( self, choices=['<',HC.UNICODE_APPROX_EQUAL,'=',HC.UNICODE_NOT_EQUAL,'>'] )
self._bytes = ClientGUIControls.BytesControl( self )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( sign, size, unit ) = predicate.GetValue()
self._sign.SetStringSelection( sign )
self._bytes.SetSeparatedValue( size, unit )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:filesize'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._bytes, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
sign = '<'
size = 200
unit = 1024
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIZE, ( sign, size, unit ) )
def GetPredicates( self ):
( size, unit ) = self._bytes.GetSeparatedValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_SIZE, ( self._sign.GetStringSelection(), size, unit ) ), )
return predicates
class PanelPredicateSystemTagAsNumber( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
self._namespace = QW.QLineEdit( self )
choices = [ '<', HC.UNICODE_APPROX_EQUAL, '>' ]
self._sign = QP.RadioBox( self, choices = choices )
self._num = ClientGUICommon.BetterSpinBox( self, min=-(2 ** 31), max= (2 ** 31) - 1 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
( namespace, sign, num ) = predicate.GetValue()
self._namespace.setText( namespace )
self._namespace.setPlaceholderText( 'Leave empty for unnamespaced, \'*\' for all namespaces' )
self._namespace.setToolTip( 'Leave empty for unnamespaced, \'*\' for all namespaces. Other wildcards also supported.' )
self._sign.SetStringSelection( sign )
self._num.setValue( num )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:tag as number: '), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._namespace, CC.FLAGS_EXPAND_BOTH_WAYS )
QP.AddToLayout( hbox, self._sign, CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._num, CC.FLAGS_CENTER_PERPENDICULAR )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
namespace = 'page'
sign = '>'
num = 0
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_TAG_AS_NUMBER, ( namespace, sign, num ) )
def GetPredicates( self ):
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_TAG_AS_NUMBER, ( self._namespace.text(), self._sign.GetStringSelection(), self._num.value() ) ), )
return predicates
class PanelPredicateSystemWidth( PanelPredicateSystemSingle ):
def __init__( self, parent, predicate ):
PanelPredicateSystemSingle.__init__( self, parent )
allowed_operators = [
ClientSearch.NUMBER_TEST_OPERATOR_LESS_THAN,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_ABSOLUTE,
ClientSearch.NUMBER_TEST_OPERATOR_APPROXIMATE_PERCENT,
ClientSearch.NUMBER_TEST_OPERATOR_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_NOT_EQUAL,
ClientSearch.NUMBER_TEST_OPERATOR_GREATER_THAN
]
self._number_test = ClientGUIControls.NumberTestWidget( self, allowed_operators = allowed_operators, unit_string = 'px', appropriate_absolute_plus_or_minus_default = 200 )
#
predicate = self._GetPredicateToInitialisePanelWith( predicate )
number_test = predicate.GetValue()
self._number_test.SetValue( number_test )
#
hbox = QP.HBoxLayout()
QP.AddToLayout( hbox, ClientGUICommon.BetterStaticText(self,'system:width'), CC.FLAGS_CENTER_PERPENDICULAR )
QP.AddToLayout( hbox, self._number_test, CC.FLAGS_CENTER_PERPENDICULAR )
hbox.addStretch( 1 )
self.setLayout( hbox )
def GetDefaultPredicate( self ):
number_test = ClientSearch.NumberTest( operator = ClientSearch.NUMBER_TEST_OPERATOR_EQUAL, value = 1920 )
return ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_WIDTH, number_test )
def GetPredicates( self ):
number_test = self._number_test.GetValue()
predicates = ( ClientSearch.Predicate( ClientSearch.PREDICATE_TYPE_SYSTEM_WIDTH, number_test ), )
return predicates
Reference in New Issue View Git Blame Copy Permalink