Flask 框架

from flask import Flask, request, jsonify
import sqlite3
import random
import string
from datetime import datetime

app = Flask(__name__)
DATABASE = 'data.db'

# 初始化数据库
def init_db():
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS entries
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  content TEXT,
                  created_at DATETIME)''')
    conn.commit()
    conn.close()

# 生成随机字符串
def generate_random_string(length=12):
    letters = string.ascii_letters + string.digits
    return ''.join(random.choice(letters) for _ in range(length))

# 写接口
@app.route('/write', methods=['POST'])
def write_entry():
    # 生成随机内容
    random_content = generate_random_string()

    # 获取当前时间
    now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

    # 写入数据库
    try:
        conn = sqlite3.connect(DATABASE)
        c = conn.cursor()
        c.execute("INSERT INTO entries (content, created_at) VALUES (?, ?)",
                 (random_content, now))
        conn.commit()
        conn.close()
        return jsonify({
            "status": "success",
            "message": "Data written successfully",
            "data": {
                "content": random_content,
                "created_at": now
            }
        }), 201
    except Exception as e:
        return jsonify({"status": "error", "message": str(e)}), 500

# 读接口
@app.route('/read', methods=['GET'])
def read_entries():
    try:
        conn = sqlite3.connect(DATABASE)
        c = conn.cursor()
        c.execute("SELECT * FROM entries ORDER BY id DESC")
        entries = c.fetchall()
        conn.close()

        # 格式化结果
        results = []
        for entry in entries:
            results.append({
                "id": entry[0],
                "content": entry[1],
                "created_at": entry[2]
            })

        return jsonify({
            "status": "success",
            "count": len(results),
            "data": results
        }), 200
    except Exception as e:
        return jsonify({"status": "error", "message": str(e)}), 500

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