pve-info.py

# coding=utf-8
# 通过python脚本获取pve服务器的信息，发送到blynk服务器，然后在手机app上实时显示
# 运行环境为python3
import blynklib
# import random
import blynktimer
import platform as pl  # 查看系统信息
import psutil as ps  # 查看系统相关参数，如温度，内存等
import socket  # 获取IP地址
import subprocess as sub  # 执行系统命令
import time
import os
import shlex  # 使用类似shell的语法分割字符串

# blynk server auth code
BLYNK_AUTH = 'auth token'

# initialize blynk
# blynk = blynklib.Blynk(BLYNK_AUTH, server='2959w71z50.qicp.vip', port=26514)

# 如果你无法实现内网穿透，可以取消下面语句的注释，可以实现本地局域网内的访问
blynk = blynklib.Blynk(BLYNK_AUTH, server='192.168.50.6', port=8080)

# last command in example - just to show error handling
# for certain HW can be added specific commands. 'gpio readall' on PI3b for example
ALLOWED_COMMANDS_LIST = [
    'lsusb', 'ip', 'lspci', 'lshw', 'date', 'df', 'pveversion', 'help', 'free',
    'networkctl', 'ls', 'lsb_release', 'cat', 'osinfo', 'lsof', 'dmidecode', 'npc'
]
#
READ_PRINT_MSG = "Read Pin: V{}"
# create timers dispatcher instance
timer = blynktimer.Timer()
update_int = 5  # 数据更新间隔，默认为5秒
# message string
WRITE_EVENT_PRINT_MSG = "Write Pin: {} Value: '{}'"


# 连接到服务器时执行
@blynk.handle_event("connect")
def connect_handler():
    for pin in range(20):
        blynk.virtual_sync(pin)
    # 获取本机计算机名称
    hostname = socket.gethostname()
    # 获取本机ip
    ip = socket.gethostbyname(hostname)
    print(WRITE_EVENT_PRINT_MSG.format('IP', ip))
    blynk.virtual_write(0, ip)
    # 显示os信息到terminal
    os = pl.uname()
    blynk.virtual_write(2, 'OS info')
    blynk.virtual_write(2, '==============')
    for info in os:
        print(WRITE_EVENT_PRINT_MSG.format('OS', info))
        blynk.virtual_write(2, info)
    blynk.virtual_write(2, '==============')


class Utils(object):
    __instance = None  # 定义一个类属性做判断

    def __new__(cls):  # 单例模式

        if cls.__instance == None:
            # 如果__instance为空证明是第一次创建实例
            # 通过父类的__new__(cls)创建实例
            cls.__instance = object.__new__(cls)
            return cls.__instance
        else:
            # 返回上一个对象的引用
            return cls.__instance

    __unit_dict = {
        '0': 'B',
        '1': 'K',
        '2': 'M',
        '3': 'G',
        '4': 'T',
    }


# 显示terminal command执行结果
@blynk.handle_event('write V2')
def write_handler(pin, values):
    header = ''
    result = ''
    delimiter = '{}\n'.format('=' * 30)
    input = ''.join(values)
    cmd = shlex.split(input.strip())
    if values[0] == 'help':
        header = '[Allowed commands]\n'
        result = '{}\n'.format('\n'.join(ALLOWED_COMMANDS_LIST))
    elif values[0] == 'osinfo':
        header = '[output]\n'
        info = pl.uname()
        result = '{}\n'.format('\n'.join(info))
    else:
        # 获取命令参数
        if cmd[0] in ALLOWED_COMMANDS_LIST:
            try:
                result = sub.check_output(cmd, shell=False)
                header = '[output]\n'
            except sub.CalledProcessError as exe_err:
                header = '[error]\n' + \
                    'Return Code: {}\n'.format(exe_err.returncode)
            except Exception as g_err:
                print("Command caused '{}'".format(g_err))
        else:
            header = '[Allowed commands]\n'
            result = '{}\n'.format('\n'.join(ALLOWED_COMMANDS_LIST))

    # communicate with terminal if help or some allowed command
    if result:
        output = '{}{}{}{}'.format(header, delimiter, result, delimiter)
        print(output)
        blynk.virtual_write(pin, output)
        blynk.virtual_write(pin, '\n')


#
# 显示负载使用率
@timer.register(vpin_num=1, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    value = ps.getloadavg()
    print(WRITE_EVENT_PRINT_MSG.format('Load', value))
    blynk.virtual_write(vpin_num, value)


# 显示cpu占用率
@timer.register(vpin_num=3, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    value = ps.cpu_percent()
    print(WRITE_EVENT_PRINT_MSG.format('CPU', value))
    blynk.virtual_write(vpin_num, value)


# 显示内存使用率
# @timer.register(vpin_num=4, interval=8, run_once=False)
@timer.register(vpin_num=4, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    try:
        value = ps.virtual_memory().percent
        print(WRITE_EVENT_PRINT_MSG.format('Mem', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("get memory data error ".format(g_err))


# 获取系统运行时间
def get_os_info():
    boot_time = ps.boot_time()
    return boot_time


# get disk info
def get_disk_info(path='/'):
    '''
    获取传入路径下的硬盘信息
    :param path:
    :return:
    '''
    disk_info = ps.disk_usage(path)
    disk_data = dict()
    disk_data['size'] = Utils.__get_unit(disk_info.total)
    disk_data['available'] = Utils.__get_unit(disk_info.free)
    disk_data['usage'] = '%.2f%s' % (disk_info.percent, '%')
    return disk_data


# get physics disk list and disk size
def get_disk_list():
    disk_name = []
    disk_size = []
    disk_info = dict()
    dev_list = os.popen(
        "lsblk -d -o NAME,SIZE,TYPE|grep disk|awk '{print $1,$2}'").read().strip().split('\n')
    # result like this
    # ['sda 40G', 'sdb 5G']
    for dev in dev_list:
        temp = dev.split(' ')
        disk_name.append('/dev/' + temp[0])
        disk_size.append(temp[1])
    # disk_name like ['/dev/sda', '/dev/sdb', '/dev/sdc']
    # 两个列表转为一个字典
    disk_info = dict(zip(disk_name, disk_size))
    return disk_name, disk_size, disk_info


# 显示系统硬盘大小
@timer.register(vpin_num=5, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    try:
        # re = format(
        #    float(sub.check_output(["fdisk", "-s", dev[1]])) / 1024 / 1024,
        #    '.2f')
        #value = format(float(re), ',')
        #print(WRITE_EVENT_PRINT_MSG.format('Disk1', value))
        print(WRITE_EVENT_PRINT_MSG.format('Disk1', info['/dev/sda']))
        #blynk.virtual_write(vpin_num, value)
        blynk.virtual_write(vpin_num, info['/dev/sda'])
    except Exception as g_err:
        print("Get sdb data error ".format(g_err))


# 显示500g硬盘大小
@timer.register(vpin_num=6, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    try:
        # re = format(
        #    float(sub.check_output(["fdisk", "-s", dev[2]])) / 1024 / 1024,
        #    '.2f')
        #value = format(float(re), ',')
        print(WRITE_EVENT_PRINT_MSG.format('Disk2', info['/dev/sdc']))
        blynk.virtual_write(vpin_num, info['/dev/sdc'])
    except Exception as g_err:
        print("Get sda data error ".format(g_err))

# 显示4t硬盘大小
@timer.register(vpin_num=14, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    try:
        # re = format(
        #    float(sub.check_output(["fdisk", "-s", dev[0]])) / 1024 / 1024,
        #    '.2f')
        #value = format(float(re), ',')
        print(WRITE_EVENT_PRINT_MSG.format('Disk2', info['/dev/sdb']))
        blynk.virtual_write(vpin_num, info['/dev/sdb'])
    except Exception as g_err:
        print("Get sda data error ".format(g_err))

# 显示硬盘使用率
@timer.register(vpin_num=7, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    # 读取系统硬盘使用率
    try:
        value = ps.disk_usage('/').percent
        print(WRITE_EVENT_PRINT_MSG.format('Disk1', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("get disk data error ".format(g_err))


@timer.register(vpin_num=8, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    # 读取500g硬盘使用率
    try:
        value = ps.disk_usage(ps.disk_partitions()[2].mountpoint).percent
        print(WRITE_EVENT_PRINT_MSG.format('Disk2', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("get data error ".format(g_err))


@timer.register(vpin_num=17, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    dev, size, info = get_disk_list()
    # 读取4t硬盘使用率
    try:
        value = ps.disk_usage(ps.disk_partitions()[4].mountpoint).percent
        print(WRITE_EVENT_PRINT_MSG.format('Disk3', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("get data error ".format(g_err))


# 显示温度
# ps.sensors_temperatures()输出结果
# {'acpitz': [shwtemp(label='', current=27.8, high=105.0, critical=105.0), shwtemp(label='', current=29.8, high=105.0, critical=105.0)], 'coretemp': [shwtemp(label='Package id 0', current=49.0, high=100.0, critical=100.0), shwtemp(label='Core 0', current=49.0, high=100.0, critical=100.0), shwtemp(label='Core 1', current=45.0, high=100.0, critical=100.0), shwtemp(label='Package id 0', current=49.0, high=100.0, critical=100.0), shwtemp(label='Core 0', current=49.0, high=100.0, critical=100.0), shwtemp(label='Core 1', current=45.0, high=100.0, critical=100.0)]}


# cpu 1温度
@timer.register(vpin_num=9, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    try:
        value = ps.sensors_temperatures().get('coretemp')[1].current
        print(WRITE_EVENT_PRINT_MSG.format('tCPU1', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("Get cpu1 temp error ".format(g_err))


# cpu 2
@timer.register(vpin_num=10, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    try:
        value = ps.sensors_temperatures().get('coretemp')[2].current
        print(WRITE_EVENT_PRINT_MSG.format('tCPU2', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("Get cpu2 temp error ".format(g_err))


# 主板温度
# @timer.register(vpin_num=11, interval=update_int, run_once=False)
# def write_to_virtual_pin(vpin_num=1):
#    try:
#        value = ps.sensors_temperatures().items()[0][1][1].current
#        print(WRITE_EVENT_PRINT_MSG.format('tBoard', value))
#        blynk.virtual_write(vpin_num, value)
#    except Exception as g_err:
#        print("Get board temp error ".format(g_err))


# 系统硬盘温度
@timer.register(vpin_num=12, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    # result '/dev/sdb: Kston 64GB: 36\xc2\xb0C\n'
    dev, size, info = get_disk_list()
    try:
        re = sub.check_output(['hddtemp', dev[1]])
        value = int(re[-7:-4].strip())  # 去除空格
        print(WRITE_EVENT_PRINT_MSG.format('tHDD1', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("Get disk temp error ".format(g_err))

# 500g硬盘温度
@timer.register(vpin_num=18, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    # result '/dev/sdb: Kston 64GB: 36\xc2\xb0C\n'
    dev, size, info = get_disk_list()
    try:
        re = sub.check_output(['hddtemp', dev[2]])
        value = int(re[-7:-4].strip())  # 去除空格
        print(WRITE_EVENT_PRINT_MSG.format('tHDD1', value))
        blynk.virtual_write(vpin_num, value)
    except Exception as g_err:
        print("Get disk temp error ".format(g_err))

# 显示网路发送数据
@timer.register(vpin_num=15, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    # 获取网络发送流量
    try:
        # value1 = format(float(ps.net_io_counters()[0]) / 1024 / 1024 / 1024, '.3f')  # TB
        value1 = ps.net_io_counters(pernic=True)['enp2s0'].bytes_sent
        time.sleep(1)  # wait for 1 second
        value2 = ps.net_io_counters(pernic=True)['enp2s0'].bytes_sent
        print(str('TX %.2f' % ((value2 - value1) / 1024)) + ' kB/s')
        # 两次获取的流量相减得到每秒流量
        blynk.virtual_write(vpin_num, (value2 - value1) / 1024)
    except Exception as g_err:
        print("Get data error ".format(g_err))


# 显示网路接收数据
@timer.register(vpin_num=16, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    # 获取网络接收流量
    try:
        # value = format(float(ps.net_io_counters()[1]) / 1024 / 1024 / 1024,'.3f')  # TB
        value1 = ps.net_io_counters(pernic=True)['enp2s0'].bytes_recv
        time.sleep(1)  # wait for 1 second
        value2 = ps.net_io_counters(pernic=True)['enp2s0'].bytes_recv
        print(str('RX %.2f' % ((value2 - value1) / 1024)) + ' kB/s')
        # 两次获取的流量相减得到每秒流量
        blynk.virtual_write(vpin_num, (value2 - value1) / 1024)
    except Exception as g_err:
        print("Get data error ".format(g_err))


# 显示npc连接状态,查看是否掉线
#
@timer.register(vpin_num=13, interval=update_int, run_once=False)
def write_to_virtual_pin(vpin_num=1):
    # 读取腾讯vps的最后断线日志
    log = os.popen(
        "cat /var/log/npc.log |grep '[E]'").read().strip().split('\n')[-1]
    blynk.virtual_write(vpin_num, log)
    # 读取最新连线日志
    log = os.popen(
        "cat /var/log/npc.log |grep 'Successful '").read().strip().split('\n')[-1]
    blynk.virtual_write(vpin_num, log)


###########################################################
# infinite loop that waits for event
###########################################################
try:
    while True:
        blynk.run()
        timer.run()
except KeyboardInterrupt:
    blynk.disconnect()