From be529bf1891185698974721692b1466777417053 Mon Sep 17 00:00:00 2001
From: simplypower-bbj <bbj@simplypower.dk>
Date: Sun, 18 Dec 2022 17:56:07 +0100
Subject: [PATCH] fixed bug where database login was abused

---
 code/translate.py | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/code/translate.py b/code/translate.py
index 34d366c..4bf1b06 100644
--- a/code/translate.py
+++ b/code/translate.py
@@ -42,21 +42,27 @@ def get_untranslated():
     return val_dict
 
 def continously_translate():
-    while True:
-        val_dict = get_untranslated()
-        if val_dict is None: break
-        # clean
-        content_clean = clean_content(val_dict['content'])
-        # translate
-        try:
-            content_en = translate(content_clean)
-        except Exception as e:
-            content_en = ''
-        val_dict['content_en'] = content_en
-        # upload to database
-        insert_translated(val_dict)
-        print('translated content')
-    print('no more untranslated content located')
+    with Data_table() as dt:
+        query = (
+            f'UPDATE {dt._name} ' 
+            ' SET content_en = (%(content_en)s) '
+            'WHERE id = (%(id)s) '
+        )
+        while True:
+            val_dict = get_untranslated()
+            if val_dict is None: break
+            # clean
+            content_clean = clean_content(val_dict['content'])
+            # translate
+            try:
+                content_en = translate(content_clean)
+            except Exception as e:
+                content_en = ''
+            val_dict['content_en'] = content_en
+            # upload to database
+            dt.execute(query, val_dict)
+            print('translated content')
+        print('no more untranslated content located')
 
 if __name__ == '__main__':
     load_dotenv()