2024-05-26 14:47:45 -06:00
|
|
|
#!/bin/env python
|
|
|
|
|
2024-05-26 16:58:18 -06:00
|
|
|
import os
|
2024-05-26 20:49:53 -06:00
|
|
|
import psutil
|
2024-05-26 14:47:45 -06:00
|
|
|
import PySimpleGUI as sg
|
2024-05-26 20:49:53 -06:00
|
|
|
import subprocess
|
2024-05-26 21:20:13 -06:00
|
|
|
import sys
|
2024-05-26 20:49:53 -06:00
|
|
|
import time
|
2024-05-26 14:47:45 -06:00
|
|
|
|
|
|
|
"""
|
|
|
|
Pismotek-Remote - Remote Desktop Application
|
|
|
|
|
|
|
|
To "swap out" a portion of a window, use a Column element for that portion. Add multiple Columns
|
|
|
|
on the same row and make only 1 of them active at a time
|
2024-05-26 21:59:54 -06:00
|
|
|
|
|
|
|
Will determine if tvnserver is a running service or application
|
|
|
|
If it is not running it will be run in application mode
|
2024-05-26 14:47:45 -06:00
|
|
|
"""
|
|
|
|
|
2024-05-26 21:20:13 -06:00
|
|
|
# ----------- Utility functions -----------
|
2024-05-26 16:58:18 -06:00
|
|
|
def resource_path(relative_path):
|
|
|
|
""" Get absolute path to resource, works for dev and for PyInstaller """
|
|
|
|
try:
|
|
|
|
base_path = sys._MEIPASS
|
|
|
|
except Exception:
|
|
|
|
base_path = os.path.abspath(".")
|
|
|
|
return os.path.join(base_path, relative_path)
|
|
|
|
|
2024-05-26 21:20:13 -06:00
|
|
|
def get_service(name):
|
|
|
|
""" return information about services """
|
|
|
|
service = None
|
|
|
|
try:
|
|
|
|
service = psutil.win_service_get(name)
|
|
|
|
service = service.as_dict()
|
|
|
|
except Exception as ex:
|
|
|
|
pass
|
|
|
|
return service
|
|
|
|
|
|
|
|
def build_layout():
|
|
|
|
logo = resource_path("logo.png")
|
|
|
|
# ----------- Create the 3 layouts this Window will display -----------
|
|
|
|
layout1 = [
|
|
|
|
[sg.Push(), sg.Image(logo), sg.Push()],
|
|
|
|
[sg.Text('This program allows Will B. to remotely control your')],
|
|
|
|
[sg.Text('computer through the internet')],
|
|
|
|
[sg.Text('Is Will waiting for you - *right now* - to run this program?')],
|
|
|
|
]
|
|
|
|
|
|
|
|
layout2 = [
|
|
|
|
[sg.Text('Remote connection started')],
|
|
|
|
[sg.Text('Allow connection through firewall if asked.')],
|
|
|
|
[sg.Text('')],
|
|
|
|
[sg.Text('Click Exit when finished')],
|
|
|
|
]
|
|
|
|
|
|
|
|
layout3 = [
|
|
|
|
[sg.Text('Please run this program ONLY when Will is waiting for you')],
|
|
|
|
[sg.Text('Click Exit to continue')],
|
|
|
|
]
|
|
|
|
|
|
|
|
# ----------- Create actual layout using Columns and a row of Buttons
|
|
|
|
layout = [
|
|
|
|
[
|
|
|
|
sg.Column(layout1, key='-COL1-'),
|
|
|
|
sg.Column(layout2, visible=False, key='-COL2-'),
|
|
|
|
sg.Column(layout3, visible=False, key='-COL3-')
|
|
|
|
],
|
|
|
|
[sg.VPush()],
|
|
|
|
[sg.Push(), sg.Button('No'), sg.Button('Yes'), sg.Button('Exit', visible=False)]
|
|
|
|
|
|
|
|
]
|
|
|
|
return layout
|
2024-05-26 14:47:45 -06:00
|
|
|
|
2024-05-26 20:49:53 -06:00
|
|
|
|
|
|
|
def vncdisconnect():
|
|
|
|
print("Starting TightVNC")
|
|
|
|
subprocess.call("tvnserver.exe -run", shell=True)
|
|
|
|
time.sleep(3)
|
|
|
|
print("Starting Disconnecting TightVNC")
|
|
|
|
subprocess.call("tvnserver.exe -controlservice -disconnectall", shell=True)
|
|
|
|
print("Did we get an error")
|
|
|
|
time.sleep(15)
|
|
|
|
|
|
|
|
# def vncconnect():
|
|
|
|
# print("This one as well")
|
|
|
|
# subprocess.call("tvnserver.exe -controlservice -connect remotesupport.pismotek.com", shell=True)
|
|
|
|
# time.sleep(3)
|
|
|
|
|
2024-05-26 21:20:13 -06:00
|
|
|
def tab_no(window):
|
|
|
|
""" handle displaying the no tab/column """
|
|
|
|
window['-COL1-'].update(visible=False)
|
|
|
|
window['-COL3-'].update(visible=True)
|
|
|
|
window['No'].update(visible=False)
|
|
|
|
window['Yes'].update(visible=False)
|
|
|
|
window['Exit'].update(visible=True)
|
|
|
|
|
|
|
|
def tab_yes(window):
|
|
|
|
""" handle displaying the yes tab/column"""
|
|
|
|
window['-COL1-'].update(visible=False)
|
|
|
|
window['-COL2-'].update(visible=True)
|
|
|
|
window['No'].update(visible=False)
|
|
|
|
window['Yes'].update(visible=False)
|
|
|
|
window['Exit'].update(visible=True)
|
|
|
|
# window.perform_long_operation(vncdisconnect, '-VNCCONNECT-')
|
|
|
|
|
|
|
|
def tvn_get_state():
|
|
|
|
""" get current state of tvnserver - retrun stopped if does not exist """
|
2024-05-26 21:59:54 -06:00
|
|
|
state = "stopped"
|
2024-05-26 21:20:13 -06:00
|
|
|
service = get_service('tvnserver')
|
|
|
|
if service:
|
2024-05-26 21:59:54 -06:00
|
|
|
state = service["status"]
|
|
|
|
return state
|
|
|
|
|
|
|
|
def get_tvnserver_info():
|
|
|
|
"""
|
|
|
|
determine the state of tvnserver.exe
|
|
|
|
service running - can use as is
|
|
|
|
application running - can use as is
|
|
|
|
application stopped - need to start application
|
|
|
|
"""
|
|
|
|
mode = "application"
|
|
|
|
state = tvn_get_state()
|
|
|
|
if state == "running":
|
|
|
|
mode = "service"
|
|
|
|
elif "tvnserver.exe" in (p.name() for p in psutil.process_iter()):
|
|
|
|
state = "running"
|
|
|
|
return mode, state
|
2024-05-26 21:20:13 -06:00
|
|
|
|
|
|
|
def main():
|
2024-05-26 21:59:54 -06:00
|
|
|
tvn_mode, tvn_state = get_tvnserver_info()
|
2024-05-26 21:20:13 -06:00
|
|
|
layout = build_layout()
|
|
|
|
window = sg.Window('Pismotek', layout, size=(380, 350))
|
|
|
|
while True:
|
|
|
|
event, values = window.read()
|
|
|
|
if event in (sg.WIN_CLOSED, 'Exit'):
|
|
|
|
break
|
|
|
|
elif event == 'No':
|
|
|
|
tab_no(window)
|
|
|
|
elif event == 'Yes':
|
|
|
|
tab_yes(window)
|
|
|
|
window.close()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
main()
|