如何在Alembic中创建postgresql的序列

时间:2013-06-19 16:15:26

标签: sqlalchemy alembic

我正在使用alembic维护我的桌子。同时,我使用声明方式更新我的模型。

这是一个alembic的表:

op.create_table(
    'groups',
    Column('id', Integer, Sequence('group_id_seq'), primary_key=True),
    Column('name', Unicode(50)),
    Column('description', Unicode(250)),
)

模型如下:

class Group(Base):
    __tablename__ = 'groups'
    id = Column(Integer, Sequence('group_id_seq'), primary_key=True)
    name = Column(Unicode(50))
    description = Column(Unicode(250))

    def __init__(self, name, description):
        self.description = description
        self.name = name

你可以看到,我在alembic迁移和声明模型中都使用了Sequence

但我注意到在使用PostgreSQL(v9.1)时,alembic不会创建任何序列,因此模型无法创建实例,因为它们将使用nextval(<sequence name>)子句。

那么,我如何创建我的alembic迁移,以便在postgresql中真正生成序列?

4 个答案:

答案 0 :(得分:6)

只需在模型中添加以下内容:

field_seq = Sequence('groups_field_seq')
field = Column(Integer, field_seq, server_default=field_seq.next_value())

并在您的迁移文件中添加以下内容(在创建表格之前):

from sqlalchemy.schema import Sequence, CreateSequence
op.execute(CreateSequence(Sequence('groups_field_seq')))

答案 1 :(得分:4)

https://bitbucket.org/zzzeek/alembic/issue/60/autogenerate-for-sequences-as-well-as#comment-4100402

处找到提示

在上一个链接中找到CreateSequence之后,我仍然需要跳过几个环节才能使我的迁移工作在SQLite和PostgreSQL中。目前我有:

def dialect_supports_sequences():
    return op._proxy.migration_context.dialect.supports_sequences


def create_seq(name):
    if dialect_supports_sequences():
       op.execute(CreateSequence(Sequence(name)))

然后在我需要的时候拨打create_seq

这是最佳做法吗?

答案 2 :(得分:2)

不确定我的问题是否正确但是没有其他人选择回答,这就是我如何完全正常id

蒸馏器:

op.create_table('event',
    sa.Column('id', sa.INTEGER(), autoincrement=True, nullable=False),

班级:

class Event(SQLADeclarativeBase):
    __tablename__ = 'event'
    id = Column(Integer, primary_key = True)

答案 3 :(得分:1)

我最近也遇到了同样的问题,这是我如何解决的。

op.execute("create sequence SEQUENCE_NAME")

我在升级功能中运行了上面的命令,对于降级,分别在降级功能中运行了以下代码。

op.execute("drop sequence SEQUENCE_NAME")