Question

当我使用sqlacodegen创建models.py时，我无法使用User.query，收到警告“[AttributeError：type object'User'没有属性'query']”。我认为db.query是一个非常有用的属性，它在SQLAlchemy（）中使用：

db = SQLAlchemy()

但我需要使用sqlacodegen为我们系统中的现有表创建模型。创建代码如下：

models.py

# coding: utf-8
from sqlalchemy import Column, Date, Float, Integer, String, Table, text
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
metadata = Base.metadata

class User(Base):
    __tablename__ = u'users'
    id = Column(Integer, primary_key=True)
    email = Column(String(64))
    username = Column(String(64))
    role_id = Column(Integer)
    password_hash = Column(String(128))

现在我导入models.py：

from models import *
@main.route('/', methods=['GET', 'POST'])
def index():
    form = NameForm()
    if form.validate_on_submit():
        user = User.query.filter_by(username=form.name.data).first()
...

当我运行时，给出的警告是：

AttributeError: type object 'User' has no attribute 'query'

我想知道如何将[类User（Base）]转换为[class User（db.Model）]类型，以便它可以使用.query属性？或者是否有一些其他可用的方法来做[类User（Base）]类型？

Answer 1

听起来你已经习惯使用Flask-SQLAlchemy了，它包含了允许你提到的功能的秘诀。

Flask Docs on SQLAlchemy表示如何获得相同的功能：

from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base

engine = create_engine('sqlite:////tmp/test.db', convert_unicode=True)
db_session = scoped_session(sessionmaker(autocommit=False,
                                     autoflush=False,
                                     bind=engine))
Base = declarative_base()
Base.query = db_session.query_property()

它会启用您期望的User.query....的简写。重要的是最后一行。您也可以使用Flask-SQLAlchemy中的db.Model类而不是纯粹的＆＃39; SQLAlchemy Base类。

如何在sqlacodegen模型中使用.query属性

1 个答案: