porting to PySide2

waves/viewer.py

@@ -24,10 +24,23 @@ import tbox as tb
 # TODO: creer un objet dataset qui gere l'ugrid (comme dans Metafor)
 
 # Qt
-from PyQt5.QtCore import *
-from PyQt5.QtGui import *
-from PyQt5.QtWidgets import *
-print("PyQt5 (Qt %s) loaded!" % QT_VERSION_STR)
+try:
+    from PyQt5.QtCore import *
+    from PyQt5.QtGui import *
+    from PyQt5.QtWidgets import *
+    print("PyQt5 (Qt %s) loaded!" % QT_VERSION_STR)
+except :
+    try:
+        import  PySide2
+        from PySide2.QtCore import *
+        from PySide2.QtGui import *
+        from PySide2.QtWidgets import *
+        #print("PySide2  loaded!")
+        print("PySide2 (Qt %s) loaded!" % PySide2.__version__)
+    except :
+        print("Unable to load PyQt nor PySide2")
+        sys.exit()
+        
 
 # vtk
 import vtk