diff options
-rw-r--r-- | setup.py | 6 | ||||
-rw-r--r-- | yoyo/scripts/__init__.py | 0 | ||||
-rw-r--r-- | yoyo/scripts/migrate.py (renamed from scripts/yoyo-migrate) | 5 |
3 files changed, 9 insertions, 2 deletions
@@ -31,7 +31,6 @@ setup( author='Oliver Cope', author_email='oliver@redgecko.org', - scripts=['scripts/yoyo-migrate'], url='', license='BSD', packages=find_packages(exclude=['ez_setup', 'examples', 'tests']), @@ -47,4 +46,9 @@ setup( dependency_links=[ 'http://sourceforge.net/project/showfiles.php?group_id=22307' ], + entry_points= { + 'console_scripts': [ + 'yoyo-migrate=yoyo.scripts.migrate:main' + ], + } ) diff --git a/yoyo/scripts/__init__.py b/yoyo/scripts/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/yoyo/scripts/__init__.py diff --git a/scripts/yoyo-migrate b/yoyo/scripts/migrate.py index b2d9cf8..b1815bb 100644 --- a/scripts/yoyo-migrate +++ b/yoyo/scripts/migrate.py @@ -157,7 +157,10 @@ def set_logging(level): }[level] ) -def main(argv): +def main(argv=None): + + if argv is None: + argv = sys.argv[1:] optparser = make_optparser() opts, args = optparser.parse_args(argv) |