moved in related functions

code/database.py

@@ -171,17 +171,6 @@ def insert_list(username_list, timestamp_list, content_list):
             except Exception as e:
                 logging.warning(e)
 
-def insert_translated(val_dict):
-    assert 'id' in val_dict.keys()
-    assert 'content_en' in val_dict.keys()
-    with Data_table() as dt:
-        query = (
-            f'UPDATE {dt._name} ' 
-            ' SET content_en = (%(content_en)s) '
-            'WHERE id = (%(id)s) '
-        )
-        dt.execute(query, val_dict)
-
 def insert_translated_list(content_list, id_list):
     with Data_table() as dt:
         for id, content in zip(id_list, content_list):
@@ -210,3 +199,28 @@ def get_all():
         query = f'SELECT * FROM {dt._name} ORDER BY time DESC '
         df = pd.read_sql_query(query, dt.con)
     return df
+
+def insert_translated(dt, val_dict):
+    assert 'id' in val_dict.keys()
+    assert 'content_en' in val_dict.keys()
+    query = (
+        f'UPDATE {dt._name} ' 
+        ' SET content_en = (%(content_en)s) '
+        'WHERE id = (%(id)s) '
+    )
+    dt.execute(query, val_dict)
+
+def get_untranslated(dt):
+    query = (
+        f'SELECT * FROM {dt._name} '
+        'WHERE content_en IS NULL '
+        'ORDER BY random() '
+        'LIMIT 1 '
+    )
+    df = pd.read_sql_query(query, dt.con)
+    if len(df)==0: return None
+    val_dict = {
+        'id': int(df['id'][0]),
+        'content': str(df['content'][0])
+    }
+    return val_dict