Refactor version command

Author: BartoszCki

paperspace/cli.py

@@ -1101,6 +1101,11 @@ def login(email, password, api_token_name):
 
 @cli.command("logout", help="Log out / remove apiKey from config file")
 def logout():
-    machines_api = client.API(config.CONFIG_HOST)
-    command = login_commands.LogOutCommand(api=machines_api)
+    command = login_commands.LogOutCommand()
+    command.execute()
+
+
+@cli.command("version", help="Show the version and exit")
+def version():
+    command = login_commands.ShowVersionCommand()
     command.execute()

paperspace/commands/login.py

@@ -1,5 +1,6 @@
 from paperspace import login, logout
 from paperspace.commands import CommandBase
+from paperspace.version import version
 
 
 class LogInCommand(CommandBase):
@@ -12,3 +13,6 @@ def execute(self):
         logout()
 
 
+class ShowVersionCommand(CommandBase):
+    def execute(self):
+        self.logger.log(version)

paperspace/main.py

@@ -8,7 +8,7 @@
 
 
 def main():
-    if len(sys.argv) >= 2 and sys.argv[1] in ('experiments', 'deployments', 'machines', 'login', 'logout'):
+    if len(sys.argv) >= 2 and sys.argv[1] in ('experiments', 'deployments', 'machines', 'login', 'logout', 'version'):
         cli(sys.argv[1:])
 
     args = sys.argv[:]
@@ -26,10 +26,6 @@ def main():
         usage(prog)
         sys.exit(0)
 
-    if cmd in ['version', '--version', '-v']:
-        vers(prog)
-        sys.exit(0)
-
     if cmd == 'apikey' or cmd == 'apiKey':
         if not args or args[0] in help_opts:
             print('usage: %s' % apikey_usage(prog))