使用Python实现伪防沉迷工具

· 之后我们来实现ui界面的关键：托盘的实现，这里要用到wx.adv，依旧定义我们自己的托盘类，继承自wx.adv.TaskBarIcon；在__init__()中__初始化frame、图标设置等__，创建菜单需要重写CreatePopupMenu()方法，在该方法中我们需要使用wx.Menu()来__创建菜单__，之后通过wx.Menu.Append()方法__将菜单项添加到菜单中__，这里每个菜单项需要用wx.MenuItem()来创建，而__每个菜单项的点击事件的绑定__则使用Bind()来完成，完成之后return menu

# 托盘类
class MyTaskBarIcon(wx.adv.TaskBarIcon):
    def __init__(self, frame):
 wx.adv.TaskBarIcon.__init__(self)
 self.frame = frame
 self.SetIcon(wx.Icon(APP_ICON), APP_TITLE)

    # 获取Menu数据
    def set_menuitem_data(self):
 data_list = (('about', self.on_about), ('Show', self.on_show), ('Close', self.on_close))
 return data_list

    # 创建菜单
    def CreatePopupMenu(self):
 my_menu = wx.Menu()
 for itemName, itemHolder in self.set_menuitem_data():
     if not itemName:
  my_menu.AppendSeparator()
  continue

     # 创建每一个菜单项
     menu_item = wx.MenuItem(None, wx.ID_ANY, text=itemName, kind=wx.ITEM_NORMAL)
     my_menu.Append(menu_item)
     self.Bind(wx.EVT_MENU, itemHolder, menu_item)

 return my_menu

    # 显示app相关信息，这里加上作者信息吧还是，2333
    def on_about(self, event):
 wx.MessageBox('signed by sophistic-cat', '关于')

    # 显示主菜单
    def on_show(self, event):
 '''
 # 这里本来设计是用来显示主窗体的，但为了避免小孩子们通过任务管理器关闭，就把这里注释掉了
 if self.frame.IsIconized():
     self.frame.Iconize(False)
 if not self.frame.IsShown():
     self.frame.Show(True)
 self.frame.Raise()
 self.frame.Maximize(True)   # 最大化显示
 '''
 pass

    # 退出app
    def on_close(self, event):
 # 实例化了一个自定义的弹窗类
 myED = ExitDialog()
 # 显示弹窗
 myED.ShowModal()
 # 判断密码是否正确，正确则退出
 # 这里是类中有一个flag标志位，当密码正确时会将flag设置为true，通过get_flag()返回
 if(myED.get_flag()):
     wx.Exit()

# 定义退出时输入密码的Dialog
class ExitDialog(wx.Dialog):
    def __init__(self):
 wx.Dialog.__init__(self, None, -1, "Exit_verify", size=(300, 100))
 # 设置静态文字，提示后面输入密码
 self.static_text = wx.StaticText(self, -1, u'请输入密码：', pos=(10, 20), size=(80, -1), style=wx.ALIGN_RIGHT)
 # 设置输入框
 self.Exit_Pwd = wx.TextCtrl(self, -1, '', pos=(90, 15), size=(100, -1), style=wx.TE_PASSWORD)
 # 设置退出按钮
 self.Sure_Bt = wx.Button(self, -1, '确认', pos=(200, 12), size=(50, -1))

 self.flag = False
 self.Center()
 # 给按钮绑定“检查密码”点击事件
 self.Sure_Bt.Bind(wx.EVT_BUTTON, self.checked_pwd)

    def checked_pwd(self, evt):
 # 获取输入框的文字
 content = self.Exit_Pwd.GetValue()
 # 打开密码文件，判断是否相同
 with open("./data/pwd.txt", "r") as f:
     date = f.read()
     if content == date:
  # 密码一致，将标志位flag设置为True
  self.flag = True
  self.Destroy()
     else:
  # 密码不一致，弹出密码错误的提示框
  wx.MessageBox('密码错误！', 'Error')
    
    # 用于外部获取flag值，来判断密码输入是否正确
    def get_flag(self):
 return self.flag

def create_splash():
    # 创建开始界面
    screen = wx.Image(r'./pic/welcome.jpg').ConvertToBitmap()   # 选择一张图片（bitmap格式）来做启动界面
    
    '''这里注意下参数含义:
 第一个参数是bitmap图；
 第二个参数是格式，这里设置为居中在屏幕中间并停顿一段时间；
 第三个参数是设置停顿时间，2000即为2s
 第四个参数是parent，这里没有，设置为None即可
    '''
    wx.adv.SplashScreen(screen, wx.adv.SPLASH_CENTRE_ON_SCREEN | wx.adv.SPLASH_TIMEOUT, 2000, None, -1)

class MainApp(wx.App):
    frame = None

    # 主程序入口类
    def OnInit(self):
 self.SetAppName(APP_TITLE)
 self.frame = Myframe()
 # self.frame.Show()     # 设置启动时显示主窗体
 return True

if __name__ == "__main__":
    app = MainApp(redirect=True, filename="./log/log_debug.txt")
    AntiInsertion(daemon=True).start()
    create_splash()
    app.MainLoop()