Course4-Python ftp/ssh

1. ftp

Python 自带有模块支持ftp. 可以参看一下代码。

 #!/usr/bin/env python
 import sys
 import os
 import time
 import getopt
 import traceback
 from ftplib import FTP

 class ftp_up_down():
         def __init__(self):
                 pass
         def captureException(self,logname='ftp_log.log'):
                 logs = open(logname, 'a')
                 print >> logs, time.strftime('%Y-%m-%d %X', time.localtime())
                 err = ''.join(traceback.format_exception(*sys.exc_info()))
                 print >> logs, err
                 logs.close()

         def logging(self,mystring, filename='ftp_log.log'):

                 try:
                     print time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()) + " " + mystring
                     fso = open(filename, 'a')
                     fso.write(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()) + " " + mystring + '\n' + '\n')
                     fso.close()
                 except:
                     captureException()

         def ftp_up(self,ServerFolder, LocalLocation, FileName, ServerAddr, user, password):
             try:
                 time1= time.time()
                 bufsize = 2048#set buffer size
                 ftp=FTP()
                 #ftp.set_debuglevel(2)#open debug level, 2 display the detailed info, 0 close the debug info
                 ftp.connect(ServerAddr)#connect
                 ftp.login(user,password)#login
                 print ftp.getwelcome()#display ftp server welcome message
                 ftp.cwd(ServerFolder) #choose the work directory
                 filename = LocalLocation + '\\' + FileName
                 file_handler = open(filename,'rb')#open the file in local as readonly
                 ftp.storbinary('STOR %s' % os.path.basename(filename),file_handler,bufsize)#upload file
                 ftp.set_debuglevel(0)
                 file_handler.close()
                 ftp.quit()
                 time2= time.time()
                 print "ftp up OK"
                 self.logging('Successfully upload file: "{0}" to FTP Server: {1}'.format(str(FileName), str(ServerAddr)))
                 print round((time2-time1),3)
             except:
                 self.logging('Fail to upload file: "{0}" to FTP Server: {1}'.format(str(FileName), str(ServerAddr)))
                 self.captureException()

         def ftp_down(self,ServerFolder, LocalLocation, FileName, ServerAddr, user, password):
             try:
                 time1= time.time()
                 bufsize = 2048#set buffer size
                 ftp=FTP()
                 #ftp.set_debuglevel(2)
                 ftp.connect(ServerAddr)#connect
                 ftp.login(user,password)#login
                 print ftp.getwelcome()#display ftp server welcome message
                 ftp.cwd(ServerFolder)#choose the work directory
                 filename = LocalLocation + '\\' + FileName
                 file_handler = open(filename,'wb').write #open the file in local as write
                 ftp.retrbinary('RETR %s' % os.path.basename(filename),file_handler,bufsize)#download file
                 ftp.set_debuglevel(0)
                 ftp.quit()
                 time2= time.time()
                 print "ftp down OK"
                 self.logging('Successfully download file: "{0}" from FTP Server: {1}'.format(str(FileName), str(ServerAddr)))
                 print round((time2-time1),3)
             except:
                 self.logging('Fail to download file: "{0}" from FTP Server: {1}'.format(str(FileName), str(ServerAddr)))
                 self.captureException()
         def usage(self):        print '''
 Usage:ftp_up_down [-u|-d]|[-S serverfolder| -L localfolder| -F filename| -I unix_ip| -U unix_user| -P unix_password]|[--help]\n
 \t-u|-d -u means upload, -p means download\n
 \t-S the unix folder to store the upload/download file\n
 \t-L the Windows folder to store the upload/download file\n
 \t-F the file name to be upload/download\n
 \t-I the unix server ip\n
 \t-U the unix user\n
 \t-P the unix user password\n
 Examples:
 \tftp_up_down -d -S "/release/scripts/" -L "E:\\Daly\\Investigation\\" -F "a.txt" -I 10.100.99.1xx -U root -P pass\n
 \tftp_up_down -u -S "/release/scripts/" -L "E:\\Daly\\Investigation\\" -F "a.txt" -I 10.100.99.1xx -U root -P pass
 '''
 ABSPATH=None
 if __name__ == "__main__":
         ABSPATH=os.path.abspath(sys.argv[0])
         ABSPATH=os.path.dirname(ABSPATH)+"\\"
         #print ABSPATH
         work_folder=ABSPATH
         #print work_folder
         serverfolder=""
         localfolder=""
         filename=""
         unix_ip=""
         unix_user=""
         unix_password=""
         try:
                 opts,args = getopt.getopt(sys.argv[1:], "hudS:L:F:I:U:P:", ["help", "up", "down", "serverfolder=", "localfolder=", "filename=", "ip=", "user=", "password="])
                 #print opts
                 #print args
                 if not opts:
                     try:
                         ftp_up_down().usage()
                     except:
                         sys.exit(0)
                 for opt, arg in opts:
                         if opt in ("-h", "--help"):
                                 ftp_up_down().usage()
                                 sys.exit(0)
                         elif opt in ("-S"):
                                 serverfolder = arg
                         elif opt in ("-L"):
                                 localfolder = arg
                         elif opt in ("-F"):
                                 filename = arg
                         elif opt in ("-I"):
                                 unix_ip = arg
                         elif opt in ("-U"):
                                 unix_user = arg
                         elif opt in ("-P"):
                                 unix_password = arg
                 #print opts[0]
                 print serverfolder, localfolder, filename, unix_ip, unix_user, unix_password
                 try:
                         if("-u" in opts[0]):
                                 try:
                                         print serverfolder, localfolder, filename, unix_ip, unix_user, unix_password
                                         ftp_up_down().ftp_up(ServerFolder=serverfolder, LocalLocation=localfolder, FileName=filename, ServerAddr=unix_ip, user=unix_user,password=unix_password)
                                 except:
                                         print ''.join(traceback.format_exception(*sys.exc_info()))
                         elif("-d" in opts[0]):
                                 try:
                                         #print serverfolder, localfolder, filename, unix_ip, unix_user, unix_password
                                         ftp_up_down().ftp_down(ServerFolder=serverfolder, LocalLocation=localfolder, FileName=filename, ServerAddr=unix_ip, user=unix_user,password=unix_password)
                                 except:
                                         print ''.join(traceback.format_exception(*sys.exc_info()))
                 except:
                         print ''.join(traceback.format_exception(*sys.exc_info()))
         except getopt.GetoptError:
                 print "getopt error"
                 ftp_up_down().usage()
                 sys.exit(1)

2. ssh

Python ssh需要安装其他模块支持，支持ssh的模块有pexpect(*Nix)，paramiko等， 在此主要讲在Windows上支持的paramiko模块。可参看一下代码。

 import paramiko
 class CommonUtils(object):
     '''
     classdocs
     '''
     def __init__(self):
         '''
         Constructor
         '''
         log().paramiko_log_to_file(level="DEBUG")
     def ssh_conn(self,host,port,user,password):
         '''
         @param host: the host name or ip address
         @param port: the ssh port number
         @param user: the user to login the host
         @param password: the user's password
         example: ssh_conn("10.xx.xx.xx",22,'xxu1','password')
         '''
         try:
             client = paramiko.SSHClient()
             client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
             client.connect(host,22,user,password)
             return client
         except Exception, e:
             log().log("{0} Please check the user name and password are correct.".format(str(e)))
             return None
     def ssh_close(self,conn):
         conn.close()
     def ssh_exec_querycommand(self,conn,command):
         '''
         @param conn: the ssh connection which setup with using function ssh_conn
         @param command: the command to be executed
         '''
         stdin, stdout, stderr = conn.exec_command(command)
         stdin.flush()
         output = []
         for line in stdout:
             print '###' + line.strip('\n')
             output.append(line.strip('\n'))
         if stderr != None and len(str(stderr)) == 0:
             print 'Error info: ' + stderr.read()
         return output
     def ssh_exec_intercommand(self,conn,begincommand,promptanswer):
         '''
         @param conn: the ssh connection which setup with using function ssh_conn
         @param begincommand: the command to be executed firstly
         @param promptanswer: the command to be executed to answer the question after executed begincommand
         '''
         channel = conn.invoke_shell()
         channel.recv(1024)
         time.sleep(1)
         channel.send(begincommand + '\n')
         time.sleep(1)
         question = channel.recv(1024)
         #realquestion = (question.split(begincommand))[0:]
         realquestion = (question.split('\n'))[1]
         log().log("Question pops: {0}".format(realquestion))
         time.sleep(1)
         channel.send(promptanswer + '\n')
         time.sleep(30)
         result = channel.recv(1024)
         log().log("After input answer: {0}".format(result))
         channel.close()
         return result
     def ssh_exec_command_after_interaction(self,conn,begincommand,promptanswer,command):
         '''
         @param conn: the ssh connection which setup with using function ssh_conn
         @param begincommand: the command to be executed firstly
         @param promptanswer: the command to be executed to answer the question after executed begincommand
         @param command: the command to be executed after the interaction
         '''
         channel = conn.invoke_shell()
         channel.recv(1024)
         time.sleep(1)
         channel.send(begincommand + '\n')
         time.sleep(1)
         question = channel.recv(1024)
         realquestion = (question.split(begincommand))[1]
         #realquestion = (question.split('\n'))[1]
         log().log("Question pops: {0}".format(realquestion))
         time.sleep(1)
         channel.send(promptanswer + '\n')
         time.sleep(1)
         result = channel.recv(1024)
         log().log("After input answer: {0}".format(result))
         channel.send(command + '\n')
         time.sleep(3)
         cmdresult = channel.recv(1024)
         fullcmdresult = (cmdresult.split(command))[0]
         realcmdresult = (fullcmdresult.split('\n'))[1:-1]
         finalrestult = ''
         for i in range(len(realcmdresult)):
             finalrestult += ''.join(realcmdresult[i])
         log().log("Command result: {0}".format(finalrestult))
         channel.close()
         return str(finalrestult)