Joshua M. Boniface
b58fa06f67
1. Ensure that system_template and script are not nullable in the DB. 2. Ensure that the CLI and API enforce the above and clean up CLI arguments for profile add. 3. Ensure that, before uploading OVAs, a 'default_ova' provisioning script is present. 4. Use the 'default_ova' script for new OVA uploads. 5. Ensure that OVA details are properly added to the vm_data dict in the provisioner vmbuilder.
284 lines
8.3 KiB
Python
Executable File
284 lines
8.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# models.py - PVC Database models
|
|
# Part of the Parallel Virtual Cluster (PVC) system
|
|
#
|
|
# Copyright (C) 2018-2021 Joshua M. Boniface <joshua@boniface.me>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
|
|
from pvcapid.flaskapi import db
|
|
|
|
|
|
class DBSystemTemplate(db.Model):
|
|
__tablename__ = "system_template"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
vcpu_count = db.Column(db.Integer, nullable=False)
|
|
vram_mb = db.Column(db.Integer, nullable=False)
|
|
serial = db.Column(db.Boolean, nullable=False)
|
|
vnc = db.Column(db.Boolean, nullable=False)
|
|
vnc_bind = db.Column(db.Text)
|
|
node_limit = db.Column(db.Text)
|
|
node_selector = db.Column(db.Text)
|
|
node_autostart = db.Column(db.Boolean, nullable=False)
|
|
migration_method = db.Column(db.Text)
|
|
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=True)
|
|
|
|
def __init__(
|
|
self,
|
|
name,
|
|
vcpu_count,
|
|
vram_mb,
|
|
serial,
|
|
vnc,
|
|
vnc_bind,
|
|
node_limit,
|
|
node_selector,
|
|
node_autostart,
|
|
migration_method,
|
|
ova=None,
|
|
):
|
|
self.name = name
|
|
self.vcpu_count = vcpu_count
|
|
self.vram_mb = vram_mb
|
|
self.serial = serial
|
|
self.vnc = vnc
|
|
self.vnc_bind = vnc_bind
|
|
self.node_limit = node_limit
|
|
self.node_selector = node_selector
|
|
self.node_autostart = node_autostart
|
|
self.migration_method = migration_method
|
|
self.ova = ova
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBNetworkTemplate(db.Model):
|
|
__tablename__ = "network_template"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
mac_template = db.Column(db.Text)
|
|
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=True)
|
|
|
|
def __init__(self, name, mac_template, ova=None):
|
|
self.name = name
|
|
self.mac_template = mac_template
|
|
self.ova = ova
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBNetworkElement(db.Model):
|
|
__tablename__ = "network"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
network_template = db.Column(
|
|
db.Integer, db.ForeignKey("network_template.id"), nullable=False
|
|
)
|
|
vni = db.Column(db.Text, nullable=False)
|
|
|
|
def __init__(self, network_template, vni):
|
|
self.network_template = network_template
|
|
self.vni = vni
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBStorageTemplate(db.Model):
|
|
__tablename__ = "storage_template"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=True)
|
|
|
|
def __init__(self, name, ova=None):
|
|
self.name = name
|
|
self.ova = ova
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBStorageElement(db.Model):
|
|
__tablename__ = "storage"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
storage_template = db.Column(
|
|
db.Integer, db.ForeignKey("storage_template.id"), nullable=False
|
|
)
|
|
pool = db.Column(db.Text, nullable=False)
|
|
disk_id = db.Column(db.Text, nullable=False)
|
|
source_volume = db.Column(db.Text)
|
|
disk_size_gb = db.Column(db.Integer)
|
|
mountpoint = db.Column(db.Text)
|
|
filesystem = db.Column(db.Text)
|
|
filesystem_args = db.Column(db.Text)
|
|
|
|
def __init__(
|
|
self,
|
|
storage_template,
|
|
pool,
|
|
disk_id,
|
|
source_volume,
|
|
disk_size_gb,
|
|
mountpoint,
|
|
filesystem,
|
|
filesystem_args,
|
|
):
|
|
self.storage_template = storage_template
|
|
self.pool = pool
|
|
self.disk_id = disk_id
|
|
self.source_volume = source_volume
|
|
self.disk_size_gb = disk_size_gb
|
|
self.mountpoint = mountpoint
|
|
self.filesystem = filesystem
|
|
self.filesystem_args = filesystem_args
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBUserdata(db.Model):
|
|
__tablename__ = "userdata"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
userdata = db.Column(db.Text, nullable=False)
|
|
|
|
def __init__(self, name, userdata):
|
|
self.name = name
|
|
self.userdata = userdata
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBScript(db.Model):
|
|
__tablename__ = "script"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
script = db.Column(db.Text, nullable=False)
|
|
|
|
def __init__(self, name, script):
|
|
self.name = name
|
|
self.script = script
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBOva(db.Model):
|
|
__tablename__ = "ova"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
ovf = db.Column(db.Text, nullable=False)
|
|
|
|
def __init__(self, name, ovf):
|
|
self.name = name
|
|
self.ovf = ovf
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBOvaVolume(db.Model):
|
|
__tablename__ = "ova_volume"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=False)
|
|
pool = db.Column(db.Text, nullable=False)
|
|
volume_name = db.Column(db.Text, nullable=False)
|
|
volume_format = db.Column(db.Text, nullable=False)
|
|
disk_id = db.Column(db.Text, nullable=False)
|
|
disk_size_gb = db.Column(db.Integer, nullable=False)
|
|
|
|
def __init__(self, ova, pool, volume_name, volume_format, disk_id, disk_size_gb):
|
|
self.ova = ova
|
|
self.pool = pool
|
|
self.volume_name = volume_name
|
|
self.volume_format = volume_format
|
|
self.disk_id = disk_id
|
|
self.disk_size_gb = disk_size_gb
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBProfile(db.Model):
|
|
__tablename__ = "profile"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.Text, nullable=False, unique=True)
|
|
profile_type = db.Column(db.Text, nullable=False)
|
|
system_template = db.Column(
|
|
db.Integer, db.ForeignKey("system_template.id"), nullable=False
|
|
)
|
|
network_template = db.Column(db.Integer, db.ForeignKey("network_template.id"))
|
|
storage_template = db.Column(db.Integer, db.ForeignKey("storage_template.id"))
|
|
userdata = db.Column(db.Integer, db.ForeignKey("userdata.id"))
|
|
script = db.Column(db.Integer, db.ForeignKey("script.id"), nullable=False)
|
|
ova = db.Column(db.Integer, db.ForeignKey("ova.id"))
|
|
arguments = db.Column(db.Text)
|
|
|
|
def __init__(
|
|
self,
|
|
name,
|
|
profile_type,
|
|
system_template,
|
|
network_template,
|
|
storage_template,
|
|
userdata,
|
|
script,
|
|
ova,
|
|
arguments,
|
|
):
|
|
self.name = name
|
|
self.profile_type = profile_type
|
|
self.system_template = system_template
|
|
self.network_template = network_template
|
|
self.storage_template = storage_template
|
|
self.userdata = userdata
|
|
self.script = script
|
|
self.ova = ova
|
|
self.arguments = arguments
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|
|
|
|
|
|
class DBStorageBenchmarks(db.Model):
|
|
__tablename__ = "storage_benchmarks"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
job = db.Column(db.Text, nullable=False)
|
|
test_format = db.Column(db.Integer, nullable=False, default=0, server_default="0")
|
|
result = db.Column(db.Text, nullable=False)
|
|
|
|
def __init__(self, job, result, test_format):
|
|
self.job = job
|
|
self.result = result
|
|
self.test_format = test_format
|
|
|
|
def __repr__(self):
|
|
return "<id {}>".format(self.id)
|