Update info.py

info.py

@@ -46,12 +46,16 @@ def ownInfo(sentence):
 
 """ "add" new values to table favourite """
 def getInfo(nick, info):
-    print("I don't know yours, what is it? ")
-    ans = input("")
-    connect = connection()       # gets the details to the database
-    cursor = connect.cursor()        # establish a cursor to fetch the results wanted from the databse
-    sql = 'UPDATE favourite SET '+info+'="'+ans+'" WHERE NICKNAME="'+ nick + '"'        #sql gets the order to update all info from the table favourite in the current user row
-    cursor.execute(sql)             # order to select executed
-    connect.commit()          # tell the database to save all the changes
-    print("Ok, I will try not to forget!")
-
+    try:
+        print("I don't know yours, what is it? ")
+        ans = input("")
+        connect = connection()       # gets the details to the database
+        cursor = connect.cursor()        # establish a cursor to fetch the results wanted from the databse
+        if info in "footballplayer":
+            info = "football_player"
+        sql = 'UPDATE favourite SET '+info+'="'+ans+'" WHERE NICKNAME="'+ nick + '"'        #sql gets the order to update all info from the table favourite in the current user row
+        cursor.execute(sql)             # order to select executed
+        connect.commit()          # tell the database to save all the changes
+        print("Ok, I will try not to forget!")
+    except:
+        print("I am not able to rember that, I am sorry.")