cli refactoring: supply, remove

setup.py

@@ -78,6 +78,8 @@ setup(name=name,
           'node format = slapos.cli.format:FormatCommand',
           'node register = slapos.cli.register:RegisterCommand',
           'console = slapos.cli.console:ConsoleCommand',
+          'supply = slapos.cli.supply:SupplyCommand',
+          'remove = slapos.cli.remove:RemoveCommand',
         ]
       },
       test_suite="slapos.tests",

slapos/cli/remove.py

+# -*- coding: utf-8 -*-
+
+import logging
+
+from slapos.cli.config import ClientConfigCommand
+from slapos.client import init, do_remove, ClientConfig
+
+
+class RemoveCommand(ClientConfigCommand):
+
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        ap = super(RemoveCommand, self).get_parser(prog_name)
+
+        ap.add_argument('software_url',
+                        help='Your software url')
+
+        ap.add_argument('none',
+                        help="Target node")
+
+        return ap
+
+    def take_action(self, args):
+        configuration_parser = self.fetch_config(args)
+        config = ClientConfig(args, configuration_parser)
+        local = init(config)
+        do_remove(args.software_url, args.node, local)

slapos/cli/supply.py

+# -*- coding: utf-8 -*-
+
+import logging
+
+from slapos.cli.config import ClientConfigCommand
+from slapos.client import init, do_supply, ClientConfig
+
+
+class SupplyCommand(ClientConfigCommand):
+
+    log = logging.getLogger(__name__)
+
+    def get_parser(self, prog_name):
+        ap = super(SupplyCommand, self).get_parser(prog_name)
+
+        ap.add_argument('software_url',
+                        help='Your software url')
+
+        ap.add_argument('node',
+                        help="Target node")
+
+        return ap
+
+    def take_action(self, args):
+        configuration_parser = self.fetch_config(args)
+        config = ClientConfig(args, configuration_parser)
+        local = init(config)
+        do_supply(args.software_url, args.node, local)

slapos/client.py

@@ -220,7 +220,7 @@ def request():
         "couple of minutes to get connection informations.")
     exit(2)
 
-def _supply(software_url, computer_id, local, remove=False):
+def do_supply(software_url, computer_id, local, remove=False):
   """
   Request installation of Software Release
   'software_url' on computer 'computer_id'.
@@ -245,6 +245,9 @@ def _supply(software_url, computer_id, local, remove=False):
   )
   print 'Done.'
 
+def do_remove(software_url, node, local):
+  do_supply(software_url, node, local, remove=True)
+
 def supply():
   """
   Run when invoking slapos supply. Mostly argument parsing.
@@ -260,7 +263,8 @@ def supply():
   args = parser.parse_args()
 
   config = ClientConfig(args, get_config_parser(args.configuration_file))
-  _supply(args.software_url, args.node, init(config))
+  local = init(config)
+  do_supply(args.software_url, args.node, local)
 
 def remove():
   """
@@ -277,7 +281,8 @@ def remove():
   args = parser.parse_args()
 
   config = ClientConfig(args, get_config_parser(args.configuration_file))
-  _supply(args.software_url, args.node, init(config), remove=True)
+  local = init(config)
+  do_remove(args.software_url, args.node, local)
 
 
 def do_console(local):