update tests and docs for migrate.versioning.script.*

1 files changed, 9 insertions, 3 deletions

diff --git a/migrate/versioning/script/sql.py b/migrate/versioning/script/sql.py
index 88c3bf8..851fdf2 100644
--- a/migrate/versioning/script/sql.py
+++ b/migrate/versioning/script/sql.py
@@ -1,8 +1,15 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
 from migrate.versioning.script import base
 
+
 class SqlScript(base.BaseScript):
     """A file containing plain SQL statements."""
-    def run(self, engine, step):
+
+    # TODO: why is step parameter even here?
+    def run(self, engine, step=None):
+        """Runs SQL script through raw dbapi execute call"""
         text = self.source()
         # Don't rely on SA's autocommit here
         # (SA uses .startswith to check if a commit is needed. What if script
@@ -11,14 +18,13 @@ class SqlScript(base.BaseScript):
         try:
             trans = conn.begin()
             try:
-                # ###HACK: SQLite doesn't allow multiple statements through
+                # HACK: SQLite doesn't allow multiple statements through
                 # its execute() method, but it provides executescript() instead
                 dbapi = conn.engine.raw_connection()
                 if getattr(dbapi, 'executescript', None):
                     dbapi.executescript(text)
                 else:
                     conn.execute(text)
-                # Success
                 trans.commit()
             except:
                 trans.rollback()