Use console_scripts entrypoint in preference to scripts=[] in setup.py

Ignore-this: 3795140a12885f6ceaa43d9bd9eadd4

 * Provides better interoperability with buildout

3 files changed, 9 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 9e8c5ae..439de80 100644
--- a/setup.py
+++ b/setup.py
@@ -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)