# -*- coding: utf-8 -*- import wx class UserActionsDialog(wx.Dialog): def __init__(self, users=[], default="follow", *args, **kwargs): super(UserActionsDialog, self).__init__(parent=None, *args, **kwargs) panel = wx.Panel(self) userSizer = wx.BoxSizer() self.SetTitle(_(u"Action")) userLabel = wx.StaticText(panel, -1, _(u"User")) self.cb = wx.ComboBox(panel, -1, choices=users, value=users[0]) self.cb.SetFocus() userSizer.Add(userLabel, 0, wx.ALL, 5) userSizer.Add(self.cb, 0, wx.ALL, 5) actionSizer = wx.BoxSizer(wx.VERTICAL) label2 = wx.StaticText(panel, -1, _(u"Action")) self.follow = wx.RadioButton(panel, -1, _(u"Follow"), style=wx.RB_GROUP) self.unfollow = wx.RadioButton(panel, -1, _(u"Unfollow")) self.mute = wx.RadioButton(panel, -1, _(u"Mute")) self.unmute = wx.RadioButton(panel, -1, _(u"Unmute")) self.block = wx.RadioButton(panel, -1, _(u"Block")) self.unblock = wx.RadioButton(panel, -1, _(u"Unblock")) self.reportSpam = wx.RadioButton(panel, -1, _(u"Report as spam")) self.ignore_client = wx.RadioButton(panel, -1, _(u"Ignore tweets from this client")) self.setup_default(default) hSizer = wx.BoxSizer(wx.HORIZONTAL) hSizer.Add(label2, 0, wx.ALL, 5) actionSizer.Add(self.follow, 0, wx.ALL, 5) actionSizer.Add(self.unfollow, 0, wx.ALL, 5) actionSizer.Add(self.mute, 0, wx.ALL, 5) actionSizer.Add(self.unmute, 0, wx.ALL, 5) actionSizer.Add(self.block, 0, wx.ALL, 5) actionSizer.Add(self.unblock, 0, wx.ALL, 5) actionSizer.Add(self.reportSpam, 0, wx.ALL, 5) actionSizer.Add(self.ignore_client, 0, wx.ALL, 5) hSizer.Add(actionSizer, 0, wx.ALL, 5) sizer = wx.BoxSizer(wx.VERTICAL) ok = wx.Button(panel, wx.ID_OK, _(u"OK")) ok.SetDefault() cancel = wx.Button(panel, wx.ID_CANCEL, _(u"Close")) btnsizer = wx.BoxSizer() btnsizer.Add(ok) btnsizer.Add(cancel) sizer.Add(userSizer) sizer.Add(hSizer, 0, wx.ALL, 5) sizer.Add(btnsizer) panel.SetSizer(sizer) def get_action(self): if self.follow.GetValue() == True: return "follow" elif self.unfollow.GetValue() == True: return "unfollow" elif self.mute.GetValue() == True: return "mute" elif self.unmute.GetValue() == True: return "unmute" elif self.reportSpam.GetValue() == True: return "report" elif self.block.GetValue() == True: return "block" elif self.unblock.GetValue() == True: return "unblock" elif self.ignore_client.GetValue() == True: return "ignore_client" def setup_default(self, default): if default == "follow": self.follow.SetValue(True) elif default == "unfollow": self.unfollow.SetValue(True) elif default == "mute": self.mute.SetValue(True) elif default == "unmute": self.unmute.SetValue(True) elif default == "report": self.reportSpam.SetValue(True) elif default == "block": self.block.SetValue(True) elif default == "unblock": self.unblock.SetValue(True) elif default == "ignore_client": self.ignore_client.SetValue(True) def get_response(self): return self.ShowModal() def get_user(self): return self.cb.GetValue()