python3批量统计用户电脑配置

# -*- coding: utf-8 -*-
#author:Guoyabin
import socket,psutil,os,sys

class wininfo():
	def ip_info(self):
		hostname=socket.gethostname() #获取主机名
		ipconfig=psutil.net_if_addrs().items() #所有网卡信息
		try:
			s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
			s.connect((‘114.114.114.114‘, 80))
			addre = s.getsockname()[0] #获取ip地址
		finally:
			s.close() #关闭socket连接
		for k,v in ipconfig:
			try:
				if addre in v[1].address:
					mac=v[0].address #获取mac地址
					‘‘‘
					gateway=addre.split(‘.‘)[0]+‘.‘+addre.split(‘.‘)[1]+‘.‘+addre.split(‘.‘)[2]+‘.1‘ #获取网关
					if ‘10.8.70‘ in addre: #特殊网关
						gateway=‘10.8.70.2‘
					os.system("netsh interface ip set address name=%s source=static addr=%s mask=255.255.255.0 gateway=%s gwmetric=1" % (k,addre,gateway)) #设置网关
					‘‘‘
					os.system("netsh interface ip set dns name=%s source=static addr=114.114.114.114" % k) #设置dns
			except:
				pass
		ip=str(hostname)+‘ ‘+addre+‘ ‘+mac
		return(ip)

	def mem_info(self):
		mem_all=psutil.virtual_memory().total/1024/1024/1024 #获取内存大小
		mem=round(mem_all) #取最接近整数的值
		return(mem)

	def cpu_info(self):
		try:
			name=os.popen(‘wmic cpu get name‘).readlines()
			cpuname1=name[-4].replace(‘\n‘,‘‘)
			cpuname=cpuname1.replace(‘ ‘,‘‘) #获取Windows CPU型号
		except:
			cpuname=‘ ‘
		cpucount=psutil.cpu_count() #获取CPU核心
		cpu=str(cpuname)+‘ ‘+str(cpucount)
		return(cpu)

	def disk_info(self):
		sum=0
		for i in psutil.disk_partitions():
			try:
				sum=sum+psutil.disk_usage(i.mountpoint).total #获取硬盘大小
			except:
				pass
		disk=round(sum/1024/1024/1024) #取最接近整数的值
		return(disk)

	def getname(self):
		name=input(‘请输入您的名字:‘)
		return(name)

class linuxinfo(wininfo):
	def getname(self):
		name=‘name‘
		return(name)

	def cpu_info(self):
		cpuname1=os.popen("grep -m1 ‘model name‘ /proc/cpuinfo |awk -F ‘:‘ ‘{print $2}‘").readlines() 
		cpuname2=cpuname1[-1].replace(‘ ‘,‘‘)
		cpuname=cpuname2.replace(‘\n‘,‘‘) #获取Linux CPU型号
		cpucount=psutil.cpu_count() #获取CPU核心
		cpu=str(cpuname)+‘ ‘+str(cpucount)
		return(cpu)

	def ip_info(self):
		hostname=socket.gethostname() #获取主机名
		try:
			s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
			s.connect((‘114.114.114.114‘, 80))
			addre = s.getsockname()[0] #获取ip地址
		finally:
			s.close() #关闭socket连接
		ipconfig=psutil.net_if_addrs().items() #所有网卡信息
		for k,v in ipconfig:
			if addre in v[0].address:
				mac=v[2].address #获取mac地址
		ip=str(hostname)+‘ ‘+addre+‘ ‘+mac
		return(ip)

if __name__==‘__main__‘:
	serverip = ‘192.168.8.50‘
	port = 9999
	client=socket.socket()
	client.connect((serverip,port))
	if sys.platform==‘win32‘:
		send=wininfo()
	else:
		send=linuxinfo()
	data = str(send.getname())+‘ ‘+str(send.ip_info())+‘ ‘+str(send.cpu_info())+‘ ‘+str(send.mem_info())+‘ ‘+str(send.disk_info())
	#print(data)
	client.sendall(bytes(data,encoding=‘utf-8‘))
	client.close()

# -*- coding:utf-8 -*-
#author:Guoyabin
import socketserver
import time

class recvpasswd(socketserver.BaseRequestHandler):
	def handle(self):
		conn=self.request #封装所有对象请求
		while True:
			date=conn.recv(1024).decode()
			if date==‘exit‘:
				break
			if len(date)==0: #空数据则断开
				break
			#if ‘=>‘ not in date: #防止恶意数据写入
			#	break
			print(date)
			self.savepasswd(date)
			
	def savepasswd(self,passwd):
		now=time.strftime("%Y-%m-%d %H:%M:%S")
		errfile=open(‘passwd.txt‘,‘a‘)
		errfile.write(now+‘ ‘+str(passwd)+‘\n‘)
		errfile.close()
		
if __name__==‘__main__‘:
	server=socketserver.ThreadingTCPServer((‘0.0.0.0‘,9999),recvpasswd)
	server.serve_forever()