.query属性在sqlacodegen模型中的应用

Question

当我使用sqlacodegen创建models.py时，我不能使用User.query，得到警告"AttributeError:键入对象'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'

我想知道如何将类用户(基本)转换为类用户(db.Model)这样的类型，这样它就可以使用.query属性了？或者是否有其他可用的方法来使用类User(Base)类型来实现这一点？

Answer 1

听起来你已经习惯了使用烧瓶-SQLAlchemy，其中包含了允许您提到的功能的秘密调料。

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的db.Model类，而不是“纯”SQLAlchemy Base类。