FreeBSDAdmin/Reminder/reminder.py

#!/usr/local/bin/python
# Reminder
#
# Douglas Thrift
#
# $Id$

from ConfigParser import NoOptionError, SafeConfigParser
import gdata.calendar.service
import getpass
import imaplib
import keyring
import optparse
import os
import sys

MODULES = frozenset(('creditcards', 'facebook'))

class Config(object):
        def __init__(self, config, module):
                self.config = config
                self.module = module

        def get(self, *args, **kwargs):
                return self.config.get(self.module, *args, **kwargs)

        def getboolean(self, *args, **kwargs):
                return self.config.getboolean(self.module, *args, **kwargs)

        def getfloat(self, *args, **kwargs):
                return self.config.getfloat(self.module, *args, **kwargs)

        def getint(self, *args, **kwargs):
                return self.config.getint(self.module, *args, **kwargs)

        def getlist(self, *args, **kwargs):
                return self.config.getlist(self.module, *args, **kwargs)

        def getpassword(self):
                return self.config.getpassword(self.module)

        def getusername(self):
                return self.config.getusername(self.module)

class ConfigParser(SafeConfigParser):
        def __init__(self, *args, **kwargs):
                SafeConfigParser.__init__(self, *args, **kwargs)

        def getlist(self, *args, **kwargs):
                return self.get(*args, **kwargs).split(',')

        def getpassword(self, section):
                return keyring.get_password('reminder_%s' % section, self.getusername(section))

        def getusername(self, section):
                return self.get(section, 'username')

class OptionParser(optparse.OptionParser):
        def __init__(self, *args, **kwargs):
                optparse.OptionParser.__init__(self, *args, **kwargs)
                self.add_option('-A', '--all', action = 'store_true', dest = 'all')
                self.add_option('-d', '--dbus', action = 'store_true', dest = 'dbus')
                self.add_option('-m', '--module', action = 'callback', callback = self.__module, dest = 'modules', type = 'string')
                self.add_option('-p', '--password', action = 'callback', callback = self.__password, dest = 'passwords', type = 'string')
                self.add_option('-D', '--debug', action = 'store_true', dest = 'debug')

        def __module(self, option, opt_str, value, parser):
                if value not in MODULES:
                        raise optparse.OptionValueError, '%s unknown module %s' % (opt_str, value)

                parser.values.ensure_value(option.dest, []).append(value)

        def __password(self, option, opt_str, value, parser):
                parser.values.ensure_value(option.dest, []).append(value)

if __name__ == '__main__':
        parser = OptionParser()
        options = parser.parse_args()[0]
        config = ConfigParser()

        config.read([os.path.expanduser('~/.reminder')])

        if options.dbus:
                dbus = os.environ['DBUS_SESSION_BUS_ADDRESS']

                with open(os.path.expanduser('~/.reminder.dbus'), 'wb') as file:
                        file.write(dbus + '\n')

                sys.exit(0)

        try:
                os.environ['DBUS_SESSION_BUS_ADDRESS']
        except KeyError:
                with open(os.path.expanduser('~/.reminder.dbus'), 'rb') as file:
                        os.environ['DBUS_SESSION_BUS_ADDRESS'] = file.readline().rstrip()

        if options.passwords:
                for section in options.passwords:
                        try:
                                username = config.getusername(section)
                                password = getpass.getpass('%s password for %s: ' % (section, username))

                                keyring.set_password('reminder_%s' % section, username, password)
                        except NoOptionError:
                                pass

                sys.exit(0)

        if not options.all and not options.modules:
                parser.error('-A or -m not specified')

        calendar = gdata.calendar.service.CalendarService()
        calendar.email = config.getusername('google')
        calendar.password = config.getpassword('google')
        calendar.source = 'Reminder-0.9'

        calendar.ProgrammaticLogin()

        if options.debug:
                imaplib.Debug = 4

        imap = imaplib.IMAP4_SSL(config.get('imap', 'server'))

        imap.login(config.getusername('imap'), config.getpassword('imap'))

        for module in options.modules if not options.all else sorted(MODULES):
                exec 'import %s' % module
                exec '%s.main(calendar, imap, Config(config, module), options.debug)' % module

        imap.logout()
Revision:	1424
Committed:	2011-05-03T00:02:31-07:00 (14 years, 2 months ago) by douglas
Content type:	text/x-python
File size:	3864 byte(s)
Log Message:	New fangled email based reminders?
#	User	Rev	Content
1	douglas	1179	#!/usr/local/bin/python
2			# Reminder
3	douglas	1178	#
4			# Douglas Thrift
5			#
6			# $Id$
7
8	douglas	1229	from ConfigParser import NoOptionError, SafeConfigParser
9	douglas	1424	import gdata.calendar.service
10	douglas	1229	import getpass
11	douglas	1424	import imaplib
12	douglas	1229	import keyring
13	douglas	1178	import optparse
14	douglas	1229	import os
15			import sys
16	douglas	1178
17	douglas	1424	MODULES = frozenset(('creditcards', 'facebook'))
18	douglas	1178
19	douglas	1179	class Config(object):
20			def __init__(self, config, module):
21			self.config = config
22			self.module = module
23	douglas	1178
24	douglas	1179	def get(self, args, *kwargs):
25			return self.config.get(self.module, args, *kwargs)
26	douglas	1178
27	douglas	1179	def getboolean(self, args, *kwargs):
28			return self.config.getboolean(self.module, args, *kwargs)
29	douglas	1178
30	douglas	1179	def getfloat(self, args, *kwargs):
31			return self.config.getfloat(self.module, args, *kwargs)
32	douglas	1178
33	douglas	1179	def getint(self, args, *kwargs):
34			return self.config.getint(self.module, args, *kwargs)
35	douglas	1178
36	douglas	1179	def getlist(self, args, *kwargs):
37			return self.config.getlist(self.module, args, *kwargs)
38	douglas	1178
39	douglas	1229	def getpassword(self):
40			return self.config.getpassword(self.module)
41	douglas	1178
42	douglas	1229	def getusername(self):
43			return self.config.getusername(self.module)
44
45	douglas	1179	class ConfigParser(SafeConfigParser):
46			def __init__(self, args, *kwargs):
47			SafeConfigParser.__init__(self, args, *kwargs)
48	douglas	1178
49	douglas	1179	def getlist(self, args, *kwargs):
50			return self.get(args, *kwargs).split(',')
51	douglas	1178
52	douglas	1229	def getpassword(self, section):
53			return keyring.get_password('reminder_%s' % section, self.getusername(section))
54	douglas	1178
55	douglas	1229	def getusername(self, section):
56			return self.get(section, 'username')
57
58	douglas	1179	class OptionParser(optparse.OptionParser):
59			def __init__(self, args, *kwargs):
60			optparse.OptionParser.__init__(self, args, *kwargs)
61			self.add_option('-A', '--all', action = 'store_true', dest = 'all')
62	douglas	1229	self.add_option('-d', '--dbus', action = 'store_true', dest = 'dbus')
63	douglas	1179	self.add_option('-m', '--module', action = 'callback', callback = self.__module, dest = 'modules', type = 'string')
64	douglas	1229	self.add_option('-p', '--password', action = 'callback', callback = self.__password, dest = 'passwords', type = 'string')
65	douglas	1179	self.add_option('-D', '--debug', action = 'store_true', dest = 'debug')
66	douglas	1178
67	douglas	1179	def __module(self, option, opt_str, value, parser):
68			if value not in MODULES:
69			raise optparse.OptionValueError, '%s unknown module %s' % (opt_str, value)
70	douglas	1178
71	douglas	1179	parser.values.ensure_value(option.dest, []).append(value)
72	douglas	1178
73	douglas	1229	def __password(self, option, opt_str, value, parser):
74			parser.values.ensure_value(option.dest, []).append(value)
75
76	douglas	1179	if __name__ == '__main__':
77			parser = OptionParser()
78			options = parser.parse_args()[0]
79	douglas	1229	config = ConfigParser()
80	douglas	1178
81	douglas	1229	config.read([os.path.expanduser('~/.reminder')])
82
83			if options.dbus:
84			dbus = os.environ['DBUS_SESSION_BUS_ADDRESS']
85
86			with open(os.path.expanduser('~/.reminder.dbus'), 'wb') as file:
87			file.write(dbus + '\n')
88
89			sys.exit(0)
90
91			try:
92			os.environ['DBUS_SESSION_BUS_ADDRESS']
93			except KeyError:
94			with open(os.path.expanduser('~/.reminder.dbus'), 'rb') as file:
95			os.environ['DBUS_SESSION_BUS_ADDRESS'] = file.readline().rstrip()
96
97	douglas	1424	if options.passwords:
98			for section in options.passwords:
99			try:
100			username = config.getusername(section)
101			password = getpass.getpass('%s password for %s: ' % (section, username))
102
103			keyring.set_password('reminder_%s' % section, username, password)
104			except NoOptionError:
105			pass
106
107			sys.exit(0)
108
109	douglas	1179	if not options.all and not options.modules:
110			parser.error('-A or -m not specified')
111
112			calendar = gdata.calendar.service.CalendarService()
113	douglas	1229	calendar.email = config.getusername('google')
114			calendar.password = config.getpassword('google')
115	douglas	1179	calendar.source = 'Reminder-0.9'
116
117			calendar.ProgrammaticLogin()
118
119	douglas	1424	if options.debug:
120			imaplib.Debug = 4
121	douglas	1179
122	douglas	1424	imap = imaplib.IMAP4_SSL(config.get('imap', 'server'))
123
124			imap.login(config.getusername('imap'), config.getpassword('imap'))
125
126	douglas	1179	for module in options.modules if not options.all else sorted(MODULES):
127			exec 'import %s' % module
128	douglas	1424	exec '%s.main(calendar, imap, Config(config, module), options.debug)' % module
129
130			imap.logout()