- add support for autogenerate to include "batch"

2014-11-06 19:37:49 -05:00
parent 94024c2d2d
commit 0933f08dc5
5 changed files with 112 additions and 34 deletions
--- a/alembic/operations.py
+++ b/alembic/operations.py
@@ -190,8 +190,8 @@ class Operations(object):
                rel_t.append_column(sa_schema.Column(cname, NULLTYPE))

    @contextmanager
-    def batch_alter_table(self, table_name, recreate=None):
-        impl = batch.BatchOperationImpl(self, table_name, recreate)
+    def batch_alter_table(self, table_name, schema=None, recreate=None):
+        impl = batch.BatchOperationImpl(self, table_name, schema, recreate)
        batch_op = Operations(self.migration_context, impl=impl)
        yield batch_op
        impl.flush()