Update util/dbquery.py

util/dbquery.py

@@ -12,12 +12,12 @@ def connection():
     return connection
 
 def getDataNum():
-    conn = connection()
-    mycursor = conn.cursor()
-    query = "SELECT COUNT(id) AS coloumn_count FROM Markets"
-    mycursor.execute(query)
-    data_num = mycursor.fetchall()
-    return data_num
+    conn = connection()  
+    mycursor = conn.cursor() 
+    query = "SELECT COUNT(*) FROM Markets"
+    mycursor.execute(query)  
+    data_num = mycursor.fetchone()  
+    return data_num[0]
     
 def insert_data(json_data):
     conn = connection()