jetwork/interface.py

from dash import Dash, dcc, html, Input, Output, callback
app = Dash(__name__, title='Jetwork', update_title=None)

from db import *
from status import *

from os import walk
from os import system as sys
from platform import system
from threading import Thread

app.layout = html.Div([ html.Div([

dcc.ConfirmDialog(id='shut_mess', message='Клиент выключен!'),
html.Button("Выключить клиент", className='off_btn', n_clicks=0, id='off_btn'),
html.Div([], id='our_port', className='our_port'),
html.Div([], id='servers', className='servers'),
html.Div([], id='sites', className='sites'),
dcc.Dropdown(options=[], id='search', placeholder='Поиск...'),

dcc.Interval(id='interval-component', interval=1*1000, n_intervals=0)

], className='main')], className='content')


# Функция выключения
def shutdown():
	# Задаём код остановки
	status_set('stop')
	# Определяем платформу
	if system() == 'Linux':
		# вырубаем прокси и скрипты
		sys('killall bore')
		sys('killall python')
	elif system() == 'Windows':
		sys('taskkill /f /im bore.exe')
		sys('taskkill /f /im python.exe')

# Кнопка выключения
@callback(
  Output('shut_mess', 'displayed'),
  Input('off_btn', 'n_clicks'),
	prevent_initial_call=True
)
def shut_btn(n_clicks):
	th = Thread(target=shutdown)
	th.start()
	return True

# Обновление нашего порта (зачем?)
@callback(Output('our_port', 'children'),
					Input('interval-component', 'n_intervals'))
def update_our_port(n):
	return f"Ваш порт: {read()['our_port']}"

# Обновление доступных узлов
@callback(Output('servers', 'children'),
          Input('interval-component', 'n_intervals'))
def update_servers(n):
	res = []
	for i in read()['ports']:
		res.append(html.Div([i], className='serv_elem'))
	return res

# Обновление доступных сайтов
@callback(Output('sites', 'children'),
					Input('interval-component', 'n_intervals'),
					Input('search', 'value'))
def update_sites(n, s_val):
	# Префикс
	base_url = read()['base_url']
	# Если есть элемент в поиске
	if s_val:
		return html.Div([ dcc.Link(children=i, href=f'{base_url}/{s_val}',
		target='_blank') ], className='sites_elem')

	res = []
	for i in next(walk('cached/'), (None, None, []))[1]:
		conf = read(f'cached/{i}/config.json')
		res.append(html.Div([ dcc.Link(children=i, href=f'http://{i}',
													target='_blank') ], className='sites_elem'))
	return res

# Обновление доступных сайтов в поиске
@callback(Output('search', 'options'),
          Input('interval-component', 'n_intervals'))
def update_search(n):
	res = []
	for i in next(walk('cached/'), (None, None, []))[1]:
		res.append(i)
	return res

#app.run(debug=True, port = 5555)
app.run(debug=False, host = '0.0.0.0', port = 5555)