pvc/api-daemon/pvcapid/models.py

284 lines
8.3 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
# models.py - PVC Database models
# Part of the Parallel Virtual Cluster (PVC) system
#
# Copyright (C) 2018-2024 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)
2020-02-17 11:27:35 -05:00
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
2020-02-17 11:27:35 -05:00
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)
2020-02-17 11:27:35 -05:00
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=True)
2020-02-17 11:27:35 -05:00
def __init__(self, name, mac_template, ova=None):
self.name = name
self.mac_template = mac_template
2020-02-17 11:27:35 -05:00
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
)
2021-06-03 00:22:48 -04:00
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)
2020-02-17 11:27:35 -05:00
ova = db.Column(db.Integer, db.ForeignKey("ova.id"), nullable=True)
2020-02-17 11:27:35 -05:00
def __init__(self, name, ova=None):
self.name = name
2020-02-17 11:27:35 -05:00
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)
2020-02-17 11:27:35 -05:00
class DBOva(db.Model):
__tablename__ = "ova"
2020-02-17 11:27:35 -05:00
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)
2020-02-17 11:27:35 -05:00
2020-02-17 11:27:35 -05:00
class DBOvaVolume(db.Model):
__tablename__ = "ova_volume"
2020-02-17 11:27:35 -05:00
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)
2020-02-17 11:27:35 -05:00
class DBProfile(db.Model):
__tablename__ = "profile"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.Text, nullable=False, unique=True)
2020-02-17 11:27:35 -05:00
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)
2020-02-17 11:27:35 -05:00
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
2020-02-17 11:27:35 -05:00
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
2020-02-17 11:27:35 -05:00
self.ova = ova
self.arguments = arguments
def __repr__(self):
return "<id {}>".format(self.id)
2020-08-24 14:57:52 -04:00
2020-08-24 14:57:52 -04:00
class DBStorageBenchmarks(db.Model):
__tablename__ = "storage_benchmarks"
2020-08-24 14:57:52 -04:00
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")
2020-08-24 14:57:52 -04:00
result = db.Column(db.Text, nullable=False)
def __init__(self, job, result, test_format):
2020-08-24 14:57:52 -04:00
self.job = job
self.result = result
self.test_format = test_format
2020-08-24 14:57:52 -04:00
def __repr__(self):
return "<id {}>".format(self.id)