summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJan Dittberner <jan.dittberner@googlemail.com>2011-10-28 13:41:27 +0200
committerJan Dittberner <jan.dittberner@googlemail.com>2011-10-28 13:41:27 +0200
commite31b1cf3895aa16693aea8de73abc577a4e21918 (patch)
treeb0755173c7a3fb1f01bbf28b04a1e4e3efe952ff
parentbd6df33a08d219434217caf17eca05eb7f5f7c83 (diff)
downloadsqalchemy-migrate-e31b1cf3895aa16693aea8de73abc577a4e21918.tar.gz
generate if __name__ == "__main__" in manage.py (fixes issue 121)
-rw-r--r--migrate/versioning/templates/manage/default.py_tmpl4
-rw-r--r--migrate/versioning/templates/manage/pylons.py_tmpl3
2 files changed, 5 insertions, 2 deletions
diff --git a/migrate/versioning/templates/manage/default.py_tmpl b/migrate/versioning/templates/manage/default.py_tmpl
index cc4d0e8..f6d75c5 100644
--- a/migrate/versioning/templates/manage/default.py_tmpl
+++ b/migrate/versioning/templates/manage/default.py_tmpl
@@ -7,4 +7,6 @@ del _vars['__template_name__']
_vars.pop('repository_name', None)
defaults = ", ".join(["%s='%s'" % var for var in _vars.iteritems()])
}}
-main({{ defaults }})
+
+if __name__ == '__main__':
+ main({{ defaults }})
diff --git a/migrate/versioning/templates/manage/pylons.py_tmpl b/migrate/versioning/templates/manage/pylons.py_tmpl
index 475b8ce..cc2f788 100644
--- a/migrate/versioning/templates/manage/pylons.py_tmpl
+++ b/migrate/versioning/templates/manage/pylons.py_tmpl
@@ -26,4 +26,5 @@ conf_dict = ConfigLoader(conf_path).parser._sections['app:main']
# migrate supports passing url as an existing Engine instance (since 0.6.0)
# usage: migrate -c path/to/config.ini COMMANDS
-main(url=engine_from_config(conf_dict), repository=migrations.__path__[0],{{ defaults }})
+if __name__ == '__main__':
+ main(url=engine_from_config(conf_dict), repository=migrations.__path__[0],{{ defaults }})