数据库留言板流程

from flask import Flask, request, jsonify
import sqlite3

app = Flask(__name__)

# 创建数据库和表
def init_db():
    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS messages (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            content TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()

# 提交留言
@app.route('/submit', methods=['POST'])
def submit_message():
    content = request.json.get('content')
    if content:
        conn = sqlite3.connect('messages.db')
        cursor = conn.cursor()
        cursor.execute('INSERT INTO messages (content) VALUES (?)', (content,))
        conn.commit()
        conn.close()
        return jsonify({'status': 'success'}), 200
    else:
        return jsonify({'status': 'error', 'message': 'Content is required'}), 400

# 查看留言
@app.route('/messages', methods=['GET'])
def get_messages():
    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM messages')
    messages = cursor.fetchall()
    conn.close()
    return jsonify({'messages': messages}), 200

if __name__ == '__main__':
    init_db()
    app.run(debug=True)