Autogenerate renderer for op.ExecuteSQLOp

Asserts that the passed in sqltext is a string to simplify the
implementation. Generates a op.execute(sqltext) line.

diff --git a/alembic/autogenerate/render.py b/alembic/autogenerate/render.py
index 9a64641004af708bfd84a176a50e003e37e4d33c..f9c6bc776b052e75270e56f6ae56a3932eec4500 100644 (file)
--- a/alembic/autogenerate/render.py
+++ b/alembic/autogenerate/render.py
@@ -721,4 +721,14 @@ def _render_check_constraint(constraint, autogen_context):
     }
 
 
+@renderers.dispatch_for(ops.ExecuteSQLOp)
+def _execute_sql(autogen_context, op):
+    if not isinstance(op.sqltext, string_types):
+        raise NotImplementedError(
+            "Autogenerate rendering of SQL Expression language constructs "
+            "not supported here; please use a plain SQL string"
+        )
+    return 'op.execute(%r)' % op.sqltext
+
+
 renderers = default_renderers.branch()