#sql_tools.py

from flask import Flask, request, jsonify
import mysql.connector

# 數(shù)據(jù)庫(kù)連接配置
config = {
    'user': 'your_username',
    'password': 'your_password',
    'host': 'localhost',
    'database': 'your_database',
    'raise_on_warnings': True
}

# 初始化Flask應(yīng)用
app = Flask(__name__)

# 連接數(shù)據(jù)庫(kù)
def connect_to_database():
    try:
        conn = mysql.connector.connect(**config)
        print("Connected to MySQL database")
        return conn
    except mysql.connector.Error as err:
        print(f"Error: {err}")
        return None

# 執(zhí)行SQL查詢
def execute_query(conn, sql):
    cursor = conn.cursor()
    try:
        cursor.execute(sql)
        if sql.strip().lower().startswith("select"):
            # 如果是查詢操作，返回結(jié)果
            result = cursor.fetchall()
            return result
        else:
            # 如果是插入、更新、刪除操作，提交事務(wù)并返回受影響的行數(shù)
            conn.commit()
            return cursor.rowcount
    except mysql.connector.Error as err:
        print(f"Error executing SQL: {err}")
        return None
    finally:
        cursor.close()

# HTTP接口：執(zhí)行SQL
@app.route('/execute', methods=['POST'])
def execute_sql():
    # 獲取請(qǐng)求中的SQL語(yǔ)句
    data = request.json
    if not data or 'sql' not in data:
        return jsonify({"error": "SQL statement is required"}), 400

    sql = data['sql']
    conn = connect_to_database()
    if not conn:
        return jsonify({"error": "Failed to connect to database"}), 500

    # 執(zhí)行SQL
    result = execute_query(conn, sql)
    conn.close()

    if result is None:
        return jsonify({"error": "Failed to execute SQL"}), 500

    # 返回結(jié)果
    return jsonify({"result": result})

# 啟動(dòng)Flask應(yīng)用
if __name__ == '__main__':
    app.run(host='0.0.0.0', port=3000)

import requests
def main(sql: str) -> dict:
    # 定義API的URL
    url = "https://xxx.ngrok-free.app/execute"

    # 構(gòu)造請(qǐng)求體
    payload = {
        "sql": sql
    }

    # 發(fā)送POST請(qǐng)求
    try:
        response = requests.post(url, json=payload)
    
        # 檢查響應(yīng)狀態(tài)碼
        if response.status_code == 200:
            # 解析響應(yīng)數(shù)據(jù)
            result = response.json()
            return {
                "result": f"{result}"
            }
        else:
            return {
                "result": f"請(qǐng)求失敗，狀態(tài)碼：{response.status_code},{response.json()}"
            }
    except requests.exceptions.RequestException as e:
        return {
            "result": f"請(qǐng)求異常：{e}"
        }