Parcourir la source

使用sqlite3数据库

Gogs il y a 9 mois
Parent
commit
3ae9578610
2 fichiers modifiés avec 47 ajouts et 7 suppressions
  1. 15 7
      app.py
  2. 32 0
      data.py

+ 15 - 7
app.py

@@ -1,13 +1,21 @@
-from flask import Flask
+# app.py
 
-app = Flask(__name__)
+from flask import Flask, jsonify
+from data import initialize_db, insert_user, fetch_users
 
+# 创建 Flask 应用实例
+app = Flask(__name__)
 
+# 首页路由
 @app.route('/')
-def hello_world():  # put application's code here
-    return 'Hello World!'
+def home():
+    # 初始化数据库
+    initialize_db()
 
+    # 插入用户数据(可选,避免重复插入)
+    insert_user('Alice', 30)
+    insert_user('Bob', 25)
 
-if __name__ == '__main__':
-    #app.run(debug=True, port=50001)
-    app.run()
+    # 查询用户数据
+    users = fetch_users()
+    return jsonify(users)  # 将用户数据返回为 JSON 格式

+ 32 - 0
data.py

@@ -0,0 +1,32 @@
+# 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