update code

main.py

@@ -128,8 +128,8 @@ def update_itr(id):
     itr = ITRHandler()
 
     if request.method == 'POST':
-        # data = {k: request.form.get(k, 0) for k in request.form}
-        itr.update(id, request.form)
+        data = {k: request.form.get(k, 0) for k in request.form}
+        itr.update(id, data)
         itr.close()
         return redirect(url_for('display_itr'))
 
@@ -247,8 +247,8 @@ def update_cit(id):
         return "CIT record not found", 404
 
     if request.method == 'POST':
-        # data = {k: request.form.get(k, 0) for k in request.form}
-        cit.update_cit(id, request.form)
+        data = {k: request.form.get(k, 0) for k in request.form}
+        cit.update_cit(id, data)
         cit.close()
         return redirect(url_for('display_cit'))
 
@@ -275,8 +275,8 @@ def display_itat():
 def add_itat():
     if request.method == 'POST':
         itat = ITATHandler()
-        # data = {k: request.form.get(k, 0) for k in request.form}
-        itat.add_itat(request.form)
+        data = {k: request.form.get(k, 0) for k in request.form}
+        itat.add_itat(data)
         itat.close()
         flash("ITAT record added successfully!", "success")
         return redirect(url_for('display_itat'))
@@ -455,7 +455,24 @@ def summary_report():
 #     check_year_obj = YearGet()
 #     result = check_year_obj.CheckYearExists(table_name, year)
 #     check_year_obj.close()
+#     return result
 
+@app.route('/check_year', methods=['POST'])
+def check_year():
+    table_name = request.json.get("table")
+    year = request.json.get("year")
+
+    conn = DBConfig.get_db_connection()
+    cursor = conn.cursor()
+
+    sqlstr = f"SELECT COUNT(*) FROM {table_name} WHERE year = %s"
+    cursor.execute(sqlstr, (year,))
+    result = cursor.fetchone()[0]
+
+    cursor.close()
+    conn.close()
+
+    return {"exists": result > 0}
 
 # Mat credit from 
 @app.route("/mat_credit", methods=["GET"])