120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
import os
|
|
import glob
|
|
import datetime
|
|
from xml.dom.minidom import *
|
|
from gi.repository import GObject
|
|
|
|
from gajim.plugins import GajimPlugin
|
|
from gajim.plugins.helpers import log_calls
|
|
from gajim.notify import popup
|
|
|
|
from gajim.common import configpaths
|
|
from gajim.common import app
|
|
from gajim.common import ged
|
|
|
|
|
|
class BirthDayPlugin(GajimPlugin):
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def init(self):
|
|
|
|
self.config_dialog = None
|
|
self.description = ('Birthday reminder plugin')
|
|
self.events_handlers = {
|
|
'roster-received': (ged.GUI2, self.roster_received)}
|
|
configpath = configpaths.ConfigPaths()
|
|
cache_path = configpath.cache_root
|
|
self.vcard_path = os.path.join(cache_path, 'vcards') + os.sep
|
|
self.timeout_id = 0
|
|
self.showed_accounts = []
|
|
|
|
def check_birthdays(self, account=None):
|
|
def show_popup(account, jid):
|
|
contact_instances = app.contacts.get_contacts(account, jid)
|
|
contact = app.contacts.get_highest_prio_contact_from_contacts(
|
|
contact_instances)
|
|
if contact:
|
|
nick = GObject.markup_escape_text(contact.get_shown_name())
|
|
try:
|
|
image = os.path.dirname(__file__) + os.sep + \
|
|
'birthday_reminder_large.png'
|
|
except:
|
|
image = None
|
|
|
|
popup('Send message', contact.jid, account, msg_type='', \
|
|
path_to_image=image, title=title, text=text + ' ' + nick)
|
|
|
|
accounts = app.contacts.get_accounts()
|
|
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):][:-1]] = data
|
|
except:
|
|
pass
|
|
|
|
today = datetime.date.today()
|
|
|
|
for key, value in date_dict.items():
|
|
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"
|
|
elif time_to_bday.days == 3:
|
|
text = "3 days before BDay"
|
|
elif time_to_bday.days == 1:
|
|
text = "Tomorrow BDay"
|
|
elif time_to_bday.days == 0:
|
|
text = "Today BDay"
|
|
if not text:
|
|
continue
|
|
if account:
|
|
show_popup(account,key)
|
|
else:
|
|
for acct in accounts:
|
|
show_popup(account, key)
|
|
return True
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def activate(self):
|
|
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)
|
|
|
|
|
|
@log_calls('BirthDayPlugin')
|
|
def roster_received(self, obj):
|
|
if obj.conn.name not in self.showed_accounts:
|
|
self.check_birthdays(obj.conn.name)
|
|
self.showed_accounts.append(obj.conn.name)
|