32

我正在尝试Alembic第一次使用并想使用此处--autogenerate描述的功能

我的项目结构看起来像

project/
       configuration/
                    __init__.py
                    dev.py
                    test.py
       core/
           app/
              models/
                    __init__.py
                    user.py
       db/
          alembic/
                  versions/
                  env.py
          alembic.ini

我正在使用FlaskandSQLAlchemy和他们的Flask-SQLAlchemy扩展。我的模型User看起来像

class User(UserMixin, db.Model):
    __tablename__ = 'users'
    # noinspection PyShadowingBuiltins
    uuid = Column('uuid', GUID(), default=uuid.uuid4, primary_key=True,
                  unique=True)
    email = Column('email', String, nullable=False, unique=True)
    _password = Column('password', String, nullable=False)
    created_on = Column('created_on', sa.types.DateTime(timezone=True),
                        default=datetime.utcnow())
    last_login = Column('last_login', sa.types.DateTime(timezone=True),
                        onupdate=datetime.utcnow())

如此处所述，我修改env.py为看起来像

from configuration import app

alembic_config = config.get_section(config.config_ini_section)
alembic_config['sqlalchemy.url'] = app.config['SQLALCHEMY_DATABASE_URI']
engine = engine_from_config(
    alembic_config,
            prefix='sqlalchemy.',
            poolclass=pool.NullPool)

和

from configuration import db


target_metadata = db.metadata

configuration.__init__py看起来像哪里

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
import dev


app = Flask(__name__)
app.config.from_envvar('SETTINGS_PT')
db = SQLAlchemy(app)

现在当我运行迁移时

$alembic revision --autogenerate -m "Added user table"
INFO  [alembic.migration] Context impl PostgresqlImpl.
INFO  [alembic.migration] Will assume transactional DDL.
  Generating /Users/me/IdeaProjects/project/db/alembic/versions/55a9d5
  35d8ae_added_user_table.py...done

但文件alembic/versions/55a9d5有空upgrade()和downgrade()方法

"""Added user table

Revision ID: 1b62a62eef0d
Revises: None
Create Date: 2013-03-27 06:37:08.314177

"""

# revision identifiers, used by Alembic.
revision = '1b62a62eef0d'
down_revision = None

from alembic import op
import sqlalchemy as sa


def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    pass
    ### end Alembic commands ###


def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    pass
    ### end Alembic commands ###

怎么就看不懂有新User机型了？请帮忙

4

4 回答 4

41

根据@zzzeek，在我包含以下内容后env.py，我可以使用--autogenerate选项

在下env.py_run_migrations_online()

from configuration import app
from core.expense.models import user # added my model here

alembic_config = config.get_section(config.config_ini_section)
alembic_config['sqlalchemy.url'] = app.config['SQLALCHEMY_DATABASE_URI']
engine = engine_from_config(
    alembic_config,
    prefix='sqlalchemy.',
    poolclass=pool.NullPool)

然后我跑alembic revision --autogenerate -m "Added initial table"了

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.create_table('users',
    sa.Column('uuid', sa.GUID(), nullable=False),
    sa.Column('email', sa.String(), nullable=False),
    sa.Column('password', sa.String(), nullable=False),
    sa.Column('created_on', sa.DateTime(timezone=True), nullable=True),
    sa.Column('last_login', sa.DateTime(timezone=True), nullable=True),
    sa.PrimaryKeyConstraint('uuid'),
    sa.UniqueConstraint('email'),
    sa.UniqueConstraint('uuid')
    )
    ### end Alembic commands ###

感谢迈克尔的所有帮助！

于 2013-03-27T19:53:25.853 回答
8

我觉得这里值得指出的是，我在当前版本（0.8.4）也遇到了同样的问题，但是设置元数据的方法似乎变得更加明确了：除了导入模型，还需要设置target_metadata（其中存在env.py但默认为None)。

文档建议导入他们称为 的东西，Base但不清楚具体是什么；导入我的模型继承的 DeclarativeBase 实例对我没有任何作用（与 OP 的结果相同）。

但是，代码中的实际注释建议target_metadata使用实际模型ModelNameHere.metadata（

于 2016-02-26T04:31:57.160 回答
2

如果你不希望你的 flake8 抛出未使用的导入错误，那么你也可以将 Record 添加到__all__你的模型文件中。

在 users.py 的末尾

__all__ = Users

更多信息 - https://docs.python.org/3/tutorial/modules.html#importing-from-a-package

于 2019-10-01T05:18:17.640 回答
0

一切都简单多了，刚开始阅读文档，我就解决了自动生成迁移的问题。不要忘记 在 env.py 文件中target_metadata从 None 更改为。Base.metadata

target_metadata = Base.metadata

以及要在__init__.py模块中跟踪的所有模型的导入，这些模型包含从 Base 模型继承的所有 db 模型。然后你只需要在你的 env.py 中导入这些 Base

from app.database.models import Base

而已！

于 2021-09-07T11:16:07.553 回答