From c896ece4db8b37fb4100a9487eea849ce8eb5781 Mon Sep 17 00:00:00 2001
From: Mathias Rav <rav@cs.au.dk>
Date: Fri, 1 May 2015 17:01:56 +0200
Subject: [PATCH] No policy in Python 2

---
 emailtunnel/send.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/emailtunnel/send.py b/emailtunnel/send.py
index 819a7dd..99e7c80 100644
--- a/emailtunnel/send.py
+++ b/emailtunnel/send.py
@@ -1,3 +1,4 @@
+import six
 import sys
 import argparse
 import smtplib
@@ -97,9 +98,13 @@ def main(*args, **kwargs):
     message.set_body_text(body, args.encoding)
 
     from email.generator import Generator
-    policy = email.message.compat32
-    g = Generator(
-        sys.stdout, maxheaderlen=80, policy=policy, mangle_from_=False)
+    if six.PY3:
+        policy = email.message.compat32
+        g = Generator(
+            sys.stdout, maxheaderlen=80, policy=policy, mangle_from_=False)
+    else:
+        g = Generator(
+            sys.stdout, maxheaderlen=80, mangle_from_=False)
     try:
         g.flatten(message.message, unixfrom=False)
     except: