Switch to setuptools entry_points

3 files changed, 12 insertions, 2 deletions

diff --git a/gemato/cli.py b/gemato/cli.py
index d04b2d8..eae70fb 100644
--- a/gemato/cli.py
+++ b/gemato/cli.py
@@ -10,6 +10,7 @@ import datetime
 import io
 import logging
 import os.path
+import sys
 import timeit
 
 import gemato.find_top_level
@@ -320,3 +321,8 @@ def main(argv):
     if not hasattr(vals, 'func'):
         argp.error('No function specified')
     return vals.func(vals, argp)
+
+
+def setuptools_main():
+    logging.getLogger().setLevel(logging.INFO)
+    sys.exit(main(sys.argv))
diff --git a/gemato/recursiveloader.py b/gemato/recursiveloader.py
index c9a6380..5ef8738 100644
--- a/gemato/recursiveloader.py
+++ b/gemato/recursiveloader.py
@@ -635,7 +635,7 @@ class ManifestRecursiveLoader(object):
                 ret = all(pool.imap_unordered(verifier, _walk_directory(it),
                                    chunksize=64))
             else:
-                # in py<3.5 imap() swallows exceptions, so fall back
+                # in py<3.5.4 imap() swallows exceptions, so fall back
                 # to regular map() [it's only a little slower]
                 ret = all(pool.map(verifier, _walk_directory(it),
                                    chunksize=64))
diff --git a/setup.py b/setup.py
index aaef46e..78e9484 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,11 @@ setup(
     },
 
     packages=['gemato'],
-    scripts=['bin/gemato'],
+    entry_points={
+        'console_scripts': [
+            'gemato=gemato.cli:setuptools_main',
+        ],
+    },
 
     classifiers=[
         'Development Status :: 4 - Beta',