- more python 3.x fixes

[tld-builder.git] / TLD_Builder / notify.py

diff --git a/TLD_Builder/notify.py b/TLD_Builder/notify.py
index 5b0368f8a064a29833c9e0575cde0cdc831a103b..e13c05235f2ba80a6d05be69ba2a8d681ded61aa 100644 (file)
--- a/TLD_Builder/notify.py
+++ b/TLD_Builder/notify.py
@@ -1,6 +1,10 @@
 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
 
-import StringIO
+import sys
+if sys.version_info[0] == 2:
+    import StringIO
+else:
+    from io import StringIO
 
 import mailer
 import gpg
@@ -10,12 +14,12 @@ from config import config
 
 class Notifier:
     def __init__(self, g):
-        self.xml = StringIO.StringIO()
+        self.xml = StringIO()
         self.xml.write("<notification group-id='%s' builder='%s'>\n" % \
                         (g.id, config.builder))
 
     def send(self, r):
-        sio = StringIO.StringIO()
+        sio = StringIO()
         self.xml.write("</notification>\n")
         self.xml.seek(0)
         sio.write(gpg.sign(self.xml.read()))