X-Git-Url: https://git.tld-linux.org/?a=blobdiff_plain;f=TLD_Builder%2Fnotify.py;h=e13c05235f2ba80a6d05be69ba2a8d681ded61aa;hb=3a6296fff210d498302ce0a2da6beed689466347;hp=5b0368f8a064a29833c9e0575cde0cdc831a103b;hpb=b999f53d4bf5d44586ecf028876e8bc20b5fd2ce;p=tld-builder.git

diff --git a/TLD_Builder/notify.py b/TLD_Builder/notify.py
index 5b0368f..e13c052 100644
--- 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()))