mirror of
https://github.com/FAUSheppy/athq-vm-management
synced 2025-12-06 05:41:35 +01:00
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
import libvirt
|
|
import jinja2
|
|
|
|
class VM:
|
|
|
|
environment = jinja2.Environment(loader=jinja2.FileSystemLoader(searchpath="./templates"))
|
|
|
|
def __init__(self, args):
|
|
|
|
self.hostname = args.get("hostname")
|
|
self.subdomains = args.get("subdomains")
|
|
self.ports = args.get("ports")
|
|
self.terminateSSL = args.get("terminate-ssl")
|
|
self.network = args.get("network") or "default"
|
|
self.lease = self._get_lease_for_hostname()
|
|
self.ip = self.lease.get("ipaddr")
|
|
|
|
def _get_lease_for_hostname(self):
|
|
|
|
with libvirt.open() as con:
|
|
network = con.networkLookupByName(self.network)
|
|
leases = network.DHCPLeases()
|
|
for l in leases:
|
|
if l.get("hostname") == self.hostname:
|
|
return l
|
|
|
|
raise ValueError("Hostname {} doesn't have a DHCP lease".format(self.hostname))
|
|
|
|
def dumpStreamComponents(self):
|
|
|
|
# port forwarding components #
|
|
components = []
|
|
template = self.environment.get_template("nginx_stream_block.conf.j2")
|
|
|
|
for portStruct in self.ports:
|
|
|
|
name = str(portStruct.get("name")).replace(" ", "")
|
|
portstring = str(portStruct.get("port")).replace(" ", "")
|
|
transparent = portStruct.get("transparent")
|
|
proto = portStruct.get("proto") or "tcp"
|
|
isUDP = proto == "udp"
|
|
|
|
compositeName = "-".join((self.hostname, name, portstring, proto))
|
|
|
|
component = template.render(targetip=self.ip, udp=isUDP, portstring=portstring,
|
|
transparent=transparent)
|
|
components.append(component)
|
|
|
|
return components
|
|
|
|
def dumpServerComponents(self):
|
|
|
|
# https components #
|
|
components = []
|
|
template = self.environment.get_template("nginx_server_block.conf.j2")
|
|
targetport = 80
|
|
|
|
if all([type(e) == dict for e in self.subdomains]):
|
|
for subdomain in self.subdomains:
|
|
compositeName = "-".join((self.hostname, subdomain["name"].replace(".","-")))
|
|
targetport = subdomain["port"]
|
|
component = template.render(targetip=self.ip, targetport=targetport,
|
|
servernames=[subdomain["name"]], comment=compositeName)
|
|
components.append(component)
|
|
|
|
elif any([type(e) == dict for e in self.subdomains]):
|
|
raise ValueError("Mixed subdomains not allowed - must be all complex or all simple")
|
|
else:
|
|
compositeName = "-".join((self.hostname, self.subdomains[0].replace(".","-")))
|
|
component = template.render(targetip=self.ip, targetport=targetport,
|
|
servernames=self.subdomains, comment=compositeName)
|
|
components.append(component)
|
|
|
|
return components
|
|
|