From 0933f08dc5877e7e1bb2597d8f9326a29d3e822a Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Thu, 6 Nov 2014 19:37:49 -0500 Subject: - add support for autogenerate to include "batch" --- alembic/batch.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'alembic/batch.py') diff --git a/alembic/batch.py b/alembic/batch.py index 96e12f4..26a148d 100644 --- a/alembic/batch.py +++ b/alembic/batch.py @@ -1,7 +1,8 @@ class BatchOperationsImpl(object): - def __init__(self, operations, table_name, recreate): + def __init__(self, operations, table_name, schema, recreate): self.operations = operations self.table_name = table_name + self.schema = schema self.recreate = recreate self.batch = [] @@ -33,6 +34,7 @@ class BatchOperationsImpl(object): ) def add_column(self, *arg, **kw): + # TODO: omit table and schema names from all commands self.batch.append( ("add_column", arg, kw) ) -- cgit v1.2.1