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

Merge pull request #3 from completion/main

Browse Source 
API Tokens
This commit is contained in:
joshpatten 2022-03-09 21:14:57 -06:00 committed by GitHub
commit 4e251c31cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 24 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vdiclient.ini.example
View File

@ -17,6 +17,12 @@ auth_backend = pve
auth_totp = false
# If disabled, TLS certificate will not be checked
tls_verify = false
# User name
#user = user
# API Token Name
#token_name = dvi
#API Token Value
#token_value = xxx-x-x-x-xxx
[Hosts]
# Hosts are entered as `IP/FQDN = Port`

24
vdiclient.py
View File

@ -19,6 +19,9 @@ class G:
#########
title = 'VDI Login'
backend = 'pve'
user = ""
token_name = None
token_value = None
totp = False
imagefile = None
kiosk = False
@ -85,6 +88,12 @@ def loadconfig(config_location = None):
G.totp = config['Authentication'].getboolean('auth_totp')
if 'tls_verify' in config['Authentication']:
G.verify_ssl = config['Authentication'].getboolean('tls_verify')
if 'user' in config['Authentication']:
G.user = config['Authentication']['user']
if 'token_name' in config['Authentication']:
G.token_name = config['Authentication']['token_name']
if 'token_value' in config['Authentication']:
G.token_value = config['Authentication']['token_value']
if not 'Hosts' in config:
win_popup_button(f'Unable to read supplied configuration:\nNo `Hosts` section defined!', 'OK')
return False
@ -123,7 +132,7 @@ def setmainlayout():
layout.append([sg.Image(G.imagefile), sg.Text(G.title, size =(18, 1), justification='c', font=["Helvetica", 18])])
else:
layout.append([sg.Text(G.title, size =(30, 1), justification='c', font=["Helvetica", 18])])
layout.append([sg.Text("Username", size =(12, 1), font=["Helvetica", 12]), sg.InputText(key='-username-', font=["Helvetica", 12])])
layout.append([sg.Text("Username", size =(12, 1), font=["Helvetica", 12]), sg.InputText(default_text=G.user,key='-username-', font=["Helvetica", 12])])
layout.append([sg.Text("Password", size =(12, 1),font=["Helvetica", 12]), sg.InputText(key='-password-', password_char='*', font=["Helvetica", 12])])
if G.totp:
@ -148,9 +157,10 @@ def setvmlayout(vms):
layout.append([sg.Text(G.title, size =(30, 1), justification='c', font=["Helvetica", 18])])
layout.append([sg.Text('Please select a desktop instance to connect to', size =(40, 1), justification='c', font=["Helvetica", 10])])
for vm in vms:
connkeyname = f'-CONN|{vm["vmid"]}-'
layout.append([sg.Text(vm['name'], font=["Helvetica", 14]), sg.Button('Connect', font=["Helvetica", 14], key=connkeyname)])
layout.append([sg.HorizontalSeparator()])
if not vm["status"] == "unknown":
connkeyname = f'-CONN|{vm["vmid"]}-'
layout.append([sg.Text(vm['name'], font=["Helvetica", 14]), sg.Button('Connect', font=["Helvetica", 14], key=connkeyname)])
layout.append([sg.HorizontalSeparator()])
layout.append([sg.Button('Logout', font=["Helvetica", 14])])
return layout
@ -210,7 +220,7 @@ def vmaction(vmnode, vmid, vmtype):
confignode.write(inifile)
inifile.seek(0)
inistring = inifile.read()
pcmd = [G.cmd]
pcmd = [G.vvcmd]
if G.kiosk:
pcmd.append('--kiosk')
pcmd.append('--kiosk-quit')
@ -264,7 +274,9 @@ def pveauth(username, passwd, totp):
authenticated = False
if not connected and not authenticated:
try:
if totp:
if G.token_name and G.token_value:
G.proxmox = proxmoxer.ProxmoxAPI(host, user=f'{username}@{G.backend}',token_name=G.token_name,token_value=G.token_value, verify_ssl=G.verify_ssl, port=port)
elif totp:
G.proxmox = proxmoxer.ProxmoxAPI(host, user=f'{username}@{G.backend}', otp=totp, password=passwd, verify_ssl=G.verify_ssl, port=port)
else:
G.proxmox = proxmoxer.ProxmoxAPI(host, user=f'{username}@{G.backend}', password=passwd, verify_ssl=G.verify_ssl, port=port)