Return the script object from command.revision

diff --git a/alembic/command.py b/alembic/command.py
index a44788da61e5c6eaf09b11d32b364370d5583adf..f1c5962111b1d376a685c7cca7e44d3b95542a4e 100644 (file)
--- a/alembic/command.py
+++ b/alembic/command.py
@@ -95,7 +95,8 @@ def revision(config, message=None, autogenerate=False, sql=False):
             template_args=template_args,
         ):
             script.run_env()
-    script.generate_revision(util.rev_id(), message, **template_args)
+    return script.generate_revision(util.rev_id(), message, refresh=True,
+                                    **template_args)
 
 
 def upgrade(config, revision, sql=False, tag=None):