python 运维常用脚本

Python 批量遍历目录文件,并修改访问时间

path="D:/UASM64/include/" dirs = os.listdir(path) for file in dirs: ... print(os.path.join(path,file))

import os

path = "D:/UASM64/include/" dirs = os.listdir(path) temp=[];

for file in dirs: temp.append(os.path.join(path, file)) for x in temp: os.utime(x, (1577808000, 1577808000)) Python 实现的自动化服务器管理

import sys import os import paramiko

ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

=============================================================================================================================

def ssh_cmd(user,passwd,port,userfile,cmd):

check_ok=[]
check_er=[]

file = open(userfile, "r")
line = file.readlines()
file.close()
for i in range(len(line)):
    print("-------------------------------------------->" + line[i].strip('\n') + "<--------------------------------------------\n" )
    try:
        ssh.connect(hostname=line[i].strip('\n'),port=port,username=user,password=passwd)
        cmd=cmd
        stdin, stdout, stderr = ssh.exec_command(cmd)
        result = stdout.read()

        if not result:
            result=stderr.read()
        ssh.close()

        print(result.decode())
        check_ok.append(line[i].strip('\n'))
    except Exception:
        check_er.append(line[i].strip('\n'))
        continue


print("\n\n\n--------------------------------------------------------------------------------------------------------")
print("主机IP\t\t\t端口\t\t\t命令\t\t\t状态\t\t\t")
print("--------------------------------------------------------------------------------------------------------")

for i in range(len(check_ok)):
    print(check_ok[i] + "\t\t"+port+"\t\t\t"+cmd+"\t\t[完成]")
for i in range(len(check_er)):
    print(check_er[i] + "\t\t"+ port +"\t\t\t"+ cmd +"\t\t[失败]")
print("--------------------------------------------------------------------------------------------------------")           

=============================================================================================================================

def ssh_put(user,passwd,source,target):

check_ok=[]
check_er=[]

file=open("./ip.txt","r")
line=file.readlines()
file.close()

for i in range(len(line)):
    print("============================================>" + line[i].strip('\n') + "<============================================" )
    try:
         transport = paramiko.Transport((line[i].strip('\n'), 22))
         transport.connect(username=user, password=passwd)
         sftp = paramiko.SFTPClient.from_transport(transport)
         sftp.put(source, target)
         check_ok.append(line[i].strip('\n'))
        
    except Exception:
        check_er.append(line[i].strip('\n'))
        continue
    
print("\n\n\n--------------------------------------------------------------------------------------------------------")
print("主机IP\t\t\t端口\t\t\t传输源\t\t传输到\t\t 状态 \n")
print("--------------------------------------------------------------------------------------------------------")
for i in range(len(check_ok)):
    print(check_ok[i] + "\t\t" + "22" + "\t\t" + source + "\t\t" + target + "\t\t" + "[完成]")
for i in range(len(check_er)):
    print(check_er[i] + "\t\t" + "22" + "\t\t" + source + "\t\t" + target + "\t\t" + "[失败]")
print("--------------------------------------------------------------------------------------------------------")

while True: try: shell=str(input("[Shell] # ")) if (shell == ""): continue elif (shell == "exit"): exit() elif (shell == "put"): ssh_put("root","123123","./a.py","/root/a.py") elif (shell =="cron"): temp=input("输入一个计划任务: ") temp1="(crontab -l; echo "+ temp + ") |crontab" ssh_cmd("root","123123","22","./user_ip.conf",temp1) elif (shell == "uncron"): temp=input("输入要删除的计划任务: ") temp1="crontab -l | grep -v " "+ temp + "|crontab" ssh_cmd("root","123123","22","./user_ip.conf",temp1) else: ssh_cmd("lyshark","123123","22","./user_ip.conf",shell)

except Exception:
    continue

遍历目录和文件

import os

def list_all_files(rootdir): import os _files = [] list = os.listdir(rootdir) #列出文件夹下所有的目录与文件 for i in range(0,len(list)): path = os.path.join(rootdir,list[i]) if os.path.isdir(path): _files.extend(list_all_files(path)) if os.path.isfile(path): _files.append(path) return _files

a=list_all_files("C:/Users/LyShark/Desktop/a") print(a) python检测指定端口状态

import socket

sk = socket.socket(socket.AF_INET,socket.SOCK_STREAM) sk.settimeout(1)

for ip in range(0,254): try: sk.connect(("192.168.1."+str(ip),443)) print("192.168.1.%d server open \n"%ip) except Exception: print("192.168.1.%d server not open"%ip)

sk.close()

python实现批量执行CMD命令

import sys import os import paramiko

ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

print("------------------------------>\n") print("使用说明,在当前目录创建ip.txt写入ip地址") print("------------------------------>\n")

user=input("输入用户名:") passwd=input("输入密码:") port=input("输入端口:") cmd=input("输入执行的命令:")

file = open("./ip.txt", "r") line = file.readlines()

for i in range(len(line)): print("对IP: %s 执行"%line[i].strip('\n'))

    ssh.connect(hostname=line[i].strip('\n'),port=port,username=user,password=passwd)
    stdin, stdout, stderr = ssh.exec_command(cmd)
    result = stdout.read()

    if not result:
        result=stderr.read()
    ssh.close()

    print(result.decode())

python3-实现钉钉报警

import requests import sys import json

dingding_url = 'https://oapi.dingtalk.com/robot/send?access_token=6d11af3252812ea50410c2ccb861814a6ed11b2306606934a5d4ca9f2ec8c09'

data = {"msgtype": "markdown","markdown": {"title": "监控","text": "apche异常"}}

headers = {'Content-Type':'application/json;charset=UTF-8'}

send_data = json.dumps(data).encode('utf-8') requests.post(url=dingding_url,data=send_data,headers=headers)

coding: utf-8

import psutil import requests import time import os import json

monitor_name = set(['httpd','cobblerd']) # 用户指定监控的服务进程名称

proc_dict = {} proc_name = set() # 系统检测的进程名称 monitor_map = { 'httpd': 'systemctl restart httpd', 'cobblerd': 'systemctl restart cobblerd' # 系统在进程down掉后，自动重启 }

dingding_url = 'https://oapi.dingtalk.com/robot/send?access_token=b5258c4335ed8ab792075013c965efcbf4f8940f92e7bd936cdc7842d3bf9405'

钉钉机器人token使用参考文档：http://www.pc6.com/infoview/Article_108931.html

while True: for proc in psutil.process_iter(attrs=['pid','name']): proc_dict[proc.info['pid']] = proc.info['name'] proc_name.add(proc.info['name'])

proc_stop = monitor_name - proc_name  # 通过集合的形式来找出停掉的进程名,前者有但是后者没有的

if proc_stop: # 如果确实有监控的进程停掉了，那么我们需要告警以及自动重启功能
    for p in proc_stop:
        p_status = '停止'
        p_name = p
        data = {
            "msgtype": "markdown",
            "markdown": {
                "title": "监控信息",
                "text": "### %s\n" % time.strftime("%Y-%m-%d %X") +
                "> #### 服务名：%s \n\n" % p_name +
                "> #### 状态：%s \n\n" % p_status +
                "> #### 正在尝试启动"
            },
        }
        headers = {'Content-Type':'application/json;charset=UTF-8'}
        send_data = json.dumps(data).encode('utf-8')
        requests.post(url=dingding_url,data=send_data,headers=headers)

        os.system(monitor_map[p_name]) # 执行重启命令，然后判断是否重启成功
        proc_set = set()
        for proc_again in psutil.process_iter(attrs=['pid','name']):
            proc_set.add(proc_again.info['name'])

        if p in proc_set: # 如果进程启动成功，p是以前停掉的进程，proc_set是已经重启过一次后的所有进程集合
            p_status = '成功'
            p_name = p
            data = {
                "msgtype": "markdown",
                "markdown": {
                    "title": "监控信息",
                    "text": "### %s\n" % time.strftime("%Y-%m-%d %X") +
                    "> #### 服务名：%s \n\n" % p_name +
                    "> #### 状态：%s \n\n" % p_status +
                    "> #### 已经启动成功，服务正在运行！"
                },
            }
            headers = {'Content-Type':'application/json;charset=UTF-8'}
            send_data = json.dumps(data).encode('utf-8')
            requests.post(url=dingding_url,data=send_data,headers=headers)
        else:
            p_status = '重启失败'
            p_name = p
            data = {
                "msgtype": "markdown",
                "markdown": {
                    "title": "监控信息",
                    "text": "### %s\n" % time.strftime("%Y-%m-%d %X") +
                    "> #### 服务名：%s \n\n" % p_name +
                    "> #### 状态：%s \n\n" % p_status +
                    "> #### Sorry,服务启动失败鸟！"
                },
            }
            headers = {'Content-Type':'application/json;charset=UTF-8'}
            send_data = json.dumps(data).encode('utf-8')
            requests.post(url=dingding_url,data=send_data,headers=headers)
time.sleep(5)

判断指定端口是否开放

import socket

port_number = [135,443,80]

for index in port_number: sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) result = sock.connect_ex(('127.0.0.1', index)) if result == 0: print("Port %d is open" % index) else: print("Port %d is not open" % index) sock.close()

判断指定端口并且实现钉钉轮询报警

By LyShark

import requests import sys import json import socket import time

def dingding(title,text): dingding_url = 'https://oapi.dingtalk.com/robot/send?access_token=6d11af3252812ea50410c2ccb861814a69ed11b2306606934a5d4ca9f2c8c09' data = {"msgtype": "markdown","markdown": {"title": title,"text": text}} headers = {'Content-Type':'application/json;charset=UTF-8'} send_data = json.dumps(data).encode('utf-8') requests.post(url=dingding_url,data=send_data,headers=headers)

def net_scan(): port_number = [80,135,443] for index in port_number: sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) result = sock.connect_ex(('127.0.0.1', index)) if result == 0: print("Port %d is open" % index) else: return index sock.close()

while True: dingding("Warning",net_scan()) time.sleep(60)

python-实现SSH批量CMD执行命令

By:LyShark

import sys import os import paramiko

ssh = paramiko.SSHClient() ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

def ssh_cmd(user,passwd,port,userfile,cmd): file = open(userfile, "r") line = file.readlines() for i in range(len(line)): print("对IP: %s 执行"%line[i].strip('\n')) ssh.connect(hostname=line[i].strip('\n'),port=port,username=user,password=passwd) cmd=cmd stdin, stdout, stderr = ssh.exec_command(cmd) result = stdout.read()

    if not result:
        result=stderr.read()
    ssh.close()

    print(result.decode())

ssh_cmd("lyshark","123","22","./ip.txt","free -h |grep 'Mem:' |awk '{print $3}'")

用python写一个列举当前目录以及所有子目录下的文件，并打印出绝对路径

import sys import os

for root,dirs,files in os.walk("C://"): for name in files: print(os.path.join(root,name)) os.walk()

按照这样的日期格式（xxxx-xx-xx）每日生成一个文件，例如今天生成的文件为2013-09-23.log， 并且把磁盘的使用情况写到到这个文件中。

import os import sys import time

new_time = time.strftime("%Y-%m-%d") disk_status = os.popen("df -h").readlines()

str1 = ''.join(disk_status) f = open(new_time+'.log','w') f.write("%s"%str1)

f.flush() f.close()

统计出每个IP的访问量有多少？（从日志文件中查找）

import sys

list = []

f = open("/var/log/httpd/access_log","r") str1 = f.readlines() f.close()

for i in str1: ip=i.split()[0] list.append(ip)

list_num=set(list)

for j in list_num: num=list.count(j) print("%s -----> %s" %(num,j))

写个程序，接受用户输入数字，并进行校验，非数字给出错误提示，然后重新等待用户输入。

根据用户输入数字，输出从0到该数字之间所有的素数。(只能被1和自身整除的数为素数）

import tab import sys

while True: try: num=int(input("输入数字：").strip()) for x in range(2,num+1): for y in range(2,x): if x % y == 0: break else: print(x) except ValueError: print("您输入的不是数字") except KeyboardInterrupt: sys.exit("\n")

ps 可以查看进程的内存占用大小，写一个脚本计算一下所有进程所占用内存大小的和。

（提示，使用ps aux 列出所有进程，过滤出RSS那列，然后求和）

import sys import os

list=[] sum=0

str1=os.popen("ps aux","r").readlines()

for i in str1: str2=i.split() new_rss=str2[5] list.append(new_rss) for i in list[1:-1]: num=int(i) sum=sum+num

print("%s ---> %s"%(list[0],sum))

关于Python 命令行参数argv

import sys

if len(sys.argv) < 2: print ("没有输入任何参数") sys.exit()

if sys.argv[1].startswith("-"): option = sys.argv[1][1:]

if option == "version":
    print ("版本信息")
elif option == "help":
    print ("帮助菜单")
elif option == "option":
    print("配置菜单")
else:
    print ("异常")
    sys.exit()

利用random生成6位数字加字母随机验证码

import sys import random

rand=[]

for x in range(6): y=random.randrange(0,5) if y == 2 or y == 4: num=random.randrange(0,9) rand.append(str(num)) else: temp=random.randrange(65,91) c=chr(temp) rand.append(c) result="".join(rand) print(result)

自动化-使用pexpect非交互登陆系统

import pexpect import sys

ssh = pexpect.spawn('ssh lyshark@59.110.167.239') fout = file('sshlog.txt', 'w') ssh.logfile = fout

ssh.expect("lyshark@59.110.167.239's password:")

ssh.sendline("密码") ssh.expect('#')

ssh.sendline('ls /home') ssh.expect('#')

Python-取系统时间

import sys import time

time_str = time.strftime("日期：%Y-%m-%d",time.localtime()) print(time_str)

time_str= time.strftime("时间：%H:%M",time.localtime()) print(time_str)

psutil-获取内存使用情况

import sys import os import psutil

获取系统内存使用情况

memory_convent = 1024 * 1024 mem =psutil.virtual_memory()

print("内存容量为:"+str(mem.total/(memory_convent))+"MB\n") print("已使用内存:"+str(mem.used/(memory_convent))+"MB\n") print("可用内存:"+str(mem.total/(memory_convent)-mem.used/(1024*1024))+"MB\n") print("buffer容量:"+str(mem.buffers/( memory_convent ))+"MB\n") print("cache容量:"+str(mem.cached/(memory_convent))+"MB\n")

Python-通过SNMP协议监控CPU 注意：被监控的机器上需要支持snmp协议 yum install -y net-snmp*

!/usr/bin/python

import os

def getAllitems(host, oid): sn1 = os.popen('snmpwalk -v 2c -c public ' + host + ' ' + oid + '|grep Raw|grep Cpu|grep -v Kernel').read().split('\n')[:-1] return sn1

def getDate(host): items = getAllitems(host, '.1.3.6.1.4.1.2021.11')

    date = []
    rate = []
    cpu_total = 0

    #us = us+ni, sy = sy + irq + sirq

    for item in items:
            float_item = float(item.split(' ')[3])
            cpu_total += float_item
            if item == items[0]:
                    date.append(float(item.split(' ')[3]) + float(items[1].split(' ')[3]))
            elif item == item[2]:
                    date.append(float(item.split(' ')[3] + items[5].split(' ')[3] + items[6].split(' ')[3]))
            else:
                    date.append(float_item)

    #calculate cpu usage percentage

    for item in date:
            rate.append((item/cpu_total)*100)

    mean = ['%us','%ni','%sy','%id','%wa','%cpu_irq','%cpu_sIRQ']

    #calculate cpu usage percentage
    result = map(None,rate,mean)
    return result

if name == 'main':

    hosts = ['192.168.1.17','192.168.1.17']

    for host in hosts:
            print '==========' + host + '=========='
            result = getDate(host)
            print 'Cpu(s)',
            #print result
            for i in range(5):
                    print ' %.2f%s' % (result[i][0],result[i][1]),
            print
            print

Python-通过SNMP协议监控系统负载 注意：被监控的机器上需要支持snmp协议 yum install -y net-snmp*

!/usr/bin/python

import os import sys

def getAllitems(host, oid): sn1 = os.popen('snmpwalk -v 2c -c public ' + host + ' ' + oid).read().split('\n') return sn1

def getload(host,loid): load_oids = '1.3.6.1.4.1.2021.10.1.3.' + str(loid) return getAllitems(host,load_oids)[0].split(':')[3]

if name == 'main':

    hosts = ['192.168.1.17','192.168.1.17']

    print ('==============System Load==============')

    for host in hosts:
            load1 = getload(host, 1)
            load10 = getload(host, 2)
            load15 = getload(host, 3)
            print ('%s load(1min): %s ,load(10min): %s ,load(15min): %s' % (host,load1,load10,load15))

Python-通过SNMP协议监控内存 注意：被监控的机器上需要支持snmp协议 yum install -y net-snmp*

!/usr/bin/python

import os

def getAllitems(host, oid):

    sn1 = os.popen('snmpwalk -v 2c -c public ' + host + ' ' + oid).read().split('\n')[:-1]
    return sn1

def getSwapTotal(host):

    swap_total = getAllitems(host, 'UCD-SNMP-MIB::memTotalSwap.0')[0].split(' ')[3]
    return swap_total

def getSwapUsed(host):

    swap_avail = getAllitems(host, 'UCD-SNMP-MIB::memAvailSwap.0')[0].split(' ')[3]
    swap_total = getSwapTotal(host)
    swap_used = str(round(((float(swap_total)-float(swap_avail))/float(swap_total))*100 ,2)) + '%'
    return swap_used

def getMemTotal(host):

    mem_total = getAllitems(host, 'UCD-SNMP-MIB::memTotalReal.0')[0].split(' ')[3]
    return mem_total

def getMemUsed(host):

    mem_total = getMemTotal(host)
    mem_avail = getAllitems(host, 'UCD-SNMP-MIB::memAvailReal.0')[0].split(' ')[3]
    mem_used = str(round(((float(mem_total)-float(mem_avail))/float(mem_total))*100 ,2)) + '%'
    return mem_used

if name == 'main':

    hosts = ['192.168.1.17','192.168.1.17']
    print ("Monitoring Memory Usage")

    for host in hosts:

            mem_used = getMemUsed(host)
            swap_used = getSwapUsed(host)

            print ('==========' + host + '==========')
            print ('Mem_Used = %-15s  Swap_Used = %-15s' %(mem_used,swap_used))
            print()

Python-通过SNMP协议监控磁盘 注意：被监控的机器上需要支持snmp协议 yum install -y net-snmp*

!/usr/bin/python

import re import os

def getAllitems(host,oid):

    sn1 = os.popen('snmpwalk -v 2c -c public ' + host + ' ' + oid).read().split('\n')[:-1]
    return sn1

def getDate(source,newitem):

    for item in source[5:]:
            newitem.append(item.split(':')[3].strip())
    return newitem

def getRealDate(item1,item2,listname):

    for i in range(len(item1)):
            listname.append(int(item1[i])*int(item2[i])/1024)
    return listname

def caculateDiskUsedRate(host):

    hrStorageDescr = getAllitems(host, 'HOST-RESOURCES-MIB::hrStorageDescr')
    hrStorageUsed = getAllitems(host, 'HOST-RESOURCES-MIB::hrStorageUsed')
    hrStorageSize = getAllitems(host, 'HOST-RESOURCES-MIB::hrStorageSize')
    hrStorageAllocationUnits = getAllitems(host, 'HOST-RESOURCES-MIB::hrStorageAllocationUnits')


    disk_list = []
    hrsused = []
    hrsize = []
    hrsaunits = []
                                                         
    #get disk_list

    for item in hrStorageDescr:
            if re.search('/',item):
                    disk_list.append(item.split(':')[3])

    #print disk_list      

                                                         

    getDate(hrStorageUsed,hrsused)
    getDate(hrStorageSize,hrsize)

    #print getDate(hrStorageAllocationUnits,hrsaunits)

                                                         

    #get hrstorageAllocationUnits

    for item in hrStorageAllocationUnits[5:]:
            hrsaunits.append(item.split(':')[3].strip().split(' ')[0])

    #caculate the result

    #disk_used = hrStorageUsed * hrStorageAllocationUnits /1024 (KB)

    disk_used = []
    total_size = []
    disk_used = getRealDate(hrsused,hrsaunits,disk_used)
    total_size = getRealDate(hrsize,hrsaunits,total_size)

                                                         

    diskused_rate = []

    for i in range(len(disk_used)):
            diskused_rate.append(str(round((float(disk_used[i])/float(total_size[i])*100), 2)) + '%')

    return diskused_rate,disk_list

if name == 'main':

    hosts = ['192.168.1.17','192.168.1.17']

    for host in hosts:

            result = caculateDiskUsedRate(host)
            diskused_rate = result[0]
            partition = result[1]

            print ("==========",host,'==========')

            for i in range(len(diskused_rate)):
                    print ('%-20s used: %s' %(partition[i],diskused_rate[i]))
            print()

Python-通过SNMP协议监控网卡流量 注意：被监控的机器上需要支持snmp协议 yum install -y net-snmp*

!/usr/bin/python

import re import os

get SNMP-MIB2 of the devices

def getAllitems(host,oid): sn1 = os.popen('snmpwalk -v 2c -c public ' + host + ' ' + oid).read().split('\n')[:-1] return sn1

get network device

def getDevices(host): device_mib = getAllitems(host,'RFC1213-MIB::ifDescr') device_list = []

    for item in device_mib:
            if re.search('eth',item):
                    device_list.append(item.split(':')[3].strip())
    return device_list

get network date

def getDate(host,oid): date_mib = getAllitems(host,oid)[1:] date = []

    for item in date_mib:
            byte = float(item.split(':')[3].strip())
            date.append(str(round(byte/1024,2)) + ' KB')
    return date

if name == 'main':

    hosts = ['192.168.1.17','192.168.1.17']

    for host in hosts:
            device_list = getDevices(host)
            inside = getDate(host,'IF-MIB::ifInOctets')
            outside = getDate(host,'IF-MIB::ifOutOctets')

            print '==========' + host + '=========='

            for i in range(len(inside)):
                    print '%s : RX: %-15s   TX: %s ' % (device_list[i], inside[i], outside[i])
            print

Python-实现多级菜单

import os import sys

ps="[None]->" ip=["192.168.1.1","192.168.1.2","192.168.1.3"] flage=1

while True: ps="[None]->" temp=input(ps) if (temp=="test"): print("test page !!!!") elif(temp=="user"): while (flage == 1): ps="[User]->" temp1=input(ps) if(temp1 =="exit"): flage=0 break elif(temp1=="show"): for i in range(len(ip)): print(i)

Python实现一个没用的东西

import sys

ps="[root@localhost]# " ip=["192.168.1.1","192.168.1.2","192.168.1.3"]

while True: temp=input(ps) temp1=temp.split()

try:
    
    if(temp=="show"):
        for i in range(len(ip)):
            print(ip[i])
    elif( temp1[0] == "user" and temp1[1] != "" ):
        print("您的执行参数是："+temp1[1])
except Exception:
    continue

检查各个进程读写的磁盘IO

!/usr/bin/env python

-- coding=utf-8 --

import sys import os import time import signal import re

class DiskIO: def init(self, pname=None, pid=None, reads=0, writes=0): self.pname = pname self.pid = pid self.reads = 0 self.writes = 0

def main(): argc = len(sys.argv) if argc != 1: print ("usage: please run this script like [./lyshark.py]") sys.exit(0) if os.getuid() != 0: print ("Error: This script must be run as root") sys.exit(0) signal.signal(signal.SIGINT, signal_handler) os.system('echo 1 > /proc/sys/vm/block_dump') print ("TASK PID READ WRITE") while True: os.system('dmesg -c > /tmp/diskio.log') l = [] f = open('/tmp/diskio.log', 'r') line = f.readline() while line: m = re.match( '^(\S+)(\d+)(\d+): (READ|WRITE) block (\d+) on (\S+)', line) if m != None: if not l: l.append(DiskIO(m.group(1), m.group(2))) line = f.readline() continue found = False for item in l: if item.pid == m.group(2): found = True if m.group(3) == "READ": item.reads = item.reads + 1 elif m.group(3) == "WRITE": item.writes = item.writes + 1 if not found: l.append(DiskIO(m.group(1), m.group(2))) line = f.readline() time.sleep(1) for item in l: print ("%-10s %10s %10d %10d" % (item.pname, item.pid, item.reads, item.writes)) def signal_handler(signal, frame): os.system('echo 0 > /proc/sys/vm/block_dump') sys.exit(0)

if name=="main": main()

利用Pexpect实现自动非交互登陆linux

!/usr/bin/env python

-- coding: utf-8 --

import pexpect import sys

ssh = pexpect.spawn('ssh root@59.110.167.239') fout = file('sshlog.log', 'w') ssh.logfile = fout

ssh.expect("root@59.110.167.239's password:")

ssh.sendline("密码")

ssh.expect('#') ssh.sendline('ls /home') ssh.expect('#')

利用psutil模块获取系统的各种统计信息

import sys import psutil import time import os

获取当前时间

time_str = time.strftime( "%Y-%m-%d", time.localtime( ) ) file_name = "./" + time_str + ".log"

if os.path.exists ( file_name ) == False : os.mknod( file_name ) handle = open ( file_name , "w" ) else : handle = open ( file_name , "a" )

获取命令行参数的个数

if len( sys.argv ) == 1 : print_type = 1 else : print_type = 2

def isset ( list_arr , name ) : if name in list_arr : return True else : return False

print_str = "";

获取系统内存使用情况

if ( print_type == 1 ) or isset( sys.argv,"mem" ) : memory_convent = 1024 * 1024 mem = psutil.virtual_memory() print_str += " 内存状态如下:\n" print_str = print_str + " 系统的内存容量为: "+str( mem.total/( memory_convent ) ) + " MB\n" print_str = print_str + " 系统的内存以使用容量为: "+str( mem.used/( memory_convent ) ) + " MB\n" print_str = print_str + " 系统可用的内存容量为: "+str( mem.total/( memory_convent ) - mem.used/( 1024*1024 )) + "MB\n" print_str = print_str + " 内存的buffer容量为: "+str( mem.buffers/( memory_convent ) ) + " MB\n" print_str = print_str + " 内存的cache容量为:" +str( mem.cached/( memory_convent ) ) + " MB\n"

获取cpu的相关信息

if ( print_type == 1 ) or isset( sys.argv,"cpu" ) : print_str += " CPU状态如下:\n" cpu_status = psutil.cpu_times() print_str = print_str + " user = " + str( cpu_status.user ) + "\n" print_str = print_str + " nice = " + str( cpu_status.nice ) + "\n" print_str = print_str + " system = " + str( cpu_status.system ) + "\n" print_str = print_str + " idle = " + str ( cpu_status.idle ) + "\n" print_str = print_str + " iowait = " + str ( cpu_status.iowait ) + "\n" print_str = print_str + " irq = " + str( cpu_status.irq ) + "\n" print_str = print_str + " softirq = " + str ( cpu_status.softirq ) + "\n" print_str = print_str + " steal = " + str ( cpu_status.steal ) + "\n" print_str = print_str + " guest = " + str ( cpu_status.guest ) + "\n"

查看硬盘基本信息

if ( print_type == 1 ) or isset ( sys.argv,"disk" ) : print_str += " 硬盘信息如下:\n" disk_status = psutil.disk_partitions() for item in disk_status : print_str = print_str + " "+ str( item ) + "\n"

查看当前登录的用户信息

if ( print_type == 1 ) or isset ( sys.argv,"user" ) : print_str += " 登录用户信息如下:\n " user_status = psutil.users() for item in user_status : print_str = print_str + " "+ str( item ) + "\n"

print_str += "---------------------------------------------------------------\n" print ( print_str ) handle.write( print_str ) handle.close()

输出内存使用情况（以字节为单位）

import psutil

mem = psutil.virtual_memory() print mem.total,mem.used,mem print psutil.swap_memory() # 输出获取SWAP分区信息

输出CPU使用情况

cpu = psutil.cpu_stats() printcpu.interrupts,cpu.ctx_switches

psutil.cpu_times(percpu=True) # 输出每个核心的详细CPU信息 psutil.cpu_times().user # 获取CPU的单项数据 [用户态CPU的数据] psutil.cpu_count() # 获取CPU逻辑核心数，默认logical=True psutil.cpu_count(logical=False) # 获取CPU物理核心数

输出磁盘信息

psutil.disk_partitions() # 列出全部的分区信息 psutil.disk_usage('/') # 显示出指定的挂载点情况【字节为单位】 psutil.disk_io_counters() # 磁盘总的IO个数 psutil.disk_io_counters(perdisk=True) # 获取单个分区IO个数

输出网卡信息

psutil.net_io_counter() 获取网络总的IO，默认参数pernic=False psutil.net_io_counter(pernic=Ture)获取网络各个网卡的IO

获取进程信息

psutil.pids() # 列出所有进程的pid号 p = psutil.Process(2047) p.name() 列出进程名称 p.exe() 列出进程bin路径 p.cwd() 列出进程工作目录的绝对路径 p.status()进程当前状态[sleep等状态] p.create_time() 进程创建的时间 [时间戳格式] p.uids() p.gids() p.cputimes() 【进程的CPU时间，包括用户态、内核态】 p.cpu_affinity() # 显示CPU亲缘关系 p.memory_percent() 进程内存利用率 p.meminfo() 进程的RSS、VMS信息 p.io_counters() 进程IO信息，包括读写IO数及字节数 p.connections() 返回打开进程socket的namedutples列表 p.num_threads() 进程打开的线程数

下面的例子中，Popen类的作用是获取用户启动的应用程序进程信息，以便跟踪程序进程的执行情况

import psutil from subprocess import PIPE p =psutil.Popen(["/usr/bin/python" ,"-c","print 'helloworld'"],stdout=PIPE) p.name() p.username() p.communicate() p.cpu_times()

其它

psutil.users() # 显示当前登录的用户，和Linux的who命令差不多

获取开机时间

psutil.boot_time() 结果是个UNIX时间戳，下面我们来转换它为标准时间格式，如下： datetime.datetime.fromtimestamp(psutil.boot_time()) # 得出的结果不是str格式，继续进行转换 datetime.datetime.fromtimestamp(psutil.boot_time()).strftime('%Y-%m-%d%H:%M:%S')

Python生成一个随机密码

!/usr/bin/env python

-- coding:utf-8 --

import random, string def GenPassword(length):

#随机出数字的个数
numOfNum = random.randint(1,length-1)
numOfLetter = length - numOfNum

#选中numOfNum个数字
slcNum = [random.choice(string.digits) for i in range(numOfNum)]

#选中numOfLetter个字母
slcLetter = [random.choice(string.ascii_letters) for i in range(numOfLetter)]

#打乱这个组合
slcChar = slcNum + slcLetter
random.shuffle(slcChar)
#生成密码
genPwd = ''.join([i for i in slcChar])
return genPwd

if name == 'main': print (GenPassword(6))