X-Git-Url: https://code.communitydata.science/nu-vpn-proxy.git/blobdiff_plain/6ee0c497943d70259d6ded703665b557841f4904..b61f124435c3f5332344d15fdbdab96d495778a7:/gp-saml-gui.py

diff --git a/gp-saml-gui.py b/gp-saml-gui.py
index 54d14c2..d6f8f15 100755
--- a/gp-saml-gui.py
+++ b/gp-saml-gui.py
@@ -1,6 +1,15 @@
 #!/usr/bin/env python3
 
-import gi
+try:
+    import gi
+except ImportError:
+    try:
+        import pgi as gi
+    except ImportError:
+        gi = None
+if gi is None:
+    raise ImportError("Either gi (PyGObject) or pgi module is required.")
+
 import argparse
 import pprint
 import urllib