Improving cli arguments

jscriptcoder · jscriptcoder · commit fabb2916a6af · 2020-10-14T22:25:55.000+02:00
diff --git a/gymie/__init__.py b/gymie/__init__.py
@@ -2,4 +2,4 @@
 from gymie.api import override
 from gymie.exceptions import *
 
-__version__ = '0.1.1'
+__version__ = '0.2.0'
diff --git a/gymie/__main__.py b/gymie/__main__.py
@@ -1,4 +1,10 @@
+import argparse
 from gymie.server import start
 
 if __name__ == '__main__':
-    start()
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-l', '--host', default='0.0.0.0')
+    parser.add_argument('-p', '--port', default=5000, type=int)
+    args = parser.parse_args()
+
+    start(args.host, args.port)
diff --git a/gymie/server.py b/gymie/server.py
@@ -1,6 +1,5 @@
 import json
 import eventlet
-import argparse
 from eventlet import wsgi, websocket
 from gymie.api import public
 from gymie.exceptions import *
@@ -97,11 +96,6 @@ def start(host='0.0.0.0', port=5000):
         host (str): default value '0.0.0.0'
         port (int): default value 5000
     """
-    parser = argparse.ArgumentParser()
-    parser.add_argument('-l', '--host', default=host)
-    parser.add_argument('-p', '--port', default=port, type=int)
-    args = parser.parse_args()
-    
     try:
         listener = eventlet.listen((host, port), reuse_port=False)
     except OSError as err:
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="gymie",
-    version="0.1.1",
+    version="0.2.0",
     author="Francisco Ramos",
     author_email="francisco.ramos@researchlab.ai",
     description="WebSocket server that exposes an API to train AI agents on OpenAI Gym and gym-api-like Environments",
