我使用 SQLAlchemy 作为我的 ORM 层来访问我的 MySQL 数据库。
这是表格,映射器和模型,我正在谈论:
CREATE TABLE user (
uid INT(12) NOT NULL auto_increment,
email VARCHAR(128) NOT NULL,
password VARCHAR(128) NOT NULL,
display_name VARCHAR(128) NOT NULL,
timezone VARCHAR(96) NOT NULL,
join_date DATETIME NOT NULL,
last_login DATETIME NOT NULL,
verified INT(1) NOT NULL DEFAULT 0,
PRIMARY KEY (uid),
UNIQUE (email)
) ENGINE=InnoDB;
class User(object):
def __init__(self, id=0, email="", password="", displayName="",
timezone="UTC", joinDate=None, lastLogin=None, verified=False):
self.id = id
self.email = email
self.password = password
self.displayName = displayName
self.timezone = timezone
self.joinDate = joinDate or datetime.utcnow()
self.lastLogin = lastLogin or datetime.utcnow()
self.verified = verified
userTable = sqlalchemy.Table('user', meta, autoload=True)
sqlalchemy.orm.mapper(models.User, userTable, properties={
'id': userTable.c.uid,
'displayName': userTable.c.display_name,
'joinDate': userTable.c.join_date,
'lastLogin': userTable.c.last_login
}
问题是,每当我尝试通过 orm 层将对象插入数据库时,我希望新生成的 ID 会保存在原始对象中以供将来参考。否则这整件事对我来说毫无用处。
>>> session = Session()
>>> u = User(0, 'test@example.com', 'asdf', 'Test User')
>>> session.add(u)
>>> session.flush()
>>> u.id
0
有谁知道如何告诉 SQLALchemy 将 ID 保存到原始对象中?谢谢你。