Rudimentary support for not re-encrypting already encrypted messages

This commit is contained in:
Colin Moller 2013-06-18 18:46:41 -07:00 committed by perennate
parent 91a846beb6
commit 08fc6fa591
1 changed files with 2 additions and 0 deletions

View File

@ -39,6 +39,8 @@ def encrypt_payload( payload, gpg_to_cmdline ):
gpg = GnuPG.GPGEncryptor( cfg['gpg']['keyhome'], gpg_to_cmdline )
raw_payload = payload.get_payload(decode=True)
gpg.update( raw_payload )
if "-----BEGIN PGP MESSAGE-----" in raw_payload and "-----END PGP MESSAGE-----" in raw_payload:
return payload
payload.set_payload( gpg.encrypt() )
if payload['Content-Disposition']:
payload.replace_header( 'Content-Disposition', re.sub(r'filename="([^"]+)"', r'filename="\1.pgp"', payload['Content-Disposition']) )