Commit 1c5fcb24 authored by Marco Mariani's avatar Marco Mariani

cli refactoring: supply, remove

parent cd813da7
......@@ -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",
......
# -*- 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)
# -*- 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)
......@@ -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):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment