Support Python 2.7 and GTK+2

[~andy/csm213a-hw] / vis / test.py

diff --git a/vis/test.py b/vis/test.py
index c3e7ea2f44fa808fe01e49616aa078eb8685eab1..e18191cb5673462c7eb6bca327813e6d42caa23f 100644 (file)
--- a/vis/test.py
+++ b/vis/test.py
@@ -9,8 +9,12 @@ import time
 import datetime
 import random
 
-from gi.repository import Gtk
-from gi.repository import GLib
+if sys.version_info<(3,0,0):
+       import pygtk as PyGtk
+       import gtk   as Gtk
+else:
+       from gi.repository import Gtk
+       from gi.repository import GLib
 
 from serial import Serial