def SetBundleState(self, newstate, refresh=True, reset=False):
if newstate is None:
auto_guess = self.guiutility.utility.config.Read('use_bundle_magic', "boolean")
newstate = Bundler.ALG_OFF # default
stored_state = None
if not reset:
keywords = self.torrentsearch_manager.getSearchKeywords()[0]
if keywords != '':
try:
stored_state = self.bundle_db.getPreference(keywords)
except:
pass
#if db interaction fails, ignore
local_override = stored_state is not None
if local_override:
newstate = stored_state
elif auto_guess:
newstate = Bundler.ALG_MAGIC
self.bundlestate = newstate
self.selected_bundle_mode = None
self.Freeze()
if newstate != Bundler.ALG_OFF:
self.bundlestatetext.SetLabel(' by %s' % self.bundlestates_str[newstate])
else:
self.bundlestatetext.SetLabel(' is %s' % self.bundlestates_str[newstate])
self.torrentsearch_manager.setBundleMode(newstate, refresh)
self.bundleSizer.ShowItems(False)
self.bundleSizer.Clear(deleteWindows = True)
self.bundletexts = []
self.bundleSizer.Add(StaticText(self, -1, 'Bundle by '))
for i, state in enumerate(self.bundlestates):
if newstate == state:
text = StaticText(self, -1, self.bundlestates_str[state])
self.bundleSizer.Add(text)
self.bundletexts.append(text)
else:
link = LinkStaticText(self, self.bundlestates_str[state], "wand.png")
link.ShowIcon(False)
link.SetIconToolTipString('Selected by Magic')
link.Bind(wx.EVT_LEFT_UP, self.OnRebundle)
link.action = state
self.bundleSizer.Add(link)
self.bundletexts.append(link)
if i+1 < len(self.bundlestates):
self.bundleSizer.AddSpacer((1, -1))
self.Layout()
self.Thaw()
class FamilyFilterHeader(TitleHeader):
def __init__(self, *args, **kwargs):
self.family_filter = None
self.nrfiltered = 0
TitleHeader.__init__(self, *args, **kwargs)
@warnWxThread
def GetSubTitlePanel(self, parent):
hSizer = wx.BoxSizer(wx.HORIZONTAL)
self.ff = StaticText(parent)
self.ffbutton = LinkStaticText(parent, '', None)
self.ffbutton.Bind(wx.EVT_LEFT_UP, self.toggleFamilyFilter)
self._SetLabels()
hSizer.Add(self.ff)
hSizer.Add(self.ffbutton)
return hSizer
def SetFF(self, family_filter, nrfiltered=0):
self.family_filter = family_filter
self.nrfiltered = nrfiltered
self._SetLabels()
def SetFamilyFiltered(self, nr):
self.nrfiltered = nr
self._SetLabels()
@warnWxThread
def SetBackgroundColour(self, colour):
TitleHeader.SetBackgroundColour(self, colour)
if getattr(self, 'ffbutton', False):
self.ffbutton.SetBackgroundColour(colour)
def toggleFamilyFilter(self, event):
self.parent_list.toggleFamilyFilter()
@warnWxThread
def _SetLabels(self):
self.Freeze()
if self.family_filter:
if self.nrfiltered > 0:
self.ff.SetLabel('%d results blocked by Family Filter, ' %
self.nrfiltered)
else:
self.ff.SetLabel('Family Filter is On, ')
self.ffbutton.SetLabel('turn off')
else:
self.ff.SetLabel('Family Filter is Off, ')
self.ffbutton.SetLabel('turn on')
self.Layout()
self.Thaw()
def UpdateGrid(self, hits, noChange=False):
N = BUNDLE_NUM_ROWS * BUNDLE_NUM_COLS
items_to_add = min(N, self.nrhits)
if self.nrhits > N:
items_to_add -= 1
self.parent.Freeze()
children = self.grid.GetChildren()
didChange = len(children) < min(N, self.nrhits)
if not didChange:
if DEBUG:
print >> sys.stderr, "*** BundlePanel.UpdateGrid: total nr items did not change, updating labels only"
#total nr items did not change
for i in range(items_to_add):
link_static_text = children[i].GetWindow() or children[i].GetSizer()
if link_static_text and getattr(link_static_text, 'SetLabel', False):
link_static_text.SetLabel(hits[i].name)
link_static_text.action = hits[i]
else:
didChange = True
break
if self.nrhits > N:
more_caption = '(%s more...)' % (self.nrhits - N + 1)
link_static_text = children[i+1].GetWindow() or children[i+1].GetSizer()
if link_static_text and getattr(link_static_text, 'SetLabel', False):
link_static_text.SetLabel(more_caption)
link_static_text.Unbind(wx.EVT_LEFT_UP)
link_static_text.Bind(wx.EVT_LEFT_UP, self.OnMoreClick)
else:
didChange = True
if didChange:
if DEBUG:
print >> sys.stderr, "*** BundlePanel.UpdateGrid: something did change rebuilding grid", len(children),min(N, self.nrhits)
curRows = len(children) / BUNDLE_NUM_COLS
newRows = min(self.nrhits / BUNDLE_NUM_COLS, BUNDLE_NUM_ROWS)
rowsChanged = curRows != newRows
self.grid.ShowItems(False)
self.grid.Clear(deleteWindows = True)
for i in range(items_to_add):
hit = hits[i]
new_text = LinkStaticText(self.parent, hit.name, icon = False, icon_type = 'tree', icon_align = wx.ALIGN_LEFT, font_increment = self.font_increment, font_colour = BUNDLE_FONT_COLOR)
new_text.Bind(wx.EVT_LEFT_UP, self.OnBundleLinkClick)
new_text.SetMinSize((1,-1))
new_text.action = hit
self.grid.Add(new_text, 0, wx.EXPAND)
if self.nrhits > N:
caption = '(%s more...)' % (self.nrhits - N + 1)
more_label = LinkStaticText(self.parent, caption, icon = False, icon_align = wx.ALIGN_LEFT, font_increment = self.font_increment, font_colour = BUNDLE_FONT_COLOR)
more_label.Bind(wx.EVT_LEFT_UP, self.OnMoreClick)
self.grid.Add(more_label, 0, wx.EXPAND)
self.parent_listitem.AddEvents(self.grid)
if self.state != self.COLLAPSED:
self.ShowGrid(False)
if rowsChanged and not noChange:
self.parent_listitem.OnChange()
self.parent.Thaw()
return didChange
class SearchSideBar(wx.Panel):
INDENT = 7
HEADER_FONT_WEIGHT = wx.FONTWEIGHT_NORMAL
def __init__(self, parent, parent_list, size):
wx.Panel.__init__(self, parent, size = size)
self.SetForegroundColour(parent.GetForegroundColour())
self.guiutility = GUIUtility.getInstance()
self.torrentsearch_manager = self.guiutility.torrentsearch_manager
self.parent = parent
self.parent_list = parent_list
self.nrfiltered = 0
self.family_filter = True
self.bundlestates = [Bundler.ALG_MAGIC, Bundler.ALG_NAME, Bundler.ALG_NUMBERS, Bundler.ALG_SIZE, Bundler.ALG_OFF]
self.bundlestates_str = {Bundler.ALG_NAME: 'Name',
Bundler.ALG_NUMBERS: 'Numbers',
Bundler.ALG_SIZE: 'Size',
Bundler.ALG_MAGIC: 'Magic',
Bundler.ALG_OFF: 'Off'}
self.bundletexts = []
self.bundle_db = BundlerPreferenceDBHandler.getInstance()
self.uelog = UserEventLogDBHandler.getInstance()
self.vSizer = wx.BoxSizer(wx.VERTICAL)
hSizer = wx.BoxSizer(wx.HORIZONTAL)
header = StaticText(self, -1, 'Search')
if SearchSideBar.HEADER_FONT_WEIGHT != wx.FONTWEIGHT_NORMAL:
font = header.GetFont()
font.SetWeight(SearchSideBar.HEADER_FONT_WEIGHT)
header.SetFont(font)
hSizer.Add(header, 0, wx.ALIGN_CENTER_VERTICAL)
self.searchState = StaticText(self)
hSizer.Add(self.searchState, 1, wx.ALIGN_CENTER_VERTICAL)
ag_fname = os.path.join(self.guiutility.utility.getPath(), LIBRARYNAME, 'Main', 'vwxGUI', 'images', 'search_new.gif')
self.ag = wx.animate.GIFAnimationCtrl(self, -1, ag_fname)
self.ag.UseBackgroundColour(True)
self.ag.Hide()
hSizer.Add(self.ag, 0, wx.RESERVE_SPACE_EVEN_IF_HIDDEN)
self.vSizer.Add(hSizer, 0, wx.EXPAND|wx.BOTTOM, 3)
self.vSizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND|wx.BOTTOM, 3)
self.searchGauge = wx.Gauge(self, size = (-1, 7))
self.vSizer.Add(self.searchGauge, 0, wx.EXPAND|wx.RESERVE_SPACE_EVEN_IF_HIDDEN)
self.vSizer.AddSpacer((-1,15))
hSizer = wx.BoxSizer(wx.HORIZONTAL)
header = StaticText(self, -1, 'Family Filter')
if SearchSideBar.HEADER_FONT_WEIGHT != wx.FONTWEIGHT_NORMAL:
font = header.GetFont()
font.SetWeight(SearchSideBar.HEADER_FONT_WEIGHT)
header.SetFont(font)
hSizer.Add(header)
self.ffstate = StaticText(self)
hSizer.Add(self.ffstate)
self.vSizer.Add(hSizer, 0, wx.EXPAND|wx.BOTTOM, 3)
self.vSizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND|wx.BOTTOM, 3)
self.ffblocked = StaticText(self)
self.vSizer.Add(self.ffblocked, 0, wx.EXPAND|wx.LEFT, SearchSideBar.INDENT)
self.ffbutton = LinkStaticText(self, '', None)
self.ffbutton.Bind(wx.EVT_LEFT_UP, self.toggleFamilyFilter)
self.vSizer.Add(self.ffbutton, 0, wx.EXPAND|wx.LEFT, SearchSideBar.INDENT)
self.vSizer.AddSpacer((-1,15))
hSizer = wx.BoxSizer(wx.HORIZONTAL)
header = StaticText(self, -1, 'Bundling')
if SearchSideBar.HEADER_FONT_WEIGHT != wx.FONTWEIGHT_NORMAL:
font = header.GetFont()
font.SetWeight(SearchSideBar.HEADER_FONT_WEIGHT)
header.SetFont(font)
hSizer.Add(header)
#keep longest text in bundlestatetext, to define bestsize (width) for sidepanel
self.bundlestatetext = StaticText(self, -1, ' by Numbers')
hSizer.Add(self.bundlestatetext)
self.vSizer.Add(hSizer, 0, wx.EXPAND|wx.BOTTOM, 3)
self.vSizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND|wx.BOTTOM, 3)
self.bundleSizer = wx.FlexGridSizer(0, 2, 0, 0)
self.SetBundleState(None, reset = True)
self.vSizer.Add(self.bundleSizer, 0, wx.EXPAND|wx.LEFT, SearchSideBar.INDENT)
self.vSizer.AddSpacer((-1,15))
header = StaticText(self, -1, 'Associated Channels')
if SearchSideBar.HEADER_FONT_WEIGHT != wx.FONTWEIGHT_NORMAL:
font = header.GetFont()
font.SetWeight(SearchSideBar.HEADER_FONT_WEIGHT)
header.SetFont(font)
self.vSizer.Add(header, 0, wx.EXPAND|wx.BOTTOM, 3)
self.vSizer.Add(wx.StaticLine(self, -1), 0, wx.EXPAND|wx.BOTTOM, 3)
self.nochannels = StaticText(self, -1, 'None')
self.vSizer.Add(self.nochannels, 0, wx.EXPAND|wx.LEFT, SearchSideBar.INDENT)
self.channels = [LinkStaticText(self, '', icon = None) for _ in range(3)]
for channel in self.channels:
self.vSizer.Add(channel, 0, wx.EXPAND|wx.LEFT, SearchSideBar.INDENT)
channel.Bind(wx.EVT_LEFT_UP, self.OnChannel)
borderSizer = wx.BoxSizer(wx.VERTICAL)
borderSizer.AddSpacer((-1, 3))
borderSizer.Add(self.vSizer, 1, wx.EXPAND|wx.LEFT|wx.RIGHT, 7)
self.SetSizer(borderSizer)
self.SetMinSize((self.GetBestSize()[0], -1))
self.Reset()
def SetFF(self, family_filter, nrfiltered):
self.family_filter = family_filter
self.nrfiltered = nrfiltered
self._SetLabels()
@forceWxThread
def SetMaxResults(self, max, remotekeywords):
self.Freeze()
self.searchGauge.SetRange(max)
self.searchGauge.SetValue(0)
self.searchGauge.Show()
self.searchState.SetLabel(' in progress')
wx.CallLater(10000, self.SetFinished, remotekeywords)
self.ag.Play()
self.ag.Show()
self.Thaw()
@forceWxThread
def NewResult(self):
maxValue = self.searchGauge.GetRange()
newValue = min(self.searchGauge.GetValue() + 1, maxValue)
if newValue == maxValue:
self.SetFinished(None)
else:
self.searchGauge.SetValue(newValue)
def SetFinished(self, keywords):
curkeywords, hits, filtered = self.guiutility.torrentsearch_manager.getSearchKeywords()
if not keywords or curkeywords == keywords:
self.Freeze()
self.ag.Stop()
self.ag.Hide()
self.searchGauge.Hide()
self.searchState.SetLabel(' completed')
self.Layout()
self.Thaw()
self.guiutility.frame.searchlist.SetFinished()
@forceWxThread
def SetAssociatedChannels(self, channels):
#channels should be a list, of occurrences, name, permid
self.Freeze()
self.nochannels.Show(len(channels) == 0)
for i in range(len(self.channels)):
if i < len(channels):
tooltip = "Click to go to %s's Channel."%channels[i][-1].name
self.channels[i].SetLabel(channels[i][-1].name)
self.channels[i].SetToolTipString(tooltip)
self.channels[i].channel = channels[i][-1]
else:
self.channels[i].SetLabel('')
self.channels[i].SetToolTipString('')
self.Layout()
self.Thaw()
def toggleFamilyFilter(self, event):
self.parent_list.toggleFamilyFilter()
def _SetLabels(self):
self.Freeze()
if self.family_filter:
if self.nrfiltered > 0:
if self.nrfiltered > 1:
self.ffblocked.SetLabel('%d results blocked'%self.nrfiltered)
else:
self.ffblocked.SetLabel('1 result blocked')
self.vSizer.Detach(self.ffblocked)
self.vSizer.Insert(6, self.ffblocked, 0, wx.EXPAND|wx.LEFT, 7)
else:
self.ffblocked.SetLabel('')
self.vSizer.Detach(self.ffblocked)
self.vSizer.Insert(7, self.ffblocked)
self.ffstate.SetLabel(' is On')
self.ffbutton.SetLabel('turn off')
else:
self.ffstate.SetLabel(' is Off')
self.ffbutton.SetLabel('turn on')
self.ffblocked.SetLabel('')
self.Layout()
self.Thaw()
def Reset(self):
self.SetBundleState(None,refresh=False,reset=True)
self.nochannels.Show()
self.searchState.SetLabel(' in progress')
for channel in self.channels:
channel.SetLabel('')
channel.SetToolTipString('')
def OnRebundle(self, event):
curstate = self.bundlestate
selectedByMagic = -1
for i, text in enumerate(self.bundletexts):
if isinstance(text, LinkStaticText) and text.IsIconShown():
selectedByMagic = self.bundlestates[i]
break
newstate = event.GetEventObject().action
self.SetBundleState(newstate)
def db_callback():
keywords = self.torrentsearch_manager.getSearchKeywords()[0]
self.bundle_db.storePreference(keywords, newstate)
query = ' '.join(keywords)
selectedByMagicStr = ''
if selectedByMagic != -1:
selectedByMagicStr = self.bundlestates_str[selectedByMagic]
self.uelog.addEvent(message="Bundler GUI: %s -> %s; %s -> %s; selectedByMagic %s (%s); q=%s"
% (curstate, newstate, self.bundlestates_str[curstate],
self.bundlestates_str[newstate],
selectedByMagic, selectedByMagicStr, query), type = 3)
self.guiutility.frame.guiserver.add_task(db_callback)
def SetBundleState(self, newstate, refresh=True, reset=False):
if newstate is None:
auto_guess = self.guiutility.utility.config.Read('use_bundle_magic', "boolean")
newstate = Bundler.ALG_OFF # default
stored_state = None
if not reset:
keywords = self.torrentsearch_manager.getSearchKeywords()[0]
if keywords != '':
try:
stored_state = self.bundle_db.getPreference(keywords)
except:
pass
#if db interaction fails, ignore
local_override = stored_state is not None
if local_override:
newstate = stored_state
elif auto_guess:
newstate = Bundler.ALG_MAGIC
self.bundlestate = newstate
self.selected_bundle_mode = None
self.Freeze()
if newstate != Bundler.ALG_OFF:
self.bundlestatetext.SetLabel(' by %s' % self.bundlestates_str[newstate])
else:
self.bundlestatetext.SetLabel(' is %s' % self.bundlestates_str[newstate])
self.torrentsearch_manager.setBundleMode(newstate, refresh)
self.bundleSizer.ShowItems(False)
self.bundleSizer.Clear(deleteWindows = True)
self.bundletexts = []
self.bundleSizer.Add(StaticText(self, -1, 'Bundle by '))
for i, state in enumerate(self.bundlestates):
if newstate == state:
text = StaticText(self, -1, self.bundlestates_str[state])
self.bundleSizer.Add(text)
self.bundletexts.append(text)
else:
link = LinkStaticText(self, self.bundlestates_str[state], "wand.png")
link.ShowIcon(False)
link.SetIconToolTipString('Selected by Magic')
link.Bind(wx.EVT_LEFT_UP, self.OnRebundle)
link.action = state
self.bundleSizer.Add(link)
self.bundletexts.append(link)
if i+1 < len(self.bundlestates):
self.bundleSizer.AddSpacer((1, -1))
self.Layout()
self.Thaw()
def SetSelectedBundleMode(self, selected_bundle_mode):
if self.bundlestate == Bundler.ALG_MAGIC:
self.Freeze()
self.selected_bundle_mode = selected_bundle_mode
index = self.bundlestates.index(selected_bundle_mode)
for i in range(len(self.bundletexts)):
linkStaticText = self.bundletexts[i]
if isinstance(linkStaticText, LinkStaticText):
if i == index:
if not linkStaticText.IsIconShown():
linkStaticText.ShowIcon(True)
wx.CallAfter(linkStaticText.Blink)
else:
linkStaticText.ShowIcon(False)
self.Thaw()
def OnChannel(self, event):
label = event.GetEventObject()
channel_name = label.GetLabel()
if channel_name != '':
channel = label.channel
self.guiutility.showChannel(channel)
def SetBackgroundColour(self, colour):
wx.Panel.SetBackgroundColour(self, colour)
self.ffbutton.SetBackgroundColour(colour)
self.ag.SetBackgroundColour(colour)
for channel in self.channels:
channel.SetBackgroundColour(colour)
for sizeritem in self.bundleSizer.GetChildren():
if sizeritem.IsWindow():
child = sizeritem.GetWindow()
if isinstance(child, wx.Panel):
child.SetBackgroundColour(colour)