Fix unicode vs. bytes issue in glsa-check (#341293)
authorAndy Kittner <andkit@gmx.de>
Wed, 26 Jan 2011 23:21:21 +0000 (00:21 +0100)
committerPaul Varner <fuzzyray@gentoo.org>
Fri, 28 Jan 2011 17:26:32 +0000 (11:26 -0600)
bin/glsa-check
pym/gentoolkit/glsa/__init__.py

index a8c018888dc9ae80e6b55e79e8b8bd4d723a0381..a35375b6b0a5fcddfd8568444826b85f8781a702 100755 (executable)
@@ -358,7 +358,7 @@ if mode == "mail":
                import portage_mail
 
        import socket
-       from io import StringIO
+       from io import BytesIO
        try:
                from email.mime.text import MIMEText
        except ImportError:
@@ -380,11 +380,13 @@ if mode == "mail":
        mysubject = "[glsa-check] Summary for %s" % socket.getfqdn()
 
        # need a file object for summarylist()
-       myfd = StringIO()
-       myfd.write("GLSA Summary report for host %s\n" % socket.getfqdn())
-       myfd.write("(Command was: %s)\n\n" % " ".join(sys.argv))
+       myfd = BytesIO()
+       line = "GLSA Summary report for host %s\n" % socket.getfqdn()
+       myfd.write(line.encode("utf-8"))
+       line = "(Command was: %s)\n\n" % " ".join(sys.argv)
+       myfd.write(line.encode("utf-8"))
        summarylist(glsalist, fd1=myfd, fd2=myfd)
-       summary = str(myfd.getvalue())
+       summary = myfd.getvalue().decode("utf-8")
        myfd.close()
 
        myattachments = []
@@ -395,9 +397,10 @@ if mode == "mail":
                        if verbose:
                                sys.stderr.write(("invalid GLSA: %s (error message was: %s)\n" % (myid, e)))
                        continue
-               myfd = StringIO()
+               myfd = BytesIO()
                myglsa.dump(outstream=myfd)
-               myattachments.append(MIMEText(str(myfd.getvalue()), _charset="utf8"))
+               attachment = myfd.getvalue().decode("utf-8")
+               myattachments.append(MIMEText(attachment, _charset="utf8"))
                myfd.close()
 
        if glsalist or not quiet:
index ab0394785c7bffb9b51a5c8687166880b3a38867..af97ac6876dd7bffb55137aa2c922414a488b32f 100644 (file)
@@ -622,6 +622,7 @@ class Glsa:
                @param  outfile: Stream that should be used for writing
                                                 (defaults to sys.stdout)
                """
+               outstream = getattr(outstream, "buffer", outstream)
                outstream = codecs.getwriter(encoding)(outstream)
                width = int(self.config["PRINTWIDTH"])
                outstream.write(center("GLSA %s: \n%s" % (self.nr, self.title), width)+"\n")