diff --git a/push_to_db.py b/push_to_db.py index 81c3e7371472e5f00e6c5396f4336e19f3bace66..fbc3cd64626c03fbad1930eac8c025c392374075 100644 --- a/push_to_db.py +++ b/push_to_db.py @@ -107,6 +107,11 @@ def match_part_obj_into_db(_part, part_type): return _matched_id +def finalize_transaction(_cursor, _transaction_id, _user_id, _combination_id): + _cursor.execute("INSERT INTO transactions (id, user_id, combination_id) VALUES (%s, %s, %s);", _transaction_id, _user_id, _combination_id) + _cursor.fetchall() + + def func(): value = R.lpop("mypc:queue") if value is None: @@ -179,6 +184,13 @@ def func(): print("Committing to DB") db_conn.commit() + print("Finalizing transaction") + with db_conn.cursor() as cursor: + finalize_transaction(cursor, _transaction_id, _user_id, _combination_id) + + print("Committing to DB") + db_conn.commit() + print("Removing anchor key from Redis") R.delete(f"mypc:code:{code}:document")