Batch operation of remote host scripts written in Python (command execution, file upload and download)

Source: Internet
Author: User
Tags ssh port

Batch operation of remote host scripts (command execution, file upload and download) Written in Python 12:53:44

Tags: Servers

Script
SSH
Python
Centralized management of original works, reprinting allowed, please be sure to mark the article in hyperlink form during Reprinting
Original source, author information, and this statement. Otherwise, legal liability will be held. Http://mayulin.blog.51cto.com/1628315/555459

I recently learned Python and used the fabric module to write a script for batch server operations. I would like to share it with you. If you have any shortcomings, please let me know.

Preparations:

Install Python 2.6.5:

Yum-y install Readline *

Tar xf Python-2.6.5.tar.bz2

CD Python-2.6.5

Remove the comments before "Readline. C-lreadline-ltermcap" in the modules/setup. Dist file in the directory.

Compile and install:

./Configure -- enable-shared

Make-J8 & make install

Install setuptools

Tar xf setuptools-0.6c11.tar.gz

CD setuptools-0.6c11

Python setup. py install

Install Fabric

During installation, the software will automatically find and install the dependent installation package from the Internet

Tar xf fabric-0.9rc2.tar.gz

CD goosemo-fabric-1eacbf2

Python setup. py install

 

######################################## ####################################

Pyssh script:

#! /Usr/bin/ENV Python
#-*-Coding: UTF-8 -*-

######################################## #############
# Author: mayulin-mayulin@cy2009.com
# Last modified:
# Filename: Pssh. py
######################################## #############

From fabric. API import ENV, run, put, get
From OS import path
From re import findall
From sys import argv
From fabric. context_managers import hide
From time import sleep

User = 'root'
Host, ip_list = [], []
Port = '22'
Pri_key, password, CMD, usrc, udst, DSRC, ddst = '','','','','','',''
Timeout = 1

For I in range (1, Len (argv) + 1 ):
If argv [I-1] = '-H' or Len (argv) = 1:
Print """
Usage:
-U [user] use this argument to specify the user, default is 'root'
-H [host] The host that you want to connect
-F the file content multiple IP address you want to connect
-P [port] the ssh port, default is 22
-P [PWD | file] You can specify password or a priviate key file to connect the host
-C [command] The command you want the host (s) to run
-U [SRC, DST] the local file that you want to upload to the remote host (s)
-D [SRC, DST] the remote file that you want to download to the local host
-T [timeout] the program running timeout, default is 1 (s)
-H print this help screen
"""

If argv [I-1] = '-U ':
User = argv [I]
Env. User = '% s' % (User)
Else:
Env. User = '% s' % (User)
If argv [I-1] = '-H ':
Arg = findall ('(\ D + \. \ D + \. \ D + \. \ D + | \ s + \. {3, 4}) ', argv [I])
For J in Arg:
If type (j). _ name __! = 'Nonetype ':
Host. append (j)
Else:
Print 'the hostip input error'
If argv [I-1] = '-P ':
Port = argv [I]
If argv [I-1] = '-F ':
If path. isfile ('% s' % (argv [I]) = true:
Ip_list = open ('% s' % (argv [I]), 'R'). readlines ()
If argv [I-1] = '-P ':
If path. isfile (argv [I]) = true:
Pri_key = argv [I]
Env. key_filename = '% s' % (pri_key)
Else:
Password = argv [I]
Env. Password = '% s' % (password)
If argv [I-1] = '-C ':
Cmd = argv [I]
If argv [I-1] = '-t ':
Timeout = argv [I]

SLP = 'sleep % s' % (timeout)

If argv [I-1] = '-U ':
X = src = \ '# \' "/> usrc = x [0]
Udst = x [1]

If argv [I-1] = '-d ':
Y = src = \ '# \' "/> DSRC = Y [0]
Ddst = Y [1]

Else:
Ip_port = []
If Len (ip_list )! = 0:
For k in ip_list:
Ip_port.append (K. Strip () + ':' + port)
If Len (host )! = 0:
For k in Host:
Ip_port.append (K. Strip () + ':' + port)
If cmd! = '':
Def command ():
With hide ('running '):
Run ("% s; % s" % (CMD, SLP ))
For IP in ip_port:
Env. host_string = IP

Print "Execute Command: \" % s \ "at host: % s" % (CMD, IP. Split (':') [0])
Print "-------------------------------------------------"
Command ()
Print "-------------------------------------------------"

If usrc and udst! = '':
Def upload ():
With hide ('running '):
Put ("% s" % (usrc), "% s" % (udst ))
For IP in ip_port:
Env. host_string = IP
Print "upload local file: \" % s \ "to host: % s \" % s \ "" % (usrc, IP. split (':') [0], udst)
Print "-------------------------------------------------"
Upload ()
Print "-------------------------------------------------"

If DSRC and ddst! = '':
Def download ():
With hide ('running '):
Get ("% s" % (DSRC), "% s" % (ddst ))
For IP in ip_port:
Env. host_string = IP
Print "Download Remote File: \" % s \ "from host: % s to Local \" % s \ "" % (DSRC, IP. split (':') [0], ddst)
Print "-------------------------------------------------"
Download ()
Print "-------------------------------------------------"
 

 

######################################## ######################################## ###########

Script Execution result:

Show Help:

Run the shell command on multiple remote hosts:

Download files from the remote host to the local directory:

Upload local files to multiple remote hosts

This article is from dream ~ Starting from here "blog, be sure to keep this source http://mayulin.blog.51cto.com/1628315/555459

Related Article

Contact Us

The content source of this page is from Internet, which doesn't represent Alibaba Cloud's opinion; products and services mentioned on that page don't have any relationship with Alibaba Cloud. If the content of the page makes you feel confusing, please write us an email, we will handle the problem within 5 days after receiving your email.

If you find any instances of plagiarism from the community, please send an email to: info-contact@alibabacloud.com and provide relevant evidence. A staff member will contact you within 5 working days.

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.