thomas/vdiclient
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ability to auto-connect VMID

Browse Source
This commit is contained in:
jpattWPC 2023-10-13 14:22:08 -05:00
parent 2bac614194
commit cd83be7680
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vdiclient.ini.example
View File

@ -36,6 +36,8 @@ tls_verify = false
#token_value = xxx-x-x-x-xxx #token_value = xxx-x-x-x-xxx
# Password Reset Command Launch. Has to be full executable Command # Password Reset Command Launch. Has to be full executable Command
#pwresetcmd = start chrome --app=http://pwreset.example.com #pwresetcmd = start chrome --app=http://pwreset.example.com
# Automatically connect to a VMID upon authentication
#auto_vmid = 100
[Hosts] [Hosts]
# Hosts are entered as `IP/FQDN = Port` # Hosts are entered as `IP/FQDN = Port`

10
vdiclient.py
View File

@ -42,6 +42,7 @@ class G:
width = None width = None
height = None height = None
pwresetcmd = None pwresetcmd = None
auto_vmid = None
def loadconfig(config_location = None): def loadconfig(config_location = None):
if config_location: if config_location:
@ -125,6 +126,8 @@ def loadconfig(config_location = None):
G.token_value = config['Authentication']['token_value'] G.token_value = config['Authentication']['token_value']
if 'pwresetcmd' in config['Authentication']: if 'pwresetcmd' in config['Authentication']:
G.pwresetcmd = config['Authentication']['pwresetcmd'] G.pwresetcmd = config['Authentication']['pwresetcmd']
if 'auto_vmid' in config['Authentication']:
G.auto_vmid = config['Authentication'].getint('auto_vmid')
if not 'Hosts' in config: if not 'Hosts' in config:
win_popup_button(f'Unable to read supplied configuration:\nNo `Hosts` section defined!', 'OK') win_popup_button(f'Unable to read supplied configuration:\nNo `Hosts` section defined!', 'OK')
return False return False
@ -617,6 +620,13 @@ def main():
return 1 return 1
break break
else: else:
if G.auto_vmid:
vms = getvms()
for row in vms:
if row['vmid'] == G.auto_vmid:
vmaction(row['node'], row['vmid'], row['type'], action='connect')
return 0
win_popup_button(f'No VDI instance with ID {G.auto_vmid} found!', 'OK')
vmstat = showvms() vmstat = showvms()
if not vmstat: if not vmstat:
G.proxmox = None G.proxmox = None