split up query

app.py

@@ -81,19 +81,16 @@ def save_data(orig_user_email, constituent_email, labels, user_response):
     db_cursor = db_connection.cursor()
 
     try:
-        print("saving first email")
         db_cursor.execute(
             "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', 0)",
             (orig_user_email,),
         )
 
-        print("saving constituent email")
         db_cursor.execute(
             "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', 1)",
             (constituent_email,),
         )
 
-        print("saving user response")
         db_cursor.execute(
             "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', 2)",
             (user_response,),
@@ -102,20 +99,21 @@ def save_data(orig_user_email, constituent_email, labels, user_response):
         # insert a row into the message_categorys_associations table for each valid label in labels with the message_id of the constituent_email
         labels = labels.split(", ")
         for label in labels:
-            print("saving label: " + label)
             label_exists = db_cursor.execute(
                 "SELECT * FROM radmap_frog12.message_categorys WHERE message_category_name = %s",
                 (label,),
             )
             label_exists = db_cursor.fetchall()
-            print(label_exists)
             if label_exists:
-                print("label exists")
+                message_id = db_cursor.execute(
+                    "SELECT id FROM messages WHERE body = %s", (constituent_email,)
+                )
+                message_id = db_cursor.fetchall()
+
                 db_cursor.execute(
-                    "INSERT INTO radmap_frog12.message_category_associations (message_id, message_category_id) VALUES ((SELECT id FROM messages WHERE body = %s), %i)",
-                    (constituent_email, label_exists[0][0]),
+                    "INSERT INTO radmap_frog12.message_category_associations (message_id, message_category_id) VALUES (%i, %i)",
+                    (message_id, label_exists[0][0]),
                 )
-                print("label saved")
 
         db_connection.commit()