94 lines
2.1 KiB
Python
94 lines
2.1 KiB
Python
|
|
import os
|
|
import json
|
|
from random import randint
|
|
|
|
|
|
class TapDevice(object):
|
|
"""
|
|
Utility class - adds/removes a tap device on the linux system. Can be used as a context manager.
|
|
"""
|
|
def __init__(self):
|
|
self.num = randint(0, 100000)
|
|
|
|
def create(self):
|
|
os.system("ip tuntap add name {} mode tap".format(self))
|
|
|
|
def destroy(self):
|
|
os.system("ip link delete {}".format(self))
|
|
|
|
def __str__(self):
|
|
return "tap{}".format(self.num)
|
|
|
|
def __enter__(self):
|
|
self.create()
|
|
return str(self)
|
|
|
|
def __exit__(self, type, value, traceback):
|
|
self.destroy()
|
|
|
|
|
|
class Machine(object):
|
|
"""
|
|
All runnable types should subclass this
|
|
"""
|
|
def __init__(self, machine_spec):
|
|
self.spec = machine_spec
|
|
|
|
def run_machine(self):
|
|
"""
|
|
Run the machine and block until it exits (or was killed)
|
|
"""
|
|
raise NotImplemented()
|
|
|
|
def stop_machine(self):
|
|
"""
|
|
Ask the machine to stop nicely
|
|
"""
|
|
raise NotImplemented()
|
|
|
|
def kill_machine(self):
|
|
"""
|
|
Stop the machine, brutally
|
|
"""
|
|
raise NotImplemented()
|
|
|
|
def get_status(self):
|
|
"""
|
|
Get the machine's status (return one of "running" or "stopped")
|
|
"""
|
|
raise NotImplemented()
|
|
|
|
def get_datastore_path(self, datastore_name, *paths):
|
|
"""
|
|
Resolve the filesystem path for a path in the given datastore
|
|
"""
|
|
return self.spec.master.datastores.get(datastore_name).get_filepath(*paths)
|
|
|
|
|
|
class ZDisk(object):
|
|
def __init__(self, datastore, disk_id, spec):
|
|
self.datastore = datastore
|
|
self.disk_id = disk_id
|
|
self.properties = spec
|
|
self.validate()
|
|
|
|
def validate(self):
|
|
pass
|
|
|
|
def get_path(self):
|
|
return self.datastore.get_filepath(os.path.join("disks", self.disk_id))
|
|
|
|
def exists(self):
|
|
path = self.get_path()
|
|
return os.path.exists(path)
|
|
|
|
def init(self):
|
|
os.makedirs(self.get_path(), exist_ok=True)
|
|
|
|
def serialize(self):
|
|
return self.properties
|
|
|
|
def delete(self):
|
|
raise NotImplemented()
|