diff options
author | Jan Dittberner <jan.dittberner@googlemail.com> | 2011-10-28 13:41:27 +0200 |
---|---|---|
committer | Jan Dittberner <jan.dittberner@googlemail.com> | 2011-10-28 13:41:27 +0200 |
commit | e31b1cf3895aa16693aea8de73abc577a4e21918 (patch) | |
tree | b0755173c7a3fb1f01bbf28b04a1e4e3efe952ff | |
parent | bd6df33a08d219434217caf17eca05eb7f5f7c83 (diff) | |
download | sqalchemy-migrate-e31b1cf3895aa16693aea8de73abc577a4e21918.tar.gz |
generate if __name__ == "__main__" in manage.py (fixes issue 121)
-rw-r--r-- | migrate/versioning/templates/manage/default.py_tmpl | 4 | ||||
-rw-r--r-- | migrate/versioning/templates/manage/pylons.py_tmpl | 3 |
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 }}) |