76 lines
1.7 KiB
Python
76 lines
1.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
from threading import Thread
|
||
|
|
from time import sleep
|
||
|
|
from subprocess import PIPE
|
||
|
|
|
||
|
|
import ujson
|
||
|
|
from psutil import Popen
|
||
|
|
from psutil import Process
|
||
|
|
import psutil
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
sys.path.append(script_dir + '/src/')
|
||
|
|
|
||
|
|
from etc import onionrvalues
|
||
|
|
|
||
|
|
|
||
|
|
sub_script = script_dir + '/' + onionrvalues.SCRIPT_NAME
|
||
|
|
|
||
|
|
def show_info(p: Process):
|
||
|
|
while True:
|
||
|
|
threads = p.num_threads()
|
||
|
|
open_files = len(p.open_files())
|
||
|
|
for proc in p.children(recursive=True):
|
||
|
|
threads += proc.num_threads()
|
||
|
|
try:
|
||
|
|
open_files += len(proc.open_files())
|
||
|
|
except psutil.AccessDenied:
|
||
|
|
pass
|
||
|
|
print(f'Approximate thread count: {threads}')
|
||
|
|
print(f'Approximate open files: {open_files}')
|
||
|
|
sleep(1)
|
||
|
|
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
"--skip-onboarding", help="Skip Onionr onboarding",
|
||
|
|
type=bool, default=False)
|
||
|
|
parser.add_argument(
|
||
|
|
'--open-ui', help='Open onionr web ui after started' ,
|
||
|
|
type=bool, default=True)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
p = Popen([sub_script, 'version'])
|
||
|
|
p.wait()
|
||
|
|
from filepaths import config_file
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
with open(config_file, 'r') as cf:
|
||
|
|
config = ujson.loads(cf.read())
|
||
|
|
|
||
|
|
if args.skip_onboarding:
|
||
|
|
config['onboarding']['done'] = True
|
||
|
|
print('Disabling onboarding')
|
||
|
|
|
||
|
|
|
||
|
|
with open(config_file, 'w') as cf:
|
||
|
|
cf.write(ujson.dumps(config))
|
||
|
|
|
||
|
|
if args.open_ui:
|
||
|
|
p = Popen([sub_script, 'start'])
|
||
|
|
sleep(2)
|
||
|
|
Popen([sub_script, 'openhome'])
|
||
|
|
else:
|
||
|
|
p = Popen([sub_script, 'start'], stdout=PIPE)
|
||
|
|
|
||
|
|
p = p.children()[0]
|
||
|
|
Thread(target=show_info, args=[p], daemon=True).start()
|
||
|
|
p.wait()
|