漏洞修复双11促销活动

from flask import Flask, request, jsonify
import os

app = Flask(__name__)

# 防止SQL注入的简单示例
@app.route('/user/<username>')
def get_user(username):
    # 假设我们有一个安全的数据库查询方法
    user = safe_query_database(username)
    if user:
        return jsonify(user)
    else:
        return jsonify({"error": "User not found"}), 404

# 防止跨站脚本攻击（XSS）
@app.route('/post', methods=['POST'])
def create_post():
    data = request.get_json()
    safe_content = escape_html(data['content'])
    # 假设我们有一个安全的保存方法
    save_post(safe_content)
    return jsonify({"message": "Post created successfully"}), 201

def safe_query_database(username):
    # 这里应该使用参数化查询或其他安全措施
    # 示例代码省略具体实现
    pass

def escape_html(content):
    # 简单的HTML转义函数
    return content.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;')

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 8080)))