You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
301 lines
6.8 KiB
301 lines
6.8 KiB
import os
|
|
import platform
|
|
import multiprocessing
|
|
from datetime import timedelta
|
|
|
|
|
|
def chunks(get, n):
|
|
return [get[i:i + n] for i in range(0, len(get), n)]
|
|
|
|
|
|
def get_uptime():
|
|
"""
|
|
Get uptime
|
|
"""
|
|
try:
|
|
with open('/proc/uptime', 'r') as f:
|
|
uptime_seconds = float(f.readline().split()[0])
|
|
uptime_time = str(timedelta(seconds=uptime_seconds))
|
|
data = uptime_time.split('.', 1)[0]
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_ipaddress():
|
|
"""
|
|
Get the IP Address
|
|
"""
|
|
data = []
|
|
try:
|
|
eth = os.popen("ip addr | grep LOWER_UP | awk '{print $2}'")
|
|
iface = eth.read().strip().replace(':', '').split('\n')
|
|
eth.close()
|
|
del iface[0]
|
|
|
|
for i in iface:
|
|
pipe = os.popen(
|
|
"ip addr show " + i + "| awk '{if ($2 == \"forever\"){!$2} else {print $2}}'")
|
|
data1 = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
if len(data1) == 2:
|
|
data1.append('unavailable')
|
|
if len(data1) == 3:
|
|
data1.append('unavailable')
|
|
data1[0] = i
|
|
data.append(data1)
|
|
|
|
ips = {'interface': iface, 'itfip': data}
|
|
|
|
data = ips
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_cpus():
|
|
"""
|
|
Get the number of CPUs and model/type
|
|
"""
|
|
try:
|
|
pipe = os.popen("cat /proc/cpuinfo |" + "grep 'model name'")
|
|
data = pipe.read().strip().split(':')[-1]
|
|
pipe.close()
|
|
|
|
if not data:
|
|
pipe = os.popen("cat /proc/cpuinfo |" + "grep 'Processor'")
|
|
data = pipe.read().strip().split(':')[-1]
|
|
pipe.close()
|
|
|
|
cpus = multiprocessing.cpu_count()
|
|
|
|
data = {'cpus': cpus, 'type': data}
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_users():
|
|
"""
|
|
Get the current logged in users
|
|
"""
|
|
try:
|
|
pipe = os.popen("who |" + "awk '{print $1, $2, $6}'")
|
|
data = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
|
|
if data == [""]:
|
|
data = None
|
|
else:
|
|
data = [i.split(None, 3) for i in data]
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_traffic(request):
|
|
"""
|
|
Get the traffic for the specified interface
|
|
"""
|
|
try:
|
|
pipe = os.popen(
|
|
"cat /proc/net/dev |" + "grep " + request + "| awk '{print $1, $9}'")
|
|
data = pipe.read().strip().split(':', 1)[-1]
|
|
pipe.close()
|
|
|
|
if not data[0].isdigit():
|
|
pipe = os.popen(
|
|
"cat /proc/net/dev |" + "grep " + request + "| awk '{print $2, $10}'")
|
|
data = pipe.read().strip().split(':', 1)[-1]
|
|
pipe.close()
|
|
|
|
data = data.split()
|
|
|
|
traffic_in = int(data[0])
|
|
traffic_out = int(data[1])
|
|
|
|
all_traffic = {'traffic_in': traffic_in, 'traffic_out': traffic_out}
|
|
|
|
data = all_traffic
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_platform():
|
|
"""
|
|
Get the OS name, hostname and kernel
|
|
"""
|
|
try:
|
|
osname = " ".join(platform.linux_distribution())
|
|
uname = platform.uname()
|
|
|
|
if osname == ' ':
|
|
osname = uname[0]
|
|
|
|
data = {'osname': osname, 'hostname': uname[1], 'kernel': uname[2]}
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_disk():
|
|
"""
|
|
Get disk usage
|
|
"""
|
|
try:
|
|
pipe = os.popen(
|
|
"df -Ph | " + "grep -v Filesystem | " + "awk '{print $1, $2, $3, $4, $5, $6}'")
|
|
data = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
|
|
data = [i.split(None, 6) for i in data]
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_disk_rw():
|
|
"""
|
|
Get the disk reads and writes
|
|
"""
|
|
try:
|
|
pipe = os.popen(
|
|
"cat /proc/partitions | grep -v 'major' | awk '{print $4}'")
|
|
data = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
|
|
rws = []
|
|
for i in data:
|
|
if i.isalpha():
|
|
pipe = os.popen(
|
|
"cat /proc/diskstats | grep -w '" + i + "'|awk '{print $4, $8}'")
|
|
rw = pipe.read().strip().split()
|
|
pipe.close()
|
|
|
|
rws.append([i, rw[0], rw[1]])
|
|
|
|
if not rws:
|
|
pipe = os.popen(
|
|
"cat /proc/diskstats | grep -w '" + data[0] + "'|awk '{print $4, $8}'")
|
|
rw = pipe.read().strip().split()
|
|
pipe.close()
|
|
|
|
rws.append([data[0], rw[0], rw[1]])
|
|
|
|
data = rws
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_mem():
|
|
"""
|
|
Get memory usage
|
|
"""
|
|
try:
|
|
pipe = os.popen(
|
|
"free -tm | " + "grep 'Mem' | " + "awk '{print $2,$4,$6,$7}'")
|
|
data = pipe.read().strip().split()
|
|
pipe.close()
|
|
|
|
allmem = int(data[0])
|
|
freemem = int(data[1])
|
|
buffers = int(data[2])
|
|
cachedmem = int(data[3])
|
|
|
|
# Memory in buffers + cached is actually available, so we count it
|
|
# as free. See http://www.linuxatemyram.com/ for details
|
|
freemem += buffers + cachedmem
|
|
|
|
percent = (100 - ((freemem * 100) / allmem))
|
|
usage = (allmem - freemem)
|
|
|
|
mem_usage = {'usage': usage, 'buffers': buffers, 'cached': cachedmem, 'free': freemem, 'percent': percent}
|
|
|
|
data = mem_usage
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_cpu_usage():
|
|
"""
|
|
Get the CPU usage and running processes
|
|
"""
|
|
try:
|
|
pipe = os.popen("ps aux --sort -%cpu,-rss")
|
|
data = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
|
|
usage = [i.split(None, 10) for i in data]
|
|
del usage[0]
|
|
|
|
total_usage = []
|
|
|
|
for element in usage:
|
|
usage_cpu = element[2]
|
|
total_usage.append(usage_cpu)
|
|
|
|
total_usage = sum(float(i) for i in total_usage)
|
|
|
|
total_free = ((100 * int(get_cpus()['cpus'])) - float(total_usage))
|
|
|
|
cpu_used = {'free': total_free, 'used':
|
|
float(total_usage), 'all': usage}
|
|
|
|
data = cpu_used
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_load():
|
|
"""
|
|
Get load average
|
|
"""
|
|
try:
|
|
data = os.getloadavg()[0]
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|
|
|
|
|
|
def get_netstat():
|
|
"""
|
|
Get ports and applications
|
|
"""
|
|
try:
|
|
pipe = os.popen(
|
|
"ss -tnp | grep ESTAB | awk '{print $4, $5}'| sed 's/::ffff://g' | awk -F: '{print $1, $2}' "
|
|
"| awk 'NF > 0' | sort -n | uniq -c")
|
|
data = pipe.read().strip().split('\n')
|
|
pipe.close()
|
|
|
|
data = [i.split(None, 4) for i in data]
|
|
|
|
except Exception as err:
|
|
data = str(err)
|
|
|
|
return data
|