Merge pull request googleapis#18 from craigcitro/from_encoding

craigcitro · craigcitro · commit ab01cb9c4ff0 · 2014-07-16T09:23:32.000-07:00
Replace uses of `as_string` from the email lib.
diff --git a/googleapiclient/discovery.py b/googleapiclient/discovery.py
@@ -27,7 +27,9 @@
 
 
 # Standard library imports
+import StringIO
 import copy
+from email.generator import Generator
 from email.mime.multipart import MIMEMultipart
 from email.mime.nonmultipart import MIMENonMultipart
 import keyword
@@ -728,7 +730,12 @@ def method(self, **kwargs):
           payload = media_upload.getbytes(0, media_upload.size())
           msg.set_payload(payload)
           msgRoot.attach(msg)
-          body = msgRoot.as_string()
+          # encode the body: note that we can't use `as_string`, because
+          # it plays games with `From ` lines.
+          fp = StringIO.StringIO()
+          g = Generator(fp, mangle_from_=False)
+          g.flatten(msgRoot, unixfrom=False)
+          body = fp.getvalue()
 
           multipart_boundary = msgRoot.get_boundary()
           headers['content-type'] = ('multipart/related; '
diff --git a/googleapiclient/http.py b/googleapiclient/http.py
@@ -1229,7 +1229,12 @@ def _execute(self, http, order, requests):
       msg.set_payload(body)
       message.attach(msg)
 
-    body = message.as_string()
+    # encode the body: note that we can't use `as_string`, because
+    # it plays games with `From ` lines.
+    fp = StringIO.StringIO()
+    g = Generator(fp, mangle_from_=False)
+    g.flatten(message, unixfrom=False)
+    body = fp.getvalue()
 
     headers = {}
     headers['content-type'] = ('multipart/mixed; '
