mirror of
https://github.com/yavook/kiwi-scp.git
synced 2024-11-23 05:13:01 +00:00
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
|
# system
|
||
|
|
import logging
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
# local
|
||
|
|
from ._subcommand import SubCommand
|
||
|
|
from .utils.dockercommand import DockerCommand
|
||
|
|
|
||
|
|
|
||
|
|
def _find_net(config, args):
|
||
|
|
ps = DockerCommand('docker').run(
|
||
|
|
config, args, ['network', 'ls', '--filter', f"name={config['network:name']}", '--format', '{{.Name}}'],
|
||
|
|
stdout=subprocess.PIPE
|
||
|
|
)
|
||
|
|
|
||
|
|
net_found = str(ps.stdout, 'utf-8').strip()
|
||
|
|
|
||
|
|
return net_found == config['network:name']
|
||
|
|
|
||
|
|
|
||
|
|
class NetUpCommand(SubCommand):
|
||
|
|
"""kiwi net-up"""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__(
|
||
|
|
'net-up',
|
||
|
|
description="Create the local network hub for this instance"
|
||
|
|
)
|
||
|
|
|
||
|
|
def run(self, runner, config, args):
|
||
|
|
if _find_net(config, args):
|
||
|
|
logging.info(f"Network '{config['network:name']}' already exists")
|
||
|
|
return
|
||
|
|
|
||
|
|
try:
|
||
|
|
DockerCommand('docker').run(
|
||
|
|
config, args,
|
||
|
|
[
|
||
|
|
'network', 'create',
|
||
|
|
'--driver', 'bridge',
|
||
|
|
'--internal',
|
||
|
|
'--subnet', config['network:cidr'],
|
||
|
|
config['network:name']
|
||
|
|
],
|
||
|
|
check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
|
||
|
|
)
|
||
|
|
logging.info(f"Network '{config['network:name']}' created")
|
||
|
|
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
logging.error(f"Error creating network '{config['network:name']}'")
|
||
|
|
|
||
|
|
|
||
|
|
class NetDownCommand(SubCommand):
|
||
|
|
"""kiwi net-down"""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__(
|
||
|
|
'net-down',
|
||
|
|
description="Remove the local network hub for this instance"
|
||
|
|
)
|
||
|
|
|
||
|
|
def run(self, runner, config, args):
|
||
|
|
if not _find_net(config, args):
|
||
|
|
logging.info(f"Network '{config['network:name']}' already removed")
|
||
|
|
return
|
||
|
|
|
||
|
|
try:
|
||
|
|
DockerCommand('docker').run(
|
||
|
|
config, args,
|
||
|
|
['network', 'rm', config['network:name']],
|
||
|
|
check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
|
||
|
|
)
|
||
|
|
logging.info(f"Network '{config['network:name']}' removed")
|
||
|
|
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
logging.error(f"Error removing network '{config['network:name']}'")
|