python ssh 执行shell命令的示例

2020-10-10 0 754
# -*- coding: utf-8 -*-

import paramiko
import threading

def run(host_ip, username, password, command):
  ssh = paramiko.SSHClient()
  try:
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(host_ip, 22, username, password)

    print(\'===================exec on [%s]=====================\' % host_ip)


    stdin, stdout, stderr = ssh.exec_command(command, timeout=300)
    out = stdout.readlines()
  for o in out:
      print (o.strip(\'\\n\'))
  except Exception as ex:
    print(\'error, host is [%s], msg is [%s]\' % (host_ip, ex.message))
  finally:
    ssh.close()


if __name__ == \'__main__\':

  # 将需要批量执行命令的host ip地址填到这里
  # eg: host_ip_list = [\'IP1\', \'IP2\']

  host_ip_list = [\'147.116.20.19\']
  for _host_ip in host_ip_list:

    # 用户名,密码,执行的命令填到这里
    run(_host_ip, \'tzgame\', \'tzgame@1234\', \'df -h\')
    run(_host_ip, \'tzgame\', \'tzgame@1234\', \'ping -c 5 220.181.38.148\')

pycrypto,由于 paramiko 模块内部依赖pycrypto,所以先下载安装pycrypto

pip3 install pycrypto
pip3 install paramiko

(1)基于用户名和密码的连接

import paramiko

# 创建SSH对象
ssh = paramiko.SSHClient()

# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接服务器
ssh.connect(hostname=\'c1.salt.com\', port=22, username=\'GSuser\', password=\'123\')

# 执行命令
stdin, stdout, stderr = ssh.exec_command(\'ls\')

# 获取命令结果
result = stdout.read()

# 关闭连接
ssh.close()

(2)基于公钥秘钥连接

import paramiko

private_key = paramiko.RSAKey.from_private_key_file(\'/home/auto/.ssh/id_rsa\')

# 创建SSH对象
ssh = paramiko.SSHClient()

# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接服务器
ssh.connect(hostname=\'c1.salt.com\', port=22, username=\'wupeiqi\', key=private_key)

# 执行命令
stdin, stdout, stderr = ssh.exec_command(\'df\')

# 获取命令结果
result = stdout.read()

# 关闭连接
ssh.close()

SFTPClient:

  用于连接远程服务器并进行上传下载功能。

(1)基于用户名密码上传下载

import paramiko

transport = paramiko.Transport((\'hostname\',22))
transport.connect(username=\'GSuser\',password=\'123\')

sftp = paramiko.SFTPClient.from_transport(transport)

# 将location.py 上传至服务器 /tmp/test.py
sftp.put(\'/tmp/location.py\', \'/tmp/test.py\')

# 将remove_path 下载到本地 local_path
sftp.get(\'remove_path\', \'local_path\')

transport.close()

(2)基于公钥秘钥上传下载

import paramiko

private_key = paramiko.RSAKey.from_private_key_file(\'/home/auto/.ssh/id_rsa\')

transport = paramiko.Transport((\'hostname\', 22))
transport.connect(username=\'GSuser\', pkey=private_key )

sftp = paramiko.SFTPClient.from_transport(transport)

# 将location.py 上传至服务器 /tmp/test.py
sftp.put(\'/tmp/location.py\', \'/tmp/test.py\')

# 将remove_path 下载到本地 local_path
sftp.get(\'remove_path\', \'local_path\')

transport.close()

下面是多线程执行版本

#!/usr/bin/python
#coding:utf-8
import threading
import subprocess
import os
import sys


sshport = 13131
log_path = \'update_log\'
output = {}

def execute(s, ip, cmd, log_path_today):
  with s:   
    cmd = \'\'\'ssh -p%s root@%s -n \"%s\" \'\'\' % (sshport, ip, cmd)

    ret = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    output[ip] = ret.stdout.readlines()




if __name__ == \"__main__\":
  if len(sys.argv) != 3:
    print \"Usage: %s config.ini cmd\" % sys.argv[0]
    sys.exit(1)
                   
  if not os.path.isfile(sys.argv[1]):
    print \"Usage: %s is not file!\" % sys.argv[1]
    sys.exit(1)
                     
  cmd = sys.argv[2]
                   
  f = open(sys.argv[1],\'r\')
  list = f.readlines()
  f.close()
  today = datetime.date.today()
  log_path_today = \'%s/%s\' % (log_path,today)
  if not os.path.isdir(log_path_today):
    os.makedirs(log_path_today)
                   
  threading_num = 100
  if threading_num > len(list):
    threading_num = len(list)

  s = threading.Semaphore(threading_num)
                   
  for line in list:
    ip = line.strip()
    t = threading.Thread(target=execute,args=(s, ip,cmd,log_path_today))
    t.setDaemon(True)
    t.start()
                     
  main_thread = threading.currentThread()
  for t in threading.enumerate():
    if t is main_thread:
      continue
    t.join()
                     
  for ip,result in output.items():
    print \"%s: \" % ip
    for line in result:
      print \"  %s\" % line.strip()
                   
  print \"Done!\"

以上脚本读取两个参数,第一个为存放IP的文本,第二个为shell命令

执行效果如下

python ssh 执行shell命令的示例

# -*- coding:utf-8 -*-
import requests
from requests.exceptions import RequestException
import os, time
import re
from lxml import etree
import threading

lock = threading.Lock()
def get_html(url):

  response = requests.get(url, timeout=10)
  # print(response.status_code)
  try:
    if response.status_code == 200:

      # print(response.text)
      return response.text
    else:
       return None
  except RequestException:
    print(\"请求失败\")
    # return None


def parse_html(html_text):

  html = etree.HTML(html_text)

  if len(html) > 0:
    img_src = html.xpath(\"//img[@class=\'photothumb lazy\']/@data-original\") # 元素提取方法
    # print(img_src)
    return img_src

  else:
    print(\"解析页面元素失败\")

def get_image_pages(url):
  html_text = get_html(url) # 获取搜索url响应内容
  # print(html_text)
  if html_text is not None:
    html = etree.HTML(html_text) # 生成XPath解析对象
    last_page = html.xpath(\"//div[@class=\'pages\']//a[last()]/@href\") # 提取最后一页所在href链接
    print(last_page)
    if last_page:
      max_page = re.compile(r\'(\\d+)\', re.S).search(last_page[0]).group() # 使用正则表达式提取链接中的页码数字
      print(max_page)
      print(type(max_page))
      return int(max_page) # 将字符串页码转为整数并返回
    else:
      print(\"暂无数据\")
      return None
  else:
    print(\"查询结果失败\")


def get_all_image_url(page_number):
  base_url = \'https://imgbin.com/free-png/naruto/\'
  image_urls = []

  x = 1 # 定义一个标识,用于给每个图片url编号,从1递增
  for i in range(1, page_number):
    url = base_url + str(i) # 根据页码遍历请求url
    try:
      html = get_html(url) # 解析每个页面的内容
      if html:
        data = parse_html(html) # 提取页面中的图片url
        # print(data)
        # time.sleep(3)
        if data:
          for j in data:
            image_urls.append({
              \'name\': x,
              \'value\': j
            })
            x += 1 # 每提取一个图片url,标识x增加1
    except RequestException as f:
      print(\"遇到错误:\", f)
      continue
  # print(image_urls)
  return image_urls

def get_image_content(url):
  try:
    r = requests.get(url, timeout=15)
    if r.status_code == 200:
      return r.content
    return None
  except RequestException:
    return None

def main(url, image_name):
  semaphore.acquire() # 加锁,限制线程数
  print(\'当前子线程: {}\'.format(threading.current_thread().name))
  save_path = os.path.dirname(os.path.abspath(\'.\')) + \'/pics/\'
  try:
    file_path = \'{0}/{1}.jpg\'.format(save_path, image_name)
    if not os.path.exists(file_path): # 判断是否存在文件,不存在则爬取
      with open(file_path, \'wb\') as f:
        f.write(get_image_content(url))
        f.close()

        print(\'第{}个文件保存成功\'.format(image_name))

    else:
      print(\"第{}个文件已存在\".format(image_name))

    semaphore.release() # 解锁imgbin-多线程-重写run方法.py

  except FileNotFoundError as f:
    print(\"第{}个文件下载时遇到错误,url为:{}:\".format(image_name, url))
    print(\"报错:\", f)
    raise

  except TypeError as e:
    print(\"第{}个文件下载时遇到错误,url为:{}:\".format(image_name, url))
    print(\"报错:\", e)

class MyThread(threading.Thread):
  \"\"\"继承Thread类重写run方法创建新进程\"\"\"
  def __init__(self, func, args):
    \"\"\"

    :param func: run方法中要调用的函数名
    :param args: func函数所需的参数
    \"\"\"
    threading.Thread.__init__(self)
    self.func = func
    self.args = args

  def run(self):
    print(\'当前子线程: {}\'.format(threading.current_thread().name))
    self.func(self.args[0], self.args[1])
    # 调用func函数
    # 因为这里的func函数其实是上述的main()函数,它需要2个参数;args传入的是个参数元组,拆解开来传入


if __name__ == \'__main__\':
  start = time.time()
  print(\'这是主线程:{}\'.format(threading.current_thread().name))

  urls = get_all_image_url(5) # 获取所有图片url列表
  thread_list = [] # 定义一个列表,向里面追加线程
  semaphore = threading.BoundedSemaphore(5) # 或使用Semaphore方法
  for t in urls:
    # print(i)

    m = MyThread(main, (t[\"value\"], t[\"name\"])) # 调用MyThread类,得到一个实例

    thread_list.append(m)

  for m in thread_list:

    m.start() # 调用start()方法,开始执行

  for m in thread_list:
    m.join() # 子线程调用join()方法,使主线程等待子线程运行完毕之后才退出


  end = time.time()
  print(end-start)
  # get_image_pages(https://imgbin.com/free-png/Naruto)

以上就是python ssh 执行shell命令的示例的详细内容,更多关于python ssh 执行shell命令的资料请关注自学编程网其它相关文章!

遇见资源网 Python python ssh 执行shell命令的示例 http://www.ox520.com/27262.html

常见问题

相关文章

发表评论
暂无评论
官方客服团队

为您解决烦忧 - 24小时在线 专业服务