You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from sqlalchemy import Column, Integer, String, ForeignKey, Table
|
|
from sqlalchemy.orm import relationship, backref
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from utils.config.config import ServerConfig
|
|
|
|
config = ServerConfig()
|
|
engine = create_engine(f"sqlite:///{config.database_file}")
|
|
Base = declarative_base(bind=engine)
|
|
Base.metadata.create_all()
|
|
|
|
client_image_table = Table(
|
|
"client_image",
|
|
Base.metadata,
|
|
Column("client_mac", String, ForeignKey("clients.mac_address")),
|
|
Column("image_id", Integer, ForeignKey("vm_images.image_id"))
|
|
)
|
|
|
|
|
|
class Client(Base):
|
|
__tablename__ = "clients"
|
|
mac_address = Column(String, primary_key=True)
|
|
ip_address = Column(String(16), nullable=False)
|
|
hostname = Column(String(100), nullable=False)
|
|
client_version = Column(String(100), nullable=False)
|
|
vm_list_on_machine = relationship(
|
|
"VMImage",
|
|
secondary=client_image_table,
|
|
)
|
|
|
|
def has_vm_installed(self, vm_object):
|
|
for vm in self.vm_list_on_machine:
|
|
if vm.image_hash == vm_object.image_hash:
|
|
return True
|
|
return False
|
|
|
|
|
|
class VMImage(Base):
|
|
__tablename__ = "vm_images"
|
|
image_id = Column(Integer, primary_key=True)
|
|
image_name = Column(String(100), unique=True, nullable=False)
|
|
image_file = Column(String(500), unique=False, nullable=False)
|
|
image_version = Column(String(100), nullable=False)
|
|
image_hash = Column(String(500), nullable=False)
|
|
clients = relationship(
|
|
"Client",
|
|
secondary=client_image_table
|
|
)
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
user_id = Column(Integer, primary_key=True, autoincrement=True)
|
|
username = Column(String)
|
|
password_hash = Column(String)
|