Update prog.py

Projet/prog.py

@@ -96,23 +96,24 @@ def afficherTable(conn,table):
     sql = "SELECT * FROM "+table
     try:
         cur.execute(sql)
-    except pg.UndefinedTable:
-        print("Table doesn't exist")
-        conn.rollback()
-    except pg.InFailedSqlTransaction:
-        print("Transaction Error")
+    except psycopg2.Error:
+        print("Execute Error")
         conn.rollback()
+        return
     # Fetch data line by line
-    
-    raw = cur.fetchone()
-    if raw:
-        while raw:
-            print (raw[0])
-            print (raw[1])
-            raw = cur.fetchone()
-    else:
-        print("Empty table")
-    conn.commit()
+    try:
+        raw = cur.fetchone()
+        if raw:
+            while raw:
+                print (raw[0])
+                print (raw[1])
+                raw = cur.fetchone()
+        else:
+            print("Empty table")
+        conn.commit()
+    except psycopg2.Error:
+        print("Fetch error")
+        conn.rollback()
 
 def modifierTables(table)
 def afficherTables(table)