crafty-4/app/classes/web/panel_handler.py

77 lines
2.3 KiB
Python

import json
import logging
import tornado.web
import tornado.escape
import bleach
from app.classes.shared.console import console
from app.classes.shared.models import Users, installer
from app.classes.web.base_handler import BaseHandler
from app.classes.shared.controller import controller
from app.classes.shared.models import db_helper
from app.classes.shared.helpers import helper
logger = logging.getLogger(__name__)
class PanelHandler(BaseHandler):
@tornado.web.authenticated
def get(self, page):
# name = tornado.escape.json_decode(self.current_user)
user_data = json.loads(self.get_secure_cookie("user_data"))
template = "panel/denied.html"
defined_servers = controller.list_defined_servers()
page_data = {
'version_data': "version_data_here",
'user_data': user_data,
'server_stats': {
'total': len(defined_servers),
'running': len(controller.list_running_servers()),
'stopped': (len(controller.list_defined_servers()) - len(controller.list_running_servers()))
},
'menu_servers': defined_servers,
'hosts_data': db_helper.get_latest_hosts_stats(),
'show_contribute': helper.get_setting("show_contribute_link", True)
}
# if no servers defined, let's go to the build server area
if page_data['server_stats']['total'] == 0:
self.set_status(301)
self.redirect("/server/step1")
return False
if page == 'unauthorized':
template = "panel/denied.html"
elif page == 'credits':
template = "panel/credits.html"
elif page == 'contribute':
template = "panel/contribute.html"
elif page == 'dashboard':
page_data['servers'] = db_helper.get_all_servers_stats()
for s in page_data['servers']:
try:
data = json.loads(s['int_ping_results'])
s['int_ping_results'] = data
except:
pass
template = "panel/dashboard.html"
elif page == 'server_detail':
template = "panel/server_details.html"
self.render(
template,
data=page_data
)