28 lines
903 B
Python
28 lines
903 B
Python
from config import *
|
|
from xml.etree import ElementTree as ET
|
|
from flask import request, Response
|
|
import subprocess
|
|
import os
|
|
|
|
def get_vnc_port(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
vm_xml = dom.XMLDesc(0)
|
|
et_xml = ET.fromstring(vm_xml)
|
|
graphics = et_xml.find('./devices/graphics')
|
|
vnc_port = graphics.get('port')
|
|
return vnc_port
|
|
|
|
def kill_consoles():
|
|
subprocess.run("pkill -9 -f 'websockify'", shell=True)
|
|
subprocess.run("for i in $(pgrep -f 'pyxterm'); do kill -9 $i; done", shell=True)
|
|
|
|
def socket_connect(vm_name):
|
|
kill_consoles()
|
|
vm_port = get_vnc_port(vm_name)
|
|
subprocess.run(['websockify', '-D', '--web=/usr/share/novnc/', '6080', 'localhost:' + vm_port])
|
|
|
|
def pyxterm_connect(path,lxc_name):
|
|
kill_consoles()
|
|
cmd = ['python3', f'{path}/pyxterm.py', '--command', 'lxc-attach', '--cmd-args', lxc_name]
|
|
pyx_process = subprocess.Popen(cmd)
|