Started working on adding romanizations to DB

2025-03-10 06:33:31 -04:00
parent 93ff0d172e
commit ab4acd189d
1 changed files with 59 additions and 55 deletions
--- a/xlit_server.py
+++ b/xlit_server.py
@@ -195,61 +195,65 @@ def ulca_api():

@app.route('/romanize', methods=['POST'])
 def romanizeHandler():
-	langCodeLookup = {
-	"as": "as",
-	"bn": "bn",
-	"gom": "gom",
-	"gu": "gu",
-	"hi": "hi",
-	"kn": "kn",
-	"mai": "mai",
-	"ml": "ml",
-	"mni-Mtei": "mni",
-	"mr": "mr",
-	"ne": "ne",
-	"or": "or",
-	"pa": "pa",
-	"sd": "sd",
-	"ta": "ta",
-	"te": "te",
-	"ur": "ur",
-	"tcy": "kn", # Tulu uses Kannada script
-	"awa": "hi", # Awadhi uses Hindi script
-	"doi": "hi", # Dogri uses Hindi script
-	"bho": "hi", # Bhojpuri uses Hindi script
-	}
-	rtv = dict()
+    langCodeLookup = {
+    "hi": "hi",
+    "bn": "bn",
+    "mr": "mr",
+    "ta": "ta",
+    "te": "te",
+    "kn": "kn",
+    "ml": "ml",
+    "or": "or",
+    "gu": "gu",
+    "ur": "ur",
+    "as": "as",
+    "pa": "pa",
+    "mai": "mai",
+    "ne": "ne",
+    "gom": "gom",
+    "tcy": "kn", # Tulu uses Kannada script
+    "bho": "hi", # Bhojpuri uses Hindi script
+    "doi": "hi", # Dogri uses Hindi script
+    "mni-Mtei": "mni",
+    "sd": "sd",
+    "awa": "hi", # Awadhi uses Hindi script
+    }
+    rtv = dict()

-	data = request.get_json(force=True)
+    data = request.get_json(force=True)

-	# Check if database contains the romanizations already
-	englishWord = data['en']
-	print(englishWord)
-	con = sqlite3.connect("../translations.db")
-	cur = con.cursor()
-	cur.execute("CREATE TABLE IF NOT EXISTS romanizations AS SELECT * FROM translations WHERE 0") # Copy schema from 'translations' table
-	cur.execute('SELECT * FROM romanizations WHERE english = ?', (englishWord,))
-	romanizations = cur.fetchall()
-	columnNames = [column[0] for column in cur.description]
-	romanizationsDict = []
-	if len(romanizations) > 0:
-		for row in romanizations:
-			row_dict = {columnNames[i]: row[i] for i in range(len(columns))}
-			romanizationsDict.append(row_dict)
-		json_data = json.dumps(romanizationsdata, indent=4)
-		print(json_data)
-	#	if len(romanizations) != 0:
-			
-	# Assuming the romanizations didn't exist before
-	for key in data:
-		if key in langCodeLookup:
-			langCode = langCodeLookup[key]
-			text = data[key]
-			response = reverse_xlit_api(langCode, text)
-			responseJson = response.get_json()
-			rtv[key] = responseJson['result']
-	
-	rtvJson = jsonify(rtv)
+    # Check if database contains the romanizations already
+    englishWord = data['en']
+    rtv["en"] = englishWord
+    print(englishWord)
+    con = sqlite3.connect("../translations.db")
+    cur = con.cursor()
+    cur.execute("CREATE TABLE IF NOT EXISTS romanizations AS SELECT * FROM translations WHERE 0") # Copy schema from 'translations' table
+    cur.execute('SELECT * FROM romanizations WHERE english = ?', (englishWord,))
+    romanizations = cur.fetchall()
+    columnNames = [column[0] for column in cur.description]
+    romanizationsDict = []
+    if len(romanizations) > 0:
+        for row in romanizations:
+            row_dict = {columnNames[i]: row[i] for i in range(len(langCodeLookup)+1)} # The '+1' is because of English, which isn't in langCodeLookup
+            romanizationsDict.append(row_dict)
+        json_data = json.dumps(romanizationsdata, indent=4)
+        print(json_data)
+    #    if len(romanizations) != 0:
+            
+    # Assuming the romanizations didn't exist before
+    for key in data:
+        if key in langCodeLookup:
+            langCode = langCodeLookup[key]
+            text = data[key]
+            response = reverse_xlit_api(langCode, text)
+            responseJson = response.get_json()
+            rtv[key] = responseJson['result']
+    
+    rtvJson = jsonify(rtv)
+    rtv["en"] = englishWord
+    cur.execute("INSERT INTO romanizations VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)", tuple(rtv.values()))
+    con.commit()

-	con.close()
-	return rtvJson
+    con.close()
+    return rtvJson