modules.py
No OneTemporary
Actions

Authored By

Unknown

Size

12 KB

Referenced Files

None

Subscribers

None

modules.py
View Options

	# -- coding: utf-8 --
	# Copyright 2010-2013 Kolab Systems AG (http://www.kolabsys.com)
	#
	# Jeroen van Meeuwen (Kolab Systems) <vanmeeuwen a kolabsys.com>
	#
	# This program is free software: you can redistribute it and/or modify
	# it under the terms of the GNU General Public License as published by
	# the Free Software Foundation, either version 3 of the License, or
	# (at your option) any later version.

	# This program is distributed in the hope that it will be useful,
	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	# GNU General Public License for more details.

	# You should have received a copy of the GNU General Public License
	# along with this program. If not, see <http://www.gnu.org/licenses/>.
	#

	import os
	import sys
	import time

	from email import message_from_string
	from email.message import Message
	from email.mime.base import MIMEBase
	from email.mime.message import MIMEMessage
	from email.mime.multipart import MIMEMultipart
	from email.mime.text import MIMEText
	from email.parser import Parser
	from email.utils import COMMASPACE
	from email.utils import formatdate
	from email.utils import formataddr
	from email.utils import getaddresses
	from email.utils import parsedate_tz

	import smtplib

	import pykolab
	from pykolab import constants
	from pykolab.translate import _

	log = pykolab.getLogger('pykolab.wallace')
	conf = pykolab.getConf()

	modules = {}

	def __init__():
	# We only want the base path
	modules_base_path = os.path.dirname(__file__)

	for modules_path, dirnames, filenames in os.walk(modules_base_path):
	if not modules_path == modules_base_path:
	continue

	for filename in filenames:
	if filename.startswith('module_') and filename.endswith('.py'):
	module_name = filename.replace('.py','')
	name = module_name.replace('module_', '')
	#print "exec(\"from %s import __init__ as %s_register\"" % (module_name,name)
	exec("from %s import __init__ as %s_register" % (module_name, name))
	exec("%s_register()" % (name))

	for dirname in dirnames:
	register_group(modules_path, dirname)

	def list_modules(args, *kw):
	"""
	List modules
	"""

	__modules = {}

	for module in modules.keys():
	if isinstance(module, tuple):
	module_group, module = module
	__modules[module_group] = {
	module: modules[(module_group,module)]
	}
	else:
	__modules[module] = modules[module]

	_modules = __modules.keys()
	_modules.sort()

	for _module in _modules:
	if __modules[_module].has_key('function'):
	# This is a top-level module
	if not __modules[_module]['description'] == None:
	print "%-25s - %s" % (_module.replace('_','-'),__modules[_module]['description'])
	else:
	print "%-25s" % (_module.replace('_','-'))

	for _module in _modules:
	if not __modules[_module].has_key('function'):
	# This is a nested module
	print "\n" + _("Module Group: %s") % (_module) + "\n"
	___modules = __modules[_module].keys()
	___modules.sort()
	for __module in ___modules:
	if not __modules[_module][__module]['description'] == None:
	print "%-4s%-21s - %s" % ('',__module.replace('_','-'),__modules[_module][__module]['description'])
	else:
	print "%-4s%-21s" % ('',__module.replace('_','-'))

	def execute(name, args, *kw):
	if not modules.has_key(name):
	log.error(_("No such module %r in modules %r (1).") % (name, modules))
	sys.exit(1)

	if not modules[name].has_key('function') and \
	not modules[name].has_key('group'):
	log.error(_("No such module %r in modules %r (2).") %(name, modules))
	sys.exit(1)

	return modules[name]['function'](args, *kw)

	def cb_action_HOLD(module, filepath):
	log.info(_("Holding message in queue for manual review (%s by %s)") % (filepath, module))

	def cb_action_DEFER(module, filepath):
	log.info(_("Deferring message in %s (by module %s)") % (filepath, module))

	# parse message headers
	message = Parser().parse(open(filepath, 'r'), True)

	internal_time = parsedate_tz(message.__getitem__('Date'))
	internal_time = time.mktime(internal_time[:9]) + internal_time[9]

	now_time = time.time()

	delta = now_time - internal_time

	log.debug(_("The time when the message was sent: %r") % (internal_time), level=8)
	log.debug(_("The time now: %r") % (now_time), level=8)
	log.debug(_("The time delta: %r") % (delta), level=8)

	if delta > 432000:
	# TODO: Send NDR back to user
	log.debug(_("Message in file %s older then 5 days, deleting") % (filepath), level=8)
	os.unlink(filepath)

	# Alternative method is file age.
	#Date sent(/var/spool/pykolab/wallace/optout/DEFER/tmpIv7pDl): 'Thu, 08 Mar 2012 11:51:03 +0000'
	#(2012, 3, 8, 11, 51, 3, 0, 1, -1)
	# YYYY M D H m s weekday, yearday

	#log.debug(datetime.datetime(*), level=8)

	#import os
	#stat = os.stat(filepath)

	#fileage = datetime.datetime.fromtimestamp(stat.st_mtime)
	#now = datetime.datetime.now()
	#delta = now - fileage

	#print "file:", filepath, "fileage:", fileage, "now:", now, "delta(seconds):", delta.seconds

	#if delta.seconds > 1800:
	## TODO: Send NDR back to user
	#log.debug(_("Message in file %s older then 1800 seconds, deleting") % (filepath), level=8)
	#os.unlink(filepath)

	def cb_action_REJECT(module, filepath):
	log.info(_("Rejecting message in %s (by module %s)") % (filepath, module))

	# parse message headers
	message = Parser().parse(open(filepath, 'r'), True)

	envelope_sender = getaddresses(message.get_all('From', []))

	recipients = getaddresses(message.get_all('To', [])) + \
	getaddresses(message.get_all('Cc', [])) + \
	getaddresses(message.get_all('X-Kolab-To', []))

	_recipients = []

	for recipient in recipients:
	if not recipient[0] == '':
	_recipients.append('%s <%s>' % (recipient[0], recipient[1]))
	else:
	_recipients.append('%s' % (recipient[1]))

	# TODO: Find the preferredLanguage for the envelope_sender user.
	ndr_message_subject = "Undelivered Mail Returned to Sender"
	ndr_message_text = _("""This is the email system Wallace at %s.

	I'm sorry to inform you we could not deliver the attached message
	to the following recipients:

	- %s

	Your message is being delivered to any other recipients you may have
	sent your message to. There is no need to resend the message to those
	recipients.
	""") % (
	constants.fqdn,
	"\n- ".join(_recipients)
	)

	diagnostics = _("""X-Wallace-Module: %s
	X-Wallace-Result: REJECT
	""") % (
	module
	)

	msg = MIMEMultipart("report")
	msg['From'] = "MAILER-DAEMON@%s" % (constants.fqdn)
	msg['To'] = formataddr(envelope_sender[0])
	msg['Date'] = formatdate(localtime=True)
	msg['Subject'] = ndr_message_subject

	msg.preamble = "This is a MIME-encapsulated message."

	part = MIMEText(ndr_message_text)
	part.add_header("Content-Description", "Notification")
	msg.attach(part)

	_diag_message = Message()
	_diag_message.set_payload(diagnostics)
	part = MIMEMessage(_diag_message, "delivery-status")
	part.add_header("Content-Description", "Delivery Report")
	msg.attach(part)

	# @TODO: here I'm not sure message will contain the whole body
	# when we used headersonly argument of Parser().parse() above
	# delete X-Kolab-* headers
	del message['X-Kolab-From']
	del message['X-Kolab-To']

	part = MIMEMessage(message)
	part.add_header("Content-Description", "Undelivered Message")
	msg.attach(part)

	smtp = smtplib.SMTP("localhost", 10027)

	try:
	smtp.sendmail(
	"MAILER-DAEMON@%s" % (constants.fqdn),
	[formataddr(envelope_sender[0])],
	msg.as_string()
	)

	except smtplib.SMTPDataError, errmsg:
	# DEFER
	pass
	except smtplib.SMTPHeloError, errmsg:
	# DEFER
	pass
	except smtplib.SMTPRecipientsRefused, errmsg:
	# REJECT, send NDR
	pass
	except smtplib.SMTPSenderRefused, errmsg:
	# REJECT, send NDR
	pass
	finally:
	os.unlink(filepath)

	def cb_action_ACCEPT(module, filepath):
	log.info(_("Accepting message in %s (by module %s)") % (filepath, module))

	log.debug(_("Accepting message in: %r") %(filepath), level=8)

	# parse message headers
	message = Parser().parse(open(filepath, 'r'), True)

	sender = [formataddr(x) for x in getaddresses(message.get_all('X-Kolab-From', []))]
	recipients = [formataddr(x) for x in getaddresses(message.get_all('X-Kolab-To', []))]
	log.debug(_("recipients: %r") % (recipients))

	# delete X-Kolab-* headers
	del message['X-Kolab-From']
	del message['X-Kolab-To']

	smtp = smtplib.SMTP("localhost", 10027)

	if conf.debuglevel > 8:
	smtp.set_debuglevel(True)

	try:
	smtp.sendmail(
	sender,
	recipients,
	# - Make sure we do not send this as binary.
	# - Second, strip NUL characters - I don't know where they
	# come from (TODO)
	# - Third, a character return is inserted somewhere. It
	# divides the body from the headers - and we don't like (TODO)
	# @TODO: check if we need Parser().parse() to load the whole message
	message.as_string()
	)

	except smtplib.SMTPDataError, errmsg:
	log.error("SMTP Data Error, %r" % (errmsg))
	# DEFER
	pass
	except smtplib.SMTPHeloError, errmsg:
	log.error("SMTP HELO Error, %r" % (errmsg))
	# DEFER
	pass
	except smtplib.SMTPRecipientsRefused, errmsg:
	log.error("SMTP Recipient(s) Refused, %r" % (errmsg))
	# DEFER
	pass
	except smtplib.SMTPSenderRefused, errmsg:
	log.error("SMTP Sender Refused, %r" % (errmsg))
	# DEFER
	pass
	finally:
	os.unlink(filepath)

	def register_group(dirname, module):
	modules_base_path = os.path.join(os.path.dirname(__file__), module)

	modules[module] = {}

	for modules_path, dirnames, filenames in os.walk(modules_base_path):
	if not modules_path == modules_base_path:
	continue

	for filename in filenames:
	if filename.startswith('module_') and filename.endswith('.py'):
	module_name = filename.replace('.py','')
	name = module_name.replace('module_', '')
	# TODO: Error recovery from incomplete / incorrect modules.
	exec(
	"from %s.%s import __init__ as %s_%s_register" % (
	module,
	module_name,
	module,
	name
	)
	)

	exec("%s_%s_register()" % (module,name))

	def register(name, func, group=None, description=None, aliases=[]):
	if not group == None:
	module = "%s_%s" % (group,name)
	else:
	module = name

	if isinstance(aliases, basestring):
	aliases = [aliases]

	if modules.has_key(module):
	log.fatal(_("Module '%s' already registered") % (module))
	sys.exit(1)

	if callable(func):
	if group == None:
	modules[name] = {
	'function': func,
	'description': description
	}
	else:
	modules[group][name] = {
	'function': func,
	'description': description
	}

	modules[module] = modules[group][name]
	modules[module]['group'] = group
	modules[module]['name'] = name

	for alias in aliases:
	modules[alias] = {
	'function': func,
	'description': _("Alias for %s") % (name)
	}

File Metadata

Mime Type: text/x-script.python
Expires: Sun, Apr 5, 11:58 PM (1 w, 4 d ago)
Storage Engine: local-disk
Storage Format: Raw Data
Storage Handle: ee/b1/e42936fc7dbae3bf98f77db626ae
Default Alt Text: modules.py (12 KB)

modules.pyNo OneTemporaryActions

modules.pyView Options

File Metadata

Event Timeline

modules.py
No OneTemporary
Actions

modules.py
View Options