89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
import os
|
|
import glob
|
|
import datetime
|
|
from xml.dom.minidom import *
|
|
import gobject
|
|
|
|
from plugins import GajimPlugin
|
|
from plugins.helpers import log_calls
|
|
from notify import popup
|
|
|
|
from common import configpaths
|
|
|
|
|
|
class BirthDayPlugin(GajimPlugin):
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def init(self):
|
|
|
|
self.config_dialog = None
|
|
self.description = ('Birthday reminder plugin')
|
|
configpath = configpaths.ConfigPaths()
|
|
cache_path = configpath.cache_root
|
|
self.vcard_path = os.path.join(cache_path, 'vcards') + os.sep
|
|
self.timeout_id = 0
|
|
|
|
def check_birthdays(self):
|
|
vcards = []
|
|
date_dict = {}
|
|
for jid in glob.glob(self.vcard_path + '*@*'):
|
|
if os.path.isfile(jid):
|
|
vcards.append(jid)
|
|
|
|
for xmldoc in vcards:
|
|
try:
|
|
xml = parse(xmldoc)
|
|
except:
|
|
pass
|
|
else:
|
|
name = xml.getElementsByTagName('BDAY')
|
|
for node in name:
|
|
try:
|
|
data = node.childNodes[0].nodeValue
|
|
date_dict[xmldoc[len(self.vcard_path):]] = data
|
|
except:
|
|
pass
|
|
|
|
today = datetime.date.today()
|
|
|
|
for key, value in date_dict.iteritems():
|
|
try:
|
|
convert_date = datetime.datetime.strptime(value, "%Y-%m-%d")
|
|
user_bday = datetime.date(today.year, convert_date.month,
|
|
convert_date.day)
|
|
except:
|
|
continue
|
|
|
|
if user_bday < today:
|
|
user_bday = user_bday.replace(year=today.year+1)
|
|
|
|
time_to_bday = abs(user_bday - today)
|
|
title = "BirthDay Reminder"
|
|
text = None
|
|
|
|
if time_to_bday.days > 5:
|
|
continue
|
|
if time_to_bday.days == 5:
|
|
text = "5 days before BDay %s" % key
|
|
elif time_to_bday.days == 3:
|
|
text = "3 days before BDay %s" % key
|
|
elif time_to_bday.days == 1:
|
|
text = "Tommorrow BDay %s" % key
|
|
elif time_to_bday.days == 0:
|
|
text = "Today BDay %s" % key
|
|
if text:
|
|
popup('', key, key, title=title, text=text)
|
|
return True
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def activate(self):
|
|
self.check_birthdays()
|
|
self.timeout_id = gobject.timeout_add_seconds(24*3600,
|
|
self.check_birthdays)
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def deactivate(self):
|
|
if self.timeout_id > 0:
|
|
gobject.source_remove(self.timeout_id)
|
|
|