add chk_conn and get_tables functions to utils.py

pages/quiz.py

@@ -17,13 +17,6 @@ def app():
         c = conn.cursor()
         return c, conn
 
-    def chk_conn(conn):
-     try:
-        conn.cursor()
-        return True
-     except Exception as ex:
-        return False
-
     def form_callback(questions):
         st.session_state.form_submit = True
         num_correct = 0
@@ -57,11 +50,6 @@ def app():
     if "form_submit" not in st.session_state: 
         c, conn = db_connect(DATABASE)
 
-        st.write(chk_conn(conn))
-        
-        c.execute("SELECT name FROM sqlite_master WHERE type='table';")
-        st.write(c.fetchall())
-
         units_list = []
         for item in c.execute("SELECT DISTINCT unit FROM vocab"):
             units_list.append(item[0])

pages/utils.py

@@ -2,6 +2,7 @@ import re
 import random
 import datetime
 import string
+import sqlite3
 
 def add_blanks(word, sentence, blank = "__"):
   return re.sub(word, blank, sentence, flags=re.IGNORECASE)
@@ -14,4 +15,15 @@ def random_session_id():
   return ''.join(random.choices(alphabet, k=12))
 
 def check_answer(item, answer):
-  return item == answer
+  return item == answer
+
+def chk_conn(conn):
+  try:
+    conn.cursor()
+    return True
+  except Exception as ex:
+    return False
+
+def get_tables(cursor):      
+  cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
+  return cursor.fetchall()