Alembic / Flask-Migrate未检测到after_create事件

时间:2016-04-06 15:27:09

标签: flask sqlalchemy flask-sqlalchemy alembic flask-migrate

我有一个简单的Flask-SQLAlchemy模型(使用事件监听器来创建触发器):

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class Confirm(db.Model):
  created = db.Column(db.DateTime, default=db.func.current_timestamp(), nullable=False)
  modified = db.Column(db.DateTime, default=db.func.current_timestamp(), onupdate=db.func.current_timestamp(), nullable=False)
  id = db.Column(db.String(36), primary_key=True) 

class ConfirmOld(db.Model):
  orig_created = db.Column(db.DateTime)
  orig_modified = db.Column(db.DateTime)
  orig_id = db.Column(db.String(36))

confirm_delete = DDL('''\
CREATE TRIGGER confirm_delete 
BEFORE DELETE
  ON confirm FOR EACH ROW
  BEGIN
    INSERT INTO confirm_old ( orig_created, orig_modified, orig_id )
    VALUES ( OLD.created, OLD.modified, OLD.id );
  END;
''')

event.listen(Confirm.__table__, 'after_create', confirm_delete)

当我运行Alembic迁移和升级时,未创建TRIGGER(在MySQL中)。但是,当我使用db.create_all()时, 创建并正常工作。

是否有可能让Alembic / Flask-Migrate创建和管理我的触发器(即在after_create事件上运行的自定义DDL)?

1 个答案:

答案 0 :(得分:0)

我也遇到过相同的问题,尝试使用Replacable对象解决方案,但是没有用:

我设法通过编辑迁移脚本并执行触发器创建查询来使其工作。

这是步骤:

  • 运行flask db migrate -m 'adding custom trigger on table x,它将在迁移文件夹的版本子文件夹下为您生成一个迁移脚本。

  • 检查版本下创建的文件夹,并按以下方式对其进行编辑:

像这样创建触发器查询:

在文件中:

trigger = '''
CREATE TRIGGER confirm_delete
BEFORE DELETE
ON confirm FOR EACH ROW
BEGIN
INSERT INTO confirm_old ( orig_created, orig_modified, orig_id )
VALUES ( OLD.created, OLD.modified, OLD.id );
END;
'''

在升级方法中:

添加此行:

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

    # ### end Alembic commands ###

    ### add your queries here execute
    op.execute(trigger)

如果您运行flask db upgrade,它将执行查询并更新数据库

要降级数据库,请将其添加到降级方法中:

def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    # ### end Alembic commands ###
    op.execute('drop trigger if exists confirm_delete on confirm cascade;')

如果您检查您的数据库更改将被应用。

PS:更优雅的解决方案应该是建议here

使用Replaceable对象,尝试了一下,但是没有用,可能是我的Alembic没有更新。

这是解决方案的外观:

创建一个ReplaceableObjects类:

class ReplaceableObject(object):
    def __init__(self, name, sqltext):
        self.name = name
        self.sqltext = sqltext

使用您的查询语句实例化它。

delete_trigger = ReplaceableObject('delete_trigger', trigger)

像这样更新升级和降级功能:

def upgrade():
    op.create_sp(delete_trigger)


def downgrade():
    op.drop_sp(delete_trigger)

希望它将对其他人有帮助...