external/proxstar
1
0
Fork 0
mirror of https://github.com/ComputerScienceHouse/proxstar.git synced 2025-03-09 15:40:09 +00:00
Code Issues Releases Wiki Activity

Use Redis to link VNC tokens to VMID

Browse source
This commit is contained in:
Will Nilges 2022-07-31 00:12:06 -04:00
parent aa1c4834f4
commit 433c845a35
4 changed files with 22 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
proxstar/vnc.py
View file

@ -56,15 +56,22 @@ def add_vnc_target(node, port):
return token
def delete_vnc_target(node, port):
def delete_vnc_target(node=None, port=None, token=None):
targets = get_vnc_targets()
target = next((target for target in targets if target['host'] == f'{node}:{port}'), None)
if node is not None and port is not None:
target = next((target for target in targets if target['host'] == f'{node}:{port}'), None)
elif token is not None:
target = next((target for target in targets if target['token'] == f'{token}'), None)
else:
raise ValueError("Need either a node and port, or a token.")
if target:
targets.remove(target)
target_file = open(app.config['WEBSOCKIFY_TARGET_FILE'], 'w')
for target in targets:
target_file.write(f"{target['token']}: {target['host']}\n")
target_file.close()
else:
raise LookupError("Target does not exist")
def open_vnc_session(vmid, node, proxmox_user, proxmox_pass):