From: Jacob Magnusson <m@jacobian.se>
Date: Thu, 29 Oct 2015 11:14:20 +0000 (+0100)
Subject: Autogenerate renderer for op.ExecuteSQLOp
X-Git-Tag: rel_0_8_4~4^2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=6a763cd2c0348bbdd643f22def1f95fb4964ea95;p=thirdparty%2Fsqlalchemy%2Falembic.git

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 9a646410..f9c6bc77 100644
--- 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()