# -*- coding: utf-8 -*-

#+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#@auhor by ruiy
#
#
#
#pip install paramiko -i https://pypi.tuna.tsinghua.edu.cn/simple
#
#pip install psutil -i https://pypi.tuna.tsinghua.edu.cn/simple
#
#注意:添加openssh 账号sshadmin必须加入administrator组,否则获取相关监控指标数据为null
#+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

import paramiko
import datetime
import time
import os
import psutil

dt = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())
#print(type(dt))
print("时间%s"%dt)

#psutil

def get_cpu_infos():
cpu_count = psutil.cpu_count(logical=False)
xc_count = psutil.cpu_count()
cpu_percent = round((psutil.cpu_percent(1)),2)
cpu_info = (cpu_count,xc_count,cpu_percent)
return cpu_info


def get_mem_infos():
memory = psutil.virtual_memory()
total_mem = round((float(memory.total) / 1024 / 1024 / 1024),2)
used_mem = round((float(memory.used) / 1024 / 1024 / 1024),2)
free_mem = round((float(memory.free) / 1024 / 1024 / 1024),2)
percent_mem = round((float(memory.used) / float(memory.total) * 100),2)
mem_info = (total_mem,used_mem,free_mem,percent_mem)
return mem_info

def get_disk_infos():
list = psutil.disk_partitions()
ilen = len(list)
i = 0
retlist1 = []
retlist2 = []
disk_info_list = []

while i < ilen:
diskinfo = psutil.disk_usage(list[i].device)
total_disk = round((float(diskinfo.total) / 1024 / 1024 / 1024),2)
used_disk = round((float(diskinfo.used) / 1024 / 1024 / 1024),2)
free_disk = round((float(diskinfo.free) / 1024 / 1024 / 1024),2)
#percent_disk = diskinfo.percent()
retlist1 = [i,list[i].device,total_disk,used_disk,free_disk]
disk_info_list.append(retlist1)
i = i + 1
return disk_info_list

#windows


#linux

if __name__ == '__main__':
hosts = [
#'hostname="127.0.0.1",port=3389,username="administrator",password=""',
'hostname="1.2.3.172",port=22,username="administrator",password="qbkj"',
'hostname="1.2.3.225",port=22,username="sshadmin",password="321"',
'hostname="172.16.28.122",port=22,username="sshadmin",password="qb"',
]
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
for i in hosts:
#print(i)
splitx = i.split(',')
ipx = splitx[0].split('=')[1].replace('"','')
print(ipx)
portx = int(splitx[1].split('=')[1])
#print(portx)
unamex = splitx[2].split('=')[1].replace('"','')
#print(unamex)
pwdx = splitx[3].split('=')[1].replace('"','')
#print(pwdx)
try:
#建立连接
ssh.connect(hostname=ipx,port=portx,username=unamex,password=pwdx)
print("建立连接")
#ssh.connect(hostname='1.2.3.172',port=22,username='sshadmin',password='321',timeout=3)
#ssh.connect(hostname='1.2.3.172',port=22,username='sshadmin',password='321')
stdin_init,stdout_init,stderr_init = ssh.exec_command("hostname")
print((stdout_init.read()).decode('gbk'))

time.sleep(1)
#'''
#windows
#cpu利用率:wmic cpu get loadpercentage
stdin,stdout,stderr = ssh.exec_command("wmic cpu get loadpercentage")
#print(stdout.read())
#time.sleep(100)
#cpu_percent = stdout.read().decode('utf-8').replace('LoadPercentage','').replace('\n','')
cpu_percent = stdout.read().decode('utf-8').replace('LoadPercentage','')
#print("cpu利用率{x}".format(x = cpu_percent))
xx = cpu_percent
#print(type(xx))
#print(xx)
#print("rui%s"%(xx))
#xr = xx
#print("亲%s"%xr)
print("cpu利用率%:{var}".format(var=xx))
time.sleep(1)

#占用cpu前5 进程
stdin,stdout,stderr = ssh.exec_command('Powershell.exe -Command " Get-Process | select *,@{N=\'CPUPC\';E={[float]$_.cpu}}|sort CPUPC -desc|select Id,name -first 5"')
cpu5 = stdout.read().decode('utf-8')
print(cpu5)

#mem内存信息
#systeminfo | findstr "物理内存总量:"
stdin,stdout,stderr = ssh.exec_command('systeminfo | findstr "物理内存总量:"')
mem_total = stdout.read().decode('gbk').replace('物理内存总量:','').replace('\t','').replace('MB','').replace(',','').replace('\n','').replace(' ','')
print("总内存M:%s"%mem_total)
time.sleep(1)

#systeminfo | findstr "可用的物理内存:"
stdin,stdout,stderr = ssh.exec_command('systeminfo | findstr "可用的物理内存:"')
mem_free = stdout.read().decode('gbk').replace('可用的物理内存:','').replace('\t','').replace('MB','').replace(',','').replace('\n','').replace(' ','')
print("可用内存M:%s"%mem_free)

mem_use = int(mem_total) - int(mem_free)
print("已使用内存M:%s"%mem_use)
mem_use_percent = round(mem_use / int(mem_total),2)
print("内存使用率%:{varx}".format(varx = float(mem_use_percent) * 100))
#print()
time.sleep(1)

#占用mem前5
stdin,stdout,stderr = ssh.exec_command('Powershell.exe -Command " Get-Process | select *,@{N=\'PrivateMemorySizePC\';E={[float]$_.PrivateMemorySize}}|sort PrivateMemorySize -desc|select Id,name -first 5"')
mem5 = stdout.read().decode('utf-8')
print(mem5)

#disk
#fsutil volume diskfree c:
#wmic logicaldisk get caption

stdin,stdout,stderr = ssh.exec_command('wmic logicaldisk get caption')
#disk_partion_label = stdout.read().decode('gbk')
#print(disk_partion_label)
time.sleep(1)
disk_partion_l = bytes.decode(stdout.read()).split('Caption')[1].replace('\r\r\n','').replace(' ','').split(":")
#print(disk_partion_l)
#print(len(disk_partion_l))
#print(disk_partion_l[0])
#x = "{}:".format(disk_partion_l[0])
#print(x)
#fsutil volume diskfree x
#剔除list中空元素
while ' ' in disk_partion_l:
disk_partion_l.remove(' ')
#print(disk_partion_l)
for i in disk_partion_l:
lab = "{}:".format(i)
#print(lab)
#stdin,stdout,stderr = ssh.exec_command("{}".format(lab))
#print("fsutil volume diskfree %s"%(lab))
stdin,stdout,stderr = ssh.exec_command("fsutil volume diskfree %s"%(lab))
print(stdout.read().decode('gbk'))
time.sleep(1)
#disk_partion_label = stdout.read().decode('gbk').replace('Caption','').replace('\r\r','').split('\n')
#disk_partion_label = stdout.read().decode('gbk').replace('Caption','').replace('\t','').replace("\r\r\n\r\r\n",'').replace('\r\r\n','').split(":")
#print(disk_partion_label)
#print(disk_partion_label[0])
"""
print(len(disk_partion_label))
for i in disk_partion_label:
if i =='':
print("null")
else:
print(i)
"""
#mem_total = bytes.decode(stdout.read())
#print(mem_total)
#print(os.environ['SYSTEMDRIVE'])
#stdin_init,stdout_init,stderr_init = ssh.exec_command("wmic os get TotalVisibleMemorySize")
#mem_total = stdout_init.read().decode('utf-8').replace('TotalVisibleMemorySize','').replace('\t','').replace('\n','').replace('MB','')
#mem_total = stdout_init.read()
#print(float(float(int(mem_total)) / 1024 / 1024),2)
#print(int(mem_total))
#print((stdout_init.read()).decode('utf-8'))
#print((stdout_init.read()))

#infos = bytes.decode(stdout_init.read())
#infos
#print(infos)
#infos = stdout_init.read()
#print(infos)
#'''
except Exception as paramiko_err:
print(paramiko_err)





"""
cpuInfos = get_cpu_infos()
print(cpuInfos)

memInfos = get_mem_infos()
print(memInfos)

diskInfos = get_disk_infos()
print(diskInfos)
"""