haproxy-wi/app/overview.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import sys

import psutil
from jinja2 import Environment, FileSystemLoader

import modules.db.sql as sql
import modules.roxywi.logs as roxy_logs
import modules.roxywi.auth as roxywi_auth
import modules.roxywi.common as roxywi_common
import modules.server.server as server_mod

env = Environment(loader=FileSystemLoader('templates/'), autoescape=True)
template = env.get_template('ovw.html')

print('Content-type: text/html\n')

grafana = 0
metrics_worker = 0
checker_worker = 0
is_checker_worker = 0
is_metrics_worker = 0
servers_group = []
host = os.environ.get('HTTP_HOST', '')

user_params = roxywi_common.get_users_params()

try:
	roxywi_auth.check_login(user_params['user_uuid'], user_params['token'])
except Exception as e:
	print(f'error {e}')
	sys.exit()

try:
	groups = sql.select_groups()
	user_group = roxywi_common.get_user_group(id=1)

	if (user_params['role'] == 2 or user_params['role'] == 3) and int(user_group) != 1:
		for s in user_params['servers']:
			servers_group.append(s[2])

	is_checker_worker = len(sql.select_all_alerts(group=user_group))
	is_metrics_worker = len(sql.select_servers_metrics_for_master(group=user_group))

	for pids in psutil.pids():
		if pids < 300:
			continue
		try:
			pid = psutil.Process(pids)
			cmdline_out = pid.cmdline()
			if len(cmdline_out) > 2:
				if 'checker_' in cmdline_out[1]:
					if len(servers_group) > 0:
						if cmdline_out[2] in servers_group:
							checker_worker += 1
					else:
						checker_worker += 1
				elif 'metrics_' in cmdline_out[1]:
					if len(servers_group) > 0:
						if cmdline_out[2] in servers_group:
							metrics_worker += 1
					else:
						metrics_worker += 1
				if len(servers_group) == 0:
					if 'grafana' in cmdline_out[1]:
						grafana += 1
		except psutil.NoSuchProcess:
			pass

	cmd = "systemctl is-active roxy-wi-metrics"
	metrics_master, stderr = server_mod.subprocess_execute(cmd)
	cmd = "systemctl is-active roxy-wi-checker"
	checker_master, stderr = server_mod.subprocess_execute(cmd)
	cmd = "systemctl is-active roxy-wi-keep_alive"
	keep_alive, stderr = server_mod.subprocess_execute(cmd)
	cmd = "systemctl is-active roxy-wi-smon"
	smon, stderr = server_mod.subprocess_execute(cmd)
	cmd = "systemctl is-active roxy-wi-portscanner"
	port_scanner, stderr = server_mod.subprocess_execute(cmd)
	cmd = "systemctl is-active roxy-wi-socket"
	socket, stderr = server_mod.subprocess_execute(cmd)

except Exception as e:
	groups = ''
	roles = ''
	metrics_master = ''
	checker_master = ''
	keep_alive = ''
	smon = ''
	socket = ''
	stderr = ''
	print(e)

rendered_template = template.render(
	h2=1, autorefresh=1, title="Overview", role=user_params['role'], user=user_params['user'], groups=groups,
	roles=sql.select_roles(), metrics_master=''.join(metrics_master), metrics_worker=metrics_worker,
	checker_master=''.join(checker_master), checker_worker=checker_worker, keep_alive=''.join(keep_alive),
	smon=''.join(smon), port_scanner=''.join(port_scanner), grafana=grafana, socket=''.join(socket),
	roxy_wi_log_id=roxy_logs.roxy_wi_log(log_id=1, file="roxy-wi-"),
	metrics_log_id=roxy_logs.roxy_wi_log(log_id=1, file="metrics"),
	checker_log_id=roxy_logs.roxy_wi_log(log_id=1, file="checker"),
	keep_alive_log_id=roxy_logs.roxy_wi_log(log_id=1, file="keep_alive"),
	socket_log_id=roxy_logs.roxy_wi_log(log_id=1, file="socket"), error=stderr,
	roxy_wi_log=roxy_logs.roxy_wi_log(), servers=user_params['servers'], is_checker_worker=is_checker_worker,
	is_metrics_worker=is_metrics_worker, host=host, user_services=user_params['user_services'],
	token=user_params['token'], guide_me=1
)
print(rendered_template)