bitbar-offlineimapNotification/offlineimap-notification.1s.py
Andros Fenollosa 3c874ade1a first commit
2016-07-06 18:57:35 +02:00

103 lines
3.0 KiB
Python
Executable File

#!/usr/bin/env python
# - * -coding: utf - 8 - * -
#
# <bitbar.title>Offlineimap Checker and notification</bitbar.title>
# <bitbar.version>v1.0</bitbar.version>
# <bitbar.author>Andros Fenollosa</bitbar.author>
# <bitbar.author.github>tanrax</bitbar.author.github>
# <bitbar.desc>Alert new emails and displays inbox count</bitbar.desc>
# <bitbar.image>https://programadorwebvalencia.com/wp-content/uploads/2016/07/Screen-Shot-2016-07-06-at-18.42.35.jpg</bitbar.image>
import os
import pickle
import tempfile
import re
from os.path import expanduser
# Location
PATH_MAIL = os.path.join(expanduser("~"), 'Mail')
DIR_NEW = 'INBOX/new'
ICON_EMPTY = '📪'
ICON_NEWS = '📬'
# Variables
SAVE_LOCATION = os.path.join(tempfile.gettempdir(),
'offlineimap-notification.pkl')
data_save = False
num_news = 0
data_temp = {}
try:
dateFile = open(SAVE_LOCATION)
data_temp = pickle.load(dateFile)
except:
pass
def send_alert_osx(mail_from, mail_subject):
os.system('osascript -e \'display notification \
"{mail_subject}" with title "{mail_from}" sound name "Blow"\''
.format(mail_from=mail_from, mail_subject=mail_subject))
def get_fields(path_mail):
'''
Function get elements: email from, email to and Subject
return list(mail_from, mail_to, mail_subject)
'''
mail_from = False
mail_from_temp = False
mail_to = False
mail_to_temp = False
mail_subject = False
mail_subject_temp = False
f = open(path_mail, 'r')
for line in f:
mail_from_temp = re.search('(?<=From\: )(.*)(?=)', line)
mail_to_temp = re.search('(?<=To\: )(.*)(?=)', line)
mail_subject_temp = re.search('(?<=Subject\: )(.*)(?=)', line)
if mail_from_temp:
mail_from = mail_from_temp.group(1)
if mail_to_temp:
mail_to = mail_to_temp.group(1)
if mail_subject_temp:
mail_subject = mail_subject_temp.group(1)
return {'mail_from': mail_from,
'mail_to': mail_to,
'mail_subject': mail_subject}
# Get info INGOX news
# Loop folders INBOX
list_dirs = os.listdir(os.path.join(PATH_MAIL))
data_news_emails = []
for name in list_dirs:
path_news_INBOX = os.path.join(PATH_MAIL, name, DIR_NEW)
if os.path.isdir(path_news_INBOX):
# Loop mails news
for name in os.listdir(path_news_INBOX):
mail = os.path.join(path_news_INBOX, name)
if os.path.isfile(mail):
num_news += 1
data_news_emails.append(get_fields(mail))
# Send notifications
# Check changes
if data_news_emails != data_temp:
# Send native alert
for item in data_news_emails:
if item not in data_temp:
print(item)
send_alert_osx(item['mail_from'], item['mail_subject'])
# Save
data_save = open(SAVE_LOCATION, 'w+')
pickle.dump(data_news_emails, data_save)
# Print
icon = ICON_EMPTY
if num_news > 0:
icon = ICON_NEWS
print('{icon}{num_news}'.format(icon=icon, num_news=num_news))