2025-04-12 17:39:38 +08:00

57 lines
989 B
Python

# -*- coding: utf-8 -*-
# Copyright (c) 2020, JINGROW
# For license information, please see license.txt
import click
from backbone.hypervisor import Hypervisor, Shell
from backbone.tests import run_tests
@click.group()
def cli():
pass
@cli.group()
def hypervisor():
pass
@hypervisor.command()
def setup():
shell = Shell()
hypervisor = Hypervisor(shell=shell)
hypervisor.setup()
@hypervisor.command()
@click.option("--size", default=16384, type=int)
@click.option("--scaleway", is_flag=True)
def build(size, scaleway):
shell = Shell()
hypervisor = Hypervisor(shell=shell)
if scaleway:
hypervisor.build_scaleway(size=size)
else:
hypervisor.build(size=size)
@hypervisor.command()
def up():
shell = Shell()
hypervisor = Hypervisor(shell=shell)
hypervisor.up()
@hypervisor.command()
@click.option("-c", "--command")
def ssh(command):
shell = Shell()
hypervisor = Hypervisor(shell=shell)
hypervisor.ssh(command=command)
@cli.command()
def tests():
run_tests()