使用mysql数据库，进行增删查

app.py

@@ -1,21 +1,58 @@
-# app.py
+# -*- coding: utf-8 -*-
+from flask import Flask, jsonify, request
+from data import *
 
-from flask import Flask, jsonify
-from data import initialize_db, insert_user, fetch_users
-
-# 创建 Flask 应用实例
+# 初始化 Flask 应用
 app = Flask(__name__)
 
-# 首页路由
+# 初始化数据库和表
+initialize_database()
+initialize_table()
+
 @app.route('/')
 def home():
-    # 初始化数据库
-    initialize_db()
+    return "Welcome to the Fruit API!"
+
+# 添加水果数据
+@app.route('/add_fruit', methods=['POST'])
+def add_fruit():
+    data = request.json
+    name = data.get('name')
+    origin = data.get('origin')
+
+    if not name or not origin:
+        return jsonify({"error": "Missing name or origin"}), 400
+
+    try:
+        insert_data(name, origin)
+        return jsonify({"message": f"Fruit {name} from {origin} added successfully!"}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+# 获取所有水果数据
+@app.route('/fruits', methods=['GET'])
+def get_fruits():
+    try:
+        fruits = fetch_data()
+        return jsonify({"fruits": fruits})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route('/delete_fruit', methods=['POST'])
+def delete_fruit():
+    data = request.json
+    name = data.get('name')
+
+    if not name:
+        return jsonify({"error": "Missing name"}), 400
+
+    try:
+        delete_data(name)
+        return jsonify({"message": f"Fruit {name} deleted successfully!"}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
-    # 插入用户数据（可选，避免重复插入）
-    insert_user('Alice', 30)
-    insert_user('Bob', 25)
 
-    # 查询用户数据
-    users = fetch_users()
-    return jsonify(users)  # 将用户数据返回为 JSON 格式
+# 运行 Flask 应用
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000, debug=True)

data.py

@@ -1,32 +1,141 @@
-# data.py
-
-import sqlite3
-
-# 创建和初始化数据库
-def initialize_db():
-    conn = sqlite3.connect('example.db')
-    cursor = conn.cursor()
-    cursor.execute('''CREATE TABLE IF NOT EXISTS users (
-                        id INTEGER PRIMARY KEY,
-                        name TEXT,
-                        age INTEGER
-                     )''')
-    conn.commit()
-    conn.close()
-
-# 插入用户数据
-def insert_user(name, age):
-    conn = sqlite3.connect('example.db')
-    cursor = conn.cursor()
-    cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", (name, age))
-    conn.commit()
-    conn.close()
-
-# 查询所有用户数据
-def fetch_users():
-    conn = sqlite3.connect('example.db')
-    cursor = conn.cursor()
-    cursor.execute("SELECT * FROM users")
-    rows = cursor.fetchall()
-    conn.close()
-    return rows
+# -*- coding: utf-8 -*-
+import mysql.connector
+from mysql.connector import Error
+
+# 配置数据库连接
+DB_CONFIG = {
+    'host': '60.204.139.57',
+    'user': 'root',
+    'password': '1',
+    'port': 6003,
+    'database': 'fruit'  # 替换为数据库名称
+}
+
+# 初始化数据库
+def initialize_database():
+    try:
+        # 连接 MySQL 服务器（不指定数据库）
+        connection = mysql.connector.connect(
+            host=DB_CONFIG['host'],
+            user=DB_CONFIG['user'],
+            password=DB_CONFIG['password'],
+            port=DB_CONFIG['port']
+        )
+
+        if connection.is_connected():
+            print("成功连接到 MySQL 服务器")
+            cursor = connection.cursor()
+
+            # 检查数据库是否存在
+            cursor.execute(f"SHOW DATABASES LIKE '{DB_CONFIG['database']}';")
+            result = cursor.fetchone()
+
+            if result:
+                print(f"数据库 {DB_CONFIG['database']} 已存在。")
+            else:
+                print(f"数据库 {DB_CONFIG['database']} 不存在，正在创建...")
+                cursor.execute(f"CREATE DATABASE {DB_CONFIG['database']};")
+                print(f"数据库 {DB_CONFIG['database']} 创建成功。")
+
+    except Error as err:
+        print(f"连接 MySQL 时发生错误：{err}")
+    finally:
+        if connection.is_connected():
+            cursor.close()
+            connection.close()
+            print("MySQL 服务器连接已关闭")
+
+# 初始化表
+def initialize_table():
+    try:
+        connection = mysql.connector.connect(**DB_CONFIG)
+        if connection.is_connected():
+            cursor = connection.cursor()
+
+            # 创建表
+            cursor.execute('''CREATE TABLE IF NOT EXISTS fruit (
+                                id INT AUTO_INCREMENT PRIMARY KEY,
+                                name VARCHAR(100),
+                                origin VARCHAR(100)
+                              )''')
+            print("--- 创建 fruit 表 ---")
+    except Error as err:
+        print(f"初始化表时发生错误：{err}")
+    finally:
+        if connection.is_connected():
+            cursor.close()
+            connection.close()
+            print("MySQL 连接已关闭")
+
+# 插入数据
+def insert_data(name, origin):
+    try:
+        connection = mysql.connector.connect(**DB_CONFIG)
+        if connection.is_connected():
+            cursor = connection.cursor()
+
+            # 插入数据
+            cursor.execute("INSERT INTO fruit (name, origin) VALUES (%s, %s)", (name, origin))
+            connection.commit()
+            print(f"插入数据成功：({name}, {origin})")
+    except Error as err:
+        print(f"插入数据时发生错误：{err}")
+    finally:
+        if connection.is_connected():
+            cursor.close()
+            connection.close()
+            print("MySQL 连接已关闭")
+
+def delete_data(name):
+    try:
+        connection = mysql.connector.connect(**DB_CONFIG)
+        if connection.is_connected():
+            cursor = connection.cursor()
+
+            # 删除数据
+            cursor.execute("DELETE FROM fruit WHERE name = %s", (name,))
+            connection.commit()
+            print(f"成功删除名称为 {name} 的记录")
+    except Error as err:
+        print(f"删除数据时发生错误：{err}")
+    finally:
+        if connection.is_connected():
+            cursor.close()
+            connection.close()
+            print("MySQL 连接已关闭")
+
+
+# 查询数据
+def fetch_data():
+    try:
+        connection = mysql.connector.connect(**DB_CONFIG)
+        if connection.is_connected():
+            cursor = connection.cursor()
+
+            # 查询数据
+            cursor.execute("SELECT * FROM fruit")
+            rows = cursor.fetchall()
+            print("--- 查询结果 ---")
+            for row in rows:
+                print(row)
+            return rows
+    except Error as err:
+        print(f"查询数据时发生错误：{err}")
+    finally:
+        if connection.is_connected():
+            cursor.close()
+            connection.close()
+            print("MySQL 连接已关闭")
+
+# 主函数
+if __name__ == "__main__":
+    # 初始化数据库和表
+    initialize_database()
+    initialize_table()
+
+    # 插入示例数据
+    insert_data('Apple', 'China')
+    insert_data('Banana', 'Ecuador')
+
+    # 查询数据
+    fetch_data()