Changeset View
Changeset View
Standalone View
Standalone View
wallace/module_footer.py
| Show All 16 Lines | |||||
| # along with this program. If not, see <http://www.gnu.org/licenses/>. | # along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
| # | # | ||||
| import os | import os | ||||
| import tempfile | import tempfile | ||||
| import time | import time | ||||
| from email import message_from_file | from email import message_from_file | ||||
| from email.encoders import encode_quopri | |||||
| import modules | import modules | ||||
| import pykolab | import pykolab | ||||
| from pykolab.translate import _ | from pykolab.translate import _ | ||||
| log = pykolab.getLogger('pykolab.wallace') | log = pykolab.getLogger('pykolab.wallace') | ||||
| conf = pykolab.getConf() | conf = pykolab.getConf() | ||||
| mybasepath = '/var/spool/pykolab/wallace/footer/' | mybasepath = '/var/spool/pykolab/wallace/footer/' | ||||
| def __init__(): | def __init__(): | ||||
| modules.register('footer', execute, description=description()) | modules.register('footer', execute, description=description()) | ||||
| def description(): | def description(): | ||||
| return """Append a footer to messages.""" | return """Append a footer to messages.""" | ||||
| def set_part_content(part, content): | |||||
| # Reset old encoding and use quoted-printable (#5414) | |||||
| del part['Content-Transfer-Encoding'] | |||||
| part.set_payload(content) | |||||
| encode_quopri(part) | |||||
| return True | |||||
| def execute(*args, **kw): | def execute(*args, **kw): | ||||
| if not os.path.isdir(mybasepath): | if not os.path.isdir(mybasepath): | ||||
| os.makedirs(mybasepath) | os.makedirs(mybasepath) | ||||
| for stage in ['incoming', 'ACCEPT' ]: | for stage in ['incoming', 'ACCEPT']: | ||||
| if not os.path.isdir(os.path.join(mybasepath, stage)): | if not os.path.isdir(os.path.join(mybasepath, stage)): | ||||
| os.makedirs(os.path.join(mybasepath, stage)) | os.makedirs(os.path.join(mybasepath, stage)) | ||||
| # TODO: Test for correct call. | # TODO: Test for correct call. | ||||
| filepath = args[0] | filepath = args[0] | ||||
| if kw.has_key('stage'): | if 'stage' in kw: | ||||
| log.debug(_("Issuing callback after processing to stage %s") % (kw['stage']), level=8) | log.debug(_("Issuing callback after processing to stage %s") % (kw['stage']), level=8) | ||||
| log.debug(_("Testing cb_action_%s()") % (kw['stage']), level=8) | log.debug(_("Testing cb_action_%s()") % (kw['stage']), level=8) | ||||
| if hasattr(modules, 'cb_action_%s' % (kw['stage'])): | if hasattr(modules, 'cb_action_%s' % (kw['stage'])): | ||||
| log.debug(_("Attempting to execute cb_action_%s()") % (kw['stage']), level=8) | log.debug(_("Attempting to execute cb_action_%s()") % (kw['stage']), level=8) | ||||
| exec('modules.cb_action_%s(%r, %r)' % (kw['stage'],'optout',filepath)) | exec('modules.cb_action_%s(%r, %r)' % (kw['stage'],'optout',filepath)) | ||||
| return | return | ||||
| log.debug(_("Executing module footer for %r, %r") % (args, kw), level=8) | log.debug(_("Executing module footer for %r, %r") % (args, kw), level=8) | ||||
| Show All 12 Lines | def execute(*args, **kw): | ||||
| footer_html_file = conf.get('wallace', 'footer_html') | footer_html_file = conf.get('wallace', 'footer_html') | ||||
| footer_text_file = conf.get('wallace', 'footer_text') | footer_text_file = conf.get('wallace', 'footer_text') | ||||
| if not os.path.isfile(footer_text_file) and not os.path.isfile(footer_html_file): | if not os.path.isfile(footer_text_file) and not os.path.isfile(footer_html_file): | ||||
| log.warning(_("No contents configured for footer module")) | log.warning(_("No contents configured for footer module")) | ||||
| exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', filepath)) | exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', filepath)) | ||||
| return | return | ||||
| if os.path.isfile(footer_text_file): | if os.path.isfile(footer_text_file): | ||||
| footer['plain'] = open(footer_text_file, 'r').read() | footer['plain'] = open(footer_text_file, 'r').read() | ||||
| if not os.path.isfile(footer_html_file): | if not os.path.isfile(footer_html_file): | ||||
| footer['html'] = '<p>' + footer['plain'] + '</p>' | footer['html'] = '<p>' + footer['plain'] + '</p>' | ||||
| else: | else: | ||||
| footer['html'] = open(footer_html_file, 'r').read() | footer['html'] = open(footer_html_file, 'r').read() | ||||
| if footer['html'] == "": | if footer['html'] == "": | ||||
| footer['html'] = '<p>' + footer['plain'] + '</p>' | footer['html'] = '<p>' + footer['plain'] + '</p>' | ||||
| if footer['plain'] == "" and footer['html'] == "<p></p>": | if footer['plain'] == "" and footer['html'] == "<p></p>": | ||||
| exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', filepath)) | exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', filepath)) | ||||
| return | return | ||||
| footer_added = False | footer_added = False | ||||
| try: | try: | ||||
| _footer_added = message.get("X-Wallace-Footer") | _footer_added = message.get("X-Wallace-Footer") | ||||
| except: | except: | ||||
| pass | pass | ||||
| if _footer_added == "YES": | if _footer_added == "YES": | ||||
| Show All 10 Lines | for part in message.walk(): | ||||
| try: | try: | ||||
| disposition = part.get("Content-Disposition") | disposition = part.get("Content-Disposition") | ||||
| except: | except: | ||||
| pass | pass | ||||
| log.debug("Walking message part: %s; disposition = %r" % (content_type, disposition), level=8) | log.debug("Walking message part: %s; disposition = %r" % (content_type, disposition), level=8) | ||||
| if not disposition == None: | if disposition is not None: | ||||
| continue | continue | ||||
| if content_type == "text/plain": | if content_type == "text/plain": | ||||
| content = part.get_payload(decode=True) | content = part.get_payload(decode=True) | ||||
| content += "\n\n-- \n%s" % (footer['plain']) | content += "\n\n-- \n%s" % (footer['plain']) | ||||
| part.set_payload(content) | footer_added = set_part_content(part, content) | ||||
| footer_added = True | |||||
| log.debug("Text footer attached.", level=6) | |||||
| elif content_type == "text/html": | elif content_type == "text/html": | ||||
| content = part.get_payload(decode=True) | content = part.get_payload(decode=True) | ||||
| append = "\n<!-- footer appended by Wallace -->\n" + footer['html'] | append = "\n<!-- footer appended by Wallace -->\n" + footer['html'] | ||||
| if "</body>" in content: | if "</body>" in content: | ||||
| part.set_payload(content.replace("</body>", append + "</body>")) | content = content.replace("</body>", append + "</body>") | ||||
| else: | else: | ||||
| part.set_payload("<html><body>" + content + append + "</body></html>") | content = "<html><body>" + content + append + "</body></html>" | ||||
| footer_added = set_part_content(part, content) | |||||
| footer_added = True | |||||
| log.debug("HTML footer attached.", level=6) | |||||
| if footer_added: | if footer_added: | ||||
| log.debug("Footer attached.") | log.debug("Footer attached.") | ||||
| message.add_header("X-Wallace-Footer", "YES") | message.add_header("X-Wallace-Footer", "YES") | ||||
| (fp, new_filepath) = tempfile.mkstemp(dir="/var/spool/pykolab/wallace/footer/ACCEPT") | (fp, new_filepath) = tempfile.mkstemp(dir="/var/spool/pykolab/wallace/footer/ACCEPT") | ||||
| os.write(fp, message.as_string()) | os.write(fp, message.as_string()) | ||||
| os.close(fp) | os.close(fp) | ||||
| os.unlink(filepath) | os.unlink(filepath) | ||||
| exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', new_filepath)) | exec('modules.cb_action_%s(%r, %r)' % ('ACCEPT','footer', new_filepath)) | ||||