使用Python之paramiko模块和threading实现多线程登录多台Linux服务器
admin
2023-03-18 06:41:21
0

        有时候我们需要在多台Linux服务器上面执行同样的命令,或者同样的操作,如果我们每一台单独登录上去做这样太麻烦了,所以我们可以考虑使用自动化脚本来实现。我这里使用Python多线程的方式,这样速度更快,如果使用Shell只能一台完了执行另外一台效率不高。

        针对这样的需要写了一个Python脚本来完成这样的工作,大致实现了我需要的效果


实现代码:

#!/usr/bin/python
#*-*coding:utf8*-*

"""
此脚本适用于批量登录到Linux操作系统,并执行一些简单命令
要求所有服务器的用户名密码一样,或者密钥一样,都能使用同一种方法登录
执行的命令默认只输入第一行,如果要输出多行需要修改ssh_login函数
默认脚本是以用户名密码登录,也可以改为用密钥登录
选项特殊说明:
-c:后面跟要执行命令,命令如w,uptime,hostname,date等,如果命令中有空格,需要用双引号,如"cat /etc/hosts"
-a:后面跟主机,有三种写法192.168.1.31指单台主机,192.168.1.31,192.168.132,指定多台主机,把逗号换成“-”指定区间

如果有很多主机,但用户名密码不一样,登录方法也不一样,可以考虑把这些信息写入到一个文件,
通过对文件遍历来完成复杂环境的需求
"""

import paramiko
from optparse import OptionParser
import sys
import netaddr
import threading

#使用选项帮助信息可以使用中文
reload(sys)
sys.setdefaultencoding("utf-8")

#定义选项
usage = sys.argv[0] + " [选项]"
parser = OptionParser(usage)
parser.add_option('-c', '--commond',
                  dest='commond',
                  action='store',
                  default=False,
                  help='你要执行的命令,如:uptime')
parser.add_option('-a', '--host',
                  dest='host',
                  action='store',
                  default=False,
                  help='需要执行命令的主机,跟主机IP地址,多个IP用逗号分隔,也可以用“-”连接一个主机范围')
options, args = parser.parse_args()

def ssh_login(ip, commond):
    "登录并执行命令,可以更改为使用密钥登录"
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname=ip, port=22, username='root', password='p-0p-0p-0')
    stdin, stdout, stderr = ssh.exec_command(commond)
    print('%s\t%s' %(ip, stdout.readline().rstrip()))
    ssh.close()

if __name__ == '__main__':
    if not options.commond:
        print("请指定要执行的命令")
        exit(1)
    if not options.host:
        print("请指定主机")
        exit(1)

    if ',' in options.host:
        ip = options.host.split(',')
        for i in ip:
            t = threading.Thread(target=ssh_login, args=(i, options.commond))   #使用线程方式执行,更快
            t.start()
    elif '-' in options.host:
        startip = options.host.split('-')[0]
        endip = options.host.split('-')[1]
        ip = list(netaddr.IPRange(startip, endip))        #netaddr.IPRange()用于计算IP地址区间内的所有IP
        for i in ip:
            t = threading.Thread(target=ssh_login, args=(str(i), options.commond))
            t.start()
    else:
        ip = options.host
        ssh_login(ip, options.commond)


执行方式及结果如下:

使用Python之paramiko模块和threading实现多线程登录多台Linux服务器


.

上一篇:SVN服务器升级

下一篇:vsftp服务器搭建

相关内容

热门资讯

局地可达35℃以上!今年首轮高... 热起来了!今天(10日)起至13日,北方迎来今年首场大范围高温天气,大部地区气温将超30℃,刷新今年...
炒股精神病院去年被罚117万,... 精神病院作为投资主体,而且在云谲波诡的股市里打拼得很好,本身就是一个让人好奇的事情。现在既然投资行为...
这个法国人,真让人刮目相看 政坛老将,第四次冲击总统宝座。更让人刮目相看的,是他对华的清醒表态他就是梅朗雄,法国左翼政党的领导人...
台“军购条例”大幅缩水,美国果... 【文/观察者网 王恺雯】台立法机构5月8日表决通过蓝白版“军购特别条例”,预算上限由赖清德当局要求的...
奋斗华章丨平凡亦有光 奋斗即力... 奋斗华章丨平凡亦有光 奋斗即力量 五月的风,裹挟着劳动的滚烫与青春的赤诚,吹遍华夏大地。 当五一劳...
中国移动发布Token运营生态... 5月8日,中国移动在2026移动云大会期间举办“词元聚力 智享未来”Token运营发展论坛,联合腾讯...
台当局滥发文化补助,涉事歌词低... 叶元之海峡导报综合报道 台民意机构“教育及文化委员会”7日邀台当局文化主管部门进行业务报告并备询,国...
国乒男女团今天冲冠,一起迎战日... 北京时间5月10日(星期日),2026伦敦世乒赛团体赛决赛将迎来巅峰对决,男团、女团决赛将分别进行,...
瑞幸,杀到农夫山泉家门口 便利店的饮料货架上,出现了瑞幸这个熟悉又陌生的名字。近日,瑞幸正式推出全新瓶装即饮咖啡,一口气上架生...