DeepSeek+Dify查询数据库

#  创建班级表CREATE TABLE classes (    class_id INT AUTO_INCREMENT PRIMARY KEY,    class_name VARCHAR(50) NOT NULL);
#  创建用户表CREATE TABLE users (    user_id INT AUTO_INCREMENT PRIMARY KEY,    username VARCHAR(50) NOT NULL,    email VARCHAR(100) NOT NULL,    class_id INT,    FOREIGN KEY (class_id) REFERENCES classes(class_id));
#  插入班级数据INSERT INTO classes (class_name) VALUES ('Class A'),('Class B'),('Class C'),('Class D'); #  插入用户数据INSERT INTO users (username, email, class_id) VALUES ('Alice', 'alice@example.com', 1),('Bob', 'bob@example.com', 2),('Charlie', 'charlie@example.com', 3),('David', 'david@example.com', 4),('Eve', 'eve@example.com', 1),('Frank', 'frank@example.com', 2),('Grace', 'grace@example.com', 3),('Hank', 'hank@example.com', 4),('Ivy', 'ivy@example.com', 1),('Jack', 'jack@example.com', 2);

from flask import Flask, request, jsonifyimport pymysqlimport re
# 创建 Flask 应用app = Flask(__name__)
# 数据库连接配置DB_CONFIG = {    'host': 'localhost',    'user': 'test',  # 替换为你的数据库用户名    'password': '123456',  # 替换为你的数据库密码    'database': 'test',  # 替换为你的数据库名称    'cursorclass': pymysql.cursors.DictCursor}
# 定义接口：执行 SQL 查询@app.route('/query', methods=['POST'])def execute_query():      # 获取客户端传递的 SQL 语句      data = request.json      if not data or 'sql' not in data:            return jsonify({'error': 'Missing SQL statement in request body'}), 400          sql = data['sql']        # 简单的 SQL 语句验证（防止恶意操作）      if not re.match(r'^SELECT\s', sql, re.IGNORECASE):            return jsonify({'error': 'Only SELECT queries are allowed'}), 400        try:        # 连接数据库          connection = pymysql.connect(**DB_CONFIG)          with connection.cursor() as cursor:                # 执行 SQL 查询                cursor.execute(sql)                           result = cursor.fetchall()                return jsonify(result)    except pymysql.MySQLError as e:          return jsonify({'error': str(e)}), 500    finally:          if connection:              connection.close()  # 启动服务if __name__ == '__main__':      app.run(debug=True)

python .\server.py

你是一个数据分析师，根据用户输入的{content}需求以及检索到{上下文}，生成对应的查询SQL，SQL必须经过严格的校验。## 你可以使用的其他方法用户输入类似于“求和”或“总和”时，则在sql语句中使用SUM()。用户输入类似于“平均数”或“平均”时，在在sql语句中使用AVG()。## 要求1.如果用户输入的内容无法生成为sql语句，请直接说“抱歉，该命令无法形成数据库查询操作”。2.当可以生成sql语句时，请确保输出的内容为完整正确的sql语句，不要输出此外的其他任何字符，确保你生成的内容用户可以直接执行查询操作。3.对于字符串内容的查询请使用LIKE操作而不是等于操作。4.请不要在回复中包括除sql语句之外的任何内容。5.禁止中间过程输出。6.输出的sql为一整行。