Add password reset command

Update vdiclient.py

README.md

@@ -10,7 +10,7 @@ This project's focus is to create a simple VDI client intended for mass deployme
 
 ## Windows Installation
 
-You **MUST** install virt-viewer prior to using PVE VDI client, you may download it from the [official Virtual Machine Manager](https://virt-manager.org/download) site.
+You **MUST** install virt-viewer prior to using PVE VDI client, you may download it from the [official Virtual Machine Manager](https://virt-manager.org/download.html) site.
 
 Please visit the [releases](https://github.com/joshpatten/PVE-VDIClient/releases) section to download a prebuilt MSI package
 

dist/vdiclient.json

@@ -1,6 +1,6 @@
 {
 	"upgrade_guid" : "46cbad92-353e-4b28-9bee-83950991dad8",
-	"version" : "1.2.01",
+	"version" : "1.2.05",
 	"product_name" : "VDI Client",
 	"manufacturer" : "Josh Patten",
 	"name" : "VDI Client",

vdiclient.ini.example

@@ -17,6 +17,9 @@ inidebug = False
 guest_type = both
 # Show VM option for resetting VM
 #show_reset = True
+# Set Window Dimensions. Only use if window isn't sizing properly
+#window_width = 800
+#window_height = 600
 
 [Authentication]
 # This is the authentication backend that will be used to authenticate
@@ -31,6 +34,8 @@ tls_verify = false
 #token_name = dvi
 # API Token Value
 #token_value = xxx-x-x-x-xxx
+# Password Reset Command Launch. Has to be full executable Command
+#pwresetcmd = start chrome --app=http://pwreset.example.com
 
 [Hosts]
 # Hosts are entered as `IP/FQDN = Port`

vdiclient.py

@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+#!/usr/bin/env python3
 import proxmoxer # pip install proxmoxer
 import PySimpleGUI as sg # pip install PySimpleGUI
 gui = 'TK'
@@ -39,6 +39,9 @@ class G:
 	addl_params = None
 	theme = 'LightBlue'
 	guest_type = 'both'
+	width = None
+	height = None
+	pwresetcmd = None
 
 def loadconfig(config_location = None):
 	if config_location:
@@ -100,6 +103,10 @@ def loadconfig(config_location = None):
 			G.guest_type = config['General']['guest_type']
 		if 'show_reset' in config['General']:
 			G.show_reset = config['General'].getboolean('show_reset')
+		if 'window_width' in config['General']:
+			G.width = config['General'].getint('window_width')
+		if 'window_height' in config['General']:
+			G.height = config['General'].getint('window_height')
 	if not 'Authentication' in config:
 		win_popup_button(f'Unable to read supplied configuration:\nNo `Authentication` section defined!', 'OK')
 		return False
@@ -111,11 +118,13 @@ def loadconfig(config_location = None):
 		if 'tls_verify' in config['Authentication']:
 			G.verify_ssl = config['Authentication'].getboolean('tls_verify')
 		if 'user' in config['Authentication']:
-				G.user = config['Authentication']['user']
+			G.user = config['Authentication']['user']
 		if 'token_name' in config['Authentication']:
-				G.token_name = config['Authentication']['token_name']
+			G.token_name = config['Authentication']['token_name']
 		if 'token_value' in config['Authentication']:
-				G.token_value = config['Authentication']['token_value']
+			G.token_value = config['Authentication']['token_value']
+		if 'pwresetcmd' in config['Authentication']:
+			G.pwresetcmd = config['Authentication']['pwresetcmd']
 	if not 'Hosts' in config:
 		win_popup_button(f'Unable to read supplied configuration:\nNo `Hosts` section defined!', 'OK')
 		return False
@@ -174,6 +183,8 @@ def setmainlayout():
 		layout.append([sg.Button("Log In", font=["Helvetica", 14], bind_return_key=True)])
 	else:
 		layout.append([sg.Button("Log In", font=["Helvetica", 14], bind_return_key=True), sg.Button("Cancel", font=["Helvetica", 14])])
+	if G.pwresetcmd:
+		layout[-1].append(sg.Button('Password Reset', font=["Helvetica", 14]))
 	return layout
 
 def getvms(listonly = False):
@@ -204,6 +215,9 @@ def getvms(listonly = False):
 	except proxmoxer.core.ResourceException as e:
 		win_popup_button(f"Unable to display list of VMs:\n {e!r}", 'OK')
 		return False
+	except requests.exceptions.ConnectionError as e:
+		print(f"Encountered error when querying proxmox: {e!r}")
+		return False
 
 def setvmlayout(vms):
 	layout = []
@@ -215,7 +229,6 @@ def setvmlayout(vms):
 	layoutcolumn = []
 	for vm in vms:
 		if not vm["status"] == "unknown":
-			print(vm)
 			vmkeyname = f'-VM|{vm["vmid"]}-'
 			connkeyname = f'-CONN|{vm["vmid"]}-'
 			resetkeyname = f'-RESET|{vm["vmid"]}-'
@@ -247,7 +260,7 @@ def setvmlayout(vms):
 			layoutcolumn.append(tmplayout)
 			layoutcolumn.append([sg.HorizontalSeparator()])
 	if len(vms) > 5: # We need a scrollbar
-		layout.append([sg.Column(layoutcolumn, scrollable = True, size = [450*G.scaling, None] )])
+		layout.append([sg.Column(layoutcolumn, scrollable = True, size = [None, None] )])
 	else:
 		for row in layoutcolumn:
 			layout.append(row)
@@ -475,6 +488,11 @@ def loginwindow():
 			if event == 'Cancel' or event == sg.WIN_CLOSED:
 				window.close()
 				return False
+			elif event == 'Password Reset':
+				try:
+					subprocess.check_call(G.pwresetcmd, shell=True)
+				except Exception as e:
+					win_popup_button(f'Unable to open password reset command.\n\nError Info:\n{e}', 'OK')
 			else:
 				if event in ('Log In', '\r', 'special 16777220', 'special 16777221'):
 					popwin = win_popup("Please wait, authenticating...")
@@ -505,45 +523,47 @@ def showvms():
 		win_popup_button('No desktop instances found, please consult with your system administrator', 'OK')
 		return False
 	layout = setvmlayout(vms)
-
 	if G.icon:
-		window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, no_titlebar=G.kiosk, icon=G.icon)
+		window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, no_titlebar=G.kiosk, size=(G.width, G.height), icon=G.icon)
 	else:
-		window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, no_titlebar=G.kiosk)
+		window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, size=(G.width, G.height), no_titlebar=G.kiosk)
 	timer = datetime.now()
 	while True:
 		if (datetime.now() - timer).total_seconds() > 5:
 			timer = datetime.now()
 			newvmlist = getvms(listonly = True)
-			if vmlist != newvmlist:
-				vmlist = newvmlist.copy()
-				vms = getvms()
-				layout = setvmlayout(vms)
-				window.close()
-				if G.icon:
-					window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, no_titlebar=G.kiosk, icon=G.icon)
-				else:
-					window = sg.Window(G.title, layout, return_keyboard_events=True,finalize=True, resizable=False, no_titlebar=G.kiosk)
-				window.bring_to_front()
-			else: # Refresh existing vm status
-				newvms = getvms()
-				for vm in newvms:
-					vmkeyname = f'-VM|{vm["vmid"]}-'
-					connkeyname = f'-CONN|{vm["vmid"]}-'
-					state = 'stopped'
-					if vm['status'] == 'running':
-						if 'lock' in vm:
-							state = vm['lock']
-							if state in ('suspending', 'suspended'):
-								window[connkeyname].update(disabled=True)
-								if state == 'suspended':
-									state = 'starting'
+			if newvmlist:
+				if vmlist != newvmlist:
+					vmlist = newvmlist.copy()
+					vms = getvms()
+					if vms:
+						layout = setvmlayout(vms)
+						window.close()
+						if G.icon:
+							window = sg.Window(G.title, layout, return_keyboard_events=True, finalize=True, resizable=False, no_titlebar=G.kiosk, size=(G.width, G.height), icon=G.icon)
 						else:
-							state = vm['status']
-							window[connkeyname].update(disabled=False)
-					else:
-						window[connkeyname].update(disabled=False)
-					window[vmkeyname].update(f"State: {state}")
+							window = sg.Window(G.title, layout, return_keyboard_events=True,finalize=True, resizable=False, no_titlebar=G.kiosk, size=(G.width, G.height))
+					window.bring_to_front()
+				else: # Refresh existing vm status
+					newvms = getvms()
+					if newvms:
+						for vm in newvms:
+							vmkeyname = f'-VM|{vm["vmid"]}-'
+							connkeyname = f'-CONN|{vm["vmid"]}-'
+							state = 'stopped'
+							if vm['status'] == 'running':
+								if 'lock' in vm:
+									state = vm['lock']
+									if state in ('suspending', 'suspended'):
+										window[connkeyname].update(disabled=True)
+										if state == 'suspended':
+											state = 'starting'
+								else:
+									state = vm['status']
+									window[connkeyname].update(disabled=False)
+							else:
+								window[connkeyname].update(disabled=False)
+							window[vmkeyname].update(f"State: {state}")
 
 		event, values = window.read(timeout = 1000)
 		if event in ('Logout', None):