Python Flask 如何保存页面提交的注册表单数据，部署后写入失效如何解决？

原有代码的错误点 #

• open()的第一个参数是文件路径字符串，必须包裹引号，你写的file.txt未加引号属于语法错误，程序会将其识别为未定义变量直接报错
• open()默认使用只读模式r打开文件，没有写入权限，要写入数据必须指定写入模式，常用的有a（追加模式，每次写入新增到文件末尾）、w（覆盖模式，每次写入清空原有内容）
• 未使用上下文管理器管理文件句柄，程序中途报错会导致文件泄漏，且未处理权限类异常，上线后服务器环境的目录读写权限限制会直接导致写入失败
• 明文拼接存储用户名和密码属于严重安全漏洞，既无法后续拆分用户名和密码的边界，也不符合数据安全存储规范

修复方案 #

方案1：修复文件写入逻辑（适合小型测试场景） #

首先确保Flask正确接收表单参数，按以下逻辑调整代码，可解决部署后的写入失败问题：

from flask import Flask, request
import os
from werkzeug.security import generate_password_hash

app = Flask(__name__)
# 用绝对路径定位存储文件，避免相对路径在不同部署环境下找不到目标文件
DATA_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'user_data.txt')

@app.route('/register', methods=['POST'])
def register():
    # 接收表单提交的参数
    username = request.form.get('username')
    password = request.form.get('password')
    # 基础参数校验
    if not username or not password:
        return "用户名和密码不能为空", 400
    
    try:
        # with上下文管理器会自动关闭文件，避免句柄泄漏，指定utf-8编码防止乱码
        with open(DATA_FILE, 'a+', encoding='utf-8') as f:
            # 密码哈希加密后存储，用|分隔字段方便后续读取拆分，换行区分不同用户
            encrypted_pwd = generate_password_hash(password)
            f.write(f"{username}|{encrypted_pwd}\n")
        return "注册成功"
    except PermissionError:
        return "服务器存储权限不足，请检查存储目录对Flask进程用户开放读写权限", 500
    except Exception as e:
        return f"存储失败：{str(e)}", 500

if __name__ == '__main__':
    app.run()

部署上线额外注意：

• Linux环境下给存储文件所在目录开放对应权限，可执行chmod 755 你的项目根目录，单独给存储文件设置chmod 644 user_data.txt即可
• 不要将存储文件放在web可访问的静态资源目录下，避免被外部直接下载泄露数据

方案2：数据库存储（适合正式项目） #

纯文本文件存储仅适合临时测试，正式项目更推荐使用轻量数据库SQLite，无需额外部署服务，读写效率、数据安全性远高于纯文本文件，示例代码如下：

import sqlite3
from werkzeug.security import generate_password_hash, check_password_hash

# 初始化数据库，仅需运行一次
def init_db():
    conn = sqlite3.connect('users.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS users (
            username TEXT PRIMARY KEY,
            encrypted_pwd TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()

# 存储用户数据
def save_user(username, password):
    encrypted_pwd = generate_password_hash(password)
    conn = sqlite3.connect('users.db')
    cursor = conn.cursor()
    cursor.execute("INSERT INTO users VALUES (?, ?)", (username, encrypted_pwd))
    conn.commit()
    conn.close()

# 后续登录校验示例
def verify_user(username, input_password):
    conn = sqlite3.connect('users.db')
    cursor = conn.cursor()
    cursor.execute("SELECT encrypted_pwd FROM users WHERE username = ?", (username,))
    result = cursor.fetchone()
    conn.close()
    if not result:
        return False
    return check_password_hash(result[0], input_password)

在注册接口中调用save_user(username, password)即可完成数据存储，后续读取查询也比逐行读取文本文件高效很多。

内容的提问来源于stack exchange，提问作者ImPelo