Merge pull request #23 from napalm-automation/fix-script

Fix script

napalm_logs/scripts/main.py

@@ -100,6 +100,7 @@ def prepare(self):
         self.add_option(
             '-p', '--port',
             dest='port',
+            type=int,
             help=('Listener bind port. Default: {0}'.format(defaults.PORT))
         )
         self.add_option(
@@ -116,6 +117,7 @@ def prepare(self):
         self.add_option(
             '--publish-port',
             dest='publish_port',
+            type=int,
             help=('Publisher bind port. Default: {0}'.format(defaults.PUBLISH_PORT))
         )
         self.add_option(
@@ -180,7 +182,7 @@ def parse(self, screen_handler):
         return cfg
 
 
-if __name__ == '__main__':
+def main():
     if '' in sys.path:
         sys.path.remove('')
     # Temporarily will forward the log entries to the screen
@@ -193,3 +195,7 @@ def parse(self, screen_handler):
     config = nlop.parse(screen)
     nl = napalm_logs.NapalmLogs(**config)
     nl.start_engine()
+
+if __name__ == '__main__':
+    main()
+

setup.py

@@ -36,7 +36,7 @@
     install_requires=reqs,
     entry_points={
         'console_scripts': [
-            'napalm-logs=napalm_logs.scripts.main'
+            'napalm-logs=napalm_logs.scripts.main:main'
         ],
     }
 )