mirror of
https://github.com/ComputerScienceHouse/proxstar.git
synced 2025-02-12 13:01:51 +00:00
737c9f9367
I made an "oopsie" and hardcoded the port, so now that's configurable, and also fixed my local .env file. I should update the README
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import os
|
|
import subprocess
|
|
|
|
from flask import Flask
|
|
app = Flask(__name__)
|
|
if os.path.exists(os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config_local.py")):
|
|
config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config_local.py")
|
|
else:
|
|
config = os.path.join(app.config.get('ROOT_DIR', os.getcwd()), "config.py")
|
|
app.config.from_pyfile(config)
|
|
|
|
timeout = app.config['TIMEOUT']
|
|
|
|
|
|
def start_websockify(websockify_path, target_file):
|
|
result = subprocess.run(['pgrep', 'websockify'], stdout=subprocess.PIPE)
|
|
if not result.stdout:
|
|
print("Websockify is stopped. Starting websockify.")
|
|
proxstar_port = app.config.get('VNC_PORT')
|
|
subprocess.call(
|
|
[
|
|
websockify_path,
|
|
proxstar_port,
|
|
'--token-plugin',
|
|
'TokenFile',
|
|
'--token-source',
|
|
target_file,
|
|
'-D',
|
|
],
|
|
stdout=subprocess.PIPE,
|
|
)
|
|
else:
|
|
print("Websockify started.")
|
|
|
|
|
|
def on_starting(server):
|
|
print("Booting Websockify server in daemon mode...")
|
|
start_websockify(app.config['WEBSOCKIFY_PATH'], app.config['WEBSOCKIFY_TARGET_FILE'])
|