forked from uncloud/uncloud
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
|
from uncloud.common.parser import BaseParser
|
||
|
|
from uncloud.cli.helper import make_request, get_otp_parser
|
||
|
|
|
||
|
|
|
||
|
|
class VMParser(BaseParser):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__('vm')
|
||
|
|
|
||
|
|
def start(self, **args):
|
||
|
|
p = self.subparser.add_parser('start', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
|
||
|
|
def stop(self, **args):
|
||
|
|
p = self.subparser.add_parser('stop', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
|
||
|
|
def status(self, **args):
|
||
|
|
p = self.subparser.add_parser('status', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
|
||
|
|
def delete(self, **args):
|
||
|
|
p = self.subparser.add_parser('delete', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
|
||
|
|
def migrate(self, **args):
|
||
|
|
p = self.subparser.add_parser('migrate', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
p.add_argument('--destination', required=True)
|
||
|
|
|
||
|
|
def create(self, **args):
|
||
|
|
p = self.subparser.add_parser('create', parents=[get_otp_parser()], **args)
|
||
|
|
p.add_argument('--cpu', required=True)
|
||
|
|
p.add_argument('--ram', required=True)
|
||
|
|
p.add_argument('--os-ssd', required=True)
|
||
|
|
p.add_argument('--hdd', action='append', default=list())
|
||
|
|
p.add_argument('--image', required=True)
|
||
|
|
p.add_argument('--network', action='append', default=[])
|
||
|
|
p.add_argument('--vm-name', required=True)
|
||
|
|
|
||
|
|
|
||
|
|
parser = VMParser()
|
||
|
|
arg_parser = parser.arg_parser
|
||
|
|
|
||
|
|
|
||
|
|
def main(**kwargs):
|
||
|
|
subcommand = kwargs.pop('vm_subcommand')
|
||
|
|
if not subcommand:
|
||
|
|
arg_parser.print_help()
|
||
|
|
else:
|
||
|
|
data = kwargs
|
||
|
|
endpoint = subcommand
|
||
|
|
if subcommand in ['start', 'stop', 'delete']:
|
||
|
|
endpoint = 'action'
|
||
|
|
data['action'] = subcommand
|
||
|
|
elif subcommand == 'create':
|
||
|
|
kwargs['specs'] = {
|
||
|
|
'cpu': kwargs.pop('cpu'),
|
||
|
|
'ram': kwargs.pop('ram'),
|
||
|
|
'os-ssd': kwargs.pop('os_ssd'),
|
||
|
|
'hdd': kwargs.pop('hdd')
|
||
|
|
}
|
||
|
|
make_request('vm', endpoint, data=data)
|