zoukankan      html  css  js  c++  java
  • Python实现分发数据块到多台服务器上

    代码如下:

    # coding: utf-8
     
    import paramiko
    import re
    import os
    from time import sleep
     
    # 定义一个类,表示一台远端linux主机
    # 参考https://www.cnblogs.com/haigege/p/5517422.html wyc
    class Linux(object):
        # 通过IP, 用户名,密码,超时时间初始化一个远程Linux主机
        def __init__(self, ip, username, password, timeout=3000):
            self.ip = ip
            self.username = username
            self.password = password
            self.timeout = timeout
            # transport和chanel
            self.t = ''
            self.chan = ''
            # 链接失败的重试次数
            self.try_times = 3
     
        # 调用该方法连接远程主机
        def connect(self):
            while True:
                # 连接过程中可能会抛出异常,比如网络不通、链接超时
                try:
                    self.t = paramiko.Transport(sock=(self.ip, 22))
                    self.t.connect(username=self.username, password=self.password)
                    self.chan = self.t.open_session()
                    self.chan.settimeout(self.timeout)
                    self.chan.get_pty()
                    self.chan.invoke_shell()
                    # 如果没有抛出异常说明连接成功,直接返回
                    print (u'连接%s成功' % self.ip)
                    # 接收到的网络数据解码为str
                    print (self.chan.recv(65535).decode('utf-8'))
                    return
                # 这里不对可能的异常如socket.error, socket.timeout细化,直接一网打尽
                except Exception as e1:
                    if self.try_times != 0:
                        print (u'连接%s失败,进行重试' %self.ip)
                        self.try_times -= 1
                    else:
                        print (u'重试3次失败,结束程序')
                        exit(1)
     
        # 断开连接
        def close(self):
            self.chan.close()
            self.t.close()
     
        # 发送要执行的命令
        def send(self, cmd):
            cmd += '
    '
            # 通过命令执行提示符来判断命令是否执行完成
            p = re.compile(r':~ #')
     
            result = ''
            # 发送要执行的命令
            self.chan.send(cmd)
            # 回显很长的命令可能执行较久,通过循环分批次取回回显
            while True:
                sleep(0.5)
                ret = self.chan.recv(65535)
                ret = ret.decode('utf-8')
                result += ret
                if p.search(ret):
                    print (result)
                    return result
        # ------获取本地指定目录及其子目录下的所有文件------
        def __get_all_files_in_local_dir(self, local_dir):
            # 保存所有文件的列表
            all_files = list()
     
            # 获取当前指定目录下的所有目录及文件,包含属性值
            files = os.listdir(local_dir)
            for x in files:
                # local_dir目录中每一个文件或目录的完整路径
                filename = os.path.join(local_dir, x)
                # 如果是目录,则递归处理该目录
                if os.path.isdir(x):
                    all_files.extend(self.__get_all_files_in_local_dir(filename))
                else:
                    all_files.append(filename)
            return all_files
     
        def sftp_put_dir(self, local_dir, remote_dir):
            t = paramiko.Transport(sock=(self.ip, 22))
            t.connect(username=self.username, password=self.password)
            sftp = paramiko.SFTPClient.from_transport(t)
     
            # 去掉路径字符穿最后的字符'/',如果有的话
            if remote_dir[-1] == '/':
                remote_dir = remote_dir[0:-1]
     
            # 获取本地指定目录及其子目录下的所有文件
            all_files = self.__get_all_files_in_local_dir(local_dir)
            # 依次put每一个文件
            for x in all_files:
                filename = os.path.split(x)[-1]
                remote_filename = remote_dir + '/' + filename
                print (x)
                print (remote_filename)
                print (u'Put文件%s传输到%s中...' % (filename,self.ip))
                sftp.put(x, remote_filename)
    if __name__ == '__main__':
        # host = Linux('172.16.10.40', 'root', 'imlytek!40')
        # host.connect()
        # host.send('ls -l')
        remote_path = r'/imlytek/smpt'
        local_path = r'D:TSTP'
        # host.sftp_put_dir(local_path, remote_path)
        hostArray=[['172.16.10.40','root','iflstek!40'],['172.16.10.51','root','x1f@2013']]
        for x in hostArray:
          host = Linux(x[0], x[1], x[2])
          host.sftp_put_dir(local_path, remote_path)
        #host.close()
  • 相关阅读:
    如何避免delete和delete[]的尴尬?
    笔面集锦:判断单链表里面是否有环及相关扩展题(转)
    各排序算法的C++实现与性能测试(转)
    五个好的C语言编程实践
    Reviewboard管理员指南(5.5)—— Permission Groups(重要)
    Jenkins CLI getjob与RoleBased Strategy的那点事
    Reviewboard管理员指南(5.4)—— Default Reviewers(重要)
    Reviewboard管理员指南(4.2)—— Administrator Dashboard
    Maven Nexus admin密码重置的方法
    Reviewboard管理员指南(5.2)—— Access Control(重要)
  • 原文地址:https://www.cnblogs.com/lijinze-tsinghua/p/10003842.html
Copyright © 2011-2022 走看看