Flask-SQLAlchemy自增ID获取及手动添加ID方法咨询

一、获取自增ID的正确方式 #

你当前的__init__方法强制要求传入id参数，这和SQLAlchemy默认的自增主键逻辑冲突——Integer类型搭配primary_key=True时，数据库会自动生成自增ID，不需要手动传入。要正确获取自增ID，需做以下调整：

1. 修改__init__方法，让id参数可选
   把id的默认值设为None，创建用户实例时无需传id，由数据库自动生成：

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)
    hashed_password = db.Column(db.String(128), nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)
    
    confirmation_email = db.Column(db.Boolean, default=False, nullable=False) 
    reset_email_password = db.Column(db.Boolean, default=False, nullable=False)
    
    def __init__(self, username: str, email: str, plaintext_password: str, confirmation_email=False, reset_email_password=False, id: int = None): 
        self.id = id  # 传入则用手动值，否则留空由数据库生成
        self.username = username
        self.email = email
        self.hashed_password = bcrypt.hashpw(plaintext_password.encode('utf-8'), bcrypt.gensalt())
        self.confirmation_email = confirmation_email 
        self.reset_email_password = reset_email_password

2. 提交会话后获取ID
   创建实例并添加到会话后，必须执行db.session.commit()，此时数据库会生成自增ID并同步到实例对象中：

# 创建用户实例，无需传id
new_user = User(username="test_user", email="test@example.com", plaintext_password="your_password")
db.session.add(new_user)
db.session.commit()  # 提交后ID才会被赋值

print(new_user.id)  # 此时就能拿到数据库生成的自增ID

二、手动向数据库添加ID的操作 #

如果需要手动指定ID，只需在创建实例时传入合法的id值，但要注意两点：

• 传入的ID必须在数据库中唯一，否则会触发主键冲突错误；
• 若数据库自增序列当前值小于你传入的ID，后续自增ID会从传入ID的下一个值开始（不同数据库逻辑略有差异，比如MySQL会自动更新自增序列）。

示例代码：

# 手动指定ID创建用户
manual_id_user = User(id=1001, username="manual_user", email="manual@example.com", plaintext_password="your_password")
db.session.add(manual_id_user)
db.session.commit()

print(manual_id_user.id)  # 输出1001，即手动传入的ID

额外提醒 #

• 你代码里硬编码的plaintext_password = 'pojkp[kjpj[pj'是测试用的吧？实际使用要保留传入的参数值，否则密码逻辑会失效；
• 手动指定ID前，建议先查询数据库当前的自增序列值（比如MySQL用SELECT AUTO_INCREMENT FROM information_schema.TABLES WHERE TABLE_SCHEMA = '你的数据库名' AND TABLE_NAME = 'user';），避免冲突。

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