Change gridmap_web to just be an automatically created entry point.

Author: dan-blanchard

scripts/gridmap_web gridmap/web.pyscripts/gridmap_web renamed to gridmap/web.py

@@ -124,11 +124,17 @@ def job_to_html(job):
         return body_text.encode()
 
 
-def _main():
+def main(argv=None):
     """
     Parse the command line inputs and start web monitor.
+
+    :param argv: List of arguments, as if specified on the command-line.
+                 If None, ``sys.argv`` is used instead.
+    :type argv: list of str
     """
     # Get command line arguments
+    if argv is None:
+        argv = sys.argv
     parser = argparse.ArgumentParser(description="Provides a web interface to \
                                                   0MQ job monitor.")
     parser.add_argument('module_dir',
@@ -138,7 +144,7 @@ def _main():
     parser.add_argument('-p', '--port',
                         help='Port for server to listen on.', type=int,
                         default=8076)
-    args = parser.parse_args()
+    args = parser.parse_args(argv)
 
     # Make warnings from built-in warnings module get formatted more nicely
     logging.captureWarnings(True)
@@ -161,4 +167,4 @@ def _main():
 
 
 if __name__ == "__main__":
-    _main()
+    main()

setup.py

@@ -25,11 +25,18 @@
 exec(compile(open('gridmap/version.py').read(), 'gridmap/version.py', 'exec'))
 # (we use the above instead of execfile for Python 3.x compatibility)
 
+
 def readme():
     with open('README.rst') as f:
         return f.read()
 
 
+def requirements():
+    with open('requirements.txt') as f:
+        reqs = f.read().splitlines()
+    return reqs
+
+
 setup(name='gridmap',
       version=__version__,
       description=('Easily map Python functions onto a cluster using a ' +
@@ -41,8 +48,8 @@ def readme():
       author_email='dblanchard@ets.org',
       license='GPL',
       packages=['gridmap'],
-      install_requires=['drmaa', 'pyzmq'],
-      scripts=['scripts/gridmap_web'],
+      install_requires=requirements(),
+      entry_points={'console_scripts': ['gridmap_web = gridmap.web:main']},
       classifiers=['Intended Audience :: Science/Research',
                    'Intended Audience :: Developers',
                    'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',