fayewest · April 1, 2012 21:29 · Apr 1, 2012
diff --git a/gistfile1.py b/gistfile1.py
@@ -0,0 +1,239 @@
+import wx
+import poplib
+import imaplib
+import re
+import time
+import cPickle
+import os,sys
+import random
+import zlib
+
+#try: # Windows only
+#    import msvcrt
+#    msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
+#except ImportError: pass 
+
+class MainWindow(wx.MiniFrame):
+    def __init__(self,parent=None,id=-1,title="mailGuy"):
+        #TODO - Clean up algorithm to get pos to place window in lower right corner, use tuples?
+        mode = wx.Display.GetCurrentMode(wx.Display(0))
+        x = mode.GetWidth()
+        y = mode.GetHeight()
+        sizex = 300
+        sizey = 150
+        newx = x - sizex
+        # windows taskbar is 30 pixels
+        newy = (y - sizey) - 30
+
+        wx.MiniFrame.__init__(
+            self,parent,id,title,size=(sizex,sizey),
+            style=wx.DEFAULT_FRAME_STYLE | wx.STAY_ON_TOP,pos=(newx,newy))
+
+        #Values for transparency effect
+        self.amount = 255
+        self.delta = -3
+
+        p = wx.Panel(self)
+
+        #Timer to check mail account
+        self.checktimer = wx.CallLater(300000,self.onTimer)
+
+        #Show a default message on startup
+        self.st = wx.StaticText(p,-1,str("mailGuy v%s" % myVersion),(25,25))
+        self.st.SetFont(wx.Font(14,wx.SWISS,wx.NORMAL,wx.NORMAL)) 
+
+        self.timer = wx.Timer(self)
+        self.timer.Start(50)
+
+        #Event bindings for the fade effect
+        self.Bind(wx.EVT_TIMER,self.AlphaCycle)
+        p.Bind(wx.EVT_LEFT_DOWN,self.HoldAlpha)
+        p.Bind(wx.EVT_ENTER_WINDOW,self.ResetAlpha)
+        self.st.Bind(wx.EVT_LEFT_DOWN,self.HoldAlpha)
+
+        self.onTimer
+
+    #AlphaCycle,ResetAlpha,HoldAlpha all deal with making the window fade away
+    def AlphaCycle(self,evt):
+        if self.amount > 0:
+            self.amount += self.delta
+            self.SetTransparent(self.amount)
+    def ResetAlpha(self,*args):
+        self.amount = 255
+    def HoldAlpha(self,*args):
+        self.SetTransparent(255)
+        self.amount = 0
+        wx.CallLater(15000,self.ResetAlpha)    
+    def onTimer(self,*args):
+
+        params = []
+        f = open('config.dat','rb')
+        cryptparams = cPickle.load(f)
+        f.close()
+
+        for y in cryptparams:
+            params.append(ConfigWindow.crypt(myKey,y,True))
+
+        user = params[0]
+        pass_ = params[1]
+        type_ = params[2]
+        host = params[3]
+
+        mail = MailAccount(type_,host,user,pass_)
+        msgcount = mail.checkMail()
+
+        #If this is the first time we've done this, setup old message count to equal the current amount of messages
+        if not hasattr(self,"oldmsgcount"):
+            self.oldmsgcount = msgcount
+        if msgcount > self.oldmsgcount:
+            newmsg = msgcount - self.oldmsgcount
+            self.SetFocus
+            self.HoldAlpha()
+        elif msgcount <= self.oldmsgcount:
+            newmsg = 0
+
+        self.oldmsgcount = msgcount
+
+        #Just a debug message
+        #print "Mail Check! Msgcount: %s" % msgcount
+        if newmsg == 1:
+            plural = ""
+        else:
+            plural = "s"
+        if newmsg > 0:
+   	    self.st.SetLabel(str("%s:\n%s New Message%s" % (host,newmsg,plural)))
+        self.checktimer.Restart()      
+
+#MailAccount represents one particular email account we can check       
+class MailAccount:
+    def __init__(self,accttype,host,usr,pwd):
+        self.accttype = accttype
+        self.host = host
+        self.usr = usr
+        self.pwd = pwd
+    def checkMail(self):
+        if self.accttype == "POP3":
+            pop = poplib.POP3(self.host)
+            pop.user(self.usr)
+            pop.pass_(self.pwd)
+            stat = pop.stat()
+            count = stat[0]
+            pop.quit()
+            return count
+        if self.accttype == "Gmail":
+            m = imaplib.IMAP4_SSL("imap.gmail.com")
+            m.login(self.usr,self.pwd)
+            resp,msgs = m.select("Inbox",readonly=1)            
+            m.logout()
+            return int(msgs[0])
+        if self.accttype == "IMAP":
+            m = imaplib.IMAP4(self.host)
+            m.login(self.usr,self.pwd)
+            resp,msgs = m.select("Inbox",readonly=1)
+            m.logout()
+            return int(msgs[0])
+        if self.accttype == "IMAP-SSL":
+            m = imaplib.IMAP4_SSL("imap.gmail.com")
+	    m.login(self.usr,self.pwd)
+	    resp,msgs = m.select("Inbox",readonly=1)            
+            m.logout()
+class ConfigWindow(wx.Frame):
+    def __init__(self,parent=None,id=-1,title="mailGuy Config"):
+        wx.Frame.__init__(
+            self,parent,id,title,size=(400,340),
+        style=wx.DEFAULT_FRAME_STYLE & ~wx.RESIZE_BORDER)
+        p = wx.Panel(self)
+        self.st = wx.StaticText(p,-1,str("mailGuy Initial Configuration"),(25,25))
+        self.st.SetFont(wx.Font(14,wx.SWISS,wx.NORMAL,wx.NORMAL)) 
+
+        lblUser = wx.StaticText(p,-1,str("Username:"),(25,25))
+        lblUser.SetFont(wx.Font(9,wx.SWISS,wx.NORMAL,wx.NORMAL))
+
+        self.txtUser = wx.TextCtrl(p,-1,size=(125,-1))
+
+        lblPass = wx.StaticText(p,-1,str("Password:"),(25,25))
+        lblPass.SetFont(wx.Font(9,wx.SWISS,wx.NORMAL,wx.NORMAL))
+
+        self.txtPass = wx.TextCtrl(p,-1,size=(125,-1),style=wx.TE_PASSWORD)
+
+        lblSvrType= wx.StaticText(p,-1,str("Service Type:"),(25,25))
+        lblSvrType.SetFont(wx.Font(9,wx.SWISS,wx.NORMAL,wx.NORMAL))
+
+        svrTypeLst = ['POP3','IMAP','IMAP-SSL','Gmail']
+        self.cmbSvrType = wx.ComboBox(p,500,"",(90,50),(130,-1),svrTypeLst,wx.CB_DROPDOWN)
+
+        lblMailServer = wx.StaticText(p,-1,str("Mail Server:"),(25,25))
+        lblMailServer.SetFont(wx.Font(9,wx.SWISS,wx.NORMAL,wx.NORMAL))
+
+        self.txtMailServer = wx.TextCtrl(p,-1,size=(145,-1))
+
+        b = wx.Button(p,10,"Ok")
+        self.Bind(wx.EVT_BUTTON,self.createConfig,b)
+
+        sizer = wx.BoxSizer(wx.VERTICAL)
+	sizer.Add(self.st,0,wx.ALL,3)
+	sizer.Add((50,0),0,wx.ALL,3)
+	sizer.Add(lblUser,0,wx.ALL,3)
+	sizer.Add(self.txtUser,0,wx.ALL,3)
+	sizer.Add(lblPass,0,wx.ALL,3)
+	sizer.Add(self.txtPass,0,wx.ALL,3)
+	sizer.Add(lblSvrType,0,wx.ALL,3)
+	sizer.Add(self.cmbSvrType,0,wx.ALL,3)
+	sizer.Add(lblMailServer,0,wx.ALL,3)
+	sizer.Add(self.txtMailServer,0,wx.ALL,3)
+
+	sizer2 = wx.BoxSizer(wx.HORIZONTAL)
+
+	sizer2.Add((300,0),0,wx.ALL)
+	sizer2.Add(b,0,wx.ALL)
+
+	box = wx.BoxSizer(wx.VERTICAL)
+	box.Add(sizer,0,wx.ALL,20)
+	box.Add(sizer2,0,wx.ALL,-1)
+        p.SetSizer(box)
+
+        self.SetIcon(icon1)
+        self.Show(True)
+    def createConfig(self,evt):
+        parameters = [self.txtUser.GetValue(),self.txtPass.GetValue(),
+            self.cmbSvrType.GetValue(),self.txtMailServer.GetValue()
+            ]
+        paracrypt = []
+        f = open("config.dat","wb")
+
+        for x in parameters:
+            paracrypt.append(ConfigWindow.crypt(myKey,x))
+
+        cPickle.dump(paracrypt,f,1)
+
+        f.close()
+
+        frame = MainWindow()
+        frame.SetIcon(icon1)
+        frame.Show()
+        self.Close()
+    def crypt(key,text,reverse=False):
+        rand = random.Random(key).randrange
+        if not reverse:
+            text = zlib.compress(text)
+        text = ''.join([chr(ord(elem)^rand(256)) for elem in text])
+        if reverse:
+            text = zlib.decompress(text)
+        return text
+    crypt = staticmethod(crypt)    
+
+myKey = "all good things..."
+myVersion = "0.4"
+app = wx.PySimpleApp()
+config = "config.dat"
+icon1 = wx.Icon("icon.ico",wx.BITMAP_TYPE_ICO)
+
+if os.path.exists(os.path.join(os.getcwd(),config)):
+    frame = MainWindow()
+    frame.SetIcon(icon1)
+    frame.Show()
+else:
+    frame = ConfigWindow()
+    frame.Show()
+
+app.MainLoop()