1 files changed, 43 insertions, 0 deletions
diff --git a/.config/waybar/modules/mail.py b/.config/waybar/modules/mail.py
new file mode 100755
index 0000000..31958f7
--- /dev/null
+++ b/.config/waybar/modules/mail.py
@@ -0,0 +1,43 @@
+#!/usr/bin/python
+
+import os
+import imaplib
+
+import mailsecrets
+
+def getmails(username, password, server):
+    imap = imaplib.IMAP4_SSL(server, 993)
+    imap.login(username, password)
+    imap.select('INBOX')
+    ustatus, uresponse = imap.uid('search', None, 'UNSEEN')
+    if ustatus == 'OK':
+        unread_msg_nums = uresponse[0].split()
+    else:
+        unread_msg_nums = []
+
+    fstatus, fresponse = imap.uid('search', None, 'FLAGGED')
+    if fstatus == 'OK':
+        flagged_msg_nums = fresponse[0].split()
+    else:
+        flagged_msg_nums = []
+
+    return [len(unread_msg_nums), len(flagged_msg_nums)]
+
+ping = os.system("ping " + mailsecrets.server + " -c1 > /dev/null 2>&1")
+if ping == 0:
+    mails = getmails(mailsecrets.username, mailsecrets.password, mailsecrets.server)
+    text = ''
+    alt = ''
+
+    if mails[0] > 0:
+        text = alt = str(mails[0])
+        if mails[1] > 0:
+            alt = str(mails[1]) + "  " + alt
+    else:
+        exit(1)
+
+    print('{"text":"' + text + '", "alt": "' + alt + '"}')
+
+else:
+    exit(1)
+