From 5acd15836f5fc2f784bd5553c5d7aaacbae39870 Mon Sep 17 00:00:00 2001 From: Igor Rzegocki Date: Sun, 14 Apr 2013 13:13:52 +0200 Subject: [PATCH] Fixed plain text messages encryption --- gpg-mailgate.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/gpg-mailgate.py b/gpg-mailgate.py index d64146c..70302a5 100755 --- a/gpg-mailgate.py +++ b/gpg-mailgate.py @@ -2,6 +2,7 @@ from ConfigParser import RawConfigParser import email +import email.message import re import GnuPG import smtplib @@ -65,6 +66,10 @@ def encrypt_payload( payload, gpg_to_cmdline ): def encrypt_all_payloads( payloads, gpg_to_cmdline ): encrypted_payloads = list() + if type( payloads ) == str: + msg = email.message.Message() + msg.set_payload( payloads ) + return encrypt_payload( msg, gpg_to_cmdline ).as_string() for payload in payloads: if( type( payload.get_payload() ) == list ): encrypted_payloads.append( encrypt_all_payloads( payload.get_payload(), gpg_to_cmdline ) )