from flask import redirect, url_for, g, request
from .config import db
import pymysql

def pay_confirm():
    order_id = request.args.get('order_id')
    if not g.user:
        return redirect(url_for("login"))

    conn = pymysql.connect(**db)
    cursor = conn.cursor()

    # 检查订单是否存在
    check_order_sql = "SELECT ID FROM Orders WHERE ID = %s"
    cursor.execute(check_order_sql, (order_id,))
    order_exists = cursor.fetchone()

    if not order_exists:
        cursor.close()
        conn.close()
        return redirect(url_for("order_list"))

    update_order_sql = "UPDATE Orders SET Paid = 1 WHERE ID = %s"
    cursor.execute(update_order_sql, (order_id,))

    conn.commit()
    cursor.close()
    conn.close()

    return redirect(url_for('order_list', order_id=order_id))